#include <QHostAddress>

#include "streamconfigdialog.h"
#include "stream.h"
#include "abstractprotocol.h"
#include "protocollistiterator.h"

#include "modeltest.h"

// FIXME(HI) - remove
#include "../common/protocolmanager.h"
extern ProtocolManager OstProtocolManager;

int StreamConfigDialog::lastTopLevelTabIndex = 0;

StreamConfigDialog::StreamConfigDialog(Port &port, uint streamIndex,
	QWidget *parent) : QDialog (parent), mPort(port)
{
	OstProto::Stream s;
	mCurrentStreamIndex = streamIndex;
	mpStream = new Stream;
	mPort.streamByIndex(mCurrentStreamIndex)->protoDataCopyInto(s);
	mpStream->protoDataCopyFrom(s);
	_iter = mpStream->createProtocolListIterator();
	isUpdateInProgress = false;

	setupUi(this);
	setupUiExtra();

	for (int i = ProtoMin; i < ProtoMax; i++)
	{
		bgProto[i]->setProperty("ProtocolLevel", i);
		bgProto[i]->setProperty("ProtocolId", ButtonIdNone);
		connect(bgProto[i], SIGNAL(buttonClicked(int)),
			this, SLOT(updateProtocol(int)));
	}

	//! \todo causes a crash!
#if 0	
	connect(lePktLen, SIGNAL(textEdited(QString)),
		this, SLOT(updateContents()));
#endif

	// Time to play match the signals and slots!

	// If L1/FT = None, force subsequent protocol level(s) also to None
	connect(rbL1None, SIGNAL(toggled(bool)), this, SLOT(forceProtocolNone(bool)));
	connect(rbFtNone, SIGNAL(toggled(bool)), this, SLOT(forceProtocolNone(bool)));

	// Enable/Disable L3 Protocol Choices for FT Ethernet2
	connect(rbFtEthernet2, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setEnabled(bool)));
	connect(rbFtEthernet2, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setEnabled(bool)));

	// Force L3 = None if FT = 802.3 Raw
	connect(rbFt802Dot3Raw, SIGNAL(clicked(bool)), rbL3None, SLOT(click()));
	connect(rbFt802Dot3Raw, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setDisabled(bool)));
	connect(rbFt802Dot3Raw, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setDisabled(bool)));

	// Force L3 = None if FT = 802.3 LLC (to ensure a valid L3 is selected)
	connect(rbFt802Dot3Llc, SIGNAL(clicked(bool)), rbL3None, SLOT(click()));

	// Enable/Disable L3 Protocol Choices for FT 802Dot3Llc
	connect(rbFt802Dot3Llc, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setEnabled(bool)));
	connect(rbFt802Dot3Llc, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setDisabled(bool)));

	// Enable/Disable L3 Protocol Choices for FT 802.3 LLC SNAP
	connect(rbFtLlcSnap, SIGNAL(toggled(bool)), rbL3Ipv4, SLOT(setEnabled(bool)));
	connect(rbFtLlcSnap, SIGNAL(toggled(bool)), rbL3Arp, SLOT(setEnabled(bool)));

	// Force L3 = Other if FT = Other
	connect(rbFtOther, SIGNAL(toggled(bool)), rbL3Other, SLOT(setChecked(bool)));
	connect(rbFtOther, SIGNAL(toggled(bool)), gbL3Proto, SLOT(setDisabled(bool)));

	// If L3 = None, force subsequent protocol level also to None
	connect(rbL3None, SIGNAL(toggled(bool)), this, SLOT(forceProtocolNone(bool)));

	// Enable/Disable L4 Protocol Choices for L3 Protocol IPv4
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4Icmp, SLOT(setEnabled(bool)));
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4Igmp, SLOT(setEnabled(bool)));
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4Tcp, SLOT(setEnabled(bool)));
	connect(rbL3Ipv4, SIGNAL(toggled(bool)), rbL4Udp, SLOT(setEnabled(bool)));

	// Enable/Disable L4 Protocol Choices for L3 Protocol ARP
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4Icmp, SLOT(setDisabled(bool)));
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4Igmp, SLOT(setDisabled(bool)));
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4Tcp, SLOT(setDisabled(bool)));
	connect(rbL3Arp, SIGNAL(toggled(bool)), rbL4Udp, SLOT(setDisabled(bool)));

	// Force L4 Protocol = None if L3 Protocol is set to ARP
	connect(rbL3Arp, SIGNAL(clicked(bool)), rbL4None, SLOT(click()));

	// Force L4 = Other if L3 = Other
	connect(rbL3Other, SIGNAL(toggled(bool)), rbL4Other, SLOT(setChecked(bool)));
	connect(rbL3Other, SIGNAL(toggled(bool)), gbL4Proto, SLOT(setDisabled(bool)));

	// Force Payload = Other if L4 = Other
	connect(rbL4Other, SIGNAL(toggled(bool)), rbPayloadOther, SLOT(setChecked(bool)));
	connect(rbL4Other, SIGNAL(toggled(bool)), gbPayloadProto, SLOT(setDisabled(bool)));

	mpAvailableProtocolsModel = new QStringListModel(
		OstProtocolManager.protocolDatabase(), this);
	lvAllProtocols->setModel(mpAvailableProtocolsModel);
	mpSelectedProtocolsModel = new QStringListModel(this);
	lvSelectedProtocols->setModel(mpSelectedProtocolsModel);


	connect(lvAllProtocols->selectionModel(),
		SIGNAL(selectionChanged(const QItemSelection&, const QItemSelection&)),
		this, SLOT(when_lvAllProtocols_selectionChanged(
			const QItemSelection&, const QItemSelection&)));
	connect(lvSelectedProtocols->selectionModel(),
		SIGNAL(currentChanged(const QModelIndex&, const QModelIndex&)),
		this, SLOT(when_lvSelectedProtocols_currentChanged(const QModelIndex&,
			const QModelIndex&)));

	LoadCurrentStream();
	mpPacketModel = new PacketModel(this);
	tvPacketTree->setModel(mpPacketModel);
	mpPacketModelTester = new ModelTest(mpPacketModel);
	tvPacketTree->header()->hide();
	vwPacketDump->setModel(mpPacketModel);
	vwPacketDump->setSelectionModel(tvPacketTree->selectionModel());

	// TODO(MED):

	//! \todo Implement then enable these protocols - ARP, IPv6, ICMP, IGMP
	rbL3Arp->setHidden(true);
	rbL3Ipv6->setHidden(true);
	rbL4Icmp->setHidden(true);
	rbL4Igmp->setHidden(true);
	//! \todo Enable navigation of streams
	pbPrev->setDisabled(true);
	pbNext->setDisabled(true);
	//! \todo Support Goto Stream Id
	leStreamId->setDisabled(true);
	disconnect(rbActionGotoStream, SIGNAL(toggled(bool)), leStreamId, SLOT(setEnabled(bool)));
	//! \todo Support Continuous Mode
	rbModeContinuous->setDisabled(true);

	// Finally, restore the saved last selected tab for the various tab widgets
	twTopLevel->setCurrentIndex(lastTopLevelTabIndex);
}

void StreamConfigDialog::setupUiExtra()
{
	QRegExp reHex2B("[0-9,a-f,A-F]{1,4}");
	QRegExp reHex4B("[0-9,a-f,A-F]{1,8}");
	QRegExp reMac("([0-9,a-f,A-F]{2,2}[:-]){5,5}[0-9,a-f,A-F]{2,2}");

	// ---- Setup default stuff that cannot be done in designer ----
	bgProto[ProtoL1] = new QButtonGroup();
	bgProto[ProtoL1]->addButton(rbL1None, ButtonIdNone);
	bgProto[ProtoL1]->addButton(rbL1Mac, OstProto::Protocol::kMacFieldNumber);
	bgProto[ProtoL1]->addButton(rbL1Other, ButtonIdOther);

	bgProto[ProtoL2] = new QButtonGroup();
#if 0
	foreach(QRadioButton *btn, gbFrameType->findChildren<QRadioButton*>())
		bgL2Proto->addButton(btn);
#else
	bgProto[ProtoL2]->addButton(rbFtNone, ButtonIdNone);
	bgProto[ProtoL2]->addButton(rbFtEthernet2, OstProto::Protocol::kEth2FieldNumber);
	bgProto[ProtoL2]->addButton(rbFt802Dot3Raw, OstProto::Protocol::kDot3FieldNumber);
	bgProto[ProtoL2]->addButton(rbFt802Dot3Llc, OstProto::Protocol::kDot2LlcFieldNumber);
	bgProto[ProtoL2]->addButton(rbFtLlcSnap, OstProto::Protocol::kDot2SnapFieldNumber);
	bgProto[ProtoL2]->addButton(rbFtOther, ButtonIdOther);
#endif

	bgProto[ProtoVlan] = new QButtonGroup();
	bgProto[ProtoVlan]->addButton(rbVlanNone, ButtonIdNone);
	bgProto[ProtoVlan]->addButton(rbVlanSingle, OstProto::Protocol::kVlanFieldNumber);
	bgProto[ProtoVlan]->addButton(rbVlanDouble, OstProto::Protocol::kVlanStackFieldNumber);

	bgProto[ProtoL3] = new QButtonGroup();
#if 0
	foreach(QRadioButton *btn, gbL3Proto->findChildren<QRadioButton*>())
		bgProto[ProtoL3]->addButton(btn);
#else
	bgProto[ProtoL3]->addButton(rbL3None, ButtonIdNone);
	bgProto[ProtoL3]->addButton(rbL3Ipv4, OstProto::Protocol::kIp4FieldNumber);
	bgProto[ProtoL3]->addButton(rbL3Ipv6, 0xFFFF);
	bgProto[ProtoL3]->addButton(rbL3Arp, 0xFFFF);
	bgProto[ProtoL3]->addButton(rbL3Other, ButtonIdOther);
#endif

	bgProto[ProtoL4] = new QButtonGroup();
#if 0
	foreach(QRadioButton *btn, gbL4Proto->findChildren<QRadioButton*>())
		bgProto[ProtoL4]->addButton(btn);
#else
	bgProto[ProtoL4]->addButton(rbL4None, 0);
	bgProto[ProtoL4]->addButton(rbL4Tcp, OstProto::Protocol::kTcpFieldNumber);
	bgProto[ProtoL4]->addButton(rbL4Udp, OstProto::Protocol::kUdpFieldNumber);
	bgProto[ProtoL4]->addButton(rbL4Icmp, 0xFFFF);
	bgProto[ProtoL4]->addButton(rbL4Igmp, 0xFFFF);
	bgProto[ProtoL4]->addButton(rbL4Other, ButtonIdOther);
#endif

	bgProto[ProtoPayload] = new QButtonGroup();
#if 0
	foreach(QRadioButton *btn, gbPayloadProto->findChildren<QRadioButton*>())
		bgProto[ProtoPayload]->addButton(btn);
#else
	bgProto[ProtoPayload]->addButton(rbPayloadNone, ButtonIdNone);
	bgProto[ProtoPayload]->addButton(rbPayloadPattern, OstProto::Protocol::kPayloadFieldNumber);
	bgProto[ProtoPayload]->addButton(rbPayloadOther, ButtonIdOther);
#endif
	/*
	** Setup Validators
	*/	
	// Meta Data
	//! \todo - doesn't seem to work -  range validator needs a spinbox?
	//lePktLen->setValidator(new QIntValidator(MIN_PKT_LEN, MAX_PKT_LEN, this));

	/*
	** Setup Connections
	*/
	connect(rbSendPackets, SIGNAL(toggled(bool)), 
		this, SLOT(update_NumPacketsAndNumBursts()));
	connect(rbSendBursts, SIGNAL(toggled(bool)), 
		this, SLOT(update_NumPacketsAndNumBursts()));
	connect(rbModeFixed, SIGNAL(toggled(bool)), 
		this, SLOT(update_NumPacketsAndNumBursts()));
	connect(rbModeContinuous, SIGNAL(toggled(bool)), 
		this, SLOT(update_NumPacketsAndNumBursts()));

}

StreamConfigDialog::~StreamConfigDialog()
{
	delete mpPacketModelTester;
	delete mpPacketModel;

	for (int i = ProtoMin; i < ProtoMax; i++)
		delete bgProto[i];

	delete _iter;
	delete mpStream;
}

void StreamConfigDialog::on_cmbPktLenMode_currentIndexChanged(QString mode)
{
	if (mode == "Fixed")
	{
		lePktLen->setEnabled(true);
		lePktLenMin->setDisabled(true);
		lePktLenMax->setDisabled(true);
	}
	else if (mode == "Increment")
	{
		lePktLen->setDisabled(true);
		lePktLenMin->setEnabled(true);
		lePktLenMax->setEnabled(true);
	}
	else if (mode == "Decrement")
	{
		lePktLen->setDisabled(true);
		lePktLenMin->setEnabled(true);
		lePktLenMax->setEnabled(true);
	}
	else if (mode == "Random")
	{
		lePktLen->setDisabled(true);
		lePktLenMin->setEnabled(true);
		lePktLenMax->setEnabled(true);
	}
	else
	{
		qWarning("Unhandled/Unknown PktLenMode = %s", mode.toAscii().data());
	}
}

void StreamConfigDialog::on_pbPrev_clicked()
{
#if 0
	StoreCurrentStream(currStreamIdx);
	currStreamIdx--;
	LoadCurrentStream(currStreamIdx);

	pbPrev->setDisabled((currStreamIdx == 0));
	pbNext->setDisabled((currStreamIdx == 2));
#endif
}

void StreamConfigDialog::on_pbNext_clicked()
{
#if 0
	StoreCurrentStream(currStreamIdx);
	currStreamIdx++;
	LoadCurrentStream(currStreamIdx);

	pbPrev->setDisabled((currStreamIdx == 0));
	pbNext->setDisabled((currStreamIdx == 2));
#endif
}

void StreamConfigDialog::on_tbSelectProtocols_currentChanged(int index)
{
	qDebug("%s, index = %d", __FUNCTION__, index);
	switch (index)
	{
		case 0:
			updateSelectProtocolsSimpleWidget();
			break;
		case 1:
			updateSelectProtocolsAdvancedWidget();
			break;
		default:
			qFatal("%s: unexpected index = %d", __FUNCTION__, index);
	}
}

void StreamConfigDialog::when_lvAllProtocols_selectionChanged(
	const QItemSelection &selected, const QItemSelection &deselected)
{
	int size = lvAllProtocols->selectionModel()->selectedIndexes().size();

	qDebug("%s: selected.indexes().size = %d\n", __FUNCTION__, size);

	tbAdd->setEnabled(size > 0);
}

void StreamConfigDialog::when_lvSelectedProtocols_currentChanged(
	const QModelIndex &current, const QModelIndex &previous)
{
	qDebug("%s: currentRow = %d\n", __FUNCTION__, current.row());

	tbDelete->setEnabled(current.isValid());
	tbUp->setEnabled(current.isValid() && (current.row() != 0));
	tbDown->setEnabled(current.isValid() && 
		(current.row() != (current.model()->rowCount() - 1)));
}

void StreamConfigDialog::on_tbAdd_clicked()
{
	int n = 0;
	QModelIndex idx2;
	AbstractProtocol *p;
	QModelIndexList	selection;

	selection = lvAllProtocols->selectionModel()->selectedIndexes();

	// Validation
	if (selection.size() == 0)
		return;

	idx2 = lvSelectedProtocols->currentIndex();
	if (idx2.isValid())
		n = idx2.row();

	_iter->toFront();
	while (n--)
	{
		if (!_iter->hasNext())
			return;

		p = _iter->next();
	}

	foreach(QModelIndex idx, selection)
		_iter->insert(OstProtocolManager.createProtocol(
			mpAvailableProtocolsModel->stringList().at(idx.row()), mpStream));

	updateSelectProtocolsAdvancedWidget();
	lvSelectedProtocols->setCurrentIndex(idx2);
}

void StreamConfigDialog::on_tbDelete_clicked()
{
	int n;
	QModelIndex idx;
	AbstractProtocol *p;

	idx = lvSelectedProtocols->currentIndex();

	// Validation
	if (!idx.isValid())
		return;

	n = idx.row() + 1;

	_iter->toFront();
	while (n--)
	{
		if (!_iter->hasNext())
			return;

		p = _iter->next();
	}

	_iter->remove();
	delete p;

	updateSelectProtocolsAdvancedWidget();
	lvSelectedProtocols->setCurrentIndex(idx);
}

void StreamConfigDialog::on_tbUp_clicked()
{
	int m, n;
	QModelIndex idx;
	AbstractProtocol *p;

	idx = lvSelectedProtocols->currentIndex();

	// Validation
	if (!idx.isValid() || idx.row() == 0)
		return;

	m = n = idx.row() + 1;

	_iter->toFront();
	while (n--)
	{
		if (!_iter->hasNext())
			return;

		p = _iter->next();
	}

	_iter->remove();
	_iter->previous();
	_iter->insert(p);

	updateSelectProtocolsAdvancedWidget();
	lvSelectedProtocols->setCurrentIndex(idx.sibling(m-2, 0));
}

void StreamConfigDialog::on_tbDown_clicked()
{
	int m, n;
	QModelIndex idx;
	AbstractProtocol *p;

	idx = lvSelectedProtocols->currentIndex();

	// Validation
	if (!idx.isValid() || idx.row() == idx.model()->rowCount())
		return;

	m = n = idx.row() + 1;

	_iter->toFront();
	while (n--)
	{
		if (!_iter->hasNext())
			return;

		p = _iter->next();
	}

	_iter->remove();
	_iter->next();
	_iter->insert(p);

	updateSelectProtocolsAdvancedWidget();
	lvSelectedProtocols->setCurrentIndex(idx.sibling(m,0));
}

void StreamConfigDialog::updateSelectProtocolsAdvancedWidget()
{
	QStringList	selProtoList;

	qDebug("%s", __FUNCTION__);

	_iter->toFront();
	while(_iter->hasNext())
	{
		AbstractProtocol* p = _iter->next();
		qDebug("%p -- %d", p, p->protocolNumber());
		selProtoList.append(p->shortName());
	}
	mpSelectedProtocolsModel->setStringList(selProtoList);
}

void StreamConfigDialog::on_twTopLevel_currentChanged(int index)
{
	switch (index)
	{
		// Protocol Data
		case 1:
		{
			QWidget *selWidget;

			// Hide the ToolBox before modifying it - else we have a crash !!!
			tbProtocolData->hide();

			selWidget = tbProtocolData->currentWidget();

			// Remove all existing protocol widgets 
			while (tbProtocolData->count() > 0)
			{
				QWidget* w = tbProtocolData->widget(0);
				tbProtocolData->removeItem(0);
				w->setParent(0);
			}

			// Repopulate the widgets
			_iter->toFront();
			while (_iter->hasNext())
			{
				AbstractProtocol* p = _iter->next();
				tbProtocolData->addItem(p->configWidget(), p->name());
			}

			tbProtocolData->setCurrentWidget(selWidget);

			tbProtocolData->show();
			break;
		}

		// Packet View
		case 3:
		{
			StoreCurrentStream();
			mpPacketModel->setSelectedProtocols(*_iter);
			break;
		}

		default:
			break;
	}
}

