reflexive 1.2.0__py3-none-any.whl → 1.2.1__py3-none-any.whl

reflexive/visualise.py

@@ -1,3 +1,6 @@
+from graph_tool.all import *
+import cairo
+
 from reflexive import session
 from reflexive import cfg
 
@@ -183,4 +186,151 @@ class Display:
                     b = v[0]
                     e = v[1]
             #print("New offsets:",len(new_offsets[k]))
-        return new_offsets
+        return new_offsets
+    
+class RES_graph:
+    
+    #
+    gt_props = {0:{ "lbl":"RR",
+                    "pos":(0.2,6.5),
+                    "clr":"#00AEEF"},
+                1:{ "lbl":"NR",
+                    "pos":(5,10),
+                    "clr":"#ED1B23"},
+                2:{ "lbl":"AR",
+                    "pos":(9.8,6.5),
+                    "clr":"#00A64F"},
+                3:{ "lbl":"AF",
+                    "pos":(7.9,1),
+                    "clr":"#EC008C"},
+                4:{ "lbl":"EP",
+                    "pos":(2.1,1),
+                    "clr":"#FFF200"}}
+                
+    
+    # re_props = {"RR":{
+    #                 "idx": 0,
+    #                 "pos":(0.2,6.5),
+    #                 "clr":"#00AEEF"},
+    #             "NR":{
+    #                 "idx": 1,
+    #                 "pos":(5,10),
+    #                 "clr":"#ED1B23"},
+    #             "AR":{
+    #                 "idx": 2,
+    #                 "pos":(9.8,6.5),
+    #                 "clr":"#00A64F"},
+    #             "AF":{
+    #                 "idx": 3,
+    #                 "pos":(7.9,1),
+    #                 "clr":"#EC008C"},
+    #             "EP":{
+    #                 "idx": 4,
+    #                 "pos":(2.1,1),
+    #                 "clr":"#FFF200"}}
+    
+    #
+    edges = dict()
+    #
+    iso_vertices = set()
+    #
+    v_label = None
+    #
+    v_color = None
+    #
+    v_pos = None
+
+    #
+    def __init__(self,matrix=None):
+        self._setup(matrix)
+    
+    #    
+    def _setup(self,matrix):
+        if matrix:
+            # Edges from matrix
+            self.edges = self._matrix_to_dict(matrix)
+            self.graph = Graph(g=self.edges.keys(),directed=False)
+            self.e_weight = self.graph.new_ep("double",vals=self.edges.values())
+            # Handle colour of isolated vertices
+            default_clrs = self._get_prop_values('clr')
+            actual_clrs = []
+            for i in range(5):
+                if i in self.iso_vertices:
+                    clr = "#cccccc"
+                else:
+                    clr = default_clrs[i]
+                actual_clrs.append(clr)
+            self.v_color = self.graph.new_vp("string",vals=actual_clrs)
+        else:
+            # No edges
+            self.graph = Graph(g=self._empty_edge_dict(),directed=False)
+            self.e_weight = self.graph.new_ep("double")
+            self.v_color = self.graph.new_vp("string",val="#cccccc")
+        # Vertex properties common to all graphs    
+        self.v_label = self.graph.new_vp("string",vals=self._get_prop_values('lbl'))
+        self.v_pos = self.graph.new_vp("vector<double>",vals=self._get_prop_values('pos'))
+        
+        
+
+    #
+    def _matrix_to_dict(self,matrix):
+        edges = {}
+        for r,row in enumerate(matrix):
+            # if empty row, add to iso_vertices
+            if sum(row) == 0:
+                self.iso_vertices.add(r)
+            else:
+                for c,weight in enumerate(row):
+                    if weight > 0:
+                        edge = tuple(sorted((r,c)))
+                        #print("r,c:",edge," - ",weight)
+                        edges[edge] = weight
+        return edges
+    
+    #
+    def _empty_edge_dict(self):
+        empty_edges = {}
+        for idx in self.gt_props.keys():
+            empty_edges[idx] = []
+        return empty_edges
+    
+    #
+    def _get_prop_values(self,key):
+        values_list =  self.gt_props.values()
+        return [p[key] for p in values_list]
+    
+    # flip coordinates for graph-tool
+    def _flipY(self,vpositions):
+        x, y = ungroup_vector_property(vpositions, [0, 1])
+        y.fa *= -1
+        y.fa -= y.fa.min()
+        return group_vector_property([x, y])
+    
+    #
+    def show(self,inline=True):
+        graph = self.graph
+        positions = self._flipY(self.v_pos)
+        labels = self.v_label
+        colors = self.v_color
+        weights = self.e_weight
+        graph_draw(graph, inline=inline,output_size=(300,300),fit_view=0.7,
+                        pos=positions, 
+                        vertex_text=labels,
+                        vertex_font_family="sans serif",
+                        vertex_font_size=18,
+                        vertex_font_weight=cairo.FONT_WEIGHT_BOLD,
+                        vertex_fill_color=colors,
+                        vertex_size = 50,
+                        vertex_halo=False,
+                        vertex_pen_width=1.2,
+                        vertex_color="#999999",
+                        edge_pen_width=weights)
+        
+    def get_vertex_labels(self):
+        return self._get_prop_values('lbl')
+    
+    def get_vertex_colours(self):
+        return self._get_prop_values('clr')
+    
+    def get_vertex_positions(self):
+        return self._get_prop_values('pos')

{reflexive-1.2.0.dist-info → reflexive-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reflexive
-Version: 1.2.0
+Version: 1.2.1
 Summary: Supports AWS Reflexive Expressions Systems (RES) Analysis
 Project-URL: Repository, https://github.com/nlytx/reflexive.git
 Author-email: Andrew Gibson <andrew@nlytx.io>

{reflexive-1.2.0.dist-info → reflexive-1.2.1.dist-info}/RECORD

@@ -3,8 +3,8 @@ reflexive/analyse.py,sha256=UzWwgjAFNjeWFkCQ2o99g2vWajf17_OtSq4dFCvuPYU,17489
 reflexive/cfg.py,sha256=Ges35G234P2lvOQHgPZQae5hMSOGyBsmp1bY_yQEKkk,4303
 reflexive/session.py,sha256=MbqwTsYTgq_e_gw3mb1eRv6USs-zZ2cTCrvUNWuKfAQ,10067
 reflexive/util.py,sha256=WQ1oyzDi1i8wQ6IBwBPk6IFy07YKhg-Ug2FsOGVJRJQ,3649
-reflexive/visualise.py,sha256=BhTdTZXqMzI-dc4h_trEKYCDyBO4kPut6Ppn28yb5qU,6553
-reflexive-1.2.0.dist-info/METADATA,sha256=RKfdY47-ewjNWlQcs_wPsDkTPkhHo2MC87PWZ4nBVbM,574
-reflexive-1.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-reflexive-1.2.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-reflexive-1.2.0.dist-info/RECORD,,
+reflexive/visualise.py,sha256=76ItFjz9KyPCxlfKuF16dmY5kLVbdTXAHyfcndMStH0,11355
+reflexive-1.2.1.dist-info/METADATA,sha256=sFL4BMnt-cMD8Ks--16hSTDmLI6F7anBF9zR_61hJDA,574
+reflexive-1.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+reflexive-1.2.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+reflexive-1.2.1.dist-info/RECORD,,