SimG4Core/HelpfulWatchers/src/G4StepStatistics.h - master - githubmirror/cmssw

SimG4Core/HelpfulWatchers/src/G4StepStatistics.h
231 строка8 KB
Mircho Rodozov
[ROOT master] Add direct deps after ROOT cleanup
05 май 2020, 18:29
05 май 2020, 18:2941a9dfb
#ifndef HelpfulWatchers_G4StepStatistics_h
#define HelpfulWatchers_G4StepStatistics_h
// -*- C++ -*-
//
// Package:     HelpfulWatchers
// Class  :     SimTracer
//
/**\class SimTracer SimTracer.h SimG4Core/HelpfulWatchers/interface/SimTracer.h

 Description: Prints a message for each Oscar signal

 Usage:
    <usage>

*/
//
// Original Author:
//         Created:  Tue Nov 22 16:41:33 EST 2005
//

// system include files
#include <iostream>

// user include files
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "G4ParticleDefinition.hh"
#include "G4Step.hh"
#include "G4VProcess.hh"
#include "SimG4Core/Notification/interface/Observer.h"
#include "SimG4Core/Watcher/interface/SimWatcher.h"
#include <CommonTools/UtilAlgos/interface/TFileService.h>
#include <FWCore/ServiceRegistry/interface/Service.h>
#include <map>

#include <TClonesArray.h>
#include <TFile.h>
#include <TROOT.h>
#include <TString.h>
#include <TTree.h>
#include <TVector.h>
#include <TObjString.h>

// forward declarations
class DDDWorld;
class BeginOfJob;
class BeginOfRun;
class BeginOfEvent;
class BeginOfTrack;
class G4Step;

class EndOfRun;
class EndOfEvent;
class EndOfTrack;

// Define a class StepID
class StepID {
private:
  // G4 Region
  G4String theG4RegionName;
  // G4 Physical Process
  G4String theG4ProcessName;
  // Particle PDG ID
  G4int theParticlePDGID;

public:
  // Constructor using G4Step
  StepID(const G4Step *theG4Step)
      : theG4RegionName("UNDEFINED"),
        theG4ProcessName("UNDEFINED"),
        theParticlePDGID(theG4Step->GetTrack()->GetDefinition()->GetPDGEncoding()) {
    std::cout << "Start" << std::endl;
    if (theG4Step->GetPreStepPoint()->GetPhysicalVolume()) {
      theG4RegionName = theG4Step->GetPreStepPoint()->GetPhysicalVolume()->GetLogicalVolume()->GetRegion()->GetName();
    }
    std::cout << "Physical Volume" << std::endl;
    if (theG4Step->GetPreStepPoint()->GetProcessDefinedStep()) {
      theG4ProcessName = theG4Step->GetPreStepPoint()->GetProcessDefinedStep()->GetProcessName();
    }
    std::cout << "Process Name" << std::endl;
  }

  // Getters
  G4String GetRegionName() const { return theG4RegionName; }
  G4String GetProcessName() const { return theG4ProcessName; }
  G4int GetParticlePDGID() const { return theParticlePDGID; }

  // Comparison Operators (necessary in order to use StepID as a key in a map)
  bool operator==(const StepID &id) const {
    return (strcmp(theG4RegionName, id.GetRegionName()) == 0 && strcmp(theG4ProcessName, id.GetProcessName()) == 0 &&
            theParticlePDGID == id.GetParticlePDGID())
               ? true
               : false;
  }

  bool operator<(const StepID &id) const {
    if (theParticlePDGID != id.GetParticlePDGID()) {
      return (theParticlePDGID > id.GetParticlePDGID());
    } else if (strcmp(theG4RegionName, id.GetRegionName()) != 0) {
      return strcmp(theG4RegionName, id.GetRegionName()) > 0 ? true : false;
    } else if (strcmp(theG4ProcessName, id.GetProcessName()) != 0) {
      return strcmp(theG4ProcessName, id.GetProcessName()) > 0 ? true : false;
    } else {  // The case in which they are all equal!
      return false;
    }
  }

  bool operator>(const StepID &id) const {
    if (theParticlePDGID != id.GetParticlePDGID()) {
      return (theParticlePDGID < id.GetParticlePDGID());
    } else if (strcmp(theG4RegionName, id.GetRegionName()) != 0) {
      return strcmp(theG4RegionName, id.GetRegionName()) < 0 ? true : false;
    } else if (strcmp(theG4ProcessName, id.GetProcessName()) != 0) {
      return strcmp(theG4ProcessName, id.GetProcessName()) < 0 ? true : false;
    } else {  // The case in which they are all equal!
      return false;
    }
  }
};

#define OBSERVES(type) \
public                 \
  Observer<const type *>