void StreamConfigDialog::update_NumPacketsAndNumBursts()
{
	if (rbSendPackets->isChecked() && rbModeFixed->isChecked())
		leNumPackets->setEnabled(true);
	else
		leNumPackets->setEnabled(false);

	if (rbSendBursts->isChecked() && rbModeFixed->isChecked())
		leNumBursts->setEnabled(true);
	else
		leNumBursts->setEnabled(false);
}

#if 0
void StreamConfigDialog::on_lePattern_editingFinished()
{
	ulong	num = 0;
	bool	isOk;
	QString	str;

	num = lePattern->text().remove(QChar(' ')).toULong(&isOk, 16);
	qDebug("editfinished (%s | %x)\n", lePattern->text().toAscii().data(), num);
	lePattern->setText(uintToHexStr(num, str, 4));
	qDebug("editfinished (%s | %x)\n", lePattern->text().toAscii().data(), num);
}
#endif

/*! 
Skip protocols upto and including the layer specified.
*/
bool StreamConfigDialog::skipProtocols(int layer)
{
	_iter->toFront();

	for (int i = ProtoMin; i <= layer; i++)
	{
		if(_iter->hasNext())
		{
			int id;
			QAbstractButton *btn;

			id = _iter->peekNext()->protocolNumber();
			btn = bgProto[i]->button(id);
			if (btn)
				_iter->next();
		}
	}

	return true;
}

/*!
Protocol choices (except "None" and "Other") for a protocol button group are disabled if checked is true, else they are enabled
*/
void StreamConfigDialog::disableProtocols(QButtonGroup *protocolGroup, bool checked)
{
	qDebug("%s: btnGrp = %p, chk? = %d", __FUNCTION__, protocolGroup, checked);
	foreach(QAbstractButton *btn, protocolGroup->buttons())
	{
		int id = protocolGroup->id(btn);

		if ((id != ButtonIdNone) && (id != ButtonIdOther))
			btn->setDisabled(checked);
	}
}

void StreamConfigDialog::forceProtocolNone(bool checked)
{
	QObject *btn;

	btn = sender();
	Q_ASSERT(btn != NULL);

	qDebug("%s: chk? = %d, btn = %p, L1 = %p, L2 = %p, L3 = %p", __FUNCTION__,
			checked, btn, rbL1None, rbFtNone, rbL3None);

	if (btn == rbL1None)
	{
		if (checked)
		{
			bgProto[ProtoVlan]->button(ButtonIdNone)->click();
			bgProto[ProtoL2]->button(ButtonIdNone)->click();
			bgProto[ProtoPayload]->button(ButtonIdNone)->click();
		}

		disableProtocols(bgProto[ProtoVlan], checked);
		disableProtocols(bgProto[ProtoL2], checked);
		disableProtocols(bgProto[ProtoPayload], checked);
	} 
	else if (btn == rbFtNone)
	{
		if (checked)
			bgProto[ProtoL3]->button(ButtonIdNone)->click();
		disableProtocols(bgProto[ProtoL3], checked);
	}
   	else if (btn == rbL3None)
	{
		if (checked)
			bgProto[ProtoL4]->button(ButtonIdNone)->click();
		disableProtocols(bgProto[ProtoL4], checked);
	}
   	else
	{
		Q_ASSERT(1 == 0); // Unreachable code!
	}
}

void StreamConfigDialog::updateProtocol(int newId)
{
	int level;
	QButtonGroup	*btnGrp;

	btnGrp = static_cast<QButtonGroup*>(sender());
	Q_ASSERT(btnGrp != NULL);

	level = btnGrp->property("ProtocolLevel").toInt();
	Q_ASSERT(btnGrp == bgProto[level]);

	__updateProtocol(level, newId);
}

