# 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
