wolfhece 2.1.126__py3-none-any.whl → 2.1.128__py3-none-any.whl

wolfhece/PyPalette.py

@@ -109,6 +109,34 @@ class wolfpalette(wx.Frame, LinearSegmentedColormap):
         else:
             return self(xloc, bytes=True)
 
+    def get_rgba_oneval(self, x: float):
+        """Récupération de la couleur en fonction de la valeur x"""
+
+        dval = self.values[-1]-self.values[0]
+        if dval == 0.:
+            dval = 1.
+        xloc = (x-self.values[0])/dval
+
+        if self.interval_cst:
+            rgba = np.ones((4), dtype=np.uint8)
+
+            if xloc < 0.:
+                rgba = self.colormin_uint8
+            elif xloc >= 1.:
+                rgba = self.colormax_uint8
+            else:
+                for i in range(self.nb-1):
+                    val1 = (self.values[i]-self.values[0])/dval
+                    val2 = (self.values[i+1]-self.values[0])/dval
+                    if (xloc >= val1) & (xloc < val2):
+                        c1 = self.colorsuint8[i]
+                        rgba = c1
+                        break
+
+            return rgba
+        else:
+            return self(xloc, bytes=True)
+
     def export_palette_matplotlib(self, name):
         cmaps = OrderedDict()
         cmaps['Perceptually Uniform Sequential'] = ['viridis', 'plasma', 'inferno', 'magma', 'cividis']
@@ -213,6 +241,14 @@ class wolfpalette(wx.Frame, LinearSegmentedColormap):
 
         self.fill_segmentdata()
 
+    def get_ScalarMappable_mpl(self):
+        """ Récupération de l'objet ScalarMappable via Matplotlib """
+        if self.interval_cst:
+            discrete_cmap = ListedColormap(self.colorsflt[:, :3])
+            colorbar = ScalarMappable(BoundaryNorm(self.values, ncolors=self.nb-1), cmap=discrete_cmap)
+        else:
+            return ScalarMappable(Normalize(self.values[0], self.values[-1]), cmap=self)
+
     def export_image(self, fn='', h_or_v: typing.Literal['h', 'v', ''] = '', figax=None):
         """
         Export image from colormap

wolfhece/PyParams.py

@@ -753,7 +753,7 @@ class Wolf_Param(wx.Frame):
     def SavetoFile(self, event:wx.MouseEvent):
         """ sauvegarde dans le fichier texte """
 
-        self.save()
+        self.Save()
 
     def Save(self, filename:str = ''):
         """ Save the parameters in a file """
@@ -1308,7 +1308,7 @@ class Wolf_Param(wx.Frame):
             posSep2 = positions[-1] #nameStr[posSep1+1:].find("$")
 
             # select the string between the two '$'
-            iterCode = nameStr[posSep1:posSep1+posSep2+1]
+            iterCode = nameStr[posSep1:posSep2+1]
 
             positions_left = [i for i, char in enumerate(iterCode) if char == "("]
             positions_right = [i for i, char in enumerate(iterCode) if char == ")"]