Added abstract method decorator and override

This commit is contained in:
Bill Xi 2022-01-17 20:27:34 +08:00 committed by GitHub
parent 3ffe300f96
commit e80dd243f1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,3 +1,4 @@
from abc import abstractmethod
import numpy as np
import numbers
@ -55,9 +56,11 @@ class CoordinateSystem():
def get_origin(self):
return self.c2p(*[0] * self.dimension)
@abstractmethod
def get_axes(self):
raise Exception("Not implemented")
@abstractmethod
def get_all_ranges(self):
raise Exception("Not implemented")
@ -334,6 +337,18 @@ class Axes(VGroup, CoordinateSystem):
def get_axes(self):
return self.axes
def get_axis(self, index):
return self.get_axes()[index]
def get_x_axis(self):
return self.get_axis(0)
def get_y_axis(self):
return self.get_axis(1)
def get_z_axis(self):
return self.get_axis(2)
def get_all_ranges(self):
return [self.x_range, self.y_range]