dialogs/BrowseLocationsDialog.py
changeset 4009 51272fe96999
parent 3789 8def429216ca
equal deleted inserted replaced
4008:f30573e98600 4009:51272fe96999
   132         self.SetSizer(main_sizer)
   132         self.SetSizer(main_sizer)
   133 
   133 
   134         self.Controller = controller
   134         self.Controller = controller
   135         self.VarType = var_type
   135         self.VarType = var_type
   136         self.BaseVarType = self.Controller.GetBaseType(self.VarType)
   136         self.BaseVarType = self.Controller.GetBaseType(self.VarType)
   137         self.VarTypeSize = LOCATION_SIZES[self.BaseVarType]
   137         self.VarTypeSize = LOCATION_SIZES[self.BaseVarType] if self.BaseVarType else None
   138         self.Locations = self.Controller.GetVariableLocationTree()
   138         self.Locations = self.Controller.GetVariableLocationTree()
   139 
   139 
   140         # Define Tree item icon list
   140         # Define Tree item icon list
   141         self.TreeImageList = wx.ImageList(16, 16)
   141         self.TreeImageList = wx.ImageList(16, 16)
   142         self.TreeImageDict = {}
   142         self.TreeImageDict = {}
   178 
   178 
   179     def FilterType(self, location_type, location_size):
   179     def FilterType(self, location_type, location_size):
   180         if self.TypeFilter == 0:
   180         if self.TypeFilter == 0:
   181             return True
   181             return True
   182 
   182 
   183         if location_size != self.VarTypeSize:
   183         if self.VarTypeSize is not None and location_size != self.VarTypeSize:
   184             return False
   184             return False
   185 
   185 
   186         if self.TypeFilter == 1:
   186         if self.TypeFilter == 1:
   187             return self.Controller.IsOfType(location_type, self.BaseVarType)
   187             return self.Controller.IsOfType(location_type, self.BaseVarType)
   188         elif self.TypeFilter == 2:
   188         elif self.TypeFilter == 2: