Changeset 725 for sans/utils/bt5/bt5plot/bt5plot.py

Timestamp:

Jul 7, 2010 1:43:59 AM (12 years ago)

Author:

ajj

Message:

Work to add auto grouping to bt5plot

File:

• sans/utils/bt5/bt5plot/bt5plot.py (modified) (4 diffs)

sans/utils/bt5/bt5plot/bt5plot.py

@@ -14 +14 @@
 import usans
 from BT5DataSet import BT5DataSet
+from BT5DataGroup import BT5DataGroup
 
 try:
@@ -33 +34 @@
     
     def __init__(self):
-        
-        gladefile = "bt5plot.glade"
+
+        runpath = os.path.dirname(os.path.realpath(__file__))
+        
+        gladefile = runpath+"/bt5plot.glade"
         self.windowname = "win_Main"
         self.wTree = gtk.glade.XML(gladefile, self.windowname)
@@ -60 +63 @@
         self.filelistview = self.wTree.get_widget("tv_filelist")
         
-        self.filelist = gtk.ListStore(str, 'gboolean', object)
+        self.filelist = gtk.TreeStore(str, 'gboolean', object)
         self.filelist.set_sort_column_id(0, gtk.SORT_ASCENDING)
 
@@ -121 +124 @@
     def FillFileList(self, filenames):
         self.filelist.clear()
+        
+        groupList = self.generateDataGroups(filenames)
+        
+        for group in groupList:
+                self.filelist.append(None, [group.groupName, 0, None])
+                
+        return
+            
+            
+    def generateDataGroups(self,filenames):
+        
+        datasetList = []
+        dataindexList = []
+        datagroupList = []
+        
+        
+        #assuming list hsa come in sorted by date...
         for filename in filenames:
-                self.filelist.append([filename, 0, None])
-        return
-            
+            #generate list of BT5DataSet objects
+            datasetList.append(BT5DataSet(filename))
+        
+        #print datasetList
+        #build list of list indices where either scanned motor is A2 or scanned motor is not A2
+        for dataset in datasetList:          
+            if dataset.scanmot == 'A2':
+                #check for 0
+                if 0 in dataset.detdata.keys():
+                    dataindexList.append(datasetList.index(dataset))
+            else:
+                dataindexList.append(datasetList.index(dataset))
+        
+        previndex = 0
+        for dataindex in dataindexList:
+            if dataindexList.index(dataindex) == len(dataindexList)-1:
+                nextindex = len(datasetList)
+            else:
+                nextindex = dataindex
+            datagroupList.append(BT5DataGroup(datasetList[previndex:nextindex]))
+            previndex = dataindex
+        
+        print len(datagroupList)
+        
+        return datagroupList
+    
+    
     def RefreshFileList(self, filenames):        
         #print len(filenames)