Your Name
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
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")
|
|
|
|
|