[proxmark3-svn] / linux / proxguiqt.cpp

#include <iostream>
#include <QPainterPath>
#include <QBrush>
#include <QPen>
#include <QTimer>
#include <QCloseEvent>
#include <QMouseEvent>
#include <QKeyEvent>
#include <math.h>
#include <limits.h>
#include "proxguiqt.h"
#include "proxgui.h"

void ProxGuiQT::ShowGraphWindow(void)
{
	emit ShowGraphWindowSignal();
}

void ProxGuiQT::RepaintGraphWindow(void)
{
	emit RepaintGraphWindowSignal();
}

void ProxGuiQT::HideGraphWindow(void)
{
	emit HideGraphWindowSignal();
}

void ProxGuiQT::_ShowGraphWindow(void)
{
	if(!plotapp)
		return;

	if (!plotwidget)
		plotwidget = new ProxWidget();

	plotwidget->show();
}

void ProxGuiQT::_RepaintGraphWindow(void)
{
	if (!plotapp || !plotwidget)
		return;

	plotwidget->update();
}

void ProxGuiQT::_HideGraphWindow(void)
{
	if (!plotapp || !plotwidget)
		return;

	plotwidget->hide();
}

void ProxGuiQT::MainLoop()
{
	plotapp = new QApplication(argc, argv);

	connect(this, SIGNAL(ShowGraphWindowSignal()), this, SLOT(_ShowGraphWindow()));
	connect(this, SIGNAL(RepaintGraphWindowSignal()), this, SLOT(_RepaintGraphWindow()));
	connect(this, SIGNAL(HideGraphWindowSignal()), this, SLOT(_HideGraphWindow()));

	plotapp->exec();
}

ProxGuiQT::ProxGuiQT(int argc, char **argv) : plotapp(NULL), plotwidget(NULL),
	argc(argc), argv(argv)
{
}

ProxGuiQT::~ProxGuiQT(void)
{
	if (plotwidget) {
		delete plotwidget;
		plotwidget = NULL;
	}

	if (plotapp) {
		plotapp->quit();
		delete plotapp;
		plotapp = NULL;
	}
}

void ProxWidget::paintEvent(QPaintEvent *event)
{
	QPainter painter(this);
	QPainterPath penPath, whitePath, greyPath, cursorAPath, cursorBPath;
	QRect r;
	QBrush brush(QColor(100, 255, 100));
	QPen pen(QColor(100, 255, 100));

	painter.setFont(QFont("Arial", 10));

	if(GraphStart < 0) {
		GraphStart = 0;
	}

	r = rect();

	painter.fillRect(r, QColor(0, 0, 0));

	whitePath.moveTo(r.left() + 40, r.top());
	whitePath.lineTo(r.left() + 40, r.bottom());

	int zeroHeight = r.top() + (r.bottom() - r.top()) / 2;

	greyPath.moveTo(r.left(), zeroHeight);
	greyPath.lineTo(r.right(), zeroHeight);
	painter.setPen(QColor(100, 100, 100));
	painter.drawPath(greyPath);
	
	int startMax =
		(GraphTraceLen - (int)((r.right() - r.left() - 40) / GraphPixelsPerPoint));
	if(startMax < 0) {
		startMax = 0;
	}
	if(GraphStart > startMax) {
		GraphStart = startMax;
	}

	int absYMax = 1;

	int i;
	for(i = GraphStart; ; i++) {
		if(i >= GraphTraceLen) {
			break;
		}
		if(fabs((double)GraphBuffer[i]) > absYMax) {
			absYMax = (int)fabs((double)GraphBuffer[i]);
		}
		int x = 40 + (int)((i - GraphStart)*GraphPixelsPerPoint);
		if(x > r.right()) {
			break;
		}
	}

	absYMax = (int)(absYMax*1.2 + 1);
	
	// number of points that will be plotted
	int span = (int)((r.right() - r.left()) / GraphPixelsPerPoint);
	// one label every 100 pixels, let us say
	int labels = (r.right() - r.left() - 40) / 100;
	if(labels <= 0) labels = 1;
	int pointsPerLabel = span / labels;
	if(pointsPerLabel <= 0) pointsPerLabel = 1;

	int yMin = INT_MAX;
	int yMax = INT_MIN;
	int yMean = 0;
	int n = 0;

	for(i = GraphStart; ; i++) {
		if(i >= GraphTraceLen) {
			break;
		}
		int x = 40 + (int)((i - GraphStart)*GraphPixelsPerPoint);
		if(x > r.right() + GraphPixelsPerPoint) {
			break;
		}

		int y = GraphBuffer[i];
		if(y < yMin) {
			yMin = y;
		}
		if(y > yMax) {
			yMax = y;
		}
		yMean += y;
		n++;

		y = (y * (r.top() - r.bottom()) / (2*absYMax)) + zeroHeight;
		if(i == GraphStart) {
			penPath.moveTo(x, y);
		} else {
			penPath.lineTo(x, y);
		}

		if(GraphPixelsPerPoint > 10) {
			QRect f(QPoint(x - 3, y - 3),QPoint(x + 3, y + 3));
			painter.fillRect(f, brush);
		}

		if(((i - GraphStart) % pointsPerLabel == 0) && i != GraphStart) {
			whitePath.moveTo(x, zeroHeight - 3);
			whitePath.lineTo(x, zeroHeight + 3);

			char str[100];
			sprintf(str, "+%d", (i - GraphStart));

			painter.setPen(QColor(255, 255, 255));
			QRect size;
			QFontMetrics metrics(painter.font());
			size = metrics.boundingRect(str);
			painter.drawText(x - (size.right() - size.left()), zeroHeight + 9, str);

			penPath.moveTo(x,y);
		}

		if(i == CursorAPos || i == CursorBPos) {
			QPainterPath *cursorPath;

			if(i == CursorAPos) {
				cursorPath = &cursorAPath;
			} else {
				cursorPath = &cursorBPath;
			}
			cursorPath->moveTo(x, r.top());
			cursorPath->lineTo(x, r.bottom());
			penPath.moveTo(x, y);
		}
	}

	if(n != 0) {
		yMean /= n;
	}

	painter.setPen(QColor(255, 255, 255));
	painter.drawPath(whitePath);
	painter.setPen(pen);
	painter.drawPath(penPath);
	painter.setPen(QColor(255, 255, 0));
	painter.drawPath(cursorAPath);
	painter.setPen(QColor(255, 0, 255));
	painter.drawPath(cursorBPath);

	char str[100];
	sprintf(str, "@%d   max=%d min=%d mean=%d n=%d/%d    dt=%d [%.3f]",
			GraphStart, yMax, yMin, yMean, n, GraphTraceLen,
			CursorBPos - CursorAPos, (CursorBPos - CursorAPos)/CursorScaleFactor);

	painter.setPen(QColor(255, 255, 255));
	painter.drawText(50, r.bottom() - 20, str);
}

