Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion examples/extracted_MDC2025.fcl
Original file line number Diff line number Diff line change
Expand Up @@ -44,9 +44,10 @@ physics.analyzers.Mu2eEventDisplay.filler.addCrvClusters : true
physics.analyzers.Mu2eEventDisplay.filler.addCrvTrack : true
physics.analyzers.Mu2eEventDisplay.filler.addCrvRecoPulse : false
physics.analyzers.Mu2eEventDisplay.filler.addCaloDigis : false
physics.analyzers.Mu2eEventDisplay.filler.addClusters : false
physics.analyzers.Mu2eEventDisplay.filler.addClusters : true
physics.analyzers.Mu2eEventDisplay.filler.addKalSeeds : true
physics.analyzers.Mu2eEventDisplay.filler.addTrackerHist : true
physics.analyzers.Mu2eEventDisplay.filler.addCaloHist : false
physics.analyzers.Mu2eEventDisplay.extracted : true
physics.analyzers.Mu2eEventDisplay.specifyTag : false

Expand Down
1 change: 1 addition & 0 deletions fcl/extracted.fcl
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ Mu2eEventDisplay.addTrkCaloHits : true
Mu2eEventDisplay.filler.addCosmicTrackSeeds : false
Mu2eEventDisplay.filler.addMCTraj : true
Mu2eEventDisplay.filler.addTrackerHist : true
Mu2eEventDisplay.filler.addCaloHist : true
Mu2eEventDisplay.filler.addHelixSeeds : false
Mu2eEventDisplay.specifyTag : false

