Changeset 2905 for branch/2frame/GSASIIplot.py
- Timestamp:
- Jul 4, 2017 12:11:29 PM (5 years ago)
- File:
-
- 1 edited
Legend:
- Unmodified
- Added
- Removed
-
branch/2frame/GSASIIplot.py
r2900 r2905 368 368 ''' 369 369 self.allowZoomReset = False 370 if pid: self.G2frame. PatternTree.SelectItem(pid)370 if pid: self.G2frame.GPXtree.SelectItem(pid) 371 371 self.allowZoomReset = True 372 372 if plotDebug: print 'invoke',str(self).split('0x')[1],str(pid) … … 555 555 G2frame.G2plotNB.status.SetStatusText('H = '+HKLtext,0) 556 556 557 Name = G2frame. PatternTree.GetItemText(G2frame.PatternId)557 Name = G2frame.GPXtree.GetItemText(G2frame.PatternId) 558 558 if not Title: 559 559 Title = Name … … 809 809 Draw('key') 810 810 811 Name = G2frame. PatternTree.GetItemText(G2frame.PatternId)811 Name = G2frame.GPXtree.GetItemText(G2frame.PatternId) 812 812 if Title and Title in G2frame.GetPhaseData(): #NB: save image as e.g. jpeg will fail if False; MyDir is unknown 813 813 generalData = G2frame.GetPhaseData()[Title]['General'] … … 1191 1191 if not histogram.startswith('PWDR'): return 1192 1192 pickId = G2frame.PickId 1193 G2frame.PickId = G2frame.PatternId = G2gd.Get PatternTreeItemId(G2frame, G2frame.root, histogram)1194 treedata = G2frame. PatternTree.GetItemPyData(G2frame.PatternId)1193 G2frame.PickId = G2frame.PatternId = G2gd.GetGPXtreeItemId(G2frame, G2frame.root, histogram) 1194 treedata = G2frame.GPXtree.GetItemPyData(G2frame.PatternId) 1195 1195 PlotPatterns(G2frame,newPlot=True,plotType='PWDR',data=[treedata[0],refdata]) 1196 1196 wx.Yield() # force a plot update (needed on Windows?) … … 1203 1203 ''' 1204 1204 if PatternName: 1205 G2frame.PatternId = G2gd.Get PatternTreeItemId(G2frame, G2frame.root, PatternName)1205 G2frame.PatternId = G2gd.GetGPXtreeItemId(G2frame, G2frame.root, PatternName) 1206 1206 if PickName == PatternName: 1207 1207 G2frame.PickId = G2frame.PatternId 1208 1208 elif PickName: 1209 G2frame.PickId = G2gd.Get PatternTreeItemId(G2frame, G2frame.PatternId, PickName)1209 G2frame.PickId = G2gd.GetGPXtreeItemId(G2frame, G2frame.PatternId, PickName) 1210 1210 # for now I am not sure how to regenerate G2frame.HKL 1211 1211 G2frame.HKL = [] # TODO … … 1237 1237 #patch 1238 1238 if data is None: 1239 data = G2frame. PatternTree.GetItemPyData(G2frame.PatternId)1239 data = G2frame.GPXtree.GetItemPyData(G2frame.PatternId) 1240 1240 if 'Offset' not in data[0] and plotType in ['PWDR','SASD','REFD']: #plot offset data 1241 1241 Ymax = max(data[1][1]) 1242 1242 data[0].update({'Offset':[0.0,0.0],'delOffset':0.02*Ymax,'refOffset':-0.1*Ymax, 1243 1243 'refDelt':0.1*Ymax,}) 1244 G2frame. PatternTree.SetItemPyData(G2frame.PickId,data)1244 G2frame.GPXtree.SetItemPyData(G2frame.PickId,data) 1245 1245 #end patch 1246 1246 def OnPlotKeyPress(event): 1247 1247 try: #one way to check if key stroke will work on plot 1248 Parms,Parms2 = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters'))1248 Parms,Parms2 = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters')) 1249 1249 except TypeError: 1250 1250 G2frame.G2plotNB.status.SetStatusText('Select '+plottype+' pattern first',1) … … 1335 1335 newPlot = True 1336 1336 elif event.key == 'f' and not G2frame.SinglePlot: 1337 choices = G2gd.Get PatternTreeDataNames(G2frame,plotType)1337 choices = G2gd.GetGPXtreeDataNames(G2frame,plotType) 1338 1338 dlg = G2G.G2MultiChoiceDialog(G2frame.dataFrame,'Select dataset to plot', 1339 1339 'Multidata plot selection',choices) … … 1373 1373 Page.canvas.SetCursor(wx.CROSS_CURSOR) 1374 1374 try: 1375 Id = G2gd.Get PatternTreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters')1375 Id = G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters') 1376 1376 if not Id: return 1377 Parms,Parms2 = G2frame. PatternTree.GetItemPyData(Id)1377 Parms,Parms2 = G2frame.GPXtree.GetItemPyData(Id) 1378 1378 if G2frame.plotStyle['qPlot'] and 'PWDR' in plottype: 1379 1379 q = xpos … … 1422 1422 if G2frame.PickId: 1423 1423 found = [] 1424 pickIdText = G2frame. PatternTree.GetItemText(G2frame.PickId)1424 pickIdText = G2frame.GPXtree.GetItemText(G2frame.PickId) 1425 1425 if pickIdText in ['Index Peak List','Unit Cells List','Reflection Lists'] or \ 1426 1426 'PWDR' in pickIdText: … … 1546 1546 1547 1547 try: 1548 Parms,Parms2 = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters'))1548 Parms,Parms2 = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters')) 1549 1549 except TypeError: 1550 1550 return … … 1570 1570 PatternId = G2frame.PatternId 1571 1571 PickId = G2frame.PickId 1572 if G2frame. PatternTree.GetItemText(PickId) == 'Peak List':1572 if G2frame.GPXtree.GetItemText(PickId) == 'Peak List': 1573 1573 if ind.all() != [0] and ObsLine[0].get_label() in str(pick): #picked a data point, add a new peak 1574 data = G2frame. PatternTree.GetItemPyData(G2frame.PickId)1574 data = G2frame.GPXtree.GetItemPyData(G2frame.PickId) 1575 1575 XY = G2mth.setPeakparms(Parms,Parms2,xy[0],xy[1]) 1576 1576 data['peaks'].append(XY) … … 1588 1588 G2frame.cid = Page.canvas.mpl_connect('motion_notify_event', OnDragLine) 1589 1589 pick.set_linestyle('--') # back to dashed 1590 elif G2frame. PatternTree.GetItemText(PickId) == 'Limits':1590 elif G2frame.GPXtree.GetItemText(PickId) == 'Limits': 1591 1591 if ind.all() != [0]: #picked a data point 1592 LimitId = G2gd.Get PatternTreeItemId(G2frame,PatternId, 'Limits')1593 data = G2frame. PatternTree.GetItemPyData(LimitId)1592 LimitId = G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Limits') 1593 data = G2frame.GPXtree.GetItemPyData(LimitId) 1594 1594 if G2frame.plotStyle['qPlot']: #qplot - convert back to 2-theta 1595 1595 xy[0] = G2lat.Dsp2pos(Parms,2*np.pi/xy[0]) … … 1607 1607 if mouse.button==3: 1608 1608 data[1][1] = max(xy[0],data[1][0]) 1609 G2frame. PatternTree.SetItemPyData(LimitId,data)1609 G2frame.GPXtree.SetItemPyData(LimitId,data) 1610 1610 G2pdG.UpdateLimitsGrid(G2frame,data,plottype) 1611 1611 wx.CallAfter(PlotPatterns,G2frame,plotType=plottype) … … 1621 1621 pick.set_linestyle('--') # back to dashed 1622 1622 1623 elif G2frame. PatternTree.GetItemText(PickId) == 'Models':1623 elif G2frame.GPXtree.GetItemText(PickId) == 'Models': 1624 1624 if ind.all() != [0]: #picked a data point 1625 LimitId = G2gd.Get PatternTreeItemId(G2frame,PatternId, 'Limits')1626 data = G2frame. PatternTree.GetItemPyData(LimitId)1625 LimitId = G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Limits') 1626 data = G2frame.GPXtree.GetItemPyData(LimitId) 1627 1627 if mouse.button==1: 1628 1628 data[1][0] = min(xy[0],data[1][1]) 1629 1629 if mouse.button==3: 1630 1630 data[1][1] = max(xy[0],data[1][0]) 1631 G2frame. PatternTree.SetItemPyData(LimitId,data)1631 G2frame.GPXtree.SetItemPyData(LimitId,data) 1632 1632 wx.CallAfter(PlotPatterns,G2frame,plotType=plottype) 1633 1633 else: #picked a limit line 1634 1634 G2frame.itemPicked = pick 1635 elif (G2frame. PatternTree.GetItemText(PickId) == 'Reflection Lists' or1636 'PWDR' in G2frame. PatternTree.GetItemText(PickId)1635 elif (G2frame.GPXtree.GetItemText(PickId) == 'Reflection Lists' or 1636 'PWDR' in G2frame.GPXtree.GetItemText(PickId) 1637 1637 ): 1638 1638 G2frame.itemPicked = pick … … 1664 1664 G2frame.cid = Page.canvas.mpl_connect('motion_notify_event', OnDragTickmarks) 1665 1665 1666 elif G2frame. PatternTree.GetItemText(PickId) == 'Background':1666 elif G2frame.GPXtree.GetItemText(PickId) == 'Background': 1667 1667 # selected a fixed background point. Can move it or delete it. 1668 1668 for mode,id in G2frame.dataFrame.wxID_BackPts.iteritems(): # what menu is selected? … … 1672 1672 if pick.get_marker() == 'D': 1673 1673 # find the closest point 1674 backDict = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Background'))[1]1674 backDict = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Background'))[1] 1675 1675 d2 = [(x-xy[0])**2+(y-xy[1])**2 for x,y in backDict['FixedPoints']] 1676 1676 G2frame.fixPtMarker = d2.index(min(d2)) … … 1708 1708 1709 1709 PickId = G2frame.PickId # points to item in tree 1710 if G2frame. PatternTree.GetItemText(PickId) == 'Background' and event.xdata:1710 if G2frame.GPXtree.GetItemText(PickId) == 'Background' and event.xdata: 1711 1711 if Page.toolbar._active: # prevent ops. if a toolbar zoom button pressed 1712 1712 return … … 1714 1714 if G2frame.SubBack or G2frame.Weight or G2frame.Contour or not G2frame.SinglePlot: 1715 1715 return 1716 backDict = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Background'))[1]1716 backDict = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Background'))[1] 1717 1717 if 'FixedPoints' not in backDict: backDict['FixedPoints'] = [] 1718 1718 try: 1719 Parms,Parms2 = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters'))1719 Parms,Parms2 = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters')) 1720 1720 except TypeError: 1721 1721 return … … 1745 1745 if G2frame.itemPicked is None: return 1746 1746 if DifLine[0] is G2frame.itemPicked: # respond to dragging of the difference curve 1747 data = G2frame. PatternTree.GetItemPyData(PickId)1747 data = G2frame.GPXtree.GetItemPyData(PickId) 1748 1748 ypos = event.ydata 1749 1749 Pattern[0]['delOffset'] = -ypos … … 1751 1751 wx.CallAfter(PlotPatterns,G2frame,plotType=plottype) 1752 1752 return 1753 Parms,Parms2 = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters'))1753 Parms,Parms2 = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters')) 1754 1754 xpos = event.xdata 1755 if G2frame. PatternTree.GetItemText(PickId) in ['Peak List','Limits'] and xpos:1755 if G2frame.GPXtree.GetItemText(PickId) in ['Peak List','Limits'] and xpos: 1756 1756 lines = [] 1757 1757 for line in G2frame.Lines: … … 1763 1763 nxcl = len(exclLines) 1764 1764 if lineNo in [0,1] or lineNo in exclLines: 1765 LimitId = G2gd.Get PatternTreeItemId(G2frame,G2frame.PatternId, 'Limits')1766 limits = G2frame. PatternTree.GetItemPyData(LimitId)1765 LimitId = G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Limits') 1766 limits = G2frame.GPXtree.GetItemPyData(LimitId) 1767 1767 id = lineNo/2+1 1768 1768 id2 = lineNo%2 … … 1777 1777 limits[1][0] = min(max(limits[0][0],limits[1][0]),limits[1][1]) 1778 1778 limits[1][1] = max(min(limits[0][1],limits[1][1]),limits[1][0]) 1779 if G2frame. PatternTree.GetItemText(G2frame.PickId) == 'Limits':1779 if G2frame.GPXtree.GetItemText(G2frame.PickId) == 'Limits': 1780 1780 G2pdG.UpdateLimitsGrid(G2frame,limits,plottype) 1781 1781 elif lineNo > 1+nxcl: 1782 PeakId = G2gd.Get PatternTreeItemId(G2frame,G2frame.PatternId, 'Peak List')1783 peaks = G2frame. PatternTree.GetItemPyData(PeakId)1782 PeakId = G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Peak List') 1783 peaks = G2frame.GPXtree.GetItemPyData(PeakId) 1784 1784 if event.button == 3: 1785 1785 del peaks['peaks'][lineNo-2-nxcl] … … 1793 1793 peaks['sigDict'] = {} #no longer valid 1794 1794 G2pdG.UpdatePeakGrid(G2frame,peaks) 1795 elif G2frame. PatternTree.GetItemText(PickId) in ['Models',] and xpos:1795 elif G2frame.GPXtree.GetItemText(PickId) in ['Models',] and xpos: 1796 1796 lines = [] 1797 1797 for line in G2frame.Lines: … … 1802 1802 lineNo = -1 1803 1803 if lineNo in [0,1]: 1804 LimitId = G2gd.Get PatternTreeItemId(G2frame,G2frame.PatternId, 'Limits')1805 data = G2frame. PatternTree.GetItemPyData(LimitId)1804 LimitId = G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Limits') 1805 data = G2frame.GPXtree.GetItemPyData(LimitId) 1806 1806 data[1][lineNo] = xpos 1807 1807 data[1][0] = min(max(data[0][0],data[1][0]),data[1][1]) 1808 1808 data[1][1] = max(min(data[0][1],data[1][1]),data[1][0]) 1809 elif (G2frame. PatternTree.GetItemText(PickId) == 'Reflection Lists' or \1810 'PWDR' in G2frame. PatternTree.GetItemText(PickId)) and xpos:1811 Id = G2gd.Get PatternTreeItemId(G2frame,PatternId,'Reflection Lists')1809 elif (G2frame.GPXtree.GetItemText(PickId) == 'Reflection Lists' or \ 1810 'PWDR' in G2frame.GPXtree.GetItemText(PickId)) and xpos: 1811 Id = G2gd.GetGPXtreeItemId(G2frame,PatternId,'Reflection Lists') 1812 1812 # GSASIIpath.IPyBreak() 1813 1813 if Id: 1814 #Phases = G2frame. PatternTree.GetItemPyData(Id)1814 #Phases = G2frame.GPXtree.GetItemPyData(Id) 1815 1815 pick = str(G2frame.itemPicked).split('(',1)[1][:-1] 1816 1816 if 'line' not in pick: #avoid data points, etc. 1817 data = G2frame. PatternTree.GetItemPyData(G2frame.PatternId)1817 data = G2frame.GPXtree.GetItemPyData(G2frame.PatternId) 1818 1818 num = Page.phaseList.index(pick) 1819 1819 if num: … … 1840 1840 Page.canvas.mpl_connect('button_release_event', OnRelease) 1841 1841 Page.canvas.mpl_connect('button_press_event',OnPress) 1842 if 'PWDR' in G2frame. PatternTree.GetItemText(G2frame.PickId):1842 if 'PWDR' in G2frame.GPXtree.GetItemText(G2frame.PickId): 1843 1843 Histograms,Phases = G2frame.GetUsedHistogramsAndPhasesfromTree() 1844 1844 refColors=['b','r','c','g','m','k'] 1845 1845 Page.phaseColors = {p:refColors[i%len(refColors)] for i,p in enumerate(Phases)} 1846 Phases = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId,'Reflection Lists'))1846 Phases = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId,'Reflection Lists')) 1847 1847 Page.phaseList = sorted(Phases.keys()) # define an order for phases (once!) 1848 1848 G2frame.dataWindow.Bind(wx.EVT_MENU, onMoveDiffCurve, id=G2frame.dataWindow.moveDiffCurve.GetId()) … … 1852 1852 G2frame.dataWindow.moveTickLoc.Enable(False) 1853 1853 G2frame.dataWindow.moveTickSpc.Enable(False) 1854 elif G2frame. PatternTree.GetItemText(G2frame.PickId) == 'Peak List':1854 elif G2frame.GPXtree.GetItemText(G2frame.PickId) == 'Peak List': 1855 1855 G2frame.dataFrame.Bind(wx.EVT_MENU, onMovePeak, id=G2frame.dataWindow.movePeak.GetId()) 1856 1856 # save information needed to reload from tree and redraw 1857 kwargs={'PatternName':G2frame. PatternTree.GetItemText(G2frame.PatternId)}1857 kwargs={'PatternName':G2frame.GPXtree.GetItemText(G2frame.PatternId)} 1858 1858 if G2frame.PickId: 1859 kwargs['PickName'] = G2frame. PatternTree.GetItemText(G2frame.PickId)1859 kwargs['PickName'] = G2frame.GPXtree.GetItemText(G2frame.PickId) 1860 1860 #G2frame.G2plotNB.RegisterRedrawRoutine('Powder Patterns',ReplotPattern, 1861 1861 G2frame.G2plotNB.RegisterRedrawRoutine(G2frame.G2plotNB.lastRaisedPlotTab,ReplotPattern, … … 1916 1916 exclLines = [] 1917 1917 if G2frame.SinglePlot and PatternId: 1918 Pattern = G2frame. PatternTree.GetItemPyData(PatternId)1919 Pattern.append(G2frame. PatternTree.GetItemText(PatternId))1918 Pattern = G2frame.GPXtree.GetItemPyData(PatternId) 1919 Pattern.append(G2frame.GPXtree.GetItemText(PatternId)) 1920 1920 PlotList = [Pattern,] 1921 Parms,Parms2 = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,1921 Parms,Parms2 = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame, 1922 1922 G2frame.PatternId, 'Instrument Parameters')) 1923 Sample = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Sample Parameters'))1923 Sample = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Sample Parameters')) 1924 1924 ParmList = [Parms,] 1925 1925 SampleList = [Sample,] … … 1931 1931 SampleList = [] 1932 1932 if G2frame.selections is None: 1933 choices = G2gd.Get PatternTreeDataNames(G2frame,plotType)1933 choices = G2gd.GetGPXtreeDataNames(G2frame,plotType) 1934 1934 else: 1935 1935 choices = G2frame.selections 1936 1936 for item in choices: 1937 id = G2gd.Get PatternTreeItemId(G2frame,G2frame.root, item)1938 Pattern = G2frame. PatternTree.GetItemPyData(id)1937 id = G2gd.GetGPXtreeItemId(G2frame,G2frame.root, item) 1938 Pattern = G2frame.GPXtree.GetItemPyData(id) 1939 1939 if len(Pattern) < 3: # put name on end if needed 1940 Pattern.append(G2frame. PatternTree.GetItemText(id))1940 Pattern.append(G2frame.GPXtree.GetItemText(id)) 1941 1941 if 'Offset' not in Pattern[0]: #plot offset data 1942 1942 Ymax = max(Pattern[1][1]) 1943 1943 Pattern[0].update({'Offset':[0.0,0.0],'delOffset':0.02*Ymax,'refOffset':-0.1*Ymax,'refDelt':0.1*Ymax,}) 1944 1944 PlotList.append(Pattern) 1945 ParmList.append(G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,1945 ParmList.append(G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame, 1946 1946 id,'Instrument Parameters'))[0]) 1947 SampleList.append(G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,1947 SampleList.append(G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame, 1948 1948 id, 'Sample Parameters'))) 1949 1949 lenX = 0 … … 2011 2011 xye = ma.array(ma.getdata(Pattern[1])) 2012 2012 if PickId: 2013 ifpicked = Pattern[2] == G2frame. PatternTree.GetItemText(PatternId)2014 LimitId = G2gd.Get PatternTreeItemId(G2frame,G2frame.PatternId,'Limits')2015 limits = G2frame. PatternTree.GetItemPyData(LimitId)2013 ifpicked = Pattern[2] == G2frame.GPXtree.GetItemText(PatternId) 2014 LimitId = G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId,'Limits') 2015 limits = G2frame.GPXtree.GetItemPyData(LimitId) 2016 2016 excls = limits[2:] 2017 2017 for excl in excls: … … 2042 2042 Y = xye[1]*Sample['Scale'][0]*(1.05)**(offsetY*N) 2043 2043 if LimitId and ifpicked: 2044 limits = np.array(G2frame. PatternTree.GetItemPyData(LimitId))2044 limits = np.array(G2frame.GPXtree.GetItemPyData(LimitId)) 2045 2045 lims = limits[1] 2046 2046 if G2frame.plotStyle['qPlot'] and 'PWDR' in plottype: … … 2169 2169 Page.canvas.SetToolTipString('') 2170 2170 if PickId: 2171 if G2frame. PatternTree.GetItemText(PickId) == 'Peak List':2171 if G2frame.GPXtree.GetItemText(PickId) == 'Peak List': 2172 2172 tip = 'On data point: Pick peak - L or R MB. On line: L-move, R-delete' 2173 2173 Page.canvas.SetToolTipString(tip) 2174 data = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId, 'Peak List'))2174 data = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Peak List')) 2175 2175 selectedPeaks = list(set( 2176 2176 [row for row,col in G2frame.reflGrid.GetSelectedCells()] + … … 2190 2190 if Ni == N+1: 2191 2191 Lines[-1].set_lw(Lines[-1].get_lw()+1) 2192 if G2frame. PatternTree.GetItemText(PickId) == 'Limits':2192 if G2frame.GPXtree.GetItemText(PickId) == 'Limits': 2193 2193 tip = 'On data point: Lower limit - L MB; Upper limit - R MB. On limit: MB down to move' 2194 2194 Page.canvas.SetToolTipString(tip) 2195 data = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId, 'Limits'))2195 data = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Limits')) 2196 2196 2197 2197 else: #not picked … … 2211 2211 Plot.set_ylim(bottom=np.min(np.trim_zeros(Y))/2.,top=np.max(Y)*2.) 2212 2212 if PickId and not G2frame.Contour: 2213 Parms,Parms2 = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId, 'Instrument Parameters'))2214 if G2frame. PatternTree.GetItemText(PickId) in ['Index Peak List','Unit Cells List']:2215 peaks = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId, 'Index Peak List'))2213 Parms,Parms2 = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Instrument Parameters')) 2214 if G2frame.GPXtree.GetItemText(PickId) in ['Index Peak List','Unit Cells List']: 2215 peaks = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Index Peak List')) 2216 2216 if not len(peaks): return # are there any peaks? 2217 2217 for peak in peaks[0]: … … 2233 2233 else: 2234 2234 Plot.axvline(hkl[-2],color=clr,dashes=(5,5)) 2235 elif G2frame. PatternTree.GetItemText(PickId) in ['Reflection Lists'] or \2236 'PWDR' in G2frame. PatternTree.GetItemText(PickId):2237 Phases = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId,'Reflection Lists'))2235 elif G2frame.GPXtree.GetItemText(PickId) in ['Reflection Lists'] or \ 2236 'PWDR' in G2frame.GPXtree.GetItemText(PickId): 2237 Phases = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId,'Reflection Lists')) 2238 2238 l = GSASIIpath.GetConfigValue('Tick_length',8.0) 2239 2239 w = GSASIIpath.GetConfigValue('Tick_width',1.) … … 2268 2268 else: 2269 2269 G2frame.Lines = Lines 2270 if PickId and G2frame. PatternTree.GetItemText(PickId) == 'Background':2270 if PickId and G2frame.GPXtree.GetItemText(PickId) == 'Background': 2271 2271 # plot fixed background points 2272 backDict = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Background'))[1]2272 backDict = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Background'))[1] 2273 2273 try: 2274 Parms,Parms2 = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters'))2274 Parms,Parms2 = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,G2frame.PatternId, 'Instrument Parameters')) 2275 2275 except TypeError: 2276 2276 Parms = None … … 2310 2310 # and sqrt(-ve) in np.where usage 2311 2311 # G2frame.Pwdr = True 2312 if 'PWDR' in G2frame. PatternTree.GetItemText(G2frame.PickId):2312 if 'PWDR' in G2frame.GPXtree.GetItemText(G2frame.PickId): 2313 2313 if len(Page.tickDict.keys()) == 1: 2314 2314 G2frame.dataWindow.moveTickLoc.Enable(True) … … 2326 2326 'Produces normal probability plot for a powder or single crystal histogram' 2327 2327 if PatternName: 2328 G2frame.PatternId = G2gd.Get PatternTreeItemId(G2frame, G2frame.root, PatternName)2328 G2frame.PatternId = G2gd.GetGPXtreeItemId(G2frame, G2frame.root, PatternName) 2329 2329 new,plotNum,Page,Plot,lim = G2frame.G2plotNB.FindPlotTab('Error analysis','mpl') 2330 2330 if new: … … 2335 2335 PlotDeltSig,( 2336 2336 G2frame,kind, 2337 G2frame. PatternTree.GetItemText(G2frame.PatternId))2337 G2frame.GPXtree.GetItemText(G2frame.PatternId)) 2338 2338 ) 2339 2339 Page.Choice = None 2340 2340 PatternId = G2frame.PatternId 2341 Pattern = G2frame. PatternTree.GetItemPyData(PatternId)2342 Pattern.append(G2frame. PatternTree.GetItemText(PatternId))2341 Pattern = G2frame.GPXtree.GetItemPyData(PatternId) 2342 Pattern.append(G2frame.GPXtree.GetItemText(PatternId)) 2343 2343 wtFactor = Pattern[0]['wtFactor'] 2344 2344 if kind == 'PWDR': 2345 limits = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId, 'Limits'))[1]2345 limits = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Limits'))[1] 2346 2346 xye = np.array(Pattern[1]) 2347 2347 xmin = np.searchsorted(xye[0],limits[0]) … … 2446 2446 G2frame.Waterfall = not G2frame.Waterfall 2447 2447 elif event.key == 'f' and not G2frame.SinglePlot: 2448 choices = G2gd.Get PatternTreeDataNames(G2frame,'PDF ')2448 choices = G2gd.GetGPXtreeDataNames(G2frame,'PDF ') 2449 2449 dlg = G2G.G2MultiChoiceDialog(G2frame.dataFrame,'Select dataset to plot', 2450 2450 'Multidata plot selection',choices) … … 2596 2596 PatternId = G2frame.PatternId 2597 2597 if not PatternId: return 2598 pId = G2gd.Get PatternTreeItemId(G2frame,PatternId, 'PDF Controls')2598 pId = G2gd.GetGPXtreeItemId(G2frame,PatternId, 'PDF Controls') 2599 2599 if not pId: return 2600 PDFdata = G2frame. PatternTree.GetItemPyData(pId)2600 PDFdata = G2frame.GPXtree.GetItemPyData(pId) 2601 2601 numbDen = 0. 2602 2602 if 'ElList' in PDFdata: … … 2610 2610 PlotList = [] 2611 2611 if G2frame.PDFselections is None: 2612 choices = G2gd.Get PatternTreeDataNames(G2frame,'PDF ')2612 choices = G2gd.GetGPXtreeDataNames(G2frame,'PDF ') 2613 2613 else: 2614 2614 choices = G2frame.PDFselections 2615 2615 for item in choices: 2616 Pid = G2gd.Get PatternTreeItemId(G2frame,G2frame.root,item)2617 Id = G2gd.Get PatternTreeItemId(G2frame,Pid,'PDF Controls')2618 Pattern = G2frame. PatternTree.GetItemPyData(Id)2616 Pid = G2gd.GetGPXtreeItemId(G2frame,G2frame.root,item) 2617 Id = G2gd.GetGPXtreeItemId(G2frame,Pid,'PDF Controls') 2618 Pattern = G2frame.GPXtree.GetItemPyData(Id) 2619 2619 if Pattern: 2620 2620 PlotList.append(Pattern[plotType]) … … 3155 3155 Page.Choice = None 3156 3156 PatternId = G2frame.PatternId 3157 data = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId, 'Models'))3157 data = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Models')) 3158 3158 Bins,Dbins,BinMag = data['Size']['Distribution'] 3159 3159 Plot.set_title('Size Distribution') … … 3185 3185 Page.canvas.SetCursor(wx.CROSS_CURSOR) 3186 3186 G2frame.G2plotNB.status.SetStatusText('2-theta =%9.3f '%(xpos,),1) 3187 if G2frame.PickId and G2frame. PatternTree.GetItemText(G2frame.PickId) in ['Index Peak List','Unit Cells List']:3187 if G2frame.PickId and G2frame.GPXtree.GetItemText(G2frame.PickId) in ['Index Peak List','Unit Cells List']: 3188 3188 found = [] 3189 3189 if len(G2frame.HKL): … … 3204 3204 Plot.set_xlabel(r'$\mathsf{2\theta}$',fontsize=14) 3205 3205 PatternId = G2frame.PatternId 3206 peaks = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId, 'Index Peak List'))[0]3206 peaks = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Index Peak List'))[0] 3207 3207 for peak in peaks: 3208 3208 Plot.axvline(peak[0],color='b') … … 3232 3232 # gamFW2 = lambda s,g: math.sqrt(s**2+(0.4654996*g)**2)+.5345004*g #Ubaldo Bafile - private communication 3233 3233 if PatternName: 3234 G2frame.PatternId = G2gd.Get PatternTreeItemId(G2frame, G2frame.root, PatternName)3234 G2frame.PatternId = G2gd.GetGPXtreeItemId(G2frame, G2frame.root, PatternName) 3235 3235 PatternId = G2frame.PatternId 3236 limitID = G2gd.Get PatternTreeItemId(G2frame,PatternId, 'Limits')3236 limitID = G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Limits') 3237 3237 if limitID: 3238 limits = G2frame. PatternTree.GetItemPyData(limitID)[:2]3238 limits = G2frame.GPXtree.GetItemPyData(limitID)[:2] 3239 3239 else: 3240 3240 return 3241 Parms,Parms2 = G2frame. PatternTree.GetItemPyData( \3242 G2gd.Get PatternTreeItemId(G2frame,PatternId, 'Instrument Parameters'))3241 Parms,Parms2 = G2frame.GPXtree.GetItemPyData( \ 3242 G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Instrument Parameters')) 3243 3243 if 'PKS' in Parms['Type'][0]: 3244 3244 return … … 3248 3248 lam = G2mth.getWave(Parms) 3249 3249 try: # PATCH: deal with older peak lists, before changed to dict to implement TOF 3250 peaks = G2frame. PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,PatternId, 'Peak List'))['peaks']3250 peaks = G2frame.GPXtree.GetItemPyData(G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Peak List'))['peaks'] 3251 3251 except TypeError: 3252 3252 print "Your peak list needs reformatting...", 3253 item = G2gd.Get PatternTreeItemId(G2frame,PatternId, 'Peak List')3254 G2frame. PatternTree.SelectItem(item)3255 item = G2gd.Get PatternTreeItemId(G2frame,PatternId, 'Instrument Parameters')3256 G2frame. PatternTree.SelectItem(item)3253 item = G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Peak List') 3254 G2frame.GPXtree.SelectItem(item) 3255 item = G2gd.GetGPXtreeItemId(G2frame,PatternId, 'Instrument Parameters') 3256 G2frame.GPXtree.SelectItem(item) 3257 3257 print "done" 3258 3258 return … … 3264 3264 # save information needed to reload from tree and redraw 3265 3265 G2frame.G2plotNB.RegisterRedrawRoutine(G2frame.G2plotNB.lastRaisedPlotTab, 3266 PlotPeakWidths,(G2frame,G2frame. PatternTree.GetItemText(G2frame.PatternId))3266 PlotPeakWidths,(G2frame,G2frame.GPXtree.GetItemText(G2frame.PatternId)) 3267 3267 ) 3268 3268 3269 TreeItemText = G2frame. PatternTree.GetItemText(G2frame.PatternId)3269 TreeItemText = G2frame.GPXtree.GetItemText(G2frame.PatternId) 3270 3270 G2frame.G2plotNB.status.SetStatusText('histogram: '+TreeItemText,1) 3271 3271 Page.Choice = None … … 3571 3571 Plot.set_ylabel('MRD',fontsize=14) 3572 3572 elif plotType in ['Inv. pole figure',]: 3573 Id = G2gd.Get PatternTreeItemId(G2frame,G2frame.root,hist)3574 rId = G2gd.Get PatternTreeItemId(G2frame,Id,'Reflection Lists')3575 RefData = G2frame. PatternTree.GetItemPyData(rId)[phase]3573 Id = G2gd.GetGPXtreeItemId(G2frame,G2frame.root,hist) 3574 rId = G2gd.GetGPXtreeItemId(G2frame,Id,'Reflection Lists') 3575 RefData = G2frame.GPXtree.GetItemPyData(rId)[phase] 3576 3576 Type = RefData['Type'] 3577 3577 Refs = RefData['RefList'].T … … 4358 4358 :param wx.Frame G2frame: The main GSAS-II tree "window" 4359 4359 ''' 4360 Masks = G2frame. PatternTree.GetItemPyData(4361 G2gd.Get PatternTreeItemId(G2frame,G2frame.Image, 'Masks'))4360 Masks = G2frame.GPXtree.GetItemPyData( 4361 G2gd.GetGPXtreeItemId(G2frame,G2frame.Image, 'Masks')) 4362 4362 if G2frame.MaskKey == 'f': 4363 4363 new,plotNum,Page,Plot,lim = G2frame.G2plotNB.FindPlotTab('2D Powder Image','mpl',newImage=False) … … 4396 4396 ''' 4397 4397 G2frame.GetStatusBar().SetStatusText('Add strain ring active - LB pick d-zero value',0) 4398 G2frame.PickId = G2gd.Get PatternTreeItemId(G2frame,G2frame.Image, 'Stress/Strain')4399 data = G2frame. PatternTree.GetItemPyData(G2frame.PickId)4398 G2frame.PickId = G2gd.GetGPXtreeItemId(G2frame,G2frame.Image, 'Stress/Strain') 4399 data = G2frame.GPXtree.GetItemPyData(G2frame.PickId) 4400 4400 return data 4401 4401 … … 4467 4467 global Data,Masks,StrSta # RVD: these are needed for multiple image controls/masks 4468 4468 colors=['b','g','r','c','m','k'] 4469 Data = G2frame. PatternTree.GetItemPyData(4470 G2gd.Get PatternTreeItemId(G2frame,G2frame.Image, 'Image Controls'))4469 Data = G2frame.GPXtree.GetItemPyData( 4470 G2gd.GetGPXtreeItemId(G2frame,G2frame.Image, 'Image Controls')) 4471 4471 # patch 4472 4472 if 'invert_x' not in Data: … … 4474 4474 Data['invert_y'] = True 4475 4475 # end patch 4476 Masks = G2frame. PatternTree.GetItemPyData(4477 G2gd.Get PatternTreeItemId(G2frame,G2frame.Image, 'Masks'))4476 Masks = G2frame.GPXtree.GetItemPyData( 4477 G2gd.GetGPXtreeItemId(G2frame,G2frame.Image, 'Masks')) 4478 4478 try: #may be absent 4479 StrSta = G2frame. PatternTree.GetItemPyData(4480 G2gd.Get PatternTreeItemId(G2frame,G2frame.Image, 'Stress/Strain'))4479 StrSta = G2frame.GPXtree.GetItemPyData( 4480 G2gd.GetGPXtreeItemId(G2frame,G2frame.Image, 'Stress/Strain')) 4481 4481 except TypeError: #is missing 4482 4482 StrSta = {} … … 4492 4492 scalex = 1000./pixelSize[0] #microns --> 1/mm 4493 4493 scaley = 1000./pixelSize[1] 4494 if item and G2frame. PatternTree.GetItemText(G2frame.PickId) == 'Image Controls':4494 if item and G2frame.GPXtree.GetItemText(G2frame.PickId) == 'Image Controls': 4495 4495 if 'Text' in str(item): 4496 4496 Page.canvas.SetToolTipString('%8.3f %8.3fmm'%(event.xdata,event.ydata)) … … 4526 4526 def OnImPlotKeyPress(event): 4527 4527 try: 4528 treeItem = G2frame. PatternTree.GetItemText(G2frame.PickId)4528 treeItem = G2frame.GPXtree.GetItemText(G2frame.PickId) 4529 4529 except TypeError: 4530 4530 return … … 4792 4792 4793 4793 if G2frame.itemPicked is not None: return 4794 if G2frame. PatternTree.GetItemText(G2frame.PickId) == 'Image Controls':4794 if G2frame.GPXtree.GetItemText(G2frame.PickId) == 'Image Controls': 4795 4795 G2frame.itemPicked = pick = event.artist 4796 4796 G2frame.mousePicked = event.mouseevent … … 4804 4804 G2frame.cid = Page.canvas.mpl_connect('motion_notify_event', OnDragIntBound) 4805 4805 pick.set_linestyle(saveLinestyle) # back to original 4806 elif G2frame. PatternTree.GetItemText(G2frame.PickId) == 'Masks':4806 elif G2frame.GPXtree.GetItemText(G2frame.PickId) == 'Masks': 4807 4807 # prepare to animate dragging of mask 4808 4808 G2frame.itemPicked = pick = event.artist … … 4839 4839 ''' 4840 4840 try: 4841 treeItem = G2frame. PatternTree.GetItemText(G2frame.PickId)4841 treeItem = G2frame.GPXtree.GetItemText(G2frame.PickId) 4842 4842 except TypeError: 4843 4843 return … … 5079 5079 Page.canvas.mpl_connect('button_release_event', OnImRelease) 5080 5080 Page.Choice = None 5081 Title = G2frame. PatternTree.GetItemText(G2frame.Image)[4:]5081 Title = G2frame.GPXtree.GetItemText(G2frame.Image)[4:] 5082 5082 G2frame.G2plotNB.status.DestroyChildren() 5083 5083 if G2frame.logPlot: … … 5085 5085 Plot.set_title(Title) 5086 5086 try: 5087 if G2frame. PatternTree.GetItemText(G2frame.PickId) in ['Image Controls',]:5087 if G2frame.GPXtree.GetItemText(G2frame.PickId) in ['Image Controls',]: 5088 5088 Page.Choice = (' key press','l: log(I) on','d: set dmin','x: flip x','y: flip y',) 5089 5089 if G2frame.logPlot: 5090 5090 Page.Choice[1] = 'l: log(I) off' 5091 5091 Page.keyPress = OnImPlotKeyPress 5092 elif G2frame. PatternTree.GetItemText(G2frame.PickId) in ['Masks',]:5092 elif G2frame.GPXtree.GetItemText(G2frame.PickId) in ['Masks',]: 5093 5093 Page.Choice = [' key press','l: log(I) on','a: arc mask','r: ring mask', 5094 5094 'p: polygon mask','f: frame mask', … … 5099 5099 Page.Choice[1] = 'l: log(I) off' 5100 5100 Page.keyPress = OnImPlotKeyPress 5101 elif G2frame. PatternTree.GetItemText(G2frame.PickId) in ['Stress/Strain',]:5101 elif G2frame.GPXtree.GetItemText(G2frame.PickId) in ['Stress/Strain',]: 5102 5102 Page.Choice = (' key press','a: add new ring',) 5103 5103 Page.keyPress = OnImPlotKeyPress 5104 5104 except TypeError: 5105 5105 pass 5106 size,imagefile,imagetag = G2frame. PatternTree.GetImageLoc(G2frame.Image)5106 size,imagefile,imagetag = G2frame.GPXtree.GetImageLoc(G2frame.Image) 5107 5107 5108 5108 imScale = 1 … … 5142 5142 5143 5143 Plot.plot(xcent,ycent,'x') 5144 #G2frame. PatternTree.GetItemText(item)5144 #G2frame.GPXtree.GetItemText(item) 5145 5145 if Data['showLines']: # draw integration range arc/circles/lines 5146 5146 LRAzim = Data['LRazimuth'] #NB: integers … … 5185 5185 Plot.plot([arcxI[ind],arcxO[ind]],[arcyI[ind],arcyO[ind]],color='k',dashes=(5,5)) 5186 5186 5187 if G2frame.PickId and G2frame. PatternTree.GetItemText(G2frame.PickId) in ['Image Controls',]:5187 if G2frame.PickId and G2frame.GPXtree.GetItemText(G2frame.PickId) in ['Image Controls',]: 5188 5188 for xring,yring in Data['ring']: 5189 5189 Plot.plot(xring,yring,'r+',picker=3) … … 5199 5199 Plot.add_artist(Ellipse([cent[0],cent[1]],2*width,2*height,phi,ec=col,fc='none')) 5200 5200 Plot.text(cent[0],cent[1],'+',color=col,ha='center',va='center') 5201 if G2frame.PickId and G2frame. PatternTree.GetItemText(G2frame.PickId) in ['Stress/Strain',]:5201 if G2frame.PickId and G2frame.GPXtree.GetItemText(G2frame.PickId) in ['Stress/Strain',]: 5202 5202 for N,ring in enumerate(StrSta['d-zero']): 5203 5203 if 'ImxyCalc' in ring: … … 5327 5327 Page.Choice = None 5328 5328 5329 Data = G2frame. PatternTree.GetItemPyData(5330 G2gd.Get PatternTreeItemId(G2frame,G2frame.Image, 'Image Controls'))5329 Data = G2frame.GPXtree.GetItemPyData( 5330 G2gd.GetGPXtreeItemId(G2frame,G2frame.Image, 'Image Controls')) 5331 5331 image = G2frame.Integrate[0] 5332 5332 xsc = G2frame.Integrate[1] … … 5334 5334 Imin,Imax = Data['range'][1] 5335 5335 acolor = mpl.cm.get_cmap(Data['color']) 5336 Plot.set_title(G2frame. PatternTree.GetItemText(G2frame.Image)[4:])5336 Plot.set_title(G2frame.GPXtree.GetItemText(G2frame.Image)[4:]) 5337 5337 Plot.set_ylabel('azimuth',fontsize=12) 5338 5338 Plot.set_xlabel('2-theta',fontsize=12) … … 5381 5381 Page.views = False 5382 5382 Page.Choice = None 5383 Data = G2frame. PatternTree.GetItemPyData(5384 G2gd.Get PatternTreeItemId(G2frame,G2frame.Image, 'Image Controls'))5383 Data = G2frame.GPXtree.GetItemPyData( 5384 G2gd.GetGPXtreeItemId(G2frame,G2frame.Image, 'Image Controls')) 5385 5385 Imin,Imax = Data['range'][1] 5386 5386 step = (Imax-Imin)/5. 5387 5387 V = np.arange(Imin,Imax,step) 5388 5388 acolor = mpl.cm.get_cmap(Data['color']) 5389 Plot.set_title(G2frame. PatternTree.GetItemText(G2frame.Image)[4:])5389 Plot.set_title(G2frame.GPXtree.GetItemText(G2frame.Image)[4:]) 5390 5390 Plot.set_xlabel('azimuth',fontsize=12) 5391 5391 Plot.set_ylabel('2-theta',fontsize=12)
Note: See TracChangeset
for help on using the changeset viewer.