diff tools/myTools/bin/sfa/plot/table_condition.py @ 1:7e5c71b2e71f draft default tip

Uploaded
author laurenmarazzi
date Wed, 22 Dec 2021 16:00:34 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tools/myTools/bin/sfa/plot/table_condition.py	Wed Dec 22 16:00:34 2021 +0000
@@ -0,0 +1,44 @@
+from .base import BaseTable
+from .tableaxis import ConditionTableAxis
+
+
+class ConditionTable(BaseTable):
+
+    def __init__(self, conds, *args, **kwargs):
+        self._dfc = conds  # DataFrame of condition cases
+        super().__init__(*args, **kwargs)
+        """
+        Add labels using x and y axes.
+        The default values should be assigned before adding labels.
+        """
+        self.row_tick_fontsize = 5
+        self.column_tick_fontsize = 5
+        self._add_labels()
+    # end of def __init__
+
+    def _set_colors(self, colors):
+        """Assign default color values, if it is not defined.
+        """
+        self._set_default_color('cond_up_cell', 'blue')
+        self._set_default_color('cond_dn_cell', 'white')
+        
+    def _create_axes(self):
+        super()._create_axes()
+        ax = self._axes['base']
+        self._axes['condition'] = ax
+        del self._axes['base']
+
+    def _create_tables(self):
+        self._tables = []
+        tb = ConditionTableAxis(self._axes['condition'],
+                                self._dfc, self._colors)
+        tb.fontsize = 4
+        tb.linewidth = 0.5
+        self._tables.append(tb)
+
+    def _add_labels(self):
+        tb = self._tables[0]
+        tb.add_row_labels()
+        tb.add_column_labels()
+
+