cvs.zerfleddert.de Git - proxmark3-svn/blob

1 //-----------------------------------------------------------------------------

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 //-----------------------------------------------------------------------------

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"

25 int GridOffset= 0;

26 bool GridLocked= 0;

27 int startMax;

28 int PageWidth;

 void ProxGuiQT::ShowGraphWindow(void)

31 {

32 emit ShowGraphWindowSignal();

33 }

 void ProxGuiQT::RepaintGraphWindow(void)

36 {

37 emit RepaintGraphWindowSignal();

38 }

 void ProxGuiQT::HideGraphWindow(void)

41 {

42 emit HideGraphWindowSignal();

43 }

 void ProxGuiQT::_ShowGraphWindow(void)

46 {

47 if(!plotapp)

48 return;

50 if (!plotwidget)

51 plotwidget = new ProxWidget();

53 plotwidget->show();

54 }

 void ProxGuiQT::_RepaintGraphWindow(void)

57 {

58 if (!plotapp || !plotwidget)

59 return;

61 plotwidget->update();

62 }

 void ProxGuiQT::_HideGraphWindow(void)

65 {

66 if (!plotapp || !plotwidget)

67 return;

69 plotwidget->hide();

70 }

 void ProxGuiQT::MainLoop()

73 {

         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()));

80 plotapp->exec();

81 }

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

         argc(argc), argv(argv)

85 {

86 }

 ProxGuiQT::~ProxGuiQT(void)

89 {

90 if (plotwidget) {

91 delete plotwidget;

92 plotwidget = NULL;

93 }

95 if (plotapp) {

96 plotapp->quit();

97 delete plotapp;

98 plotapp = NULL;

99 }

100 }

101

 void ProxWidget::paintEvent(QPaintEvent *event)

103 {

104 QPainter painter(this);

         QPainterPath penPath, whitePath, greyPath, lightgreyPath, cursorAPath, cursorBPath;

106 QRect r;

         QBrush brush(QColor(100, 255, 100));

         QPen pen(QColor(100, 255, 100));

109

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

111

         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

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

124

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

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

127

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

129

         greyPath.moveTo(r.left(), zeroHeight);

         greyPath.lineTo(r.right(), zeroHeight);

         painter.setPen(QColor(100, 100, 100));

133 painter.drawPath(greyPath);

134

         PageWidth= (int)((r.right() - r.left() - 40) / GraphPixelsPerPoint);

136

137 // plot X and Y grid lines

138 int i;

         if ((PlotGridX > 0) && ((PlotGridX * GraphPixelsPerPoint) > 1)) {

                 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);

                         lightgreyPath.moveTo(r.left()+i,r.top());

                         lightgreyPath.lineTo(r.left()+i,r.bottom());

146 painter.drawPath(lightgreyPath);

147 }

148 }

         if ((PlotGridY > 0) && ((PlotGridY * GraphPixelsPerPoint) > 1)){

                 for(i = 0; i < ((r.top() + r.bottom())>>1); i += (int)(PlotGridY * GraphPixelsPerPoint)) {

                         lightgreyPath.moveTo(r.left() + 40,zeroHeight + i);

                         lightgreyPath.lineTo(r.right(),zeroHeight + i);

153 painter.drawPath(lightgreyPath);

                         lightgreyPath.moveTo(r.left() + 40,zeroHeight - i);

                         lightgreyPath.lineTo(r.right(),zeroHeight - i);

156 painter.drawPath(lightgreyPath);

157 }

158 }

159

         startMax = (GraphTraceLen - (int)((r.right() - r.left() - 40) / GraphPixelsPerPoint));

         if(startMax < 0) {

162 startMax = 0;

163 }

164 if(GraphStart > startMax) {

165 GraphStart = startMax;

166 }

167

168 int absYMax = 1;

169

         for(i = GraphStart; ; i++) {

171 if(i >= GraphTraceLen) {

172 break;

173 }

                 if(fabs((double)GraphBuffer[i]) > absYMax) {

                         absYMax = (int)fabs((double)GraphBuffer[i]);

176 }

                 int x = 40 + (int)((i - GraphStart)*GraphPixelsPerPoint);

                 if(x > r.right()) {

179 break;

180 }

181 }

