cutter/src/widgets/GraphLayout.h
2019-08-03 15:10:44 +02:00

48 lines
1.1 KiB
C++

#ifndef GRAPHLAYOUT_H
#define GRAPHLAYOUT_H
#include "core/Cutter.h"
#include <unordered_map>
class GraphLayout
{
public:
struct GraphEdge {
ut64 target;
QPolygonF polyline;
enum ArrowDirection {
Down, Left, Up, Right, None
};
ArrowDirection arrow = ArrowDirection::Down;
explicit GraphEdge(ut64 target): target(target) {}
};
struct GraphBlock {
int x = 0;
int y = 0;
int width = 0;
int height = 0;
// This is a unique identifier, e.g. offset in the case of r2 blocks
ut64 entry;
// Edges
std::vector<GraphEdge> edges;
};
using Graph = std::unordered_map<ut64, GraphBlock>;
struct LayoutConfig {
int block_vertical_margin = 40;
int block_horizontal_margin = 10;
};
GraphLayout(const LayoutConfig &layout_config) : layoutConfig(layout_config) {}
virtual ~GraphLayout() {}
virtual void CalculateLayout(Graph &blocks, ut64 entry, int &width,
int &height) const = 0;
protected:
LayoutConfig layoutConfig;
};
#endif // GRAPHLAYOUT_H