ChanMap.cpp

#include "ChanMap.h"
#include <QTextStream>
#include <QRegExp>
#include <QStringList>
#include <QMap>
#include <QVector>

QString ChanMapDesc::toTerseString() const
{
	return  QString("i:") + QString::number(intan)
	+ ",ic:" + QString::number(intanCh)
	+ ",e:" + QString::number(electrodeId);
}

QString ChanMapDesc::toString() const
{
	return  QString("intan:") + QString::number(intan)
			+ ", intanCh:" + QString::number(intanCh)
			+ ", electrodeId:" + QString::number(electrodeId);
}

QString ChanMapDesc::toWSDelimString() const
{
	return QString("%1 %2 %3").arg(intan).arg(intanCh).arg(electrodeId); 
}


//#include "Util.h"

/*static*/ ChanMapDesc ChanMapDesc::fromString(const QString & s_in)
{
	ChanMapDesc ret;
	
	const QStringList sl = s_in.split(QRegExp(",\\s*"), QString::SkipEmptyParts);
	//Debug() << "s_in: " << s_in;
	foreach (QString s, sl) {
		const QStringList il = s.split(QRegExp(":\\s*"));
		if (il.size() >= 2) {
			QString name = il.at(0);
			unsigned value = il.at(1).toUInt();
			//Debug() << "name: " << name << " value: " << value;
			if (name == "intan") ret.intan = value;
			if (name == "intanCh") ret.intanCh = value;
			if (name == "electrodeId") ret.electrodeId = value;
		}
	}
	return ret;
}

/*static*/ ChanMapDesc ChanMapDesc::fromTerseString(const QString & s_in)
{
	ChanMapDesc ret;
	
	const QStringList sl = s_in.split(QRegExp(",\\s*"), QString::SkipEmptyParts);
	//Debug() << "s_in: " << s_in;
	foreach (QString s, sl) {
		const QStringList il = s.split(QRegExp(":\\s*"));
		if (il.size() >= 2) {
			QString name = il.at(0);
			unsigned value = il.at(1).toUInt();
			//Debug() << "name: " << name << " value: " << value;
			if (name == "i") ret.intan = value;
			if (name == "ic") ret.intanCh = value;
			if (name == "e") ret.electrodeId = value;
		}
	}
	return ret;
}


/*static*/ ChanMapDesc ChanMapDesc::fromWSDelimString(const QString & s_in) 
{
	ChanMapDesc ret;

	const QStringList sl = s_in.split(QRegExp("\\s+"), QString::SkipEmptyParts);
	if (sl.size() >= 3) {
		bool ok;
		unsigned v = sl.at(0).toUInt(&ok);
		if (ok) ret.intan = v;
		v = sl.at(1).toUInt(&ok);
		if (ok) ret.intanCh = v;
		v = sl.at(2).toUInt(&ok);
		if (ok) ret.electrodeId = v;
	}
	return ret;
}

QString ChanMap::toString() const
{
	QString ret;
	QTextStream ts(&ret, QIODevice::WriteOnly|QIODevice::Truncate);
	const int n = size();
	for (int i = 0; i < n; ++i) {
		if (i) ts << ",";
		ts << "(" << (*this)[i].toString() << ")";
	}
	ts.flush();
	return ret;
}

QString ChanMap::toTerseString(const QBitArray & bm) const
{
	QString ret;
	QTextStream ts(&ret, QIODevice::WriteOnly|QIODevice::Truncate);
	const int n = size(), n2 = bm.size();
	ts << "(" << n << ")";
	for (int i = 0; i < n && i < n2; ++i) {
		if (!bm.testBit(i)) continue;
		ts << ",(" << (*this)[i].toTerseString() << ")";
	}
	ts.flush();
	return ret;
}

QString ChanMap::toWSDelimFlatFileString() const 
{
	QString ret;
	QTextStream ts(&ret, QIODevice::WriteOnly|QIODevice::Truncate);
	const int n = size();
	for (int i = 0; i < n; ++i) {
		ts << ((*this)[i].toWSDelimString()) << "\n";
	}
	ts.flush();
	return ret;
}

/*static*/ ChanMap ChanMap::fromTerseString(const QString & s_in) 
{
	const QStringList sl = s_in.split(QRegExp("(^\\()|(\\),)|(\\)$)"), QString::SkipEmptyParts);
	ChanMap ret;
	int i = 0;
	
	foreach(QString s, sl) {	
		const int sp = s.startsWith("(") ? 1 : 0, ep = s.endsWith(")") ? 1 : 0;
		//Debug() << "s: " << s;
		QString inParens = s.length() > 2 ? s.mid(sp, s.length()-(sp+ep)) : "";
		if (0==i) {
			// first thing in the list is the size of the total chanmap
			bool ok;
			unsigned chanMapSize = inParens.toUInt(&ok);
			if (!ok) ret.reserve(128);
			else ret.reserve(chanMapSize);
		} else {
			// rest of the items are the chan map tuples, only the "on" channels exist
			ChanMapDesc cmd (ChanMapDesc::fromTerseString(inParens));
			//ret[cmd.graphNum] = cmd;
			ret.push_back(cmd);
		}
		++i;
	}
	return ret;
}

/*static*/ ChanMap ChanMap::fromString(const QString & s_in)
{
	const QStringList sl = s_in.split(QRegExp("(^\\()|(\\),)|(\\)$)"), QString::SkipEmptyParts);
	ChanMap ret;
	ret.reserve(sl.size());

	foreach(QString s, sl) {	
		const int sp = s.startsWith("(") ? 1 : 0, ep = s.endsWith(")") ? 1 : 0;
		//Debug() << "s: " << s;
		QString inParens = s.length() > 2 ? s.mid(sp, s.length()-(sp+ep)) : "";
		ret.push_back(ChanMapDesc::fromString(inParens));
	}
	return ret;
}

/*static*/ ChanMap ChanMap::fromWSDelimFlatFileString(const QString & s_in)
{
	const QStringList sl = s_in.split("\n", QString::SkipEmptyParts);
	ChanMap ret;
	ret.reserve(sl.size());

	foreach(QString s, sl) {
		QStringList sl2 = s.split(QRegExp("\\s+"), QString::SkipEmptyParts);
		if (sl2.size() >= 3) {
			ChanMapDesc desc = ChanMapDesc::fromWSDelimString(s);
			ret.push_back(desc);
		}
	}
	return ret;
}

void ChanMap::scrambleToPreJuly2011Demux()
{
	// sort it by intan then by channel..
	QMap<QString, ChanMapDesc> m;
	for (int i = 0; i < (int)size(); ++i) {
		ChanMapDesc & d = (*this)[i];
		m[QString().sprintf("%03d_%03d",int(d.intanCh),int(d.intan))] = d;
	}
	int i = 0;
	for (QMap<QString, ChanMapDesc>::iterator it = m.begin(); it != m.end(); ++it, ++i) {
		(*this)[i] = it.value();
	}
}