182

         absYMax = (int)(absYMax*1.2 + 1);

184

185 // number of points that will be plotted

         int span = (int)((r.right() - r.left()) / GraphPixelsPerPoint);

187 // one label every 100 pixels, let us say

         int labels = (r.right() - r.left() - 40) / 100;

         if(labels <= 0) labels = 1;

190 int pointsPerLabel = span / labels;

         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

         for(i = GraphStart; ; i++) {

199 if(i >= GraphTraceLen) {

200 break;

201 }

                 int x = 40 + (int)((i - GraphStart)*GraphPixelsPerPoint);

                 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

                 y = (y * (r.top() - r.bottom()) / (2*absYMax)) + zeroHeight;

218 if(i == GraphStart) {

                         penPath.moveTo(x, y);

220 } else {

                         penPath.lineTo(x, y);

222 }

223

                 if(GraphPixelsPerPoint > 10) {

                         QRect f(QPoint(x - 3, y - 3),QPoint(x + 3, y + 3));

                         painter.fillRect(f, brush);

227 }

228

                 if(((i - GraphStart) % pointsPerLabel == 0) && i != GraphStart) {

                         whitePath.moveTo(x, zeroHeight - 3);

                         whitePath.lineTo(x, zeroHeight + 3);

232

233 char str[100];

                         sprintf(str, "+%d", (i - GraphStart));

235

                         painter.setPen(QColor(255, 255, 255));

237 QRect size;

                         QFontMetrics metrics(painter.font());

                         size = metrics.boundingRect(str);

                         painter.drawText(x - (size.right() - size.left()), zeroHeight + 9, str);

241

                         penPath.moveTo(x,y);

243 }

244

                 if(i == CursorAPos || i == CursorBPos) {

246 QPainterPath *cursorPath;

247

248 if(i == CursorAPos) {

249 cursorPath = &cursorAPath;

250 } else {

251 cursorPath = &cursorBPath;

252 }

                         cursorPath->moveTo(x, r.top());

                         cursorPath->lineTo(x, r.bottom());

                         penPath.moveTo(x, y);

256 }

257 }

258

         if(n != 0) {

260 yMean /= n;

261 }

262

         painter.setPen(QColor(255, 255, 255));

264 painter.drawPath(whitePath);

265 painter.setPen(pen);

266 painter.drawPath(penPath);

         painter.setPen(QColor(255, 255, 0));

268 painter.drawPath(cursorAPath);

         painter.setPen(QColor(255, 0, 255));

270 painter.drawPath(cursorBPath);

271

272 char str[200];

         sprintf(str, "@%d   max=%d min=%d mean=%d n=%d/%d    dt=%d [%.3f] zoom=%.3f CursorA=%d [%d] CursorB=%d [%d]    GridX=%d GridY=%d (%s)",

                         GraphStart, yMax, yMin, yMean, n, GraphTraceLen,

                         CursorBPos - CursorAPos, (CursorBPos - CursorAPos)/CursorScaleFactor,GraphPixelsPerPoint,CursorAPos,GraphBuffer[CursorAPos],CursorBPos,GraphBuffer[CursorBPos],PlotGridXdefault,PlotGridYdefault,GridLocked?"Locked":"Unlocked");

276

         painter.setPen(QColor(255, 255, 255));

         painter.drawText(50, r.bottom() - 20, str);

279 }

280

 ProxWidget::ProxWidget(QWidget *parent) : QWidget(parent), GraphStart(0), GraphPixelsPerPoint(1)

282 {

         resize(600, 300);

284

         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));

289 setPalette(palette);

290 setAutoFillBackground(true);

291 CursorAPos = 0;

292 CursorBPos = 0;

293 }

294

 void ProxWidget::closeEvent(QCloseEvent *event)

296 {

297 event->ignore();

298 this->hide();

299 }

300

 void ProxWidget::mouseMoveEvent(QMouseEvent *event)

302 {

         int x = event->x();

304 x -= 40;

         x = (int)(x / GraphPixelsPerPoint);

306 x += GraphStart;

         if((event->buttons() & Qt::LeftButton)) {

308 CursorAPos = x;

         } else if (event->buttons() & Qt::RightButton) {

310 CursorBPos = x;

311 }

312

313

314 this->update();

315 }

