SimG4Core/CheckSecondary/src/StoreSecondary.cc - master - githubmirror/cmssw

SimG4Core/CheckSecondary/src/StoreSecondary.cc
92 строки3 KB
Slava Krutelyov
suggestions from code review
04 май 2024, 01:04
04 май 2024, 01:042c903c9
#include "SimG4Core/CheckSecondary/interface/StoreSecondary.h"
#include "SimG4Core/CheckSecondary/interface/TreatSecondary.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"

#include "SimG4Core/Notification/interface/BeginOfEvent.h"
#include "SimG4Core/Notification/interface/BeginOfTrack.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include <CLHEP/Units/GlobalPhysicalConstants.h>
#include <CLHEP/Units/SystemOfUnits.h>
#include "G4HCofThisEvent.hh"
#include "G4Step.hh"
#include "G4Track.hh"

#include <cmath>
#include <iomanip>
#include <iostream>

StoreSecondary::StoreSecondary(const edm::ParameterSet &p) {
  edm::ParameterSet m_p = p.getParameter<edm::ParameterSet>("StoreSecondary");
  treatSecondary = new TreatSecondary(m_p);

  produces<std::vector<math::XYZTLorentzVector>>("SecondaryMomenta");
  produces<std::vector<int>>("SecondaryParticles");

  edm::LogInfo("CheckSecondary") << "Instantiate StoreSecondary to store "
                                 << "secondaries after 1st hadronic inelastic"
                                 << " interaction";
}

StoreSecondary::~StoreSecondary() { delete treatSecondary; }

void StoreSecondary::produce(edm::Event &e, const edm::EventSetup &) {
  std::unique_ptr<std::vector<math::XYZTLorentzVector>> secMom(new std::vector<math::XYZTLorentzVector>);
  *secMom = secondaries;
  e.put(std::move(secMom), "SecondaryMomenta");

  std::unique_ptr<std::vector<int>> secNumber(new std::vector<int>);
  *secNumber = nsecs;
  e.put(std::move(secNumber), "SecondaryParticles");

  LogDebug("CheckSecondary") << "StoreSecondary:: Event " << e.id() << " with " << nsecs.size()
                             << " hadronic collisions with "
                             << "secondaries produced in each step";
  for (unsigned int i = 0; i < nsecs.size(); i++) {
    LogDebug("CheckSecondary") << " " << nsecs[i] << " from " << procs[i];
  }
  LogDebug("CheckSecondary") << " and " << secondaries.size() << " secondaries"
                             << " produced in the first interactions:";
  for (unsigned int i = 0; i < secondaries.size(); i++)
    LogDebug("CheckSecondary") << "Secondary " << i << " " << secondaries[i];
}

void StoreSecondary::update(const BeginOfEvent *) {
  nsecs.clear();
  procs.clear();
  secondaries.clear();
}

void StoreSecondary::update(const BeginOfTrack *trk) {
  const G4Track *thTk = (*trk)();
  treatSecondary->initTrack(thTk);
  if (nsecs.empty() && thTk->GetParentID() <= 0)
    storeIt = true;
  else
    storeIt = false;
  nHad = 0;
}

void StoreSecondary::update(const G4Step *aStep) {
  std::string name;
  int procID;
  bool hadrInt;
  double deltaE;
  std::vector<int> charge;
  std::vector<math::XYZTLorentzVector> tracks = treatSecondary->tracks(aStep, name, procID, hadrInt, deltaE, charge);
  if (hadrInt) {
    ++nHad;
    if (storeIt) {
      int sec = (int)(tracks.size());
      nsecs.push_back(sec);
      procs.push_back(name);
      if (nHad == 1) {
        for (int i = 0; i < sec; i++)
          secondaries.push_back(tracks[i]);
      }
    }
  }
}