def readobject(directory=None):
import makeGrid
if directory == None:
directory = sys.argv[1]
fname = os.path.abspath(directory) + os.sep + 'batch.pyobj'
if not os.path.exists(fname):
if makeGrid.pathIsGrid(directory):
return makeGrid.makeGrid(directory, readOnly=True, interactive=False)
return None
try:
with open(fname, 'rb') as inp:
return pickle.load(inp)
except:
if makeGrid.pathIsGrid(directory):
return makeGrid.makeGrid(directory, readOnly=True, interactive=False)
raise
def openDirectory(self, dirName):
# Check if it's a grid
if makeGrid.pathIsGrid(dirName):
self.rootdirname = dirName
self.lineEditDirectory.setText(self.rootdirname)
self._readGridChains(self.rootdirname)
return
else:
if self.is_grid:
self._resetGridData()
root_list = MCSamples.GetChainRootFiles(dirName)
if not len(root_list):
QMessageBox.critical(self, "Open chains", "No chains or grid found in that directory")
return
self.rootdirname = dirName
self.lineEditDirectory.setText(self.rootdirname)
self.getPlotter(chain_dir=self.rootdirname)
self._updateComboBoxRootname(root_list)
