Old SandTable project

thr2png.py

@@ -0,0 +1,54 @@
+from matplotlib import pyplot as plt
+import sys
+
+
+# Creates a .png file from .thr file
+def drawFile(file):
+    # Craetes a base with almost no borders
+    plt.figure(figsize=(20, 20), frameon=False)
+    ax = plt.axes([0, 0, 1, 1], projection='polar')
+    # Turns off grids and axis markings
+    plt.axis('off')
+    plt.grid(visible=None)
+
+    # Inverting the y axis and turining the graph 90 degrees, because .thr does this
+    ax.set_theta_direction(-1)
+    ax.set_theta_zero_location("N")
+
+    with open(file) as f:
+
+        theta = []
+        rho = []
+        counter = 0
+        for line in f:
+            # discard if line commented
+            if "#" in line or len(line) < 5 or "/" in line:
+                continue
+            
+            else:
+                comp = line.split()
+                theta.append(float(comp[0]))
+                rho.append(float(comp[1]))
+                counter += 1
+
+
+        plt.polar(theta, rho, marker='o', color='blue', markersize=1, linewidth=3)
+        # Draws a green dot as starting point
+        plt.polar(float(theta[0]), rho[0], marker='o', color='green', markersize=30)
+        # Draws a red dot as end point
+        plt.polar(float(theta[-1]), rho[-1], marker='o', color='red', markersize=30)
+
+        name = file.split('.')
+        plt.savefig(name[0]+'.png')
+
+        return counter
+
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        name = sys.argv[1]
+        drawFile(name)
+    else:
+        drawFile("media/tracks/sunburst.thr")
+    
+