Changeset 2899 for branch/2frame/GSASIIctrls.py

Timestamp:

Jul 3, 2017 4:12:45 PM (6 years ago)

Author:

toby

Message:

partial reorg

File:

• branch/2frame/GSASIIctrls.py (modified) (1 diff)

branch/2frame/GSASIIctrls.py

@@ -2776 +2776 @@
         parent.Raise()
         self.EndModal(wx.ID_OK)
+
+################################################################################
+class DisAglDialog(wx.Dialog):
+    '''Distance/Angle Controls input dialog. After
+    :meth:`ShowModal` returns, the results are found in
+    dict :attr:`self.data`, which is accessed using :meth:`GetData`.
+
+    :param wx.Frame parent: reference to parent frame (or None)
+    :param dict data: a dict containing the current
+      search ranges or an empty dict, which causes default values
+      to be used.
+      Will be used to set element `DisAglCtls` in 
+      :ref:`Phase Tree Item <Phase_table>`
+    :param dict default:  A dict containing the default
+      search ranges for each element.
+    :param bool Reset: if True (default), show Reset button
+    :param bool Angle: if True (default), show angle radii
+    '''
+    def __init__(self,parent,data,default,Reset=True,Angle=True):
+        text = 'Distance Angle Controls'
+        if not Angle:
+            text = 'Distance Controls'
+        wx.Dialog.__init__(self,parent,wx.ID_ANY,text, 
+            pos=wx.DefaultPosition,style=wx.DEFAULT_DIALOG_STYLE)
+        self.default = default
+        self.Reset = Reset
+        self.Angle = Angle
+        self.panel = wx.Panel(self)         #just a dummy - gets destroyed in Draw!
+        self._default(data,self.default)
+        self.Draw(self.data)
+                
+    def _default(self,data,default):
+        '''Set starting values for the search values, either from
+        the input array or from defaults, if input is null
+        '''
+        if data:
+            self.data = copy.deepcopy(data) # don't mess with originals
+        else:
+            self.data = {}
+            self.data['Name'] = default['Name']
+            self.data['Factors'] = [0.85,0.85]
+            self.data['AtomTypes'] = default['AtomTypes']
+            self.data['BondRadii'] = default['BondRadii'][:]
+            self.data['AngleRadii'] = default['AngleRadii'][:]
+
+    def Draw(self,data):
+        '''Creates the contents of the dialog. Normally called
+        by :meth:`__init__`.
+        '''
+        self.panel.Destroy()
+        self.panel = wx.Panel(self)
+        mainSizer = wx.BoxSizer(wx.VERTICAL)
+        mainSizer.Add(wx.StaticText(self.panel,-1,'Controls for phase '+data['Name']),
+            0,WACV|wx.LEFT,10)
+        mainSizer.Add((10,10),1)
+        
+        ncol = 3
+        if not self.Angle:
+            ncol=2
+        radiiSizer = wx.FlexGridSizer(0,ncol,5,5)
+        radiiSizer.Add(wx.StaticText(self.panel,-1,' Type'),0,WACV)
+        radiiSizer.Add(wx.StaticText(self.panel,-1,'Bond radii'),0,WACV)
+        if self.Angle:
+            radiiSizer.Add(wx.StaticText(self.panel,-1,'Angle radii'),0,WACV)
+        self.objList = {}
+        for id,item in enumerate(self.data['AtomTypes']):
+            radiiSizer.Add(wx.StaticText(self.panel,-1,' '+item),0,WACV)
+            bRadii = ValidatedTxtCtrl(self.panel,data['BondRadii'],id,nDig=(10,3))
+            radiiSizer.Add(bRadii,0,WACV)
+            if self.Angle:
+                aRadii = ValidatedTxtCtrl(self.panel,data['AngleRadii'],id,nDig=(10,3))
+                radiiSizer.Add(aRadii,0,WACV)
+        mainSizer.Add(radiiSizer,0,wx.EXPAND)
+        if self.Angle:
+            factorSizer = wx.FlexGridSizer(0,2,5,5)
+            Names = ['Bond','Angle']
+            for i,name in enumerate(Names):
+                factorSizer.Add(wx.StaticText(self.panel,-1,name+' search factor'),0,WACV)
+                bondFact = ValidatedTxtCtrl(self.panel,data['Factors'],i,nDig=(10,3))
+                factorSizer.Add(bondFact)
+            mainSizer.Add(factorSizer,0,wx.EXPAND)
+        
+        OkBtn = wx.Button(self.panel,-1,"Ok")
+        OkBtn.Bind(wx.EVT_BUTTON, self.OnOk)
+        btnSizer = wx.BoxSizer(wx.HORIZONTAL)
+        btnSizer.Add((20,20),1)
+        btnSizer.Add(OkBtn)
+        if self.Reset:
+            ResetBtn = wx.Button(self.panel,-1,'Reset')
+            ResetBtn.Bind(wx.EVT_BUTTON, self.OnReset)
+            btnSizer.Add(ResetBtn)
+        btnSizer.Add((20,20),1)
+        mainSizer.Add(btnSizer,0,wx.EXPAND|wx.BOTTOM|wx.TOP, 10)
+        self.panel.SetSizer(mainSizer)
+        self.panel.Fit()
+        self.Fit()
+    
+    def GetData(self):
+        'Returns the values from the dialog'
+        return self.data
+        
+    def OnOk(self,event):
+        'Called when the OK button is pressed'
+        parent = self.GetParent()
+        parent.Raise()
+        self.EndModal(wx.ID_OK)              
+        
+    def OnReset(self,event):
+        'Called when the Reset button is pressed'
+        data = {}
+        self._default(data,self.default)
+        self.Draw(self.data)
+                
+################################################################################
+class ShowLSParms(wx.Dialog):
+    '''Create frame to show least-squares parameters
+    '''
+    def __init__(self,parent,title,parmDict,varyList,fullVaryList,
+                 size=(375,430)):
+        
+        wx.Dialog.__init__(self,parent,wx.ID_ANY,title,size=size,
+                           style=wx.DEFAULT_DIALOG_STYLE|wx.RESIZE_BORDER)
+        self.panel = wxscroll.ScrolledPanel(self)         #just a dummy - gets destroyed in DrawPanel!
+        self.parmChoice = 'Phase'
+        self.parmDict = parmDict
+        self.varyList = varyList
+        self.fullVaryList = fullVaryList
+
+        self.parmNames = parmDict.keys()
+        self.parmNames.sort()
+        splitNames = [item.split(':') for item in self.parmNames if len(item) > 3 and not isinstance(self.parmDict[item],basestring)]
+        self.globNames = [':'.join(item) for item in splitNames if not item[0] and not item[1]]
+        self.globVars = list(set([' ',]+[item[2] for item in splitNames if not item[0] and not item[1]]))
+        self.globVars.sort()
+        self.hisNames = [':'.join(item) for item in splitNames if not item[0] and item[1]]
+        self.hisNums = list(set([int(item.split(':')[1]) for item in self.hisNames]))
+        self.hisNums.sort()
+        self.hisNums = [' ',]+[str(item) for item in self.hisNums]
+        self.hisVars = list(set([' ',]+[item[2] for item in splitNames if not item[0]]))
+        self.hisVars.sort()
+        self.phasNames = [':'.join(item) for item in splitNames if not item[1] and 'is' not in item[2]]
+        self.phasNums = [' ',]+list(set([item.split(':')[0] for item in self.phasNames]))
+        if '' in self.phasNums: self.phasNums.remove('')
+        self.phasVars = list(set([' ',]+[item[2] for item in splitNames if not item[1] and 'is' not in item[2]]))
+        self.phasVars.sort()
+        self.phasNums.sort()
+        self.hapNames = [':'.join(item) for item in splitNames if item[0] and item[1]]
+        self.hapVars = list(set([' ',]+[item[2] for item in splitNames if item[0] and item[1]]))
+        self.hapVars.sort()
+        self.hisNum = ' '
+        self.phasNum = ' '
+        self.varName = ' '
+        self.listSel = 'Refined'
+        self.DrawPanel()
+        
+            
+    def DrawPanel(self):
+            
+        def _OnParmSel(event):
+            self.parmChoice = parmSel.GetStringSelection()
+            self.varName = ' '
+            wx.CallLater(100,self.DrawPanel)
+            
+        def OnPhasSel(event):
+            event.Skip()
+            self.phasNum = phasSel.GetValue()
+            self.varName = ' '
+            wx.CallLater(100,self.DrawPanel)
+
+        def OnHistSel(event):
+            event.Skip()
+            self.hisNum = histSel.GetValue()
+            self.varName = ' '
+            wx.CallLater(100,self.DrawPanel)
+            
+        def OnVarSel(event):
+            self.varName = varSel.GetValue()
+            self.phasNum = ' '
+            self.hisNum = ' '
+            wx.CallLater(100,self.DrawPanel)
+            
+        def OnListSel(event):
+            self.listSel = listSel.GetStringSelection()
+            wx.CallLater(100,self.DrawPanel)
+
+        if self.panel:
+            #self.panel.DestroyChildren() # Bad on Mac: deletes scroll bars
+            sizer = self.panel.GetSizer()
+            if sizer: sizer.DeleteWindows()
+
+        mainSizer = wx.BoxSizer(wx.VERTICAL)
+        num = len(self.varyList)
+        mainSizer.Add(wx.StaticText(self.panel,label=' Number of refined variables: '+str(num)),0)
+        if len(self.varyList) != len(self.fullVaryList):
+            num = len(self.fullVaryList) - len(self.varyList)
+            mainSizer.Add(wx.StaticText(self.panel,label=' + '+str(num)+' parameters are varied via constraints'))
+        choiceDict = {'Global':self.globNames,'Phase':self.phasNames,'Phase/Histo':self.hapNames,'Histogram':self.hisNames}
+        choice = ['Phase','Phase/Histo','Histogram']
+        if len(self.globNames):
+            choice += ['Global',]
+        parmSizer = wx.FlexGridSizer(0,3,5,5)
+        parmSel = wx.RadioBox(self.panel,wx.ID_ANY,'Parameter type:',choices=choice,
+            majorDimension=1,style=wx.RA_SPECIFY_COLS)
+        parmSel.Bind(wx.EVT_RADIOBOX,_OnParmSel)
+        parmSel.SetStringSelection(self.parmChoice)
+        parmSizer.Add(parmSel,0)
+        numSizer = wx.BoxSizer(wx.VERTICAL)
+        numSizer.Add((5,25),0)
+        if self.parmChoice in ['Phase','Phase/Histo'] and len(self.phasNums) > 1:
+            numSizer.Add(wx.StaticText(self.panel,label='Phase'),0)
+            phasSel = wx.ComboBox(self.panel,choices=self.phasNums,value=self.phasNum,
+                style=wx.CB_READONLY|wx.CB_DROPDOWN)
+            phasSel.Bind(wx.EVT_COMBOBOX,OnPhasSel)
+            numSizer.Add(phasSel,0)
+        if self.parmChoice in ['Histogram','Phase/Histo'] and len(self.hisNums) > 1:
+            numSizer.Add(wx.StaticText(self.panel,label='Histogram'),0)
+            histSel = wx.ComboBox(self.panel,choices=self.hisNums,value=self.hisNum,
+                style=wx.CB_READONLY|wx.CB_DROPDOWN)
+            histSel.Bind(wx.EVT_COMBOBOX,OnHistSel)
+#            histSel = wx.TextCtrl(self.panel,size=(50,25),value='0',style=wx.TE_PROCESS_ENTER)
+#            histSel.Bind(wx.EVT_TEXT_ENTER,OnHistSel)
+#            histSel.Bind(wx.EVT_KILL_FOCUS,OnHistSel)
+            numSizer.Add(histSel,0)
+        parmSizer.Add(numSizer)
+        varSizer = wx.BoxSizer(wx.VERTICAL)
+        if self.parmChoice in ['Phase',]:
+            varSel = wx.ComboBox(self.panel,choices=self.phasVars,value=self.varName,
+                style=wx.CB_READONLY|wx.CB_DROPDOWN)
+            varSel.Bind(wx.EVT_COMBOBOX,OnVarSel)
+        elif self.parmChoice in ['Histogram',]:
+            varSel = wx.ComboBox(self.panel,choices=self.hisVars,value=self.varName,
+                style=wx.CB_READONLY|wx.CB_DROPDOWN)
+            varSel.Bind(wx.EVT_COMBOBOX,OnVarSel)
+        elif self.parmChoice in ['Phase/Histo',]:
+            varSel = wx.ComboBox(self.panel,choices=self.hapVars,value=self.varName,
+                style=wx.CB_READONLY|wx.CB_DROPDOWN)
+            varSel.Bind(wx.EVT_COMBOBOX,OnVarSel)
+        if self.parmChoice != 'Global': 
+            varSizer.Add(wx.StaticText(self.panel,label='Parameter'))
+            varSizer.Add(varSel,0)
+        parmSizer.Add(varSizer,0)
+        mainSizer.Add(parmSizer,0)
+        listChoice = ['All','Refined']
+        listSel = wx.RadioBox(self.panel,wx.ID_ANY,'Parameter type:',choices=listChoice,
+            majorDimension=0,style=wx.RA_SPECIFY_COLS)
+        listSel.SetStringSelection(self.listSel)
+        listSel.Bind(wx.EVT_RADIOBOX,OnListSel)
+        mainSizer.Add(listSel,0)
+        subSizer = wx.FlexGridSizer(cols=4,hgap=2,vgap=2)
+        subSizer.Add((-1,-1))
+        subSizer.Add(wx.StaticText(self.panel,wx.ID_ANY,'Parameter name  '))
+        subSizer.Add(wx.StaticText(self.panel,wx.ID_ANY,'refine?'))
+        subSizer.Add(wx.StaticText(self.panel,wx.ID_ANY,'value'),0,wx.ALIGN_RIGHT)
+        explainRefine = False
+        for name in choiceDict[self.parmChoice]:
+            # skip entries without numerical values
+            if isinstance(self.parmDict[name],basestring): continue
+            if 'Refined' in self.listSel and (name not in self.fullVaryList
+                                              ) and (name not in self.varyList):
+                continue
+            if 'Phase' in self.parmChoice:
+                if self.phasNum != ' ' and name.split(':')[0] != self.phasNum: continue
+            if 'Histo' in self.parmChoice:
+                if self.hisNum != ' ' and name.split(':')[1] != self.hisNum: continue
+            if (self.varName != ' ') and (self.varName not in name): continue
+            try:
+                value = G2py3.FormatSigFigs(self.parmDict[name])
+            except TypeError:
+                value = str(self.parmDict[name])+' -?' # unexpected
+                #continue
+            v = G2obj.getVarDescr(name)
+            if v is None or v[-1] is None:
+                subSizer.Add((-1,-1))
+            else:                
+                ch = HelpButton(self.panel,G2obj.fmtVarDescr(name))
+                subSizer.Add(ch,0,wx.LEFT|wx.RIGHT|WACV|wx.ALIGN_CENTER,1)
+            subSizer.Add(wx.StaticText(self.panel,wx.ID_ANY,str(name)))
+            if name in self.varyList:
+                subSizer.Add(wx.StaticText(self.panel,label='R'))   #TODO? maybe a checkbox for one stop refinemnt flag setting?
+            elif name in self.fullVaryList:
+                subSizer.Add(wx.StaticText(self.panel,label='C'))
+                explainRefine = True
+            else:
+                subSizer.Add((-1,-1))
+            subSizer.Add(wx.StaticText(self.panel,label=value),0,wx.ALIGN_RIGHT)
+
+        mainSizer.Add(subSizer,0)
+        if explainRefine:
+            mainSizer.Add(
+                wx.StaticText(self.panel,label='"R" indicates a refined variable\n'+
+                    '"C" indicates generated from a constraint'),0, wx.ALL,0)
+        # make OK button 
+        btnsizer = wx.BoxSizer(wx.HORIZONTAL)
+        btn = wx.Button(self.panel, wx.ID_CLOSE,"Close") 
+        btn.Bind(wx.EVT_BUTTON,self._onClose)
+        btnsizer.Add(btn)
+        mainSizer.Add(btnsizer, 0, wx.ALIGN_CENTER|wx.ALL, 5)
+        # Allow window to be enlarged but not made smaller
+        self.panel.SetSizer(mainSizer)
+        self.panel.SetAutoLayout(1)
+        self.panel.SetupScrolling()
+        self.panel.SetMinSize(self.GetSize())
+
+    def _onClose(self,event):
+        self.EndModal(wx.ID_CANCEL)
 
 ################################################################################