class DataManager(object):
def __init__(self, argv):
self.iom = IOManager(argv)
self.keys = {}
self.IF = ImageFactory()
# get keys from rootfile, iterate over the enum
# and see what's in the root file
for i in xrange(larcv.kProductUnknown):
product = larcv.ProductName(i)
self.keys[product] = []
producers = self.iom.iom.producer_list(i)
for p in producers:
self.keys[product].append(p)
# run subrun and event start at zero
self.run = -1
self.subrun = -1
self.event = -1
def get_nchannels(self, ii, imgprod):
# Sorry Vic I hacked this
# --> it's ok
self.iom.read_entry(ii)
imdata = self.iom.get_data(larcv.kProductImage2D, imgprod)
return imdata.Image2DArray().size()
def get_event_image(self, ii, imgprod, roiprod, planes, refresh=True):
#Load data in TChain
self.iom.read_entry(ii)
# there may be no ROI
hasroi = False
roidata = None
if roiprod is not None:
roidata = self.iom.iom.get_data(larcv.kProductROI, roiprod)
roidata = roidata.ROIArray()
hasroi = True
# get the EventImage2D
imdata = self.iom.get_data(larcv.kProductImage2D,
imgprod) # goes to disk
self.run = imdata.run()
self.subrun = imdata.subrun()
self.event = imdata.event()
# get the std::vector<larcv::Image2D>
imdata = imdata.Image2DArray()
if imdata.size() == 0: return (None, False)
# hand it off to the factory, the producer name should query the
# the correct subclass of PlotImage
image = self.IF.get(imdata, roidata, planes,
imgprod) # returns PlotImgae
# return it to rgbviewer
return (image, hasroi)
class WhiteDataManager(object):
def __init__(self,argv):
self.iom = IOManager(argv)
self.keys ={}
self.IF = WhiteImageFactory()
# get keys from rootfile, iterate over the enum
# and see what's in the root file
for i in xrange(larcv.kProductUnknown):
product = larcv.ProductName(i)
self.keys[product] = []
producers=self.iom.iom.producer_list(i)
for p in producers:
self.keys[product].append(p)
# run subrun and event start at zero
self.run = -1
self.subrun = -1
self.event = -1
def get_nchannels(self,ii,imgprod) :
# Sorry Vic I hacked this
# --> it's ok
self.iom.read_entry(ii)
imdata = self.iom.get_data(larcv.kProductImage2D,imgprod)
return imdata.Image2DArray().size()
def get_event_image(self,ii,imgprod,roiprod,planes, refresh=True) :
#Load data in TChain
self.iom.read_entry(ii)
# there may be no ROI
hasroi = False
roidata = None
if roiprod is not None:
roidata = self.iom.iom.get_data(larcv.kProductROI,roiprod)
roidata = roidata.ROIArray()
hasroi = True
# get the EventImage2D
imdata = self.iom.get_data(larcv.kProductImage2D,imgprod) # goes to disk
self.run = imdata.run()
self.subrun = imdata.subrun()
self.event = imdata.event()
# get the std::vector<larcv::Image2D>
imdata = imdata.Image2DArray()
if imdata.size() == 0 : return (None, False)
# hand it off to the factory, the producer name should query the
# the correct subclass of PlotImage
image = self.IF.get(imdata,roidata,planes,imgprod) # returns PlotImgae
# return it to rgbviewer
return ( image, hasroi )
class DataManager(object):
def __init__(self,argv):
self.iom = IOManager(argv)
self.keys ={}
self.IF = ImageFactory()
# get keys from rootfile, iterate over the enum
# and see what's in the root file
for i in xrange(larcv.kProductUnknown):
product = larcv.ProductName(i)
self.keys[product] = []
producers=self.iom.iom.producer_list(i)
for p in producers:
self.keys[product].append(p)
# run subrun and event start at zero
self.run = -1
self.subrun = -1
self.event = -1
def get_nchannels(self,ii,imgprod) :
# Sorry Vic I hacked this
# --> it's ok
self.iom.read_entry(ii)
imdata = self.iom.get_data(larcv.kProductImage2D,imgprod)
return imdata.Image2DArray().size()
def get_event_image(self,ii,imgprod,roiprod,planes, refresh=True) :
#Load data in TChain
self.iom.read_entry(ii)
# there may be no ROI
hasroi = False
roidata = None
if roiprod is not None:
roidata = self.iom.iom.get_data(larcv.kProductROI,roiprod)
roidata = roidata.ROIArray()
hasroi = True
# get the EventImage2D
imdata = self.iom.get_data(larcv.kProductImage2D,imgprod) # goes to disk
self.run = imdata.run()
self.subrun = imdata.subrun()
self.event = imdata.event()
# get the std::vector<larcv::Image2D>
imdata = imdata.Image2DArray()
if imdata.size() == 0 : return (None, False)
# hand it off to the factory, the producer name should query the
# the correct subclass of PlotImage
image = self.IF.get(imdata,roidata,planes,imgprod) # returns PlotImgae
# return it to rgbviewer
return ( image, hasroi )
# -----------------------------------------------------------------------------
# Erez, July-21, 2016 - get an image using R/S/E navigation
# -----------------------------------------------------------------------------
def get_all_images(self,imgprod,event_base_and_images,rse_map) :
for entry in range(self.iom.get_n_entries()):
read_entry = self.iom.read_entry(entry)
event_base = self.iom.get_data(larcv.kProductImage2D,imgprod)
event_base_and_images[entry] = event_base
rse = ( int(event_base.run()),int(event_base.subrun()),int(event_base.event()) )
#print rse
#rse_map[entry] = [event_base.run(),event_base.subrun(),event_base.event()]
rse_map[ rse ] = entry
# print rse_map[entry]
print "collected %d images...\nready for RSE navigation"%len(event_base_and_images)
return
# -----------------------------------------------------------------------------
# Erez, July-21, 2016 - get an image using R/S/E navigation
# -----------------------------------------------------------------------------
def get_rse_image(self,event_base_and_images,rse_map,wanted_rse,imgprod,roiprod,planes, refresh=True) :
if wanted_rse in rse_map:
return self.get_event_image(rse_map[wanted_rse],imgprod,roiprod,planes,refresh)
else:
print "i couldn't find this R/S/E..."
return None, False
ii = -1
for i in range(len(event_base_and_images)):
if rse_map[i] == wanted_rse:
ii = i
break
if (ii==-1):
print "i couldn't find this R/S/E..."
return self.get_event_image(ii,imgprod,roiprod,planes,refresh)