]> git.zerfleddert.de Git - proxmark3-svn/blob - client/proxguiqt.cpp
4e57fecc7ff8754cb46f30f84d64b2decc6daee3
[proxmark3-svn] / client / proxguiqt.cpp
1 //-----------------------------------------------------------------------------
2 // Copyright (C) 2009 Michael Gernoth <michael at gernoth.net>
3 //
4 // This code is licensed to you under the terms of the GNU GPL, version 2 or,
5 // at your option, any later version. See the LICENSE.txt file for the text of
6 // the license.
7 //-----------------------------------------------------------------------------
8 // GUI (QT)
9 //-----------------------------------------------------------------------------
10
11 #include <iostream>
12 #include <QPainterPath>
13 #include <QBrush>
14 #include <QPen>
15 #include <QTimer>
16 #include <QCloseEvent>
17 #include <QMouseEvent>
18 #include <QKeyEvent>
19 #include <math.h>
20 #include <limits.h>
21 #include <stdio.h>
22 #include "proxguiqt.h"
23 #include "proxgui.h"
24
25 int GridOffset= 0;
26 bool GridLocked= 0;
27 int startMax;
28
29 void ProxGuiQT::ShowGraphWindow(void)
30 {
31 emit ShowGraphWindowSignal();
32 }
33
34 void ProxGuiQT::RepaintGraphWindow(void)
35 {
36 emit RepaintGraphWindowSignal();
37 }
38
39 void ProxGuiQT::HideGraphWindow(void)
40 {
41 emit HideGraphWindowSignal();
42 }
43
44 void ProxGuiQT::_ShowGraphWindow(void)
45 {
46 if(!plotapp)
47 return;
48
49 if (!plotwidget)
50 plotwidget = new ProxWidget();
51
52 plotwidget->show();
53 }
54
55 void ProxGuiQT::_RepaintGraphWindow(void)
56 {
57 if (!plotapp || !plotwidget)
58 return;
59
60 plotwidget->update();
61 }
62
63 void ProxGuiQT::_HideGraphWindow(void)
64 {
65 if (!plotapp || !plotwidget)
66 return;
67
68 plotwidget->hide();
69 }
70
71 void ProxGuiQT::MainLoop()
72 {
73 plotapp = new QApplication(argc, argv);
74
75 connect(this, SIGNAL(ShowGraphWindowSignal()), this, SLOT(_ShowGraphWindow()));
76 connect(this, SIGNAL(RepaintGraphWindowSignal()), this, SLOT(_RepaintGraphWindow()));
77 connect(this, SIGNAL(HideGraphWindowSignal()), this, SLOT(_HideGraphWindow()));
78
79 plotapp->exec();
80 }
81
82 ProxGuiQT::ProxGuiQT(int argc, char **argv) : plotapp(NULL), plotwidget(NULL),
83 argc(argc), argv(argv)
84 {
85 }
86
87 ProxGuiQT::~ProxGuiQT(void)
88 {
89 if (plotwidget) {
90 delete plotwidget;
91 plotwidget = NULL;
92 }
93
94 if (plotapp) {
95 plotapp->quit();
96 delete plotapp;
97 plotapp = NULL;
98 }
99 }
100
101 void ProxWidget::paintEvent(QPaintEvent *event)
102 {
103 QPainter painter(this);
104 QPainterPath penPath, whitePath, greyPath, lightgreyPath, cursorAPath, cursorBPath;
105 QRect r;
106 QBrush brush(QColor(100, 255, 100));
107 QPen pen(QColor(100, 255, 100));
108
109 painter.setFont(QFont("Arial", 10));
110
111 if(GraphStart < 0) {
112 GraphStart = 0;
113 }
114
115 if (CursorAPos > GraphTraceLen)
116 CursorAPos= 0;
117 if(CursorBPos > GraphTraceLen)
118 CursorBPos= 0;
119
120 r = rect();
121
122 painter.fillRect(r, QColor(0, 0, 0));
123
124 whitePath.moveTo(r.left() + 40, r.top());
125 whitePath.lineTo(r.left() + 40, r.bottom());
126
127 int zeroHeight = r.top() + (r.bottom() - r.top()) / 2;
128
129 greyPath.moveTo(r.left(), zeroHeight);
130 greyPath.lineTo(r.right(), zeroHeight);
131 painter.setPen(QColor(100, 100, 100));
132 painter.drawPath(greyPath);
133
134 // plot X and Y grid lines
135 int i;
136 if ((PlotGridX > 0) && ((PlotGridX * GraphPixelsPerPoint) > 1)) {
137 for(i = 40 + (GridOffset * GraphPixelsPerPoint); i < r.right(); i += (int)(PlotGridX * GraphPixelsPerPoint)) {
138 //SelectObject(hdc, GreyPenLite);
139 //MoveToEx(hdc, r.left + i, r.top, NULL);
140 //LineTo(hdc, r.left + i, r.bottom);
141 lightgreyPath.moveTo(r.left()+i,r.top());
142 lightgreyPath.lineTo(r.left()+i,r.bottom());
143 painter.drawPath(lightgreyPath);
144 }
145 }
146 if ((PlotGridY > 0) && ((PlotGridY * GraphPixelsPerPoint) > 1)){
147 for(i = 0; i < ((r.top() + r.bottom())>>1); i += (int)(PlotGridY * GraphPixelsPerPoint)) {
148 lightgreyPath.moveTo(r.left() + 40,zeroHeight + i);
149 lightgreyPath.lineTo(r.right(),zeroHeight + i);
150 painter.drawPath(lightgreyPath);
151 lightgreyPath.moveTo(r.left() + 40,zeroHeight - i);
152 lightgreyPath.lineTo(r.right(),zeroHeight - i);
153 painter.drawPath(lightgreyPath);
154 }
155 }
156
157 startMax = (GraphTraceLen - (int)((r.right() - r.left() - 40) / GraphPixelsPerPoint));
158 if(startMax < 0) {
159 startMax = 0;
160 }
161 if(GraphStart > startMax) {
162 GraphStart = startMax;
163 }
164
165 int absYMax = 1;
166
167 for(i = GraphStart; ; i++) {
168 if(i >= GraphTraceLen) {
169 break;
170 }
171 if(fabs((double)GraphBuffer[i]) > absYMax) {
172 absYMax = (int)fabs((double)GraphBuffer[i]);
173 }
174 int x = 40 + (int)((i - GraphStart)*GraphPixelsPerPoint);
175 if(x > r.right()) {
176 break;
177 }
178 }
179
180 absYMax = (int)(absYMax*1.2 + 1);
181
182 // number of points that will be plotted
183 int span = (int)((r.right() - r.left()) / GraphPixelsPerPoint);
184 // one label every 100 pixels, let us say
185 int labels = (r.right() - r.left() - 40) / 100;
186 if(labels <= 0) labels = 1;
187 int pointsPerLabel = span / labels;
188 if(pointsPerLabel <= 0) pointsPerLabel = 1;
189
190 int yMin = INT_MAX;
191 int yMax = INT_MIN;
192 int yMean = 0;
193 int n = 0;
194
195 for(i = GraphStart; ; i++) {
196 if(i >= GraphTraceLen) {
197 break;
198 }
199 int x = 40 + (int)((i - GraphStart)*GraphPixelsPerPoint);
200 if(x > r.right() + GraphPixelsPerPoint) {
201 break;
202 }
203
204 int y = GraphBuffer[i];
205 if(y < yMin) {
206 yMin = y;
207 }
208 if(y > yMax) {
209 yMax = y;
210 }
211 yMean += y;
212 n++;
213
214 y = (y * (r.top() - r.bottom()) / (2*absYMax)) + zeroHeight;
215 if(i == GraphStart) {
216 penPath.moveTo(x, y);
217 } else {
218 penPath.lineTo(x, y);
219 }
220
221 if(GraphPixelsPerPoint > 10) {
222 QRect f(QPoint(x - 3, y - 3),QPoint(x + 3, y + 3));
223 painter.fillRect(f, brush);
224 }
225
226 if(((i - GraphStart) % pointsPerLabel == 0) && i != GraphStart) {
227 whitePath.moveTo(x, zeroHeight - 3);
228 whitePath.lineTo(x, zeroHeight + 3);
229
230 char str[100];
231 sprintf(str, "+%d", (i - GraphStart));
232
233 painter.setPen(QColor(255, 255, 255));
234 QRect size;
235 QFontMetrics metrics(painter.font());
236 size = metrics.boundingRect(str);
237 painter.drawText(x - (size.right() - size.left()), zeroHeight + 9, str);
238
239 penPath.moveTo(x,y);
240 }
241
242 if(i == CursorAPos || i == CursorBPos) {
243 QPainterPath *cursorPath;
244
245 if(i == CursorAPos) {
246 cursorPath = &cursorAPath;
247 } else {
248 cursorPath = &cursorBPath;
249 }
250 cursorPath->moveTo(x, r.top());
251 cursorPath->lineTo(x, r.bottom());
252 penPath.moveTo(x, y);
253 }
254 }
255
256 if(n != 0) {
257 yMean /= n;
258 }
259
260 painter.setPen(QColor(255, 255, 255));
261 painter.drawPath(whitePath);
262 painter.setPen(pen);
263 painter.drawPath(penPath);
264 painter.setPen(QColor(255, 255, 0));
265 painter.drawPath(cursorAPath);
266 painter.setPen(QColor(255, 0, 255));
267 painter.drawPath(cursorBPath);
268
269 char str[200];
270 sprintf(str, "@%d max=%d min=%d mean=%d n=%d/%d dt=%d [%.3f] zoom=%.3f CursorA=%d [%d] CursorB=%d [%d]",
271 GraphStart, yMax, yMin, yMean, n, GraphTraceLen,
272 CursorBPos - CursorAPos, (CursorBPos - CursorAPos)/CursorScaleFactor,GraphPixelsPerPoint,CursorAPos,GraphBuffer[CursorAPos],CursorBPos,GraphBuffer[CursorBPos]);
273
274 painter.setPen(QColor(255, 255, 255));
275 painter.drawText(50, r.bottom() - 20, str);
276 }
277
278 ProxWidget::ProxWidget(QWidget *parent) : QWidget(parent), GraphStart(0), GraphPixelsPerPoint(1)
279 {
280 resize(600, 500);
281
282 QPalette palette(QColor(0,0,0,0));
283 palette.setColor(QPalette::WindowText, QColor(255,255,255));
284 palette.setColor(QPalette::Text, QColor(255,255,255));
285 palette.setColor(QPalette::Button, QColor(100, 100, 100));
286 setPalette(palette);
287 setAutoFillBackground(true);
288 }
289
290 void ProxWidget::closeEvent(QCloseEvent *event)
291 {
292 event->ignore();
293 this->hide();
294 }
295
296 void ProxWidget::mouseMoveEvent(QMouseEvent *event)
297 {
298 int x = event->x();
299 x -= 40;
300 x = (int)(x / GraphPixelsPerPoint);
301 x += GraphStart;
302 if((event->buttons() & Qt::LeftButton)) {
303 CursorAPos = x;
304 } else if (event->buttons() & Qt::RightButton) {
305 CursorBPos = x;
306 }
307
308
309 this->update();
310 }
311
312 void ProxWidget::keyPressEvent(QKeyEvent *event)
313 {
314 switch(event->key()) {
315 case Qt::Key_Down:
316 if(GraphPixelsPerPoint <= 50) {
317 GraphPixelsPerPoint *= 2;
318 }
319 break;
320
321 case Qt::Key_Up:
322 if(GraphPixelsPerPoint >= 0.02) {
323 GraphPixelsPerPoint /= 2;
324 }
325 break;
326
327 case Qt::Key_Right:
328 if(GraphPixelsPerPoint < 20) {
329 if (GridLocked && GraphStart < startMax)
330 GridOffset -= (int)(20 / GraphPixelsPerPoint);
331 GraphStart += (int)(20 / GraphPixelsPerPoint);
332 } else {
333 if (GridLocked && GraphStart < startMax)
334 GridOffset--;
335 GraphStart++;
336 }
337 if(GridOffset < 0)
338 GridOffset += PlotGridX;
339 if (PlotGridX)
340 GridOffset %= PlotGridX;
341 break;
342
343 case Qt::Key_Left:
344 if(GraphPixelsPerPoint < 20) {
345 if (GridLocked && GraphStart > 0)
346 GridOffset += (int)(20 / GraphPixelsPerPoint);
347 GraphStart -= (int)(20 / GraphPixelsPerPoint);
348 } else {
349 if (GridLocked && GraphStart > 0)
350 GridOffset++;
351 GraphStart--;
352 }
353 if (PlotGridX)
354 GridOffset %= PlotGridX;
355 break;
356
357 case Qt::Key_G:
358 if(PlotGridX || PlotGridY) {
359 PlotGridX= 0;
360 PlotGridY= 0;
361 } else {
362 PlotGridX= PlotGridXdefault;
363 PlotGridY= PlotGridYdefault;
364 }
365 break;
366
367 case Qt::Key_H:
368 puts("Plot Window Keystrokes:\n");
369 puts(" Key Action\n");
370 puts(" DOWN Zoom in");
371 puts(" G Toggle grid display");
372 puts(" H Show help");
373 puts(" LEFT Move left");
374 puts(" L Toggle lock grid relative to samples");
375 puts(" Q Hide window");
376 puts(" RIGHT Move right");
377 puts(" UP Zoom out");
378 puts("");
379 puts("Use client window 'data help' for more plot commands\n");
380 break;
381
382 case Qt::Key_L:
383 GridLocked= !GridLocked;
384 break;
385
386 case Qt::Key_Q:
387 this->hide();
388 break;
389
390 default:
391 QWidget::keyPressEvent(event);
392 return;
393 break;
394 }
395
396 this->update();
397 }
Impressum, Datenschutz