move to a more standard python packaging structure

src/damaris/data/Drawable.py

@@ -0,0 +1,188 @@
+# -*- coding: iso-8859-1 -*-
+
+#############################################################################
+#                                                                           #
+# Name: Class Drawable                                                      #
+#                                                                           #
+# Purpose: Base class of everything plottable                               #
+#                                                                           #
+#############################################################################
+
+class Drawable:
+    def __init__(self):
+
+        # Will be set correctly in one of the subclasses
+        self.x = []
+        self.y = []
+
+        self.styles = { }
+
+        self.xlabel = None
+        self.ylabel = None
+
+        self.title = None
+
+        self.legend = { }
+
+        self.text = {}
+
+        self.xmin = 0
+        self.xmax = 0
+        self.ymin = 0
+        self.ymax = 0
+
+
+    def get_xdata(self):
+        "Returns a reference to the x-Plotdata (array)"
+        return self.x
+
+
+    def set_xdata(self, pos, value):
+        "Sets a point in x"
+        try:
+            self.x[pos] = value
+        except:
+            raise
+
+
+    def get_ydata(self, channel):
+        "Returns the y-Plotdata of channel n (array)"
+        try:
+            return self.y[channel]
+        except:
+            raise
+
+
+    def set_ydata(self, channel, pos, value):
+        "Sets a point in y"
+        try:
+            self.y[channel][pos] = value
+        except:
+            raise
+
+    def get_number_of_channels(self):
+        "Returns the number of channels in y"
+        return len(self.y)
+
+
+    def get_style(self):
+        "Returns a reference to plot-styles (dictionary)"
+        return self.styles
+
+
+    def set_style(self, channel, value):
+        "Sets a channel to a certain plot-style"
+        if channel in self.styles:
+            print("Drawable Warning: Style key \"%s\" will be overwritten with \"%s\"" % (str(channel), str(value)))
+        self.styles[channel] = str(value)
+
+
+    def get_xlabel(self):
+        "Returns the label for the x-axis"
+        return self.xlabel
+
+
+    def set_xlabel(self, label):
+        "Sets the label for the x-axis"
+        self.xlabel = str(label)
+
+
+    def get_ylabel(self):
+        "Gets the label for the y-axis"
+        return self.ylabel
+
+
+    def set_ylabel(self, label):
+        "Sets the label for the y-axis"
+        self.ylabel = str(label)
+
+
+    def get_text(self, index):
+        "Returns labels to be plotted (List)"
+        if index in self.text:
+            return self.text[index]
+        else:
+            return None
+
+
+    def set_text(self, index, text):
+        "Sets labels to be plotted "
+        self.text[index] = str(text)
+
+
+    def get_title(self):
+        "Returns the title of the plot"
+        return self.title
+
+
+    def set_title(self, title):
+        "Sets the title of the plot"
+        self.title = str(title)
+
+
+    def get_legend(self):
+        "Returns the legend of the plot (Dictionary)"
+        return self.legend
+
+
+    def set_legend(self, channel, value):
+        "Sets the legend of the plot"
+        if channel in self.legend:
+            print("Drawable Warning: Legend key \"%s\" will be overwritten with \"%s\"" % (str(channel), str(value)))
+        self.legend[channel] = str(value)
+
+    def get_xmin(self):
+        "Returns minimun of x"
+        return self.x.min()
+
+    def get_xminpos(self):
+        "Returns smallest positive value of x"
+        mask = self.x > 0
+        return self.x[mask].min()
+
+    def set_xmin(self, xmin):
+        "Sets minimum of x"
+        self.xmin = xmin
+
+    def get_xmax(self):
+        "Returns maximum of x"
+        return self.x.max()
+
+    def set_xmax(self, xmax):
+        "Sets maximum of x"
+        self.xmax = xmax
+
+    def get_ymin(self):
+        "Returns minimum of y"
+        if isinstance(self.y, list):
+            return min([yarr.min() for yarr in self.y])
+        else:
+            return self.y.min()
+
+    def get_yminpos(self):
+        "Returns smallest positive value of y"
+        if isinstance(self.y, list):
+            ymins = []
+            for ys in self.y:
+                mask = ys > 0
+                ymins.append(ys[mask].min())
+            ymin = min(ymins)
+        else:
+            mask = self.y > 0
+            ymin = self.y[mask].min()
+        return ymin
+
+    def set_ymin(self, ymin):
+        "Sets minimum of y"
+        self.ymin = ymin
+
+    def get_ymax(self):
+        "Returns maximimum of y"
+        if isinstance(self.y, list):
+            return max([yarr.max() for yarr in self.y])
+        else:
+            return self.y.max()
+
+    def set_ymax(self, ymax):
+        "Sets maximum of y"
+        self.ymax = ymax