def convert(self, coordinate_sys):
if(coordinate_sys=='cartesian'):
self.points=np.array([S3d_f.sphere2cart(i) for i in self.points])
elif(coordinate_sys=='spherical'):
self.points=np.array([S3d_f.cart2sphere(i) for i in self.points])
else:
print("Conversion failed.")
def draw_axis(self, d=10):
#XZ RED/YZ GREEN/XY BLUE
#X GREEN/Y RED/Z BLUE
vertices , colors = [], []
l=0
if self.axis_type == 0:
l=2
vertices.extend([0,-d,0,0,d,0])
vertices.extend([-d,0,0,d,0,0])
vertices.extend([0,0,-d,0,0,d])
elif self.axis_type == 1:
l=4
vertices.extend([-d, 0, -d, d, 0, -d, d, 0, d, -d, 0, d])
vertices.extend([0, -d, -d, 0, -d, d, 0, d, d, 0, d, -d])
vertices.extend([-d, -d, 0, d, -d, 0, d, d, 0, -d, d, 0])
for i in range (0, l):
colors.extend([1, 0, 0, 0.5])
for i in range (0, l):
colors.extend([0, 1, 0, 0.5])
for i in range (0, l):
colors.extend([0, 0, 1, 0.5])
vertices, colors = S3d_f.x_array(vertices), S3d_f.x_array(colors)
glEnable(GL_DEPTH_TEST)
if len(vertices) == 18:
S3d_f.draw_vertex_array(vertices, colors, GL_LINES)
elif len(vertices) == 36:
S3d_f.draw_vertex_array(vertices, colors, GL_QUADS)
glDisable(GL_DEPTH_TEST)