ProxWidget::ProxWidget(QWidget *parent) : QWidget(parent), GraphStart(0), GraphPixelsPerPoint(1)
{
	resize(600, 500);

	QPalette palette(QColor(0,0,0,0));
	palette.setColor(QPalette::WindowText, QColor(255,255,255));
	palette.setColor(QPalette::Text, QColor(255,255,255));
	palette.setColor(QPalette::Button, QColor(100, 100, 100));
	setPalette(palette);
	setAutoFillBackground(true);
}

void ProxWidget::closeEvent(QCloseEvent *event)
{
	event->ignore();
	this->hide();
}

void ProxWidget::mouseMoveEvent(QMouseEvent *event)
{
	int x = event->x();
	x -= 40;
	x = (int)(x / GraphPixelsPerPoint);
	x += GraphStart;
	if((event->buttons() & Qt::LeftButton)) {
		CursorAPos = x;
	} else if (event->buttons() & Qt::RightButton) {
		CursorBPos = x;
	}


	this->update();
}

void ProxWidget::keyPressEvent(QKeyEvent *event)
{
	switch(event->key()) {
		case Qt::Key_Down:
			if(GraphPixelsPerPoint <= 50) {
				GraphPixelsPerPoint *= 2;
			}
			break;

		case Qt::Key_Up:
			if(GraphPixelsPerPoint >= 0.02) {
				GraphPixelsPerPoint /= 2;
			}
			break;

		case Qt::Key_Right:
			if(GraphPixelsPerPoint < 20) {
				GraphStart += (int)(20 / GraphPixelsPerPoint);
			} else {
				GraphStart++;
			}
			break;

		case Qt::Key_Left:
			if(GraphPixelsPerPoint < 20) {
				GraphStart -= (int)(20 / GraphPixelsPerPoint);
			} else {
				GraphStart--;
			}
			break;

		default:
			QWidget::keyPressEvent(event);
			return;
			break;
	}

	this->update();
}
Commit	Line	Data
6658905f	1	#include <iostream>
	2	#include <QPainterPath>
	3	#include <QBrush>
	4	#include <QPen>
	5	#include <QTimer>
	6	#include <QCloseEvent>
	7	#include <QMouseEvent>
	8	#include <QKeyEvent>
	9	#include <math.h>
	10	#include <limits.h>
	11	#include "proxguiqt.h"
	12	#include "proxgui.h"
	13
	14	void ProxGuiQT::ShowGraphWindow(void)
	15	{
	16	emit ShowGraphWindowSignal();
	17	}
	18
	19	void ProxGuiQT::RepaintGraphWindow(void)
	20	{
	21	emit RepaintGraphWindowSignal();
	22	}
	23
	24	void ProxGuiQT::HideGraphWindow(void)
	25	{
	26	emit HideGraphWindowSignal();
	27	}
	28
	29	void ProxGuiQT::_ShowGraphWindow(void)
	30	{
	31	if(!plotapp)
	32	return;
	33
	34	if (!plotwidget)
	35	plotwidget = new ProxWidget();
	36
	37	plotwidget->show();
	38	}
	39
	40	void ProxGuiQT::_RepaintGraphWindow(void)
	41	{
	42	if (!plotapp \|\| !plotwidget)
	43	return;
	44
	45	plotwidget->update();
	46	}
	47
	48	void ProxGuiQT::_HideGraphWindow(void)
	49	{
	50	if (!plotapp \|\| !plotwidget)
	51	return;
	52
	53	plotwidget->hide();
	54	}
	55
	56	void ProxGuiQT::MainLoop()
	57	{
	58	plotapp = new QApplication(argc, argv);
	59
	60	connect(this, SIGNAL(ShowGraphWindowSignal()), this, SLOT(_ShowGraphWindow()));
	61	connect(this, SIGNAL(RepaintGraphWindowSignal()), this, SLOT(_RepaintGraphWindow()));
	62	connect(this, SIGNAL(HideGraphWindowSignal()), this, SLOT(_HideGraphWindow()));
	63
	64	plotapp->exec();
65	}
66
67	ProxGuiQT::ProxGuiQT(int argc, char **argv) : plotapp(NULL), plotwidget(NULL),
68	argc(argc), argv(argv)
69	{
70	}
71
72	ProxGuiQT::~ProxGuiQT(void)
73	{
74	if (plotwidget) {
75	delete plotwidget;
76	plotwidget = NULL;
77	}
78
79	if (plotapp) {
80	plotapp->quit();
81	delete plotapp;
82	plotapp = NULL;
83	}
84	}
85
86	void ProxWidget::paintEvent(QPaintEvent *event)
87	{
88	QPainter painter(this);
89	QPainterPath penPath, whitePath, greyPath, cursorAPath, cursorBPath;
90	QRect r;
91	QBrush brush(QColor(100, 255, 100));
92	QPen pen(QColor(100, 255, 100));
93
94	painter.setFont(QFont("Arial", 10));
95
96	if(GraphStart < 0) {
97	GraphStart = 0;
98	}
99
100	r = rect();
101
102	painter.fillRect(r, QColor(0, 0, 0));
103
104	whitePath.moveTo(r.left() + 40, r.top());
105	whitePath.lineTo(r.left() + 40, r.bottom());
106
107	int zeroHeight = r.top() + (r.bottom() - r.top()) / 2;
108
109	greyPath.moveTo(r.left(), zeroHeight);
110	greyPath.lineTo(r.right(), zeroHeight);
111	painter.setPen(QColor(100, 100, 100));
112	painter.drawPath(greyPath);
113
114	int startMax =
115	(GraphTraceLen - (int)((r.right() - r.left() - 40) / GraphPixelsPerPoint));
116	if(startMax < 0) {
117	startMax = 0;
118	}
119	if(GraphStart > startMax) {
120	GraphStart = startMax;
121	}
122
123	int absYMax = 1;
124
125	int i;
126	for(i = GraphStart; ; i++) {
127	if(i >= GraphTraceLen) {
128	break;
129	}
130	if(fabs((double)GraphBuffer[i]) > absYMax) {
131	absYMax = (int)fabs((double)GraphBuffer[i]);
132	}
133	int x = 40 + (int)((i - GraphStart)*GraphPixelsPerPoint);
134	if(x > r.right()) {
135	break;
136	}
137	}
138
139	absYMax = (int)(absYMax*1.2 + 1);
140
141	// number of points that will be plotted
142	int span = (int)((r.right() - r.left()) / GraphPixelsPerPoint);
143	// one label every 100 pixels, let us say
144	int labels = (r.right() - r.left() - 40) / 100;
145	if(labels <= 0) labels = 1;
146	int pointsPerLabel = span / labels;
147	if(pointsPerLabel <= 0) pointsPerLabel = 1;
148
149	int yMin = INT_MAX;
150	int yMax = INT_MIN;
151	int yMean = 0;
152	int n = 0;
153
154	for(i = GraphStart; ; i++) {
155	if(i >= GraphTraceLen) {
156	break;
157	}
158	int x = 40 + (int)((i - GraphStart)*GraphPixelsPerPoint);
159	if(x > r.right() + GraphPixelsPerPoint) {
160	break;
161	}
162
163	int y = GraphBuffer[i];
164	if(y < yMin) {
165	yMin = y;
166	}
167	if(y > yMax) {
168	yMax = y;
169	}
170	yMean += y;
171	n++;
172
173	y = (y * (r.top() - r.bottom()) / (2*absYMax)) + zeroHeight;
174	if(i == GraphStart) {
175	penPath.moveTo(x, y);
176	} else {
177	penPath.lineTo(x, y);
178	}
179
180	if(GraphPixelsPerPoint > 10) {
181	QRect f(QPoint(x - 3, y - 3),QPoint(x + 3, y + 3));
182	painter.fillRect(f, brush);
183	}
184
185	if(((i - GraphStart) % pointsPerLabel == 0) && i != GraphStart) {
186	whitePath.moveTo(x, zeroHeight - 3);
187	whitePath.lineTo(x, zeroHeight + 3);
188
189	char str[100];
190	sprintf(str, "+%d", (i - GraphStart));
191
192	painter.setPen(QColor(255, 255, 255));
193	QRect size;
194	QFontMetrics metrics(painter.font());
195	size = metrics.boundingRect(str);
196	painter.drawText(x - (size.right() - size.left()), zeroHeight + 9, str);
197
198	penPath.moveTo(x,y);
199	}
200
201	if(i == CursorAPos \|\| i == CursorBPos) {
202	QPainterPath *cursorPath;
203
204	if(i == CursorAPos) {
205	cursorPath = &cursorAPath;
206	} else {
207	cursorPath = &cursorBPath;
208	}
209	cursorPath->moveTo(x, r.top());
210	cursorPath->lineTo(x, r.bottom());
211	penPath.moveTo(x, y);
212	}
213	}
214
215	if(n != 0) {
216	yMean /= n;
217	}
218
219	painter.setPen(QColor(255, 255, 255));
220	painter.drawPath(whitePath);
221	painter.setPen(pen);
222	painter.drawPath(penPath);
223	painter.setPen(QColor(255, 255, 0));
224	painter.drawPath(cursorAPath);
225	painter.setPen(QColor(255, 0, 255));
226	painter.drawPath(cursorBPath);
227
228	char str[100];
229	sprintf(str, "@%d max=%d min=%d mean=%d n=%d/%d dt=%d [%.3f]",
230	GraphStart, yMax, yMin, yMean, n, GraphTraceLen,
231	CursorBPos - CursorAPos, (CursorBPos - CursorAPos)/CursorScaleFactor);
232
233	painter.setPen(QColor(255, 255, 255));
234	painter.drawText(50, r.bottom() - 20, str);
235	}
236
237	ProxWidget::ProxWidget(QWidget *parent) : QWidget(parent), GraphStart(0), GraphPixelsPerPoint(1)
238	{
239	resize(600, 500);
240
241	QPalette palette(QColor(0,0,0,0));
242	palette.setColor(QPalette::WindowText, QColor(255,255,255));
243	palette.setColor(QPalette::Text, QColor(255,255,255));
244	palette.setColor(QPalette::Button, QColor(100, 100, 100));
245	setPalette(palette);
246	setAutoFillBackground(true);
247	}
248
249	void ProxWidget::closeEvent(QCloseEvent *event)
250	{
251	event->ignore();
252	this->hide();
253	}
254
255	void ProxWidget::mouseMoveEvent(QMouseEvent *event)
256	{
257	int x = event->x();
258	x -= 40;
259	x = (int)(x / GraphPixelsPerPoint);
260	x += GraphStart;
261	if((event->buttons() & Qt::LeftButton)) {
262	CursorAPos = x;
263	} else if (event->buttons() & Qt::RightButton) {
264	CursorBPos = x;
265	}
266
267
268	this->update();
269	}
270
271	void ProxWidget::keyPressEvent(QKeyEvent *event)
272	{
273	switch(event->key()) {
274	case Qt::Key_Down:
275	if(GraphPixelsPerPoint <= 50) {
276	GraphPixelsPerPoint *= 2;
277	}
278	break;
279
280	case Qt::Key_Up:
281	if(GraphPixelsPerPoint >= 0.02) {
282	GraphPixelsPerPoint /= 2;
283	}
284	break;
285
286	case Qt::Key_Right:
287	if(GraphPixelsPerPoint < 20) {
288	GraphStart += (int)(20 / GraphPixelsPerPoint);
289	} else {
290	GraphStart++;
291	}
292	break;
293
294	case Qt::Key_Left:
295	if(GraphPixelsPerPoint < 20) {
296	GraphStart -= (int)(20 / GraphPixelsPerPoint);
297	} else {
298	GraphStart--;
299	}
300	break;
301
302	default:
303	QWidget::keyPressEvent(event);
304	return;
305	break;
306	}
307
308	this->update();
309	}