void pmtremove(TString infile, TString outfile, double removefrac) { // read old geometry from the input rootfile TFile *f2 = new TFile(infile); TTree *oldtree = (TTree*)f2->Get("wcsimGeoT"); WCSimRootGeom* oldgeom = new WCSimRootGeom(); TBranch *gb = oldtree->GetBranch("wcsimrootgeom"); gb->SetAddress(&oldgeom); oldtree->GetEntry(0); int oldpmtnum = oldgeom->GetWCNumPMT(); Printf("Initial Number of PMTs: %d", oldpmtnum); // set up the new output file TFile *f3 = new TFile(outfile,"RECREATE"); TTree *newtree = oldtree->CloneTree(0); WCSimRootGeom* newgeom = new WCSimRootGeom(); newtree->SetBranchAddress("wcsimrootgeom",&newgeom); newgeom->SetWCCylRadius(oldgeom->GetWCCylRadius()); newgeom->SetWCCylLength(oldgeom->GetWCCylLength()); newgeom->SetMailBox_x(oldgeom->GetMailBox_x()); newgeom->SetMailBox_y(oldgeom->GetMailBox_y()); newgeom->SetMailBox_z(oldgeom->GetMailBox_z()); newgeom->SetGeo_Type(oldgeom->GetGeo_Type()); newgeom->SetWCOffset(oldgeom->GetWCOffset(0), oldgeom->GetWCOffset(1), oldgeom->GetWCOffset(2)); newgeom->SetOrientation(oldgeom->GetOrientation()); newgeom->SetWCPMTRadius(oldgeom->GetWCPMTRadius()); // associates old PMTs with new PMTs Index into it is the old pmt number, // gives back new PMT number. If PMT is removed, this array has -1 in it. const int asize = oldpmtnum + 1; int pmtassoc[asize]; int newpmtnum = 0; int ict = 0; // old floor count for (int i=0;i<oldpmtnum;i++) { WCSimRootPMT pmt = oldgeom->GetPMT(i); int oldtubeno = pmt.GetTubeNo(); double ft = removefrac*( (double) i); int icc = (int) floor(ft); // check to see when this rolls around another integer if ( ict != icc) { ict = icc; pmtassoc[oldtubeno] = -1; // and remove this tube } else { pmtassoc[oldtubeno] = newpmtnum+1; float rot[3]; float pos[3]; for (int j=0;j<3;j++) { rot[j] = pmt.GetOrientation(j); pos[j] = pmt.GetPosition(j); } newgeom->SetPMT(newpmtnum,newpmtnum+1,pmt.GetCylLoc(),rot,pos,true); newpmtnum++; } //if (i<100) std::cout << "input pmt: " << i << " " << oldtubeno << " assoc: " << pmtassoc[oldtubeno] << std::endl; } newgeom->SetWCNumPMT(newpmtnum); std::cout << "New PMT count: " << newpmtnum << std::endl; newtree->Fill(); newtree->Write(); //------------------------------------------------------------------------------------------- //---------- Go through events -- drop PMT hits and renumber PMTs with the ones we have. //------------------------------------------------------------------------------------------- TTree *oldevtree = (TTree*) f2->Get("wcsimT"); int nevent = oldevtree->GetEntries(); std::cout << "Number of events: " << nevent << std::endl; // Create a WCSimRootEvent to put stuff from the old tree in WCSimRootEvent* oldsuperevent = new WCSimRootEvent(); TBranch *branch = oldevtree->GetBranch("wcsimrootevent"); branch->SetAddress(&oldsuperevent); // Force deletion to prevent memory leak oldevtree->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE); // Initialize the output tree TTree *newevtree = oldevtree->CloneTree(0); WCSimRootEvent *newsuperevent = new WCSimRootEvent(); newsuperevent->Initialize(); newevtree->SetBranchAddress("wcsimrootevent",&newsuperevent); // Now loop over events for (int ev=0; ev<nevent; ev++) { // Read the event from the tree into the WCSimRootEvent instance oldevtree->GetEntry(ev); // new output event -- allocate memory here and let it go out of scope at the end of the loop. // performance isn't so much an issue here. for (int itrigger = 0; itrigger < oldsuperevent->GetNumberOfEvents(); itrigger++) { //std::cout << "about to get a trigger: Event: " << ev << " Trigger: " << itrigger << std::endl; WCSimRootTrigger *oldtrigger; oldtrigger = oldsuperevent->GetTrigger(itrigger); if (itrigger > 0) newsuperevent->AddSubEvent(); // don't have to add the first trigger, done by initialize WCSimRootTrigger *newtrigger = newsuperevent->GetTrigger(itrigger); WCSimRootEventHeader *oldheader = oldtrigger->GetHeader(); newtrigger->SetHeader(oldheader->GetEvtNum(),oldheader->GetRun(),oldheader->GetDate(),oldheader->GetSubEvtNumber()); newtrigger->SetMode(oldtrigger->GetMode()); newtrigger->SetVtxvol(oldtrigger->GetVtxvol()); for (int i=0;i<3;i++) newtrigger->SetVtx(i,oldtrigger->GetVtx(i)); newtrigger->SetJmu(oldtrigger->GetJmu()); newtrigger->SetJp(oldtrigger->GetJp()); newtrigger->SetNpar(oldtrigger->GetNpar()); WCSimRootPi0 *pi0info = oldtrigger->GetPi0Info(); float pi0vtx[3],gammae[2],gammavtx[2][3]; int gammaid[2]; for (int i=0;i<3;i++) { pi0vtx[i] = pi0info->GetPi0Vtx(i); gammavtx[0][i] = pi0info->GetGammaVtx(0,i); gammavtx[1][i] = pi0info->GetGammaVtx(1,i); } for (int i=0;i<2;i++) { gammaid[i] = pi0info->GetGammaID(i); gammae[i] = pi0info->GetGammaE(i); } newtrigger->SetPi0Info(pi0vtx,gammaid,gammae,gammavtx); // Get the number of tracks int ntrack = oldtrigger->GetNtrack(); //printf("ntracks=%d\n",ntrack); // Loop through elements in the TClonesArray of WCSimTracks for (int i=0; i<ntrack; i++) { TObject *element = (oldtrigger->GetTracks())->At(i); WCSimRootTrack *track = dynamic_cast<WCSimRootTrack*>(element); float dir[3],pdir[3],stop[3],start[3]; for (int j=0;j<3;j++) { dir[j] = track->GetDir(j); pdir[j] = track->GetPdir(j); stop[j] = track->GetStop(j); start[j] = track->GetStart(j); } newtrigger->AddTrack(track->GetIpnu(), track->GetFlag(), track->GetM(), track->GetP(), track->GetE(), track->GetStartvol(), track->GetStopvol(), dir, pdir, stop, start, track->GetParenttype(), track->GetTime(), track->GetId()); } // End of loop over tracks // stopped here March 22 -- keep copying info from the old trigger to the new trigger // 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 chapter 5 of ../doc/DetectorDocumentation.pdf // for more information on the structure of the root file. // // For digitized info (one time/charge tube after a trigger) use // the digitized information. // int ncherenkovhits = oldtrigger->GetNcherenkovhits(); int ncherenkovdigihits = oldtrigger->GetNcherenkovdigihits(); // Grab the big arrays of times and parent IDs TClonesArray *timeArray = oldtrigger->GetCherenkovHitTimes(); // Loop through elements in the TClonesArray of WCSimRootCherenkovHits and copy info for non-removed PMT's for (int i=0; i< ncherenkovhits; i++) { TObject *Hit = (oldtrigger->GetCherenkovHits())->At(i); WCSimRootCherenkovHit *wcsimrootcherenkovhit = dynamic_cast<WCSimRootCherenkovHit*>(Hit); int tubeNumber = wcsimrootcherenkovhit->GetTubeID(); int timeArrayIndex = wcsimrootcherenkovhit->GetTotalPe(0); int peForTube = wcsimrootcherenkovhit->GetTotalPe(1); if (tubeNumber < 1 || tubeNumber > oldpmtnum) { std::cout << "Error in pmtremove: tube number out of range: " << tubeNumber << " max: " << oldpmtnum << std::endl; exit(0); // die if we get a bad tube number } int newtubenumber = pmtassoc[tubeNumber]; if (newtubenumber > 0) // keep the tube's hits { vector<float> truetime; vector<int> ppid; for (int j=0; j<peForTube; j++) { WCSimRootCherenkovHitTime *HitTime = dynamic_cast<WCSimRootCherenkovHitTime*>(timeArray->At(timeArrayIndex+j)); truetime.push_back(HitTime->GetTruetime()); ppid.push_back(HitTime->GetParentID()); } newtrigger->AddCherenkovHit(newtubenumber,truetime,ppid); } } // End of loop over Cherenkov hits // Copy digitized hits for non-removed PMT's, and recompute sumq float sumq = 0; for (int i=0;i<ncherenkovdigihits;i++) { // Loop through elements in the TClonesArray of WCSimRootCherenkovDigiHits TObject *element = (oldtrigger->GetCherenkovDigiHits())->At(i); WCSimRootCherenkovDigiHit *wcsimrootcherenkovdigihit = dynamic_cast<WCSimRootCherenkovDigiHit*>(element); int tubeNumber = wcsimrootcherenkovdigihit->GetTubeId(); if (tubeNumber < 1 || tubeNumber > oldpmtnum) { std::cout << "Error in pmtremove: tube number out of range: " << tubeNumber << " max: " << oldpmtnum << std::endl; exit(0); // die if we get a bad tube number } int newtubenumber = pmtassoc[tubeNumber]; if (newtubenumber > 0) // keep the tube's hits { newtrigger->AddCherenkovDigiHit(wcsimrootcherenkovdigihit->GetQ(), wcsimrootcherenkovdigihit->GetT(), newtubenumber); sumq += wcsimrootcherenkovdigihit->GetQ(); } //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 newtrigger->SetSumQ(sumq); //if (ev<100) { std::cout << "Sum charge: " << ev << " " << itrigger << " " << oldtrigger->GetSumQ() << " " << sumq << std::endl; } } // End of loop over triggers in the event newevtree->Fill(); newevtree->Write(); // reinitialize super events between loops. oldsuperevent->ReInitialize(); newsuperevent->ReInitialize(); } // End of loop over events delete f2; delete f3; }
// 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"; }
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(); }