blob: 498210c7d49a935b796698d8a9416639067ec090 [file] [log] [blame]
Zhenkai Zhu5c6ad572012-06-04 19:50:08 -07001#ifndef TREELAYOUT_H
2#define TREELAYOUT_H
3#include <vector>
4
5class TreeLayout
6{
7public:
8 struct Coordinate
9 {
10 double x;
11 double y;
12 };
13 TreeLayout(){}
14 virtual void setOneLevelLayout(std::vector<Coordinate> &childNodesCo){};
15 void setSiblingDistance(int d) {m_siblingDistance = d;}
16 void setLevelDistance(int d) {m_levelDistance = d;}
17 int getSiblingDistance() {return m_siblingDistance;}
18 int getLevelDistance() {return m_levelDistance;}
19 virtual ~TreeLayout(){}
20private:
21 int m_siblingDistance;
22 int m_levelDistance;
23};
24
25class OneLevelTreeLayout: public TreeLayout
26{
27public:
28 OneLevelTreeLayout(){}
Zhenkai Zhu21d75f92012-06-04 21:23:34 -070029 virtual void setOneLevelLayout(std::vector<Coordinate> &childNodesCo);
Zhenkai Zhu5c6ad572012-06-04 19:50:08 -070030 virtual ~OneLevelTreeLayout(){}
31};
32#endif