# In[ ]: # In[ ]: # In[52]: mgr = fmwk.storage_manager() mgr.set_io_mode(fmwk.storage_manager.kWRITE) mgr.set_out_filename("temp.root") mgr.open() mgr.set_id(1,0,1) mgr.event_id() event_ophit = mgr.get_data(fmwk.data.kOpHit,'OpHitFinder') for h in ophits: event_ophit.push_back(h) mgr.next_event() mgr.close()
w.setWindowTitle('LArFlow Visualization') # default, turn off cryostat w.changeComponentState("OuterCryostatPC", False) w.changeComponentState("InnerCryostatPC", False) w.changeComponentState("TPCFrame", False) w.changeComponentState("CathodePlate", False) w.changeComponentState("TPCPlaneVert", False) w.changeComponentState("TPCPlane", False) w.changeComponentState("FieldCageTubeZ", False) w.changeComponentState("FieldCageTubeY", False) w.changeComponentState("TPCActive", False) # import data inputfile = args.input io = larlite.storage_manager(larlite.storage_manager.kREAD) io.add_in_filename(inputfile) io.open() # modes modes = ["core", "core-only", "spheres"] # get larflow clusters io.go_to(args.entry) ev_larflow = io.get_data(larlite.data.kLArFlowCluster, args.treename) nclusters = ev_larflow.size() print "Number of larflow clusters: ", nclusters ev_pca = None if args.pca: ev_pca = io.get_data(larlite.data.kPCAxis, args.treename)
import matplotlib import matplotlib.pyplot as plt import matplotlib.image as mpimg import matplotlib.cm as cm import matplotlib.collections as mc import numpy as np import cv2 import sys from larlite import larlite as fmwk storage = fmwk.storage_manager() filename = str(sys.argv[1]) storage.add_in_filename(filename) storage.set_io_mode(storage.kREAD) storage.open() while ( storage.next_event() ): clusters = storage.get_data(fmwk.data.kCluster,"hullcluster") hits = storage.get_data(fmwk.data.kHit,"gaushit") ass_data = storage.get_data(fmwk.data.kAssociation,"hullcluster") cluster_to_hit_ass = ass_data.association(clusters.id(),hits.id()); #fig, ax = plt.subplots(figsize=(10,6))
action='store_true', default=False, help="If provided, makes a debug image, showing fit in 2D planes") args = parser.parse_args() import ROOT as rt from ROOT import std from larlite import larlite from larcv import larcv from larflow import larflow #print larflow.reco.cluster_t #print larflow.reco.cluster_larflow3dhits io = larlite.storage_manager(larlite.storage_manager.kBOTH) io.add_in_filename(args.input_dlmerged) io.add_in_filename(args.input_larflow) io.set_out_filename(args.output) io.set_data_to_read(larlite.data.kLArFlow3DHit, "larmatch") io.set_data_to_read(larlite.data.kCRTTrack, "crttrack") io.set_data_to_read(larlite.data.kCRTHit, "crthitcorr") io.set_data_to_read(larlite.data.kMCTrack, "mcreco") io.set_data_to_read(larlite.data.kMCShower, "mcreco") io.set_data_to_read(larlite.data.kMCTruth, "generator") io.set_data_to_read(larlite.data.kOpFlash, "simpleFlashBeam") io.set_data_to_read(larlite.data.kOpFlash, "simpleFlashCosmic") #io.set_data_to_read( larlite.data.kTrack, "boundarycosmicnoshift" ) #io.set_data_to_read( larlite.data.kTrack, "containedcosmic" ) #io.set_data_to_read( larlite.data.kTrack, "nutrack" ) io.open()
help="Number of entries to run [default: -1 (all)]") args = parser.parse_args() import ROOT as rt from ROOT import std from larcv import larcv from larlite import larlite from larflow import larflow from ublarcvapp import ublarcvapp """ test script for the PrepKeypointData class """ rt.gStyle.SetOptStat(0) ioll = larlite.storage_manager(larlite.storage_manager.kREAD) ioll.add_in_filename(args.input_larlite) ioll.open() if args.tick_backward: iolcv = larcv.IOManager(larcv.IOManager.kREAD, "larcv", larcv.IOManager.kTickBackward) else: iolcv = larcv.IOManager(larcv.IOManager.kREAD, "larcv", larcv.IOManager.kTickForward) iolcv.add_in_file(args.input_larcv) iolcv.reverse_all_products() iolcv.initialize() nentries = iolcv.get_n_entries() print "Number of entries: ", nentries
tickdir = larcv.IOManager.kTickForward if args.tickbackwards: tickdir = larcv.IOManager.kTickBackward io = larcv.IOManager(larcv.IOManager.kBOTH, "larcvio", tickdir) io.add_in_file(args.supera) io.set_out_file("%s_larcv.root" % (outfilestem)) io.set_verbosity(1) io.specify_data_read(larcv.kProductImage2D, "larflow") io.specify_data_read(larcv.kProductImage2D, args.adc_name) io.specify_data_read(larcv.kProductChStatus, args.chstatus_name) if args.has_wirecell: io.specify_data_read(larcv.kProductChStatus, "thrumu") io.reverse_all_products() io.initialize() out = larlite.storage_manager(larlite.storage_manager.kWRITE) out.set_out_filename("%s_larlite.root" % (outfilestem)) out.open() sigmoid = torch.nn.Sigmoid() NENTRIES = io.get_n_entries() if args.num_events > 0 and args.num_events < NENTRIES: NENTRIES = args.num_events dt_prep = 0. dt_chunk = 0. dt_net = 0. dt_save = 0.
fastototal) ophits.append(hit) # In[ ]: # In[ ]: # In[ ]: # In[ ]: # In[ ]: # In[52]: mgr = fmwk.storage_manager() mgr.set_io_mode(fmwk.storage_manager.kWRITE) mgr.set_out_filename("temp.root") mgr.open() mgr.set_id(1, 0, 1) mgr.event_id() event_ophit = mgr.get_data(fmwk.data.kOpHit, 'OpHitFinder') for h in ophits: event_ophit.push_back(h) mgr.next_event() mgr.close() # In[53]:
def PerformPMTPrecuts(s_dlmerged, ophittree="ophitBeam", PP_WINDOW_LENGTH=130, PP_COINC_WINDOW=6, PP_PE_THRESH=20.0, PP_PMT_MAX_FRAC_CUTOFF=0.60, PP_WIN_START=190, PP_WIN_END=320, PP_PORCH_WIN_START=60, PP_PORCH_WIN_END=190, PP_TICK_SIZE=0.015625): """ calculate PMT precuts TICK_SIZE units in 15.625 ns per tick """ # make config from ROOT import fcllite fcllite.PSet fcllite.CreatePSetFromFile cfg = """OpHitProducer:\"%s\" BinTickWidth:%d WinStartTick:%d WinEndTick:%d PEThreshold:%.2f VetoPEThreshold:%.2f MaxVetoPE:%.2f VetoStartTick:%.2f VetoEndTick:%.2f PMTMaxFrac:%.2f """ % (ophittree, PP_COINC_WINDOW, PP_WIN_START, PP_WIN_END, PP_PE_THRESH, PP_PE_THRESH, PP_PE_THRESH, PP_PORCH_WIN_START, PP_PORCH_WIN_END, PP_PMT_MAX_FRAC_CUTOFF) fout = open("precut.cfg", 'w') print >> fout, cfg fout.close() pset = fcllite.CreatePSetFromFile("precut.cfg", "PMTPreCut") #print pset.dump() pset = fcllite.PSet("PMTPreCut", cfg) precutalgo = larlite.LEEPreCut() precutalgo.configure(pset) print "PMT Precut algo configured" # ---------------------------------------------- # #print('<EVID: %s> -- First, we will figure out the PMT Precut info.'%_tag) #gotta do this first for io reasons PMTPrecutDict = {} # Load up larlite ll_manager = larlite.storage_manager() ll_manager.set_io_mode(ll_manager.kREAD) ll_manager.add_in_filename(s_dlmerged) ll_manager.set_in_rootdir("") ll_manager.open() while ll_manager.next_event(): id_rse = tuple((ll_manager.run_id(), ll_manager.subrun_id(), ll_manager.event_id())) ev_ophits = ll_manager.get_data(larlite.data.kOpHit, ophittree) print "[event {}] number of ophits: {}".format(id_rse, ev_ophits.size()) passcuts = precutalgo.apply(ev_ophits) PMTPrecutDict[id_rse] = dict(_totpe=precutalgo.beamPE(), _porchtotpe=precutalgo.vetoPE(), _maxpefrac=precutalgo.maxFrac(), _passpmtprecut=passcuts, _beamFirstTick=precutalgo.beamFirstTick(), _vetoFirstTick=precutalgo.vetoFirstTick()) return PMTPrecutDict
import lardly #input_larlite = sys.argv[1] # input_larlite = "/home/jmills/workdir/michel_files/tracker_reco_Run000001-SubRun000586.root" input_larlite = "/home/jmills/workdir/michel_files/tracker_reco_Run000001-SubRun000001.root" input_mc = "/home/jmills/workdir/michel_files/mcinfo-Run000001-SubRun000001.root" # input_larcv = "/home/jmills/workdir/michel_files/supera-Run000001-SubRun000586.root" input_larcv = "/home/jmills/workdir/michel_files/supera-Run000001-SubRun000001.root" input_pgraph = "/home/jmills/workdir/michel_files/pgraph_file-Run000001-SubRun000001.root" detdata = lardly.DetectorOutline() entry = 12 # LARLITE io_ll = larlite.storage_manager(larlite.storage_manager.kREAD) io_ll.add_in_filename(input_larlite) io_ll.open() io_ll.go_to(entry) io_ll_mc = larlite.storage_manager(larlite.storage_manager.kREAD) io_ll_mc.add_in_filename(input_mc) io_ll_mc.open() io_ll_mc.go_to(entry) # TRACK evtrack = io_ll.get_data(larlite.data.kTrack, "trackReco") evmctrack = io_ll_mc.get_data(larlite.data.kMCTrack, "mcreco") evmcshower = io_ll_mc.get_data(larlite.data.kMCShower, "mcreco") print("number of tracks: ", evtrack.size())
print larflow.reco.cluster_larflow3dhits iolcv = larcv.IOManager(larcv.IOManager.kBOTH, "larcv", larcv.IOManager.kTickBackward) #iolcv.add_in_file( "../../../../testdata/mcc9_v29e_intrinsic_nue_LowE/merged_dlreco_4c558c3b-344b-4f5a-b319-6ac339aa82b3.root" ) #iolcv.add_in_file( "../../../../testdata/mcc9_v13_bnbnue_corsika/larcvtruth-Run000001-SubRun000001.root" ) iolcv.add_in_file( "../../../../testdata/mcc9_v13_nueintrinsic_overlay_run1/supera-Run004999-SubRun000006.root" ) iolcv.set_out_file("outtest_showerbuilder_larcv.root") iolcv.reverse_all_products() iolcv.addto_storeonly_list(larcv.kProductImage2D, "trueshoweradc") iolcv.addto_storeonly_list(larcv.kProductImage2D, "segment") iolcv.initialize() io = larlite.storage_manager(larlite.storage_manager.kBOTH) #io.add_in_filename( "../../../../testdata/mcc9_v29e_intrinsic_nue_LowE/merged_dlreco_4c558c3b-344b-4f5a-b319-6ac339aa82b3.root" ) #io.add_in_filename( "../../../../testdata/mcc9_v29e_intrinsic_nue_LowE/larmatch_wckps_intrinsic_nue_LowE_4c558c3b-344b-4f5a-b319-6ac339aa82b3_larlite.root" ) #io.add_in_filename( "../../../../testdata/mcc9_v13_bnbnue_corsika/mcinfo-Run000001-SubRun000001.root" ) io.add_in_filename( "../../../../testdata/mcc9_v13_nueintrinsic_overlay_run1/reco2d-Run004999-SubRun000006.root" ) io.set_out_filename("outtest_showerbuilder.root") io.set_data_to_read(larlite.data.kMCShower, "mcreco") io.set_data_to_read(larlite.data.kMCTrack, "mcreco") io.set_data_to_read(larlite.data.kMCTruth, "generator") io.set_data_to_read(larlite.data.kLArFlow3DHit, "larmatch") io.set_data_to_write(larlite.data.kLArFlow3DHit, "trueshowerhits") io.set_data_to_write(larlite.data.kMCShower, "truthshower") io.set_data_to_write(larlite.data.kPCAxis, "truthshower") io.open()
import sys from larlite import larlite as fmwk from ROOT import * man = fmwk.storage_manager() man.set_io_mode(man.kWRITE) man.set_out_filename("aho.root") man.set_verbosity(fmwk.msg.kDEBUG) man.open() run, subrun, event = (0, 0, 0) for x in xrange(10): if x % 5 == 0: run = x / 5 subrun = (x % 5) / 3 event = x - run * 5 - subrun * 3 for y in xrange(fmwk.data.kDATA_TYPE_MAX): if y in [ fmwk.data.kUndefined, fmwk.data.kSeed, fmwk.data.kEvent, fmwk.data.kMCTrajectory, fmwk.data.kMCNeutrino, fmwk.data.kDATA_TYPE_MAX ]: continue man.get_data(y, "test") for y in xrange(fmwk.data.kSUBRUNDATA_TYPE_MAX): if y in [fmwk.data.kSUBRUNDATA_Undefined]:
import os,sys,time import ROOT as rt from ROOT import std from larlite import larlite from larcv import larcv from larflow import larflow print larflow.crtmatch.CRTHitMatch io = larlite.storage_manager( larlite.storage_manager.kREAD ) io.add_in_filename( "larflow_reco_extbnb_run3.root" ) io.add_in_filename( "merged_dlreco_extbnb_run3_821c2dfc-96b0-4725-a187-6b628cbbbea3.root" ) #io.set_out_filename( "crtmatch_reco_extbnb_run3.root" ) io.open() outio = larlite.storage_manager( larlite.storage_manager.kWRITE ) outio.set_out_filename( "crtmatch_reco_extbnb_run3.root" ) outio.open() iolcv = larcv.IOManager( larcv.IOManager.kREAD, "larcv", larcv.IOManager.kTickBackward ) iolcv.add_in_file( "merged_dlreco_extbnb_run3_821c2dfc-96b0-4725-a187-6b628cbbbea3.root" ) iolcv.reverse_all_products() iolcv.initialize() crtmatch = larflow.crtmatch.CRTHitMatch() crtmatch.set_verbosity(1) nentries = iolcv.get_n_entries() print "Number of entries: ",nentries nentries = 1