Add source/sink pts attribute

compass-stc · Jun 13, 2024 · 1863a7a · 1863a7a
1 parent 8b470d8
commit 1863a7a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/StructuralGT/electronic.py b/StructuralGT/electronic.py
@@ -53,6 +53,8 @@ def compute(self, network, R_j, axis, boundary_conditions, source=-1,
         print("Sink coordinate is ", sink_coord)
         network.graph_connected.vs[source_id]["o"] = source_coord
         network.graph_connected.vs[sink_id]["o"] = sink_coord
+        network.graph_connected.vs[source_id]["pts"] = source_coord
+        network.graph_connected.vs[sink_id]["pts"] = sink_coord
 
         for node in network.graph_connected.vs:
             if node["o"][axis] >= boundary1[0] and node["o"][axis] <= boundary1[1]: