src/gui/widgets/CFGScene.hxx

   1 #ifndef INCLUDE__CFGScene_hxx
   2 #define INCLUDE__CFGScene_hxx
   3
   4 #include "gui/qt.hxx"
   5 #include "BasicBlockWidget.hxx"
   6
   7 #include <vector>
   8
   9 class CFGScene : public QGraphicsScene {
  10 public:
  11         CFGScene(QWidget * parent = 0)
  12                 : QGraphicsScene(parent), highlightedBlock(NULL) {}
  13
  14         // Take special care when adding a BasicBlock to the scene as we
  15         // need to draw arrows for it later on
  16         void addItem(BasicBlockWidget* block) {
  17                 widgets.push_back(block);
  18                 QGraphicsScene::addItem(block);
  19         }
  20
  21         virtual void drawBackground(QPainter* painter, const QRectF & rect);
  22         void highlightBlock(BasicBlockWidget* block);
  23 private:
  24         std::vector<BasicBlockWidget*> widgets;
  25
  26         void drawLine(QPainter* painter, BasicBlockWidget * from, BasicBlockWidget * to, int8_t side = 0);
  27
  28         void spaceWidgets();
  29
  30         BasicBlockWidget* highlightedBlock;
  31 };
  32
  33 #endif