~mhall119/ubuntu-app-reviews/qtranscribe

/*
 *  Copyright © 2012 Matt Pharoah (mr.exuberant@gmail.com)
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#ifndef FOOTPEDALWIDGETMODELS_HH_
#define FOOTPEDALWIDGETMODELS_HH_

#include <QAbstractListModel>
#include <QAbstractTableModel>
#include <QVariant>
#include <QString>
#include <mutex>
#include <stdexcept>
#include <cstdio>
#include <cstring>
#include <bitset>
#include "Joystick.hh"
#include "Actions.hh"

class FootPedalListModel : public QAbstractListModel {
  private:
	char *pedals[10]; //first character of each entry is joystick port + 1, 2nd is number of pedals, remaining characters are device name string
	unsigned char numPedals;
	mutable mutex safety;

  public:
	FootPedalListModel() : QAbstractListModel() {
		numPedals = 0;
		recheck();
	}

	void recheck() {
		safety.lock();
		//clear old data
		for (int i = 0; i < numPedals; i++) {
			delete[] pedals[i];
		}
		numPedals = 0;
		//add new data
		for (unsigned char i = 0; i <= 9; i++) {
			char *info = new char[258];
			info[0] = i + 1;
			try {
				Joystick::getNameAndButtons(i, &info[2], 256, (unsigned char *) &info[1]);
				pedals[numPedals++] = info;
			} catch (invalid_argument &e) {
				delete[] info;
			}
		}
		safety.unlock();
		emit dataChanged(index(0,0), index(9,0));
	}

	QModelIndex find(char *ID) {
		for (short i = 0; i < numPedals; i++) {
			if (strcmp(pedals[i], ID) == 0) return index(i, 0);
		}
		return QModelIndex();
	}

	short getButtons(int index) {
		safety.lock();
		short ret;
		if (index >= 0 && index < numPedals) {
			ret = (short) (unsigned char) pedals[index][1]; //double cast so it's unsigned... not that there will ever be 128 or more pedals anyways
		} else {
			ret = -1;
		}
		safety.unlock();
		return ret;
	}

	unsigned char getPort(int index) {
		safety.lock();
		unsigned char ret;
		if (index >= 0 && index < numPedals) {
			ret = ((unsigned char) pedals[index][0]) - 1; //double cast so it's unsigned... not that there will ever be 128 or more pedals anyways
		} else {
			ret = 0xff;
		}
		safety.unlock();
		return ret;
	}

	void getStringID(int index, char *dest) {
		safety.lock();
		if (index >= 0 && index < numPedals) {
			strcpy(dest, pedals[index]);
		} else {
			strcpy(dest, "");
		}
		safety.unlock();
	}

	bool stringIDequals(int index, char *compareTo) {
		bool ret;
		safety.lock();
		if (index >= 0 && index < numPedals) {
			ret = (strcmp(pedals[index], compareTo) == 0);
		} else {
			ret = (strcmp(pedals[index], "") == 0);
		}
		safety.unlock();
		return ret;
	}

	int rowCount(const QModelIndex &parent) const {
		return numPedals;
	}

	QVariant data(const QModelIndex &index, int role) const {
		QVariant ret;
		safety.lock();
		if (role == Qt::DisplayRole) {
			const register int i = index.row();
			if (i >= 0 && i < numPedals) {
				ret = QVariant(QString(&pedals[i][2]));
			} else {
				ret = QVariant(QString());
			}
		} else {
			ret = QVariant();
		}
		safety.unlock();
		return ret;
	}

	~FootPedalListModel() {
		safety.lock(); //shouldn't be accessing it after delete anyways
		for (int i = 0; i < numPedals; i++) {
			delete[] pedals[i];
		}
	}
  protected:
	void timerEvent(QTimerEvent *ev) {
		recheck();
	}
};

class PedalCommandTable : public QAbstractTableModel {
  private:
	Joystick *currentPedal;
	unsigned char numButtons;
	unsigned char pedalID;
	mutable mutex safety;
	bitset<256> pedalState;
	Action commands[256];

	void updateCell(unsigned char row, unsigned char col) {
		emit(dataChanged(index(row, col), index(row, col)));
	}

	static void pedalEventHandler(const JSEvent &ev, void *ptr) {
		PedalCommandTable *me = (PedalCommandTable *) ptr;
		if (ev.type == Joystick::BUTTON_EVENT && (ev.value == Joystick::PRESSED || ev.value == Joystick::RELEASED)) {
			me->safety.lock();
			me->pedalState[ev.button] = (ev.value == Joystick::PRESSED);
			me->safety.unlock();
			me->updateCell(ev.button, 2);
		}
	}

  public:
	PedalCommandTable() : QAbstractTableModel() {
		currentPedal = NULL;
		pedalID = 0xff;
		numButtons = 0;
		for (short i = 0; i < 256; i++) commands[i] = A_NOOP;
	}

	void changeJoystick(unsigned char jsnum, unsigned char nb) {
		safety.lock();
		bool REMOVE = (numButtons > 0);
		if (REMOVE) beginRemoveRows(QModelIndex(), 0, numButtons - 1);
		pedalID = jsnum;
		numButtons = nb;
		if (currentPedal != NULL) {
			delete currentPedal;
			currentPedal = NULL;
		}
		pedalState.reset();
		if (REMOVE) endRemoveRows();
		if (numButtons > 0 && jsnum <= 9) beginInsertRows(QModelIndex(), 0, numButtons - 1);
		if (jsnum <= 9) currentPedal = new Joystick(jsnum, pedalEventHandler, NULL, this);
		if (numButtons > 0 && jsnum <= 9) endInsertRows();
		safety.unlock();
		emit(dataChanged(index(0,0), index(255, 2)));
	}

	void changeCommand(const QModelIndex &index, Action cmd) {
		commands[index.row()] = cmd;
		updateCell(index.row(), 1);
	}

	void changeCommand(int index, Action cmd) {
		commands[index] = cmd;
		updateCell(index, 1);
	}

	inline Action getCommand(const QModelIndex &index) {
		return commands[index.row()];
	}

	inline Action getCommand(int index) {
		return commands[index];
	}

	int rowCount(const QModelIndex &parent = QModelIndex()) const {
		return numButtons;
	}

	int columnCount(const QModelIndex &parent) const {
		return 3;
	}

	QVariant headerData(int col, Qt::Orientation orientation, int role) const {
		if (role != Qt::DisplayRole || orientation != Qt::Orientation::Horizontal) return QVariant();
		switch (col) {
			case 0: return QVariant(QString("Button"));
			case 1: return QVariant(QString("Command"));
			case 2: return QVariant(QString("Pressed?"));
			default: return QVariant();
		}
	}

	QVariant data(const QModelIndex &index, int role) const {
		if (role != Qt::DisplayRole) return QVariant();
		safety.lock();
		QVariant ret;
		if (pedalID <= 9 && index.row() >= 0 && index.row() < rowCount() && index.column() <= 256) {
			switch (index.column()) {
				case 0: {
					char str[10];
					sprintf(str, "Pedal %d", index.row() + 1);
					ret = QVariant(QString(str));
				}; break;
				case 1: {
					ret = QVariant(QString(ACTION_STRING(commands[index.row()])));
				}; break;
				case 2: {
					if (pedalState[index.row()]) ret = QVariant(QString("YES"));
				}; break;
			}
		}
		safety.unlock();
		return ret;
	}

	~PedalCommandTable() {
		safety.lock(); //shouldn't be accessing it after delete anyways
		if (currentPedal != NULL) delete currentPedal;
	}
};


#endif /* FOOTPEDALWIDGETMODELS_HH_ */