#define UPDATE(type) \
  void update(const type *) override { std::cout << "++ signal " #type << std::endl; }
class G4StepStatistics : public SimWatcher,
                         OBSERVES(DDDWorld),
                         OBSERVES(BeginOfJob),
                         OBSERVES(BeginOfRun),
                         OBSERVES(BeginOfEvent),
                         OBSERVES(BeginOfTrack),
                         OBSERVES(G4Step),
                         OBSERVES(EndOfRun),
                         OBSERVES(EndOfEvent),
                         OBSERVES(EndOfTrack) {
public:
  G4StepStatistics(const edm::ParameterSet &pSet)
      : m_verbose(pSet.getUntrackedParameter<bool>("verbose", false)), Event(0) {
    // Adding TFile Service output
    G4StepTree = fs->make<TTree>("G4StepTree", "G4Step Tree ");
    G4StepTree->Branch("Event", &Event, "Event/I");
    G4StepTree->Branch("PDGID", &PDGID, "PDGID[100000]/I");
    Region = new TClonesArray("TObjString", 100000);
    G4StepTree->Branch("Region", &Region);
    Process = new TClonesArray("TObjString", 100000);
    G4StepTree->Branch("Process", &Process);
    G4StepTree->Branch("G4StepFreq", &G4StepFreq, "G4StepFreq[100000]/I");
  }
  UPDATE(DDDWorld)
  UPDATE(BeginOfJob)
  UPDATE(BeginOfRun)
  //    void update(const BeginOfRun* iRun) {
  // std::cout <<"++ signal BeginOfRun " <<std::endl;
  //}
  UPDATE(BeginOfEvent)
  UPDATE(BeginOfTrack)
  void update(const G4Step *iStep) override {
    std::cout << "++ signal G4Step ";
    // Dump the relevant information from the G4 step in the object mysteptest
    StepID mysteptest(iStep);
    // Add the StepID to the map, or increment if it already exists:
    if (G4StatsMap.find(mysteptest) == G4StatsMap.end()) {
      // Allocating new memory for a pointer to associate with the key
      // mysteptest in our map. Initializing it to 1,will be incremented working
      // on the value of the pointer.
      unsigned int *MyValue = new unsigned int(1);
      // Inserting the new key,value pair
      G4StatsMap.insert(std::make_pair(mysteptest, MyValue));
    } else {
      // Incrementing the value of the pointer by 1
      *G4StatsMap[mysteptest] = *G4StatsMap[mysteptest] + 1;
    }

    // If the verbose flag is set, then dump the information
    if (m_verbose) {
      std::cout << " StepID RegionName: " << mysteptest.GetRegionName();
      std::cout << " StepID ProcessName: " << mysteptest.GetProcessName();
      std::cout << " StepID ParticlePDGID: " << mysteptest.GetParticlePDGID();
    }
    std::cout << std::endl;
  }
  // UPDATE(G4Step)
  UPDATE(EndOfRun)
  //  void update(const EndOfRun* iRun) {
  // std::cout <<"++ signal EndOfRun " <<std::endl;
  //}

  // UPDATE(EndOfEvent)
  void update(const EndOfEvent *iRun) override {
    std::cout << "++ signal EndOfEvent " << std::endl;
    Event++;

    // Dumping the map in the log if verbose is chosen:
    if (m_verbose) {
      std::cout << " G4StatsMap size is: " << G4StatsMap.size() << std::endl;
    }
    int index(0);
    for (std::map<const StepID, unsigned int *>::const_iterator step = G4StatsMap.begin(); step != G4StatsMap.end();
         ++step, ++index) {
      if (m_verbose) {
        std::cout << " G4StatsMap step is: " << step->first.GetRegionName() << " " << step->first.GetProcessName()
                  << " " << step->first.GetParticlePDGID();  //<<" "<<step->first.GetTrackID() ;
        std::cout << " Number of such steps: " << *step->second << std::endl;
      }
      // Rolling the map into 5 "arrays", containing the StepID information and
      // the G4Step statistics
      PDGID[index] = step->first.GetParticlePDGID();
      new ((*Region)[index]) TObjString(step->first.GetRegionName());
      new ((*Process)[index]) TObjString(step->first.GetProcessName());
      G4StepFreq[index] = *step->second;
    }

    G4StepTree->Fill();
  }
  UPDATE(EndOfTrack)

private:
  bool m_verbose;

  // Adding the G4StatsMap to keep track of statistics in terms of step
  // information...
  std::map<const StepID, unsigned int *> G4StatsMap;
  edm::Service<TFileService> fs;
  TTree *G4StepTree;
  unsigned int Event;
  Int_t PDGID[100000];
  TClonesArray *Region;
  TClonesArray *Process;
  Int_t G4StepFreq[100000];
};

#endif