// Simple example of reading a generated Root file
void sample_readfile(char *filename=NULL, bool verbose=false)
{
// Clear global scope
//gROOT->Reset();
/*
gStyle->SetOptStat(0);
gStyle->SetCanvasColor(0);
gStyle->SetTitleColor(1);
gStyle->SetStatColor(0);
gStyle->SetFrameFillColor(0);
gStyle->SetPadColor(0);
gStyle->SetPadTickX(1);
gStyle->SetPadTickY(1);
gStyle->SetTitleSize(0.04);
gStyle->SetCanvasBorderMode(0);
gStyle->SetFrameBorderMode(0);
gStyle->SetFrameLineWidth(2);
gStyle->SetPadBorderMode(0);
gStyle->SetPalette(1);
gStyle->SetTitleAlign(23);
gStyle->SetTitleX(.5);
gStyle->SetTitleY(0.99);
gStyle->SetTitleBorderSize(0);
gStyle->SetTitleFillColor(0);
gStyle->SetHatchesLineWidth(2);
gStyle->SetLineWidth(1.5);
gStyle->SetTitleFontSize(0.07);
gStyle->SetLabelSize(0.05,"X");
gStyle->SetLabelSize(0.05,"Y");
gStyle->SetTitleSize(0.04,"X");
gStyle->SetTitleSize(0.04,"Y");
gStyle->SetTitleBorderSize(0);
gStyle->SetCanvasBorderMode(0);
*/
// Load the library with class dictionary info
// (create with "gmake shared")
char* wcsimdirenv;
wcsimdirenv = getenv ("WCSIMDIR");
if(wcsimdirenv != NULL){
gSystem->Load("${WCSIMDIR}/libWCSimRoot.so");
}else{
gSystem->Load("../libWCSimRoot.so");
}
TFile *file;
// Open the file
if (filename==NULL){
file = new TFile("../wcsim.root","read");
}else{
file = new TFile(filename,"read");
}
if (!file->IsOpen()){
cout << "Error, could not open input file: " << filename << endl;
return -1;
}
// Get the a pointer to the tree from the file
TTree *tree = (TTree*)file->Get("wcsimT");
// Get the number of events
int nevent = tree->GetEntries();
if(verbose) printf("nevent %d\n",nevent);
// Create a WCSimRootEvent to put stuff from the tree in
WCSimRootEvent* wcsimrootsuperevent = new WCSimRootEvent();
// Set the branch address for reading from the tree
TBranch *branch = tree->GetBranch("wcsimrootevent");
branch->SetAddress(&wcsimrootsuperevent);
// Force deletion to prevent memory leak
tree->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE);
// Geometry tree - only need 1 "event"
TTree *geotree = (TTree*)file->Get("wcsimGeoT");
WCSimRootGeom *geo = 0;
geotree->SetBranchAddress("wcsimrootgeom", &geo);
if(verbose) std::cout << "Geotree has " << geotree->GetEntries() << " entries" << std::endl;
if (geotree->GetEntries() == 0) {
exit(9);
}
geotree->GetEntry(0);
// start with the main "subevent", as it contains most of the info
// and always exists.
WCSimRootTrigger* wcsimrootevent;
TH1F *h1 = new TH1F("PMT Hits", "PMT Hits", 8000, 0, 8000);
TH1F *hvtx0 = new TH1F("Event VTX0", "Event VTX0", 200, -1500, 1500);
TH1F *hvtx1 = new TH1F("Event VTX1", "Event VTX1", 200, -1500, 1500);
TH1F *hvtx2 = new TH1F("Event VTX2", "Event VTX2", 200, -1500, 1500);
int num_trig=0;
// Now loop over events
for (int ev=0; ev<nevent; ev++)
{
// Read the event from the tree into the WCSimRootEvent instance
tree->GetEntry(ev);
wcsimrootevent = wcsimrootsuperevent->GetTrigger(0);
if(verbose){
printf("********************************************************");
printf("Evt, date %d %d\n", wcsimrootevent->GetHeader()->GetEvtNum(),
wcsimrootevent->GetHeader()->GetDate());
printf("Mode %d\n", wcsimrootevent->GetMode());
printf("Number of subevents %d\n",
wcsimrootsuperevent->GetNumberOfSubEvents());
printf("Vtxvol %d\n", wcsimrootevent->GetVtxvol());
printf("Vtx %f %f %f\n", wcsimrootevent->GetVtx(0),
wcsimrootevent->GetVtx(1),wcsimrootevent->GetVtx(2));
}
hvtx0->Fill(wcsimrootevent->GetVtx(0));
hvtx1->Fill(wcsimrootevent->GetVtx(1));
hvtx2->Fill(wcsimrootevent->GetVtx(2));
if(verbose){
printf("Jmu %d\n", wcsimrootevent->GetJmu());
printf("Npar %d\n", wcsimrootevent->GetNpar());
printf("Ntrack %d\n", wcsimrootevent->GetNtrack());
}
// Now read the tracks in the event
// Get the number of tracks
int ntrack = wcsimrootevent->GetNtrack();
if(verbose) printf("ntracks=%d\n",ntrack);
int i;
// Loop through elements in the TClonesArray of WCSimTracks
for (i=0; i<ntrack; i++)
{
TObject *element = (wcsimrootevent->GetTracks())->At(i);
WCSimRootTrack *wcsimroottrack = dynamic_cast<WCSimRootTrack*>(element);
if(verbose){
printf("Track ipnu: %d\n",wcsimroottrack->GetIpnu());
printf("Track parent ID: %d\n",wcsimroottrack->GetParenttype());
for (int j=0; j<3; j++)
printf("Track dir: %d %f\n",j, wcsimroottrack->GetDir(j));
}
} // End of loop over tracks
// Now look at the Cherenkov hits
// Get the number of Cherenkov hits.
// Note... this is *NOT* the number of photons that hit tubes.
// It is the number of tubes hit with Cherenkov photons.
// The number of digitized tubes will be smaller because of the threshold.
// Each hit "raw" tube has several photon hits. The times are recorded.
// See http://nwg.phy.bnl.gov/DDRD/cgi-bin/private/ShowDocument?docid=245
// for more information on the structure of the root file.
//
// The following code prints out the hit times for the first 10 tubes and also
// adds up the total pe.
//
// For digitized info (one time/charge tube after a trigger) use
// the digitized information.
//
int ncherenkovhits = wcsimrootevent->GetNcherenkovhits();
int ncherenkovdigihits = wcsimrootevent->GetNcherenkovdigihits();
h1->Fill(ncherenkovdigihits);
if(verbose){
printf("node id: %i\n", ev);
printf("Ncherenkovhits %d\n", ncherenkovhits);
printf("Ncherenkovdigihits %d\n", ncherenkovdigihits);
cout << "RAW HITS:" << endl;
}
// Grab the big arrays of times and parent IDs
TClonesArray *timeArray = wcsimrootevent->GetCherenkovHitTimes();
int totalPe = 0;
// Loop through elements in the TClonesArray of WCSimRootCherenkovHits
for (i=0; i< ncherenkovhits; i++)
{
TObject *Hit = (wcsimrootevent->GetCherenkovHits())->At(i);
WCSimRootCherenkovHit *wcsimrootcherenkovhit =
dynamic_cast<WCSimRootCherenkovHit*>(Hit);
int tubeNumber = wcsimrootcherenkovhit->GetTubeID();
int timeArrayIndex = wcsimrootcherenkovhit->GetTotalPe(0);
int peForTube = wcsimrootcherenkovhit->GetTotalPe(1);
WCSimRootPMT pmt = geo->GetPMT(tubeNumber-1);
totalPe += peForTube;
if ( i < 10 ) // Only print first XX=10 tubes
{
if(verbose) printf("Total pe: %d times( ",peForTube);
for (int j = timeArrayIndex; j < timeArrayIndex + peForTube; j++)
{
WCSimRootCherenkovHitTime HitTime =
dynamic_cast<WCSimRootCherenkovHitTime>(timeArray->At(j));
if(verbose) printf("%6.2f ", HitTime.GetTruetime() );
}
if(verbose) cout << ")" << endl;
}
} // End of loop over Cherenkov hits
if(verbose) cout << "Total Pe : " << totalPe << endl;
// Look at digitized hit info
// Get the number of digitized hits
// Loop over sub events
if(verbose) cout << "DIGITIZED HITS:" << endl;
for (int index = 0 ; index < wcsimrootsuperevent->GetNumberOfEvents(); index++)
{
wcsimrootevent = wcsimrootsuperevent->GetTrigger(index);
if(verbose) cout << "Sub event number = " << index << "\n";
int ncherenkovdigihits = wcsimrootevent->GetNcherenkovdigihits();
if(verbose) printf("Ncherenkovdigihits %d\n", ncherenkovdigihits);
if(ncherenkovdigihits>0)
num_trig++;
//for (i=0;i<(ncherenkovdigihits>4 ? 4 : ncherenkovdigihits);i++){
for (i=0;i<ncherenkovdigihits;i++)
{
// Loop through elements in the TClonesArray of WCSimRootCherenkovDigHits
TObject *element = (wcsimrootevent->GetCherenkovDigiHits())->At(i);
WCSimRootCherenkovDigiHit *wcsimrootcherenkovdigihit =
dynamic_cast<WCSimRootCherenkovDigiHit*>(element);
if(verbose){
if ( i < 10 ) // Only print first XX=10 tubes
printf("q, t, tubeid: %f %f %d \n",wcsimrootcherenkovdigihit->GetQ(),
wcsimrootcherenkovdigihit->GetT(),wcsimrootcherenkovdigihit->GetTubeId());
}
} // End of loop over Cherenkov digihits
} // End of loop over trigger
// reinitialize super event between loops.
wcsimrootsuperevent->ReInitialize();
} // End of loop over events
// TCanvas c1("c1");
float win_scale = 0.75;
int n_wide(2);
int n_high(2);
TCanvas* c1 = new TCanvas("c1", "First canvas", 500*n_wide*win_scale, 500*n_high*win_scale);
c1->Draw();
c1->Divide(2,2);
c1->cd(1); hvtx0->Draw();
c1->cd(2); hvtx1->Draw();
c1->cd(3); hvtx2->Draw();
c1->cd(4); h1->Draw();
std::cout<<"num_trig "<<num_trig<<"\n";
}
// Simple example of reading a generated Root file
void verification_HitsChargeTime(char *filename=NULL, char *filename2, bool verbose=false)
{
// Clear global scope
//gROOT->Reset();
gStyle->SetOptStat(0);
gStyle->SetCanvasColor(0);
gStyle->SetTitleColor(1);
gStyle->SetStatColor(0);
gStyle->SetFrameFillColor(0);
gStyle->SetPadColor(0);
gStyle->SetPadTickX(1);
gStyle->SetPadTickY(1);
gStyle->SetTitleSize(0.04);
gStyle->SetCanvasBorderMode(0);
gStyle->SetFrameBorderMode(0);
gStyle->SetFrameLineWidth(2);
gStyle->SetPadBorderMode(0);
gStyle->SetPalette(1);
gStyle->SetTitleAlign(23);
gStyle->SetTitleX(.5);
gStyle->SetTitleY(0.99);
gStyle->SetTitleBorderSize(0);
gStyle->SetTitleFillColor(0);
gStyle->SetHatchesLineWidth(2);
gStyle->SetLineWidth(1.5);
gStyle->SetTitleFontSize(0.07);
gStyle->SetLabelSize(0.05,"X");
gStyle->SetLabelSize(0.05,"Y");
gStyle->SetTitleSize(0.04,"X");
gStyle->SetTitleSize(0.04,"Y");
gStyle->SetTitleBorderSize(0);
gStyle->SetCanvasBorderMode(0);
// Load the library with class dictionary info
// (create with "gmake shared")
char* wcsimdirenv;
wcsimdirenv = getenv ("WCSIMDIR");
if(wcsimdirenv != NULL){
gSystem->Load("${WCSIMDIR}/libWCSimRoot.so");
}else{
gSystem->Load("../libWCSimRoot.so");
}
TFile *f;
// Open the file
if (filename==NULL){
f = new TFile("wcsimtest.root","read");
filename = "wcsimtest.root";
}else{
f = new TFile(filename,"read");
}
if (!f->IsOpen()){
cout << "Error, could not open input file: " << filename << endl;
return -1;
}
TFile *f2;
// Open the file
if (filename2==NULL){
f2 = new TFile("../../WCSim_clean/verification-test-scripts/wcsimtest.root","read");
filename2 = "../../WCSim_clean/verification-test-scripts/wcsimtest.root";
}else{
f2 = new TFile(filename2,"read");
}
if (!f2->IsOpen()){
cout << "Error, could not open input file: " << filename2 << endl;
return -1;
}
TTree *wcsimT = f->Get("wcsimT");
int nevent = wcsimT->GetEntries();
WCSimRootEvent *wcsimrootsuperevent = new WCSimRootEvent();
wcsimT->SetBranchAddress("wcsimrootevent",&wcsimrootsuperevent);
// Force deletion to prevent memory leak when issuing multiple
// calls to GetEvent()
wcsimT->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE);
wcsimT->GetEvent(0);
// In the default vis.mac, only one event is run. I suspect you could loop over more events, if they existed.
WCSimRootTrigger *wcsimrootevent = wcsimrootsuperevent->GetTrigger(0);
cout << "Stats for the first event in your version of WCSim using " << filename << endl;
cout << "Number of tube hits " << wcsimrootevent->GetNumTubesHit() << endl;
cout << "Number of digitized tube hits " << wcsimrootevent->GetNumDigiTubesHit() << endl;
cout << "Number of photoelectron hit times " << wcsimrootevent->GetCherenkovHitTimes()->GetEntries() << endl;
//Save these to compare with the clean version of the code.
int num_tubes = wcsimrootevent->GetNumTubesHit();
int num_digi_tubes = wcsimrootevent->GetNumDigiTubesHit();
int hit_times = wcsimrootevent->GetCherenkovHitTimes()->GetEntries();
// Create a WCSimRootEvent to put stuff from the tree in
WCSimRootEvent* wcsimrootsuperevent = new WCSimRootEvent();
// Set the branch address for reading from the tree
TBranch *branch = wcsimT->GetBranch("wcsimrootevent");
branch->SetAddress(&wcsimrootsuperevent);
// Force deletion to prevent memory leak
wcsimT->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE);
// start with the main "subevent", as it contains most of the info
// and always exists.
WCSimRootTrigger* wcsimrootevent;
TH1F *h1 = new TH1F("PMT Hits", "# Digitized Hits", 500, 0, 3000);
TH1F *time = new TH1F("Average time", "Average time", 600, 900, 2000);
TH1F *pe = new TH1F("Q/# Digitzed PMT", "Average Charge", 200, 0, 5);
// Now loop over events
for (int ev=0; ev<nevent; ev++)
{
// Read the event from the tree into the WCSimRootEvent instance
wcsimT->GetEntry(ev);
wcsimrootevent = wcsimrootsuperevent->GetTrigger(0);
if(verbose){
printf("********************************************************");
printf("Evt, date %d %d\n", wcsimrootevent->GetHeader()->GetEvtNum(),
wcsimrootevent->GetHeader()->GetDate());
printf("Mode %d\n", wcsimrootevent->GetMode());
printf("Number of subevents %d\n",
wcsimrootsuperevent->GetNumberOfSubEvents());
printf("Vtxvol %d\n", wcsimrootevent->GetVtxvol());
printf("Vtx %f %f %f\n", wcsimrootevent->GetVtx(0),
wcsimrootevent->GetVtx(1),wcsimrootevent->GetVtx(2));
}
for (int index = 0 ; index < wcsimrootsuperevent->GetNumberOfEvents(); index++)
{
int ncherenkovdigihits = wcsimrootevent->GetNcherenkovdigihits();
h1->Fill(ncherenkovdigihits);
float totalq = 0.;
float totalt = 0.;
// Loop through elements in the TClonesArray of WCSimRootCherenkovHits
for (int i=0; i< ncherenkovdigihits; i++)
{
TObject *Digi = (wcsimrootevent->GetCherenkovDigiHits())->At(i);
WCSimRootCherenkovDigiHit *wcsimrootcherenkovdigihit =
dynamic_cast<WCSimRootCherenkovDigiHit*>(Digi);
int tubeNumber = (wcsimrootcherenkovdigihit->GetT(),wcsimrootcherenkovdigihit->GetTubeId());
float q = wcsimrootcherenkovdigihit->GetQ();
float t = wcsimrootcherenkovdigihit->GetT();
totalq+=q;
totalt+=t;
}
float av_time = totalt/ncherenkovdigihits;
float av_q = totalq/ncherenkovdigihits;
}
pe->Fill(av_q);
time->Fill(av_time);
// reinitialize super event between loops.
wcsimrootsuperevent->ReInitialize();
}// End of loop over events
TTree *wcsimT2 = f2->Get("wcsimT");
int nevent2 = wcsimT2->GetEntries();
WCSimRootEvent *wcsimrootsuperevent = new WCSimRootEvent();
wcsimT2->SetBranchAddress("wcsimrootevent",&wcsimrootsuperevent);
// Force deletion to prevent memory leak when issuing multiple
// calls to GetEvent()
wcsimT2->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE);
wcsimT2->GetEvent(0);
// In the default vis.mac, only one event is run. I suspect you could loop over more events, if they existed.
WCSimRootTrigger *wcsimrootevent = wcsimrootsuperevent->GetTrigger(0);
cout << "***********************************************************" << endl;
cout << "Stats for the first event of WCSim version on GitHub using "<< filename2 << endl;
cout << "Number of tube hits " << wcsimrootevent->GetNumTubesHit() << endl;
cout << "Number of digitized tube hits " << wcsimrootevent->GetNumDigiTubesHit() << endl;
cout << "Number of photoelectron hit times " << wcsimrootevent->GetCherenkovHitTimes()->GetEntries() << endl;
cout << "***********************************************************" << endl;
if (abs(num_tubes- wcsimrootevent->GetNumTubesHit())>1.0e-6){cout << "FIRST EVENT TEST FAILED: Number of hit tubes do not match" << endl;}
else {cout << "FIRST EVENT TEST PASSED: Number of hit tubes matches" << endl;}
if (abs(num_digi_tubes-wcsimrootevent->GetNumDigiTubesHit())>1.0e-6){cout << "FIRST EVENT TEST FAILED: Number of digitized tubes do not match" << endl; }
else {cout << "FIRST EVENT TEST PASSED: Number of digitized tubes matches" << endl; }
if (abs(hit_times-(wcsimrootevent->GetCherenkovHitTimes()->GetEntries()))> 1.0e-6){cout << "FIRST EVENT TEST FAILED: Number of hit times do not match" << endl;}
else {cout << "FIRST EVENT TEST PASSED: Number of hit times matches" << endl;}
// Create a WCSimRootEvent to put stuff from the tree in
WCSimRootEvent* wcsimrootsuperevent = new WCSimRootEvent();
// Set the branch address for reading from the tree
TBranch *branch = wcsimT2->GetBranch("wcsimrootevent");
branch->SetAddress(&wcsimrootsuperevent);
// Force deletion to prevent memory leak
wcsimT2->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE);
// start with the main "subevent", as it contains most of the info
// and always exists.
WCSimRootTrigger* wcsimrootevent;
TH1F *h2 = new TH1F("PMT Hits 2", "Digitized Hits", 500, 0, 3000);
TH1F *time2 = new TH1F("Average time 2", "Average time", 600, 900, 2000);
TH1F *pe2 = new TH1F("Q/# Digitzed PMT 2", "Q/# Digitzed PMT", 200, 0, 5);
// Now loop over events
for (int ev=0; ev<nevent; ev++)
{
// Read the event from the tree into the WCSimRootEvent instance
wcsimT2->GetEntry(ev);
wcsimrootevent = wcsimrootsuperevent->GetTrigger(0);
if(verbose){
printf("********************************************************");
printf("Evt, date %d %d\n", wcsimrootevent->GetHeader()->GetEvtNum(),
wcsimrootevent->GetHeader()->GetDate());
printf("Mode %d\n", wcsimrootevent->GetMode());
printf("Number of subevents %d\n",
wcsimrootsuperevent->GetNumberOfSubEvents());
printf("Vtxvol %d\n", wcsimrootevent->GetVtxvol());
printf("Vtx %f %f %f\n", wcsimrootevent->GetVtx(0),
wcsimrootevent->GetVtx(1),wcsimrootevent->GetVtx(2));
}
for (int index = 0 ; index < wcsimrootsuperevent->GetNumberOfEvents(); index++)
{
int ncherenkovdigihits = wcsimrootevent->GetNcherenkovdigihits();
h2->Fill(ncherenkovdigihits);
float totalq = 0.;
float totalt = 0.;
// Loop through elements in the TClonesArray of WCSimRootCherenkovHits
for (int i=0; i< ncherenkovdigihits; i++)
{
TObject *Digi = (wcsimrootevent->GetCherenkovDigiHits())->At(i);
WCSimRootCherenkovDigiHit *wcsimrootcherenkovdigihit =
dynamic_cast<WCSimRootCherenkovDigiHit*>(Digi);
int tubeNumber = (wcsimrootcherenkovdigihit->GetT(),wcsimrootcherenkovdigihit->GetTubeId());
float q = wcsimrootcherenkovdigihit->GetQ();
float t = wcsimrootcherenkovdigihit->GetT();
totalq+=q;
totalt+=t;
}
float av_time = totalt/ncherenkovdigihits;
float av_q = totalq/ncherenkovdigihits;
}
pe2->Fill(av_q);
time2->Fill(av_time);
// reinitialize super event between loops.
wcsimrootsuperevent->ReInitialize();
}// End of loop over events
Double_t ks_hits = h1->KolmogorovTest(h2);
Double_t ks_charge = pe->KolmogorovTest(pe2);
Double_t ks_time = time->KolmogorovTest(time2);
cout << "***********************************************************" << endl;
cout << "ks test for # of digitized hits: " << ks_hits << endl;
cout << "ks test for average charge: " << ks_charge << endl;
cout << "ks test for average time: " << ks_time << endl;
// TCanvas c1("c1");
float win_scale = 0.75;
int n_wide(2);
int n_high(2);
TCanvas* c1 = new TCanvas("c1", "Test Plots", 500*n_wide*win_scale, 500*n_high*win_scale);
c1->Draw();
c1->Divide(2,2);
c1->cd(1);
h2->SetLineColor(kRed);
h1->Draw();
c1->cd(1); h2->Draw("SAME");
TLegend *leg = new TLegend(0.2,0.7,0.55,0.85, "");
leg->SetFillColor(0);
leg->SetBorderSize(0);
leg->AddEntry(h1,filename, "l");
leg->AddEntry(h2,filename2, "l");
leg->Draw();
c1->cd(2);
pe->GetXaxis()->SetTitle("Total Charge / # digitized hits");
pe->Draw();
pe2->SetLineColor(kRed);
c1->cd(2); pe2->Draw("SAME");
c1->cd(3);
time->GetXaxis()->SetTitle("Total Time / # digitized hits (ns)");
time->Draw();
time2->SetLineColor(kRed);
c1->cd(3); time2->Draw("SAME");
}
void sample_readfile(const char* rootfile)
{
// Clear global scope
gROOT->Reset();
gStyle->SetOptStat(0);
gStyle->SetCanvasColor(0);
gStyle->SetTitleColor(1);
gStyle->SetStatColor(0);
gStyle->SetFrameFillColor(0);
gStyle->SetPadColor(0);
gStyle->SetPadTickX(1);
gStyle->SetPadTickY(1);
gStyle->SetTitleSize(0.04);
gStyle->SetCanvasBorderMode(0);
gStyle->SetFrameBorderMode(0);
gStyle->SetFrameLineWidth(2);
gStyle->SetPadBorderMode(0);
gStyle->SetPalette(1);
gStyle->SetTitleAlign(23);
gStyle->SetTitleX(.5);
gStyle->SetTitleY(0.99);
gStyle->SetTitleBorderSize(0);
gStyle->SetTitleFillColor(0);
gStyle->SetHatchesLineWidth(2);
gStyle->SetLineWidth(1.5);
gStyle->SetTitleFontSize(0.07);
gStyle->SetLabelSize(0.05,"X");
gStyle->SetLabelSize(0.05,"Y");
gStyle->SetTitleSize(0.04,"X");
gStyle->SetTitleSize(0.04,"Y");
gStyle->SetTitleBorderSize(0);
gStyle->SetCanvasBorderMode(0);
// Load the library with class dictionary info
gSystem.Load("libWCSimRoot.so");
// Open the file
TFile file(rootfile);
// Get the a pointer to the tree from the file
TTree *tree = (TTree*)file->Get("wcsimT");
// Get the number of events
int nevent = tree->GetEntries();
printf("nevent %d\n",nevent);
// Create a WCSimRootEvent to put stuff from the tree in
WCSimRootEvent* wcsimrootsuperevent = new WCSimRootEvent();
// Set the branch address for reading from the tree
TBranch *branch = tree->GetBranch("wcsimrootevent");
branch->SetAddress(&wcsimrootsuperevent);
// Force deletion to prevent memory leak
tree->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE);
// start with the main "subevent", as it contains most of the info
// and always exists.
WCSimRootTrigger* wcsimrootevent;
TH1F *h1 = new TH1F("PMT Hits", "PMT Hits on 1.5 GeV Muon in Geant4", 100, 1000, 9000);
// Now loop over events
for (int ev=0; ev<nevent; ev++)
{
// Read the event from the tree into the WCSimRootEvent instance
tree->GetEntry(ev);
wcsimrootevent = wcsimrootsuperevent->GetTrigger(0);
printf("********************************************************");
printf("Evt, date %d %d\n", wcsimrootevent->GetHeader()->GetEvtNum(),
wcsimrootevent->GetHeader()->GetDate());
printf("Mode %d\n", wcsimrootevent->GetMode());
printf("Number of subevents %d\n",
wcsimrootsuperevent->GetNumberOfSubEvents());
printf("Vtxvol %d\n", wcsimrootevent->GetVtxvol());
printf("Vtx %f %f %f\n", wcsimrootevent->GetVtx(0),
wcsimrootevent->GetVtx(1),wcsimrootevent->GetVtx(2));
printf("Jmu %d\n", wcsimrootevent->GetJmu());
printf("Npar %d\n", wcsimrootevent->GetNpar());
printf("Ntrack %d\n", wcsimrootevent->GetNtrack());
// Now read the tracks in the event
// Get the number of tracks
int ntrack = wcsimrootevent->GetNtrack();
printf("ntracks=%d\n",ntrack);
int i;
// Loop through elements in the TClonesArray of WCSimTracks
for (i=0; i<ntrack; i++)
{
TObject *element = (wcsimrootevent->GetTracks())->At(i);
WCSimRootTrack *wcsimroottrack = dynamic_cast<WCSimRootTrack*>(element);
printf("Track ipnu: %d\n",wcsimroottrack->GetIpnu());
printf("Track parent ID: %d\n",wcsimroottrack->GetParenttype());
for (int j=0; j<3; j++)
printf("Track dir: %d %f\n",j, wcsimroottrack->GetDir(j));
} // End of loop over tracks
// Now look at the Cherenkov hits
// Get the number of Cherenkov hits.
// Note... this is *NOT* the number of photons that hit tubes.
// It is the number of tubes hit with Cherenkov photons.
// The number of digitized tubes will be smaller because of the threshold.
// Each hit "raw" tube has several photon hits. The times are recorded.
// See http://nwg.phy.bnl.gov/DDRD/cgi-bin/private/ShowDocument?docid=245
// for more information on the structure of the root file.
//
// The following code prints out the hit times for the first 10 tubes and also
// adds up the total pe.
//
// For digitized info (one time/charge tube after a trigger) use
// the digitized information.
//
int ncherenkovhits = wcsimrootevent->GetNcherenkovhits();
int ncherenkovdigihits = wcsimrootevent->GetNcherenkovdigihits();
h1->Fill(ncherenkovdigihits);
printf("node id: %i\n", ev);
printf("Ncherenkovhits %d\n", ncherenkovhits);
printf("Ncherenkovdigihits %d\n", ncherenkovdigihits);
cout << "RAW HITS:" << endl;
// Grab the big arrays of times and parent IDs
TClonesArray *timeArray = wcsimrootevent->GetCherenkovHitTimes();
int totalPe = 0;
// Loop through elements in the TClonesArray of WCSimRootCherenkovHits
for (i=0; i< ncherenkovhits; i++)
{
TObject *Hit = (wcsimrootevent->GetCherenkovHits())->At(i);
WCSimRootCherenkovHit *wcsimrootcherenkovhit =
dynamic_cast<WCSimRootCherenkovHit*>(Hit);
int tubeNumber = wcsimrootcherenkovhit->GetTubeID();
int timeArrayIndex = wcsimrootcherenkovhit->GetTotalPe(0);
int peForTube = wcsimrootcherenkovhit->GetTotalPe(1);
totalPe += peForTube;
if ( i < 10 ) // Only print first XX=10 tubes
{
printf("Total pe: %d times( ",peForTube);
for (int j = timeArrayIndex; j < timeArrayIndex + peForTube; j++)
{
WCSimRootCherenkovHitTime HitTime =
dynamic_cast<WCSimRootCherenkovHitTime>(timeArray->At(j));
printf("%6.2f ", HitTime->GetTruetime() );
}
cout << ")" << endl;
}
} // End of loop over Cherenkov hits
cout << "Total Pe : " << totalPe << endl;
// Look at digitized hit info
// Get the number of digitized hits
// Loop over sub events
cout << "DIGITIZED HITS:" << endl;
for (int index = 0 ; index < wcsimrootsuperevent->GetNumberOfEvents(); index++)
{
wcsimrootevent = wcsimrootsuperevent->GetTrigger(index);
cout << "Sub event number = " << index << "\n";
int ncherenkovdigihits = wcsimrootevent->GetNcherenkovdigihits();
printf("Ncherenkovdigihits %d\n", ncherenkovdigihits);
//for (i=0;i<(ncherenkovdigihits>4 ? 4 : ncherenkovdigihits);i++){
for (i=0;i<ncherenkovdigihits;i++)
{
// Loop through elements in the TClonesArray of WCSimRootCherenkovDigHits
TObject *element = (wcsimrootevent->GetCherenkovDigiHits())->At(i);
WCSimRootCherenkovDigiHit *wcsimrootcherenkovdigihit =
dynamic_cast<WCSimRootCherenkovDigiHit*>(element);
if ( i < 10 ) // Only print first XX=10 tubes
printf("q, t, tubeid: %f %f %d \n",wcsimrootcherenkovdigihit->GetQ(),
wcsimrootcherenkovdigihit->GetT(),wcsimrootcherenkovdigihit->GetTubeId());
} // End of loop over Cherenkov digihits
} // End of loop over trigger
// reinitialize super event between loops.
wcsimrootsuperevent->ReInitialize();
} // End of loop over events
h1.Draw();
}