Expand Down
11 changes: 6 additions & 5 deletions fcl/prolog.fcl
Original file line number Diff line number Diff line change
Expand Up @@ -41,16 +41,17 @@ Mu2eEventDisplay : {
addHits : false
addBkgClusters : false
addCrvRecoPulse : false
addCrvClusters : false
addCrvTrack : false
addTimeClusters : false
addCrvClusters : false
addCrvTrack : false
addTimeClusters : false
addTrkHits : false //Combo hits?
addCaloDigis : false
addClusters : true
addKalSeeds : true
addCosmicTrackSeeds : false
addMCTraj : true
addTrackerHist : false
addMCTraj : true
addTrackerHist : false
addCaloHist : false
addSurfSteps : true
addSimParts : false
addHelixSeeds : false
Expand Down
4 changes: 3 additions & 1 deletion inc/CollectionFiller.hh
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,7 @@ namespace mu2e{
fhicl::Atom<bool> addSurfSteps{Name("addSurfSteps"), Comment("set to add SurfaceStep MC"),false};
fhicl::Atom<bool> addSimParts{Name("addSimParts"), Comment("set to add SimParticles MC"),false};
fhicl::Atom<bool> addTrackerHist{Name("addTrackerHist"), Comment("set to add tracker histogram"),false};
fhicl::Atom<bool> addCaloHist{Name("addCaloHist"), Comment("set to add calorimeter histogram"),false};

// Global flag to attempt retrieval of all collections found in the event
fhicl::Atom<bool> FillAll{Name("FillAll"), Comment("to see all available products"), false};
Expand Down Expand Up @@ -129,7 +130,8 @@ namespace mu2e{
// --- Boolean Control Flags (Copied from FHiCL Config) ---
bool addHits_, addBkgClusters_, addCrvRecoPulse_, addCrvClusters_, addCrvTrack_, addTimeClusters_, addTrkHits_, addCaloDigis_,
addClusters_, addHelixSeeds_, addKalSeeds_, addCosmicTrackSeeds_, addMCTraj_,
addSurfSteps_, addSimParts_, addTrackerHist_, FillAll_;
addSurfSteps_, addSimParts_, addTrackerHist_, addCaloHist_, FillAll_;


// --- Collection Retrieval Methods ---

Expand Down
9 changes: 6 additions & 3 deletions inc/MainWindow.hh
Original file line number Diff line number Diff line change
Expand Up @@ -152,14 +152,17 @@ namespace mu2e {
bool addMCTrajectories = false;
bool addSurfaceSteps = false;
bool addSimParts = false;
bool addTrackerHist = false;
bool addTrackerHist = false;
bool addCaloHist = false;

bool addTrkErrBar = true;
bool addCrystalDraw = false;
bool addCrvBars = true;
DrawOptions(){};

DrawOptions(bool cosmictracks, bool helices, bool tracks, bool calodigis, bool clusters, bool combohits, bool bkgclusters, bool crv, bool crvclu, bool crvtrack, bool timeclusters, bool trkhits, bool mctraj, bool surfsteps, bool simparts, bool trackerhist, bool errbar, bool crys, bool crvbars)
: addCosmicTracks(cosmictracks), addHelices(helices), addTracks(tracks), addCaloDigis(calodigis), addClusters(clusters), addComboHits(combohits), addBkgClusters(bkgclusters), addCrvRecoPulse(crv), addCrvClusters(crvclu), addCrvTrack(crvtrack), addTimeClusters(timeclusters), addTrkHits(trkhits), addMCTrajectories(mctraj), addSurfaceSteps(surfsteps), addSimParts(simparts), addTrackerHist(trackerhist), addTrkErrBar(errbar), addCrystalDraw(crys), addCrvBars(crvbars) {};
DrawOptions(bool cosmictracks, bool helices, bool tracks, bool calodigis, bool clusters, bool combohits, bool bkgclusters, bool crv, bool crvclu, bool crvtrack, bool timeclusters, bool trkhits, bool mctraj, bool surfsteps, bool simparts, bool trackerhist, bool calohist, bool errbar, bool crys, bool crvbars)
: addCosmicTracks(cosmictracks), addHelices(helices), addTracks(tracks), addCaloDigis(calodigis), addClusters(clusters), addComboHits(combohits), addBkgClusters(bkgclusters), addCrvRecoPulse(crv), addCrvClusters(crvclu), addCrvTrack(crvtrack), addTimeClusters(timeclusters), addTrkHits(trkhits), addMCTrajectories(mctraj), addSurfaceSteps(surfsteps), addSimParts(simparts), addTrackerHist(trackerhist), addCaloHist(calohist), addTrkErrBar(errbar), addCrystalDraw(crys), addCrvBars(crvbars) {};


};

Expand Down
10 changes: 8 additions & 2 deletions inc/TrackerCalo2DViews.hh
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
#include <ROOT/REveManager.hxx>
#include <ROOT/REveScene.hxx>
#include "Offline/RecoDataProducts/inc/KalSeed.hh"
#include <map>

namespace REX = ROOT::Experimental;

Expand All @@ -19,12 +20,17 @@ public:
virtual ~TrackerCalo2DViews();

void createHistogramView();
void drawTrackerStation(const mu2e::KalSeedPtrCollection* seedcol);
void redrawCanvas(const mu2e::KalSeedPtrCollection* seedcol);
void drawTrackerStation(const mu2e::KalSeedPtrCollection* seedcol); //, const CaloDigiCollection* calodigicol);
void drawTrackerXYView(const mu2e::KalSeedPtrCollection* seedcol);
void drawCalorimeterDisk();
//void redrawCanvas(const mu2e::KalSeedPtrCollection* seedcol);

private:
REX::REvePointSet* fCanvasHolder{nullptr};
TCanvas* fCanvas{nullptr};
TCanvas* fCaloCanvas{nullptr};
TCanvas* fXYCanvas{nullptr};
std::map<int, TCanvas*> fPlaneCanvases;
};

} // namespace mu2e
Expand Down
Empty file added src/CLAUDE.md
Empty file.
1 change: 1 addition & 0 deletions src/CollectionFiller.cc
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ namespace mu2e{
addSurfSteps_(conf.addSurfSteps()),
addSimParts_(conf.addSimParts()),
addTrackerHist_(conf.addTrackerHist()),
addCaloHist_(conf.addCaloHist()),
FillAll_(conf.FillAll())
{}

Expand Down
21 changes: 13 additions & 8 deletions src/MainWindow.cc
Original file line number Diff line number Diff line change
Expand Up @@ -781,15 +781,19 @@ void MainWindow::showEvents(REX::REveManager *eveMng, REX::REveElement* &eventSc
std::vector<const KalSeedPtrCollection*> track_list = std::get<1>(data.track_tuple);
if(drawOpts.addTracks and track_list.size() !=0) {
pass_data->FillKinKalTrajectory(eveMng, firstLoop, eventScene, data.track_tuple, KKOpts.addKalInter, KKOpts.addTrkStrawHits, KKOpts.addTrkCaloHits, t1, t2);

auto const& track_list = std::get<1>(data.track_tuple);
const mu2e::KalSeedPtrCollection* seedcol = track_list[0];
if(drawOpts.addTrackerHist) {
//redrawCanvas(seedcol);
if(drawOpts.addTrackerHist and track_list.size() !=0) {
fTrackerCalo2DViews = new TrackerCalo2DViews();
auto const& track_list = std::get<1>(data.track_tuple);
const mu2e::KalSeedPtrCollection* seedcol = track_list[0];
//std::vector<const CaloDigiCollection*> calodigi_list = std::get<1>(data.calodigi_tuple);
//const CaloDigiCollection* calodigicol = calodigi_list[0];
//fTrackerCalo2DViews->createHistogramView();
fTrackerCalo2DViews->redrawCanvas(seedcol);
}
//redrawCanvas(seedcol);
//fTrackerCalo2DViews->redrawCanvas(seedcol);
//std::cout<<"track list size = "<<track_list.size()<<" calo digi size = "<<calodigi_list.size()<<std::endl;
fTrackerCalo2DViews->drawTrackerStation(seedcol);//, calodigicol);
fTrackerCalo2DViews->drawTrackerXYView(seedcol);
}
}
if(drawOpts.addCrvTrack) {
pass_data->AddCRVKalIntersection(eveMng, firstLoop, eventScene, data.track_tuple, KKOpts.addKalInter, KKOpts.addTrkStrawHits, KKOpts.addTrkCaloHits, t1, t2, data.crvcoin_tuple, geomOpts.extracted, drawOpts.addCrvBars);
Expand Down Expand Up @@ -822,7 +826,8 @@ void MainWindow::showEvents(REX::REveManager *eveMng, REX::REveElement* &eventSc

if(drawOpts.addClusters){
std::vector<const CaloClusterCollection*> calocluster_list = std::get<1>(data.calocluster_tuple);
if(calocluster_list.size() !=0 ) pass_data->AddCaloClusters(eveMng, firstLoopCalo, data.calocluster_tuple, eventScene, drawOpts.addCrystalDraw);
if(calocluster_list.size() !=0 )
pass_data->AddCaloClusters(eveMng, firstLoopCalo, data.calocluster_tuple, eventScene, drawOpts.addCrystalDraw);
}

std::vector<const HelixSeedCollection*> helix_list = std::get<1>(data.helix_tuple);
Expand Down
6 changes: 4 additions & 2 deletions src/Mu2eEventDisplay_module.cc
Original file line number Diff line number Diff line change
Expand Up @@ -290,7 +290,9 @@ namespace mu2e
<<" addClusters : "<<filler_.addClusters_
<<" addHelices : "<<filler_.addHelixSeeds_
<<" addTracks : "<<filler_.addKalSeeds_
<<" addTrackerHist : "<<filler_.addTrackerHist_
<<" addTrackerHist : "<<filler_.addTrackerHist_
<<" addCaloHist : "<<filler_.addCaloHist_

<<" addCosmicTrackSeeds : "<<filler_.addCosmicTrackSeeds_ << std::endl;
}

Expand Down Expand Up @@ -691,7 +693,7 @@ void Mu2eEventDisplay::FillAnyCollection(const art::Event& evt, std::vector<std:
if(diagLevel_ == 1) std::cout<<"[Mu2eEventDisplay : process_single_event] -- calls to data interface "<<std::endl;

// Create a structure defining which data products should be drawn (based on module configuration).
DrawOptions drawOpts(filler_.addCosmicTrackSeeds_, filler_.addHelixSeeds_, filler_.addKalSeeds_, filler_.addCaloDigis_, filler_.addClusters_, filler_.addHits_, filler_.addBkgClusters_, filler_.addCrvRecoPulse_, filler_.addCrvClusters_, filler_.addCrvTrack_, filler_.addTimeClusters_, filler_.addTrkHits_, filler_.addMCTraj_, filler_.addSurfSteps_, filler_.addSimParts_, filler_.addTrackerHist_, addErrBar_, addCrystalHits_, addCrvBars_);
DrawOptions drawOpts(filler_.addCosmicTrackSeeds_, filler_.addHelixSeeds_, filler_.addKalSeeds_, filler_.addCaloDigis_, filler_.addClusters_, filler_.addHits_, filler_.addBkgClusters_, filler_.addCrvRecoPulse_, filler_.addCrvClusters_, filler_.addCrvTrack_, filler_.addTimeClusters_, filler_.addTrkHits_, filler_.addMCTraj_, filler_.addSurfSteps_, filler_.addSimParts_, filler_.addTrackerHist_, filler_.addCaloHist_, addErrBar_, addCrystalHits_, addCrvBars_);

// Create a structure defining visualization options specific to Kinematic/Kalman fitting results.
KinKalOptions KKOpts(addKalInter_, addTrkStrawHits_, addTrkCaloHits_);
Expand Down
Loading