TotemTestGem.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     Forward
// Class  :     TotemTestGem
//
// Implementation:
//     <Notes on implementation>
//
// Original Author:
//         Created:  Tue May 16 10:14:34 CEST 2006
//
 
// system include files
#include <iostream>
#include <iomanip>
#include <cmath>
 
// user include files
#include "SimG4Core/Notification/interface/BeginOfEvent.h"
#include "SimG4Core/Notification/interface/EndOfEvent.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
#include "SimG4CMS/Forward/interface/TotemTestGem.h"
#include "SimG4CMS/Forward/interface/TotemG4HitCollection.h"
 
#include "G4SDManager.hh"
#include "G4HCofThisEvent.hh"
#include "CLHEP/Units/GlobalSystemOfUnits.h"
#include "CLHEP/Units/GlobalPhysicalConstants.h"
 
//
// constructors and destructor
//
 
TotemTestGem::TotemTestGem(const edm::ParameterSet& p) {
  edm::ParameterSet m_Anal = p.getParameter<edm::ParameterSet>("TotemTestGem");
  names = m_Anal.getParameter<std::vector<std::string> >("Names");
 
  edm::LogInfo("ForwardSim") << "TotemTestGem:: Initialised as observer of "
                             << "begin of job, begin/end events and of G4step";
}
 
TotemTestGem::~TotemTestGem() {}
 
//
// member functions
//
 
void TotemTestGem::produce(edm::Event& e, const edm::EventSetup&) {
  std::unique_ptr<TotemTestHistoClass> product(new TotemTestHistoClass);
  fillEvent(*product);
  e.put(std::move(product));
}
 
void TotemTestGem::update(const BeginOfEvent* evt) {
  int iev = (*evt)()->GetEventID();
  LogDebug("ForwardSim") << "TotemTestGem: Begin of event = " << iev;
  clear();
}
 
void TotemTestGem::update(const EndOfEvent* evt) {
  evtnum = (*evt)()->GetEventID();
  LogDebug("ForwardSim") << "TotemTestGem:: Fill event " << evtnum;
 
  // access to the G4 hit collections
  G4HCofThisEvent* allHC = (*evt)()->GetHCofThisEvent();
 
  int nhit = 0;
  for (unsigned int in = 0; in < names.size(); in++) {
    int HCid = G4SDManager::GetSDMpointer()->GetCollectionID(names[in]);
    TotemG4HitCollection* theHC = (TotemG4HitCollection*)allHC->GetHC(HCid);
    LogDebug("ForwardSim") << "TotemTestGem :: Hit Collection for " << names[in] << " of ID " << HCid
                           << " is obtained at " << theHC;
 
    if (HCid >= 0 && theHC != nullptr) {
      int nentries = theHC->entries();
      LogDebug("ForwardSim") << "TotemTestGem :: " << names[in] << " with " << nentries << " entries";
      for (int ihit = 0; ihit < nentries; ihit++) {
        TotemG4Hit* aHit = (*theHC)[ihit];
        hits.push_back(aHit);
      }
      nhit += nentries;
    }
  }
 
  // Writing the data to the Tree
  LogDebug("ForwardSim") << "TotemTestGem:: --- after fillTree with " << nhit << " Hits";
}
 
void TotemTestGem::fillEvent(TotemTestHistoClass& product) {
  product.setEVT(evtnum);
 
  for (unsigned ihit = 0; ihit < hits.size(); ihit++) {
    TotemG4Hit* aHit = hits[ihit];
    int UID = aHit->getUnitID();
    int Ptype = aHit->getParticleType();
    int TID = aHit->getTrackID();
    int PID = aHit->getParentId();
    float ELoss = aHit->getEnergyLoss();
    float PABS = aHit->getPabs();
    float x = aHit->getEntry().x();
    float y = aHit->getEntry().y();
    float z = aHit->getEntry().z();
    float vx = aHit->getVx();
    float vy = aHit->getVy();
    float vz = aHit->getVz();
    product.fillHit(UID, Ptype, TID, PID, ELoss, PABS, vx, vy, vz, x, y, z);
  }
}
 
void TotemTestGem::clear() {
  evtnum = 0;
  hits.clear();
}