void StreamConfigDialog::__updateProtocol(int level, int newId)
{
	int oldId;
	QButtonGroup	*btnGrp;

	Q_ASSERT((level >= ProtoMin) && (level <= ProtoMax));
	btnGrp = bgProto[level];
	oldId = btnGrp->property("ProtocolId").toInt();

	qDebug("%s: level = %d old id = %d new id = %d upd? = %d", __FUNCTION__, 
		level, oldId, newId, isUpdateInProgress);

	if (newId == oldId)
		return;

	if (!isUpdateInProgress)
	{
		int ret;
		AbstractProtocol *p;

		ret = skipProtocols(level-1);
		Q_ASSERT(ret == true);

		Q_ASSERT(oldId != newId);
		Q_ASSERT(newId != ButtonIdOther);
		
		switch (oldId)
		{
			case ButtonIdNone:
				_iter->insert(OstProtocolManager.createProtocol(
						newId, mpStream));
				break;

			case ButtonIdOther:
			default:
				Q_ASSERT(_iter->hasNext());
				p =_iter->next();

				if (newId)
					_iter->setValue(OstProtocolManager.createProtocol(
							newId, mpStream));
				else
					_iter->remove();
				delete p;
				if (level == ProtoPayload)
				{
					while (_iter->hasNext())
					{
						p = _iter->next();
						_iter->remove();
						delete p;
					}
				}
				break;
		}
	}

	btnGrp->setProperty("ProtocolId", newId);
	return;
}

void StreamConfigDialog::updateSelectProtocolsSimpleWidget()
{
	int i;
	quint32	id;
	QAbstractButton *btn;

	qDebug("%s", __FUNCTION__);

	isUpdateInProgress = true;

	// Reset to default state ...
	for (i = ProtoMin; i < ProtoMax; i++)
		bgProto[i]->button(ButtonIdNone)->click();

	// ... now iterate and update
	_iter->toFront();

	for (i = ProtoMin; i < ProtoMax; i++)
	{
		if (!_iter->hasNext())
			goto _done;

		id = _iter->next()->protocolNumber();
		btn = bgProto[i]->button(id);

		if (btn && btn->isEnabled())
			btn->click();
		else
		{
			switch (i)
			{
				case ProtoVlan:
					_iter->previous();
					break;

				case ProtoPayload:
					goto _other;

				default:
					btn = bgProto[ProtoPayload]->button(id);
					if (btn && btn->isEnabled())
					{
						btn->click();
						break;
					}
					else
						goto _other;
			}
		}
	}

	// If more protocol(s) beyond payload ...
	if (_iter->hasNext())
	{
		i = ProtoPayload;
		goto _other;
	}

	goto _done;

_other:
	for (int j = i; j < ProtoMax; j++)
	{
		// VLAN doesn't have a "Other" button
		if (j == ProtoVlan)
			continue;

		bgProto[j]->button(ButtonIdOther)->setChecked(true);
		__updateProtocol(j, ButtonIdOther);
	}

_done:
	isUpdateInProgress = false;
}