316

 void ProxWidget::keyPressEvent(QKeyEvent *event)

318 {

319 int offset;

320 int gridchanged;

321

322 gridchanged= 0;

323

         if(event->modifiers() & Qt::ShiftModifier) {

325 if (PlotGridX)

326 offset= PageWidth - (PageWidth % PlotGridX);

327 else

328 offset= PageWidth;

329 } else

                 if(event->modifiers() & Qt::ControlModifier)

331 offset= 1;

332 else

                         offset= (int)(20 / GraphPixelsPerPoint);

334

         switch(event->key()) {

336 case Qt::Key_Down:

                         if(GraphPixelsPerPoint <= 50) {

338 GraphPixelsPerPoint *= 2;

339 }

340 break;

341

342 case Qt::Key_Up:

                         if(GraphPixelsPerPoint >= 0.02) {

344 GraphPixelsPerPoint /= 2;

345 }

346 break;

347

348 case Qt::Key_Right:

                         if(GraphPixelsPerPoint < 20) {

                                 if (PlotGridX && GridLocked && GraphStart < startMax){

351 GridOffset -= offset;

352 GridOffset %= PlotGridX;

353 gridchanged= 1;

354 }

355 GraphStart += offset;

356 } else {

                                 if (PlotGridX && GridLocked && GraphStart < startMax){

358 GridOffset--;

359 GridOffset %= PlotGridX;

360 gridchanged= 1;

361 }

362 GraphStart++;

363 }

                         if(GridOffset < 0) {

365 GridOffset += PlotGridX;

366 }

367 if (gridchanged)

368 if (GraphStart > startMax) {

369 GridOffset += (GraphStart - startMax);

370 GridOffset %= PlotGridX;

371 }

372 break;

373

374 case Qt::Key_Left:

                         if(GraphPixelsPerPoint < 20) {

                                 if (PlotGridX && GridLocked && GraphStart > 0){

377 GridOffset += offset;

378 GridOffset %= PlotGridX;

379 gridchanged= 1;

380 }

381 GraphStart -= offset;

382 } else {

                                 if (PlotGridX && GridLocked && GraphStart > 0){

384 GridOffset++;

385 GridOffset %= PlotGridX;

386 gridchanged= 1;

387 }

388 GraphStart--;

389 }

390 if (gridchanged){

                                 if (GraphStart < 0)

392 GridOffset += GraphStart;

                                 if(GridOffset < 0)

394 GridOffset += PlotGridX;

395 GridOffset %= PlotGridX;

396 }

397 break;

398

399 case Qt::Key_G:

400 if(PlotGridX || PlotGridY) {

401 PlotGridX= 0;

402 PlotGridY= 0;

403 } else {

404 PlotGridX= PlotGridXdefault;

405 PlotGridY= PlotGridYdefault;

406 }

407 break;

408

409 case Qt::Key_H:

                         puts("Plot Window Keystrokes:\n");

                         puts(" Key                      Action\n");

412 puts(" DOWN Zoom in");

413 puts(" G Toggle grid display");

414 puts(" H Show help");

415 puts(" L Toggle lock grid relative to samples");

416 puts(" LEFT Move left");

417 puts(" <CTL>LEFT Move left 1 sample");

418 puts(" <SHIFT>LEFT Page left");

419 puts(" LEFT-MOUSE-CLICK Set yellow cursor");

420 puts(" Q Hide window");

421 puts(" RIGHT Move right");

422 puts(" <CTL>RIGHT Move right 1 sample");

423 puts(" <SHIFT>RIGHT Page right");

424 puts(" RIGHT-MOUSE-CLICK Set purple cursor");

425 puts(" UP Zoom out");

426 puts("");

                         puts("Use client window 'data help' for more plot commands\n");

428 break;

429

430 case Qt::Key_L:

431 GridLocked= !GridLocked;

432 break;

433

434 case Qt::Key_Q:

435 this->hide();

436 break;

437

438 default:

                         QWidget::keyPressEvent(event);

440 return;

441 break;

442 }

443

444 this->update();

445 }