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