Login

New Frontiers in 5D Calorimeter Reconstruction

→ America/Los_Angeles
    • 14:00 → 14:30
      Status Updates 30m
      Speakers: Harsh Tibrewala (SLAC), Liangyu Wu (SLAC)
      Github Link for DREAMSim
      High-granularity Dual-readout Calorimeter
      On the Use of Neural Networks for Energy Reconstruction in High-granularity Calorimeters
      Root files for 10GeV e- and pi- (10events for each)

      An example code to analyze the root file:

      #include <iostream>
      #include "TFile.h"
      #include "TTree.h"
      #include "TH2F.h"
      #include "TCanvas.h"
      #include "TDirectory.h"
      #include "TROOT.h"
      #include "TStyle.h"
      #include <string>
      #include <sstream>
      #include <vector>
      #include <filesystem>
      #include <algorithm>


      // Plot2dposition(1, 19, "./testdata/", 400, 400, -4.3, -3.9, 4.3, 4.7)

      int Plot2dposition(
          int numFiles = 1,                // Number of files to process (1-10)
          int endEventNumber = 19,         // Last event number to process in each file
          const char* baseDir = "./testdata/",  // Directory containing the ROOT files
          int xBins = 400,
          int yBins = 400,
          double xMin = -4.3,
          double xMax = -3.9,
          double yMin = 4.3,
          double yMax = 4.7
      ) {
          
          TH2F* h2 = new TH2F("h2", "Photon Distribution", 
                               xBins, xMin, xMax, 
                               yBins, yMin, yMax);
          
          long totalProcessedEvents = 0;
          long totalProcessedPhotons = 0;
          
          for (int fileNum = 1; fileNum <= numFiles; fileNum++) {
              stringstream fileNameStr;
              fileNameStr << baseDir << "mc_e+_job_rod43_layer42_run43_" << fileNum 
                          << "_Test_20evt_e+_100_100.root";
              string fileName = fileNameStr.str();
              
              cout << "Processing file " << fileNum << "/" << numFiles << ": " << fileName << endl;
              
              TFile* f = TFile::Open(fileName.c_str());
              
              TTree* tree = (TTree*)f->Get("tree");
              if (!tree) {
                  cout << "Failed to get tree 'tree' from file: " << fileName << endl;
                  f->Close();
                  continue;
              }
              
              int totalEvents = tree->GetEntries();
              cout << "  File contains " << totalEvents << " events" << endl;
              

              stringstream eventSelectStr;
              if (endEventNumber == 0) {
                  eventSelectStr << "Entry$==0";
              } else {
                  eventSelectStr << "Entry$>=" << 0 << " && Entry$<=" << endEventNumber;
              }
              string eventSelect = eventSelectStr.str();
              
              stringstream drawStr;
              drawStr << "OP_pos_final_y:OP_pos_final_x";
              string drawCondition = drawStr.str();
              
              string selectionCondition = eventSelect + " && OP_pos_final_z>80 && OP_isCoreC";
              
              TH2F* tempHist = new TH2F("tempHist", "Temp Histogram", 
                                        xBins, xMin, xMax, 
                                        yBins, yMin, yMax);
              
              tree->Draw((drawCondition + ">>tempHist").c_str(), selectionCondition.c_str(), "goff");
              
              double entriesInFile = tempHist->GetEntries();
              totalProcessedPhotons += entriesInFile;
              
              cout << "  Processed " << min(endEventNumber + 1, totalEvents) 
                   << " events, found " << entriesInFile << " photons" << endl;
              
              h2->Add(tempHist);
              
              delete tempHist;
              f->Close();
              delete f;
              
              totalProcessedEvents += min(endEventNumber + 1, totalEvents);
          }
          
          TCanvas* c = new TCanvas("c1", "Combined Histogram", 800, 600);
          
          stringstream titleStr;
          titleStr << totalProcessedEvents << " events (" 
                   << 1000/xBins << "x" << 1000/yBins << " #mu m^{2} )";
          h2->SetTitle(titleStr.str().c_str());
          h2->SetXTitle("x [cm]");
          h2->SetYTitle("y [cm]");

          h2->Draw("colz");
          
          int maxBin = h2->GetMaximumBin();
          int binX, binY, binZ;
          h2->GetBinXYZ(maxBin, binX, binY, binZ);
          
          stringstream outFileStr;
          outFileStr << "2DHistogram_" << xBins << "x" << yBins 
                     << "_" << numFiles << "files_" 
                     << totalProcessedEvents << "events.png";
          
          c->SaveAs(outFileStr.str().c_str());
          
          delete c;
          delete h2;
          
          return 0;
      }