Put data types into separate python modules in the datatypes subdirectory

Author: vitcpp

datatypes/__init__.py

@@ -0,0 +1,5 @@
+from .point import Point
+from .vector import Vector
+from .parallel import Parallel
+from .meridian import Meridian
+from .great_circle_arc import GreatCircleArc

datatypes/great_circle_arc.py

@@ -0,0 +1,9 @@
+from .point import Point
+
+class GreatCircleArc:
+
+	def __init__(self, beg : Point, end : Point):
+		self.pts = [beg, end]
+
+	def __getitem__(self, idx):
+		return self.pts[idx]

datatypes/meridian.py

@@ -0,0 +1,6 @@
+from .point import Point
+
+class Meridian:
+
+	def __init__(self, lng):
+		self.lng = lng

datatypes/parallel.py

@@ -0,0 +1,6 @@
+from .point import Point
+
+class Parallel:
+
+	def __init__(self, lat):
+		self.lat = lat

datatypes/point.py

@@ -0,0 +1,8 @@
+
+class Point:
+
+	def __init__(self, lng, lat):
+		self.coords = [lng, lat]
+
+	def __getitem__(self, idx):
+		return self.coords[idx]

datatypes/vector.py

@@ -0,0 +1,34 @@
+import numpy as np
+from .point import Point
+
+class Vector:
+
+	def __init__(self, data, color = None):
+		if isinstance(data, Point):
+			self.coords = [
+				np.cos(data[0]) * np.cos(data[1]),
+				np.sin(data[0]) * np.cos(data[1]),
+				np.sin(data[1])
+				]
+		elif isinstance(data, list):
+			self.coords = np.array(data)
+		elif isinstance(data, np.ndarray):
+			self.coords = data.copy()
+		self.color = "black" if color is None else color
+
+	def __getitem__(self, idx):
+		return self.coords[idx]
+
+	def cross_product(self, v):
+		x = self[1] * v[2] - self[2] * v[1]
+		y = self[2] * v[0] - self[0] * v[2]
+		z = self[0] * v[1] - self[1] * v[0]
+		return Vector([x, y, z])
+
+	def dot_product(self, v):
+		return self[0] * v[0] + self[1] * v[1] + self[2] * v[2];
+
+	def normalized(self):
+		norm = np.sqrt(self[0] ** 2 + self[1] ** 2 + self[2] ** 2)
+		coords = self.coords / norm
+		return Vector(coords)

sphereplot.py

@@ -1,63 +1,6 @@
 import numpy as np
 import matplotlib.pyplot as plt
-
-class Point:
-
-	def __init__(self, lng, lat):
-		self.coords = [lng, lat]
-
-	def __getitem__(self, idx):
-		return self.coords[idx]
-
-class Vector:
-
-	def __init__(self, data, color = None):
-		if isinstance(data, Point):
-			self.coords = [
-				np.cos(data[0]) * np.cos(data[1]),
-				np.sin(data[0]) * np.cos(data[1]),
-				np.sin(data[1])
-				]
-		elif isinstance(data, list):
-			self.coords = np.array(data)
-		elif isinstance(data, np.ndarray):
-			self.coords = data.copy()
-		self.color = "black" if color is None else color
-
-	def __getitem__(self, idx):
-		return self.coords[idx]
-
-	def cross_product(self, v):
-		x = self[1] * v[2] - self[2] * v[1]
-		y = self[2] * v[0] - self[0] * v[2]
-		z = self[0] * v[1] - self[1] * v[0]
-		return Vector([x, y, z])
-
-	def dot_product(self, v):
-		return self[0] * v[0] + self[1] * v[1] + self[2] * v[2];
-
-	def normalized(self):
-		norm = np.sqrt(self[0] ** 2 + self[1] ** 2 + self[2] ** 2)
-		coords = self.coords / norm
-		return Vector(coords)
-
-class GreatCircleArc:
-
-	def __init__(self, beg : Point, end : Point):
-		self.pts = [beg, end]
-
-	def __getitem__(self, idx):
-		return self.pts[idx]
-
-class Meridian:
-
-	def __init__(self, lng):
-		self.lng = lng
-
-class Parallel:
-
-	def __init__(self, lat):
-		self.lat = lat
+from datatypes import *
 
 class Sphere: