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

wolfhece/PyConfig.py

@@ -32,6 +32,9 @@ class ConfigurationKeys(Enum):
     ACTIVE_ARRAY_PALETTE_FOR_IMAGE = "Use active array palette for image"
     ACTIVE_RES2D_PALETTE_FOR_IMAGE = "Use active result palette for image"
     ASSEMBLY_IMAGES = "AssemblyImages"
+    DIRECTORY_DEM = "Default DEM directory"
+    DIRECTORY_DTM = "Default DTM directory"
+    DIRECTORY_LAZ = "Default LAZ directory"
 
 class WolfConfiguration:
     """ Holds the PyWolf configuration """
@@ -76,7 +79,10 @@ class WolfConfiguration:
             ConfigurationKeys.COLOR_BACKGROUND.value: [255, 255, 255, 255],
             ConfigurationKeys.ASSEMBLY_IMAGES.value: 0,
             ConfigurationKeys.TICKS_XROTATION.value: 30.,
-            ConfigurationKeys.TICKS_FONTSIZE.value: 12
+            ConfigurationKeys.TICKS_FONTSIZE.value: 12,
+            ConfigurationKeys.DIRECTORY_DEM.value: "",
+            ConfigurationKeys.DIRECTORY_DTM.value: "",
+            ConfigurationKeys.DIRECTORY_LAZ.value: ""
 
         }
         self._types = {
@@ -89,7 +95,10 @@ class WolfConfiguration:
             ConfigurationKeys.COLOR_BACKGROUND.value: list,
             ConfigurationKeys.ASSEMBLY_IMAGES.value: int,
             ConfigurationKeys.TICKS_XROTATION.value: float,
-            ConfigurationKeys.TICKS_FONTSIZE.value: int
+            ConfigurationKeys.TICKS_FONTSIZE.value: int,
+            ConfigurationKeys.DIRECTORY_DEM.value: str,
+            ConfigurationKeys.DIRECTORY_DTM.value: str,
+            ConfigurationKeys.DIRECTORY_LAZ.value: str
         }
 
         self._check_config()
@@ -97,7 +106,7 @@ class WolfConfiguration:
     def _check_config(self):
         assert self._config.keys() == self._types.keys()
         for idx, (key,val) in enumerate(self._config.items()):
-            assert type(val) == self._types[key]
+            assert isinstance(val, self._types[key])
 
     def load(self):
         with open(self._options_file_path, "r", encoding="utf-8") as configfile:
@@ -136,7 +145,7 @@ class GlobalOptionsDialog(wx.Dialog):
         super(GlobalOptionsDialog, self).__init__(*args, **kw)
 
         self.InitUI()
-        self.SetSize((600, 400))
+        self.SetSize((600, 500))
         self.SetTitle(_("Global options"))
 
     def push_configuration(self, configuration):
@@ -149,6 +158,9 @@ class GlobalOptionsDialog(wx.Dialog):
         self.cfg_assembly_images.SetSelection(configuration[ConfigurationKeys.ASSEMBLY_IMAGES])
         self.cfg_ticks_xrotation.SetValue(str(configuration[ConfigurationKeys.TICKS_XROTATION]))
         self.cfg_ticks_fontsize.SetValue(str(configuration[ConfigurationKeys.TICKS_FONTSIZE]))
+        self.cfg_directory_dem.SetValue(str(configuration[ConfigurationKeys.DIRECTORY_DEM]))
+        self.cfg_directory_dtm.SetValue(str(configuration[ConfigurationKeys.DIRECTORY_DTM]))
+        self.cfg_directory_laz.SetValue(str(configuration[ConfigurationKeys.DIRECTORY_LAZ]))
 
     def pull_configuration(self, configuration):
         configuration[ConfigurationKeys.PLAY_WELCOME_SOUND] = self.cfg_welcome_voice.IsChecked()
@@ -160,6 +172,9 @@ class GlobalOptionsDialog(wx.Dialog):
         configuration[ConfigurationKeys.ASSEMBLY_IMAGES] = self.cfg_assembly_images.GetSelection()
         configuration[ConfigurationKeys.TICKS_XROTATION] = float(self.cfg_ticks_xrotation.Value)
         configuration[ConfigurationKeys.TICKS_FONTSIZE] = int(self.cfg_ticks_fontsize.Value)
+        configuration[ConfigurationKeys.DIRECTORY_DEM] = str(self.cfg_directory_dem.Value)
+        configuration[ConfigurationKeys.DIRECTORY_DTM] = str(self.cfg_directory_dtm.Value)
+        configuration[ConfigurationKeys.DIRECTORY_LAZ] = str(self.cfg_directory_laz.Value)
 
     def InitUI(self):
 
@@ -246,6 +261,43 @@ class GlobalOptionsDialog(wx.Dialog):
 
         sbs.Add(locsizer, 3, wx.EXPAND, 5)
 
+        # DEM directory
+        dir_dem = wx.BoxSizer(wx.HORIZONTAL)
+        self.label_directory_dem = wx.StaticText(pnl, label=_('Default DEM directory'))
+        self.cfg_directory_dem = wx.TextCtrl(pnl, value='',style = wx.TE_CENTRE )
+        self.btn_choose_dem = wx.Button(pnl, label=_('Choose'))
+        self.btn_choose_dem.Bind(wx.EVT_BUTTON, self.OnChooseDem)
+
+        dir_dem.Add(self.label_directory_dem, 1, wx.EXPAND, 2)
+        dir_dem.Add(self.cfg_directory_dem, 1, wx.EXPAND, 5)
+        dir_dem.Add(self.btn_choose_dem, 1, wx.EXPAND, 5)
+
+        # DTM directory
+        dir_dtm = wx.BoxSizer(wx.HORIZONTAL)
+        self.label_directory_dtm = wx.StaticText(pnl, label=_('Default DTM directory'))
+        self.cfg_directory_dtm = wx.TextCtrl(pnl, value='',style = wx.TE_CENTRE )
+        self.btn_choose_dtm = wx.Button(pnl, label=_('Choose'))
+        self.btn_choose_dtm.Bind(wx.EVT_BUTTON, self.OnChooseDtm)
+
+        dir_dtm.Add(self.label_directory_dtm, 1, wx.EXPAND, 2)
+        dir_dtm.Add(self.cfg_directory_dtm, 1, wx.EXPAND, 5)
+        dir_dtm.Add(self.btn_choose_dtm, 1, wx.EXPAND, 5)
+
+        # LAZ directory
+        dir_laz = wx.BoxSizer(wx.HORIZONTAL)
+        self.label_directory_laz = wx.StaticText(pnl, label=_('Default LAZ directory'))
+        self.cfg_directory_laz = wx.TextCtrl(pnl, value='',style = wx.TE_CENTRE )
+        self.btn_choose_laz = wx.Button(pnl, label=_('Choose'))
+        self.btn_choose_laz.Bind(wx.EVT_BUTTON, self.OnChooseLaz)
+
+        dir_laz.Add(self.label_directory_laz, 1, wx.EXPAND, 2)
+        dir_laz.Add(self.cfg_directory_laz, 1, wx.EXPAND, 5)
+        dir_laz.Add(self.btn_choose_laz, 1, wx.EXPAND, 5)
+
+        sbs.Add(dir_dem, 1, wx.EXPAND, 5)
+        sbs.Add(dir_dtm, 1, wx.EXPAND, 5)
+        sbs.Add(dir_laz, 1, wx.EXPAND, 5)
+
         pnl.SetSizer(sbs)
         pnl.Layout()
 
@@ -267,6 +319,27 @@ class GlobalOptionsDialog(wx.Dialog):
         okButton.Bind(wx.EVT_BUTTON, self.OnOk)
         closeButton.Bind(wx.EVT_BUTTON, self.OnClose)
 
+    def OnChooseDem(self, e):
+        """ Choose a directory for DEM files """
+        dlg = wx.DirDialog(self, _("Choose a directory"), style=wx.DD_DEFAULT_STYLE)
+        if dlg.ShowModal() == wx.ID_OK:
+            self.cfg_directory_dem.SetValue(str(dlg.GetPath()))
+        dlg.Destroy()
+
+    def OnChooseDtm(self, e):
+        """ Choose a directory for DTM files """
+        dlg = wx.DirDialog(self, _("Choose a directory"), style=wx.DD_DEFAULT_STYLE)
+        if dlg.ShowModal() == wx.ID_OK:
+            self.cfg_directory_dtm.SetValue(str(dlg.GetPath()))
+        dlg.Destroy()
+
+    def OnChooseLaz(self, e):
+        """ Choose a directory for LAZ files """
+        dlg = wx.DirDialog(self, _("Choose a directory"), style=wx.DD_DEFAULT_STYLE)
+        if dlg.ShowModal() == wx.ID_OK:
+            self.cfg_directory_laz.SetValue(str(dlg.GetPath()))
+        dlg.Destroy()
+
     def OnOk(self, e):
         if self.IsModal():
             self.EndModal(wx.ID_OK)