wolfhece 2.2.33__py3-none-any.whl → 2.2.35__py3-none-any.whl

wolfhece/PyConfig.py

@@ -37,6 +37,7 @@ class ConfigurationKeys(Enum):
     DIRECTORY_LAZ = "Default LAZ directory"
     ACTIVE_VECTOR_COLOR = "Active vector color"
     ACTIVE_VECTOR_SIZE_SQUARE = "Active vector square size"
+    XLSX_HECE_DATABASE = "Hece Database XLSX file"
 
 class WolfConfiguration:
     """ Holds the PyWolf configuration """
@@ -86,8 +87,8 @@ class WolfConfiguration:
             ConfigurationKeys.DIRECTORY_DTM.value: "",
             ConfigurationKeys.DIRECTORY_LAZ.value: "",
             ConfigurationKeys.ACTIVE_VECTOR_COLOR.value: [0, 0, 0, 255],
-            ConfigurationKeys.ACTIVE_VECTOR_SIZE_SQUARE.value: 5
-
+            ConfigurationKeys.ACTIVE_VECTOR_SIZE_SQUARE.value: 5,
+            ConfigurationKeys.XLSX_HECE_DATABASE.value: ""
         }
         self._types = {
             ConfigurationKeys.VERSION.value: int,
@@ -104,7 +105,8 @@ class WolfConfiguration:
             ConfigurationKeys.DIRECTORY_DTM.value: str,
             ConfigurationKeys.DIRECTORY_LAZ.value: str,
             ConfigurationKeys.ACTIVE_VECTOR_COLOR.value: list,
-            ConfigurationKeys.ACTIVE_VECTOR_SIZE_SQUARE.value: int
+            ConfigurationKeys.ACTIVE_VECTOR_SIZE_SQUARE.value: int,
+            ConfigurationKeys.XLSX_HECE_DATABASE.value: str
         }
 
         self._check_config()
@@ -169,6 +171,7 @@ class GlobalOptionsDialog(wx.Dialog):
         self.cfg_directory_laz.SetValue(str(configuration[ConfigurationKeys.DIRECTORY_LAZ]))
         self.cfg_vector_color.SetColour(configuration[ConfigurationKeys.ACTIVE_VECTOR_COLOR])
         self.cfg_square_size.SetValue(str(configuration[ConfigurationKeys.ACTIVE_VECTOR_SIZE_SQUARE]))
+        self.cfg_xlsx_hece_database.SetValue(str(configuration[ConfigurationKeys.XLSX_HECE_DATABASE]))
 
     def pull_configuration(self, configuration):
         configuration[ConfigurationKeys.PLAY_WELCOME_SOUND] = self.cfg_welcome_voice.IsChecked()
@@ -185,6 +188,7 @@ class GlobalOptionsDialog(wx.Dialog):
         configuration[ConfigurationKeys.DIRECTORY_LAZ] = str(self.cfg_directory_laz.Value)
         configuration[ConfigurationKeys.ACTIVE_VECTOR_COLOR] = list(self.cfg_vector_color.GetColour())
         configuration[ConfigurationKeys.ACTIVE_VECTOR_SIZE_SQUARE] = int(self.cfg_square_size.Value)
+        configuration[ConfigurationKeys.XLSX_HECE_DATABASE] = str(self.cfg_xlsx_hece_database.Value)
 
     def InitUI(self):
 
@@ -304,9 +308,20 @@ class GlobalOptionsDialog(wx.Dialog):
         dir_laz.Add(self.cfg_directory_laz, 1, wx.EXPAND, 5)
         dir_laz.Add(self.btn_choose_laz, 1, wx.EXPAND, 5)
 
+        # XLSX HECE database
+        dir_xlsx = wx.BoxSizer(wx.HORIZONTAL)
+        self.label_xlsx_hece_database = wx.StaticText(pnl, label=_('HECE Database file'))
+        self.cfg_xlsx_hece_database = wx.TextCtrl(pnl, value='', style=wx.TE_CENTRE)
+        self.btn_choose_xlsx = wx.Button(pnl, label=_('Choose'))
+        self.btn_choose_xlsx.Bind(wx.EVT_BUTTON, self.OnChooseXLSX)
+        dir_xlsx.Add(self.label_xlsx_hece_database, 1, wx.EXPAND, 2)
+        dir_xlsx.Add(self.cfg_xlsx_hece_database, 1, wx.EXPAND, 5)
+        dir_xlsx.Add(self.btn_choose_xlsx, 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)
+        sbs.Add(dir_xlsx, 1, wx.EXPAND, 5)
 
         # Vector color
         color_vector = wx.BoxSizer(wx.HORIZONTAL)
@@ -364,6 +379,14 @@ class GlobalOptionsDialog(wx.Dialog):
             self.cfg_directory_laz.SetValue(str(dlg.GetPath()))
         dlg.Destroy()
 
+    def OnChooseXLSX(self, e):
+        """ Choose a XLSX file for HECE database """
+        dlg = wx.FileDialog(self, _("Choose a HECE database file"), style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
+        dlg.SetWildcard("Excel files (*.xlsx)|*.xlsx")
+        if dlg.ShowModal() == wx.ID_OK:
+            self.cfg_xlsx_hece_database.SetValue(str(dlg.GetPath()))
+        dlg.Destroy()
+
     def OnOk(self, e):
         if self.IsModal():
             self.EndModal(wx.ID_OK)