1 files changed, 46 insertions, 0 deletions
diff --git a/Project/GraphAutoLayout.h b/Project/GraphAutoLayout.h
new file mode 100644
index 0000000..4d42752
--- /dev/null
+++ b/Project/GraphAutoLayout.h
@@ -0,0 +1,46 @@
+// Code based in: "Weighted graphs: generate a layout in C++"
+// https://rodic.fr/blog/c-weighted-graph-layout-fruchterman-reingold/
+
+#ifndef GRAPHAUTOLAYOUT_H
+#define GRAPHAUTOLAYOUT_H
+
+#include <cmath>
+//#include <vector>
+#include <wx/gdicmn.h>
+#include <wx/progdlg.h>
+
+// class ParseMatpower;
+#include "ImportForm.h"
+
+class GraphAutoLayout
+{
+   public:
+    struct GraphLayoutNode {
+        wxRealPoint position;
+        wxRealPoint displacement;
+    };
+
+    struct GraphLayoutEdge {
+        GraphLayoutNode &node1;
+        GraphLayoutNode &node2;
+        float weight;
+    };
+
+    GraphAutoLayout();
+    GraphAutoLayout(std::vector<ParseMatpower::BusData*> busData, std::vector<ParseMatpower::BranchData*> branchData);
+    ~GraphAutoLayout();
+    
+    void CalculatePositions(int iterations, double scale);
+
+   protected:
+    void AddLink(size_t index1, size_t index2, float weight = 1.f);
+    void Compute(size_t iterations);
+
+    std::vector<GraphLayoutNode> m_nodes;
+    std::vector<GraphLayoutEdge> m_edges;
+
+    std::vector<ParseMatpower::BusData*> m_busData;
+    std::vector<ParseMatpower::BranchData*> m_branchData;
+};
+
+#endif  // GRAPHAUTOLAYOUT_H