void StreamConfigDialog::LoadCurrentStream()
{
	QString	str;

	qDebug("loading mpStream %p", mpStream);

	// Meta Data
	{
		cmbPktLenMode->setCurrentIndex(mpStream->lenMode());
		lePktLen->setText(str.setNum(mpStream->frameLen()));
		lePktLenMin->setText(str.setNum(mpStream->frameLenMin()));
		lePktLenMax->setText(str.setNum(mpStream->frameLenMax()));
	}

	// Protocols
	{
		updateSelectProtocolsSimpleWidget();
		updateSelectProtocolsAdvancedWidget();

		mpStream->loadProtocolWidgets();
	}

	// Stream Control
	{
		switch (mpStream->sendUnit())
		{
		case Stream::e_su_packets:
			rbSendPackets->setChecked(true);
			break;
		case Stream::e_su_bursts:
			rbSendBursts->setChecked(true);
			break;
		default:
			qWarning("Unhandled sendUnit = %d\n", mpStream->sendUnit());
		}

		switch (mpStream->sendMode())
		{
		case Stream::e_sm_fixed:
			rbModeFixed->setChecked(true);
			break;
		case Stream::e_sm_continuous:
			rbModeContinuous->setChecked(true);
			break;
		default:
			qWarning("Unhandled sendMode = %d\n", mpStream->sendMode());
		}

		switch(mpStream->nextWhat())
		{
		case Stream::e_nw_stop:
			rbActionStop->setChecked(true);
			break;
		case Stream::e_nw_goto_next:
			rbActionGotoNext->setChecked(true);
			break;
		case Stream::e_nw_goto_id:
			rbActionGotoStream->setChecked(true);
			break;
		default:
			qWarning("Unhandled nextAction = %d\n", mpStream->nextWhat());
		}

		leNumPackets->setText(QString().setNum(mpStream->numPackets()));
		leNumBursts->setText(QString().setNum(mpStream->numBursts()));
		lePacketsPerBurst->setText(QString().setNum(mpStream->burstSize()));
		lePacketsPerSec->setText(QString().setNum(mpStream->packetRate()));
		leBurstsPerSec->setText(QString().setNum(mpStream->burstRate()));
		// TODO(MED): Change this when we support goto to specific stream
		leStreamId->setText(QString("0"));
	}
	qDebug("loading stream done");
}

void StreamConfigDialog::StoreCurrentStream()
{
	QString	str;
	bool	isOk;
	Stream	*pStream = mpStream;

	qDebug("storing pStream %p", pStream);

	// Meta Data
	pStream->setLenMode((Stream::FrameLengthMode) cmbPktLenMode->currentIndex());
	pStream->setFrameLen(lePktLen->text().toULong(&isOk));
	pStream->setFrameLenMin(lePktLenMin->text().toULong(&isOk));
	pStream->setFrameLenMax(lePktLenMax->text().toULong(&isOk));

	// Protocols
	{
		pStream->storeProtocolWidgets();
	}

	// Stream Control
	{
		if (rbSendPackets->isChecked())
			pStream->setSendUnit(Stream::e_su_packets);
		if (rbSendBursts->isChecked())
			pStream->setSendUnit(Stream::e_su_bursts);

		if (rbModeFixed->isChecked())
			pStream->setSendMode(Stream::e_sm_fixed);
		if (rbModeContinuous->isChecked())
			pStream->setSendMode(Stream::e_sm_continuous);

		if (rbActionStop->isChecked())
			pStream->setNextWhat(Stream::e_nw_stop);
		if (rbActionGotoNext->isChecked())
			pStream->setNextWhat(Stream::e_nw_goto_next);
		if (rbActionGotoStream->isChecked())
			pStream->setNextWhat(Stream::e_nw_goto_id);

		pStream->setNumPackets(leNumPackets->text().toULong(&isOk));
		pStream->setNumBursts(leNumBursts->text().toULong(&isOk));
		pStream->setBurstSize(lePacketsPerBurst->text().toULong(&isOk));
		pStream->setPacketRate(lePacketsPerSec->text().toULong(&isOk));
		pStream->setBurstRate(leBurstsPerSec->text().toULong(&isOk));
	}
}

void StreamConfigDialog::on_pbOk_clicked()
{
	OstProto::Stream	s;

	// Store dialog contents into stream
	StoreCurrentStream();

	// Copy the data from the "local working copy of stream" to "actual stream"
	mpStream->protoDataCopyInto(s);
	mPort.streamByIndex(mCurrentStreamIndex)->protoDataCopyFrom(s);

	qDebug("stream stored");

	lastTopLevelTabIndex = twTopLevel->currentIndex();
}