Source Code for Module reflectometry.model1d.profileview.saveStajDialog

  1  import wx 
  2   
  3  # ----------------------------------------------------------------------- 
  4 -class SaveStajDialog(wx.Dialog): 
  5 -    def __init__( 
  6              self, 
  7              parent, 
  8              ID, 
  9              title, 
 10              size     = (300, 200),#wx.DefaultSize, 
 11              pos      = wx.DefaultPosition, 
 12              style    = wx.DEFAULT_DIALOG_STYLE, 
 13              useMetal = False, 
 14              model    = None 
 15              ): 
 16          self.parent = parent 
 17          self.model  = model 
 18   
 19          pre = wx.PreDialog() 
 20          pre.SetExtraStyle(wx.DIALOG_EX_CONTEXTHELP) 
 21          pre.Create(parent, ID, title, pos, size, style) 
 22   
 23          self.PostCreate(pre) 
 24   
 25          # This extra style can be set after the UI object has been created. 
 26          if 'wxMac' in wx.PlatformInfo and useMetal: 
 27              self.SetExtraStyle(wx.DIALOG_EX_METAL) 
 28   
 29          self._mgr = wx.aui.AuiManager(self) 
 30   
 31          self.txts=[r'Staj   Filename', 
 32                     r'top    layers #', 
 33                     r'middle layers #', 
 34                     r'bottom layers #'] 
 35   
 36          self.DefaultTxts = [r'mlayer.staj', 
 37                              str(self.parent.nTLayers), 
 38                              str(self.parent.nMLayers), 
 39                              str(self.parent.nBLayers) ] 
 40   
 41          for x in xrange( len(self.txts)  ): 
 42              wx.StaticText(self, -1, self.txts[x], (20, 20+x*40)) 
 43   
 44          self.textCtrls = [] 
 45   
 46          for x in xrange( len(self.txts) ): 
 47              textctl = wx.TextCtrl(self, 
 48                                    -1, 
 49                                    pos=(110, 20+x*40), 
 50                                    size=wx.Size(150, 20), 
 51                                    ) 
 52              self.textCtrls.append( textctl ) 
 53   
 54   
 55          self._mgr.Update() 
 56   
 57          for x in xrange( len(self.txts) ): 
 58              self.textCtrls[x].SetValue(  self.DefaultTxts[x] ) 
 59   
 60   
 61          self.button = wx.Button(self, 
 62                                  wx.ID_OK, 
 63                                  '', 
 64                                  (100, 175) 
 65                                  ) 
 66          self.cancelbutton = wx.Button(self, 
 67                                        wx.ID_CANCEL, 
 68                                        '', 
 69                                        (200, 175) 
 70                                        ) 
 71          self.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE, self.OnClosePage) 
 72          self.Bind(wx.EVT_CLOSE, self.OnClose) 
 73   
 74   
 75   
 76 -    def setDefaultFromModel(self): 
 77          pass 
 78   
 79   
 80   
 81 -    def getOutPutInfo(self): 
 82          ret = [] 
 83          ret.append( self.getStajFileNameValue() ) 
 84          ret.append( self.getnTLayersValue()     ) 
 85          ret.append( self.getnMLayersValue()     ) 
 86          ret.append( self.getnBLayersValue()     ) 
 87   
 88          return ret 
 89   
 90   
 91 -    def getStajFileNameValue(self): 
 92          ret = self.textCtrls[0].GetValue().strip() 
 93          return ret 
 94   
 95   
 96 -    def getnTLayersValue(self): 
 97          return int(self.textCtrls[1].GetValue().strip() ) 
 98   
 99   
100 -    def getnMLayersValue(self): 
101          return  int(self.textCtrls[2].GetValue().strip() ) 
102   
103   
104 -    def getnBLayersValue(self): 
105          return int(self.textCtrls[3].GetValue().strip() ) 
106   
107   
108 -    def GetOwnerManager(self): 
109          return self._mgr 
110   
111   
112 -    def OnClose(self, event): 
113          self._mgr.UnInit() 
114          event.Skip() 
115   
116   
117 -    def OnClosePage(self, event): 
118          event.Veto() 
119