DDG4ProductionCuts.cc

Go to the documentation of this file.

#include "SimG4Core/Physics/interface/DDG4ProductionCuts.h"

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

#include "G4ProductionCuts.hh"
#include "G4RegionStore.hh"

#include <algorithm>

DDG4ProductionCuts::DDG4ProductionCuts(const G4LogicalVolumeToDDLogicalPartMap& map, int verb, 
                                       const edm::ParameterSet & p) 
  : map_(map), m_Verbosity(verb) {
  m_KeywordRegion =  "CMSCutsRegion";
  m_protonCut = p.getUntrackedParameter<bool>("CutsOnProton",true);  
  initialize();
}

DDG4ProductionCuts::~DDG4ProductionCuts(){
}

bool dd_is_greater(const std::pair<G4LogicalVolume*, DDLogicalPart> & p1,
                   const std::pair<G4LogicalVolume*, DDLogicalPart> & p2) {
  bool result = false;
  if (p1.second.name().ns() > p2.second.name().ns()) {
    result = true;
  }
  if (p1.second.name().ns() == p2.second.name().ns()) {
    if (p1.second.name().name() > p2.second.name().name()) {
      result = true;
    }
    if (p1.second.name().name() == p2.second.name().name()) {
      if (p1.first->GetName() > p2.first->GetName()) {
    result = true;
      }
    }
  }
  return result;
}          

void DDG4ProductionCuts::update() {
  //
  // Loop over all DDLP and provide the cuts for each region
  //
  for (G4LogicalVolumeToDDLogicalPartMap::Vector::iterator tit = vec_.begin();
       tit != vec_.end(); tit++){
    setProdCuts((*tit).second,(*tit).first);
  }
}


void DDG4ProductionCuts::initialize() {

  vec_ = map_.all(m_KeywordRegion);
  // sort all root volumes - to get the same sequence at every run of the application.
  // (otherwise, the sequence will depend on the pointer (memory address) of the 
  // involved objects, because 'new' does no guarantee that you allways get a
  // higher (or lower) address when allocating an object of the same type ...
  sort(vec_.begin(),vec_.end(),&dd_is_greater);
  if ( m_Verbosity > 0 ) {
    LogDebug("Physics") <<" DDG4ProductionCuts (New) : starting\n"
            <<" DDG4ProductionCuts : Got "<<vec_.size()
            <<" region roots.\n"
            <<" DDG4ProductionCuts : List of all roots:";
    for ( size_t jj=0; jj<vec_.size(); ++jj)
      LogDebug("Physics") << "   DDG4ProductionCuts : root=" 
              << vec_[jj].second.name();
  }

  // Now generate all the regions
  for (G4LogicalVolumeToDDLogicalPartMap::Vector::iterator tit = vec_.begin();
       tit != vec_.end(); tit++) {

    std::string  regionName;
    unsigned int num= map_.toString(m_KeywordRegion,(*tit).second,regionName);
  
    if (num != 1) {
      throw cms::Exception("SimG4CorePhysics", " DDG4ProductionCuts::initialize: Problem with Region tags.");
    }
    G4Region * region = getRegion(regionName);
    region->AddRootLogicalVolume((*tit).first);
  
    if ( m_Verbosity > 0 )
      LogDebug("Physics") << " MakeRegions: added " <<((*tit).first)->GetName()
              << " to region " << region->GetName();
  }
}


void DDG4ProductionCuts::setProdCuts(const DDLogicalPart lpart, 
                     G4LogicalVolume* lvol ) {  
  
  if ( m_Verbosity > 0 ) 
    LogDebug("Physics") <<" DDG4ProductionCuts: inside setProdCuts";
  
  G4Region * region = nullptr;
  
  std::string  regionName;
  unsigned int num= map_.toString(m_KeywordRegion,lpart,regionName);
  
  if (num != 1) {
    throw cms::Exception("SimG4CorePhysics", " DDG4ProductionCuts::setProdCuts: Problem with Region tags.");
  }
  if ( m_Verbosity > 0 ) LogDebug("Physics") << "Using region " << regionName;

  region = getRegion(regionName);

  //
  // search for production cuts
  // you must have four of them: e+ e- gamma proton
  //
  double gammacut;
  double electroncut;
  double positroncut;
  double protoncut = 0.0;
  int temp =  map_.toDouble("ProdCutsForGamma",lpart,gammacut);
  if (temp != 1){
    throw cms::Exception("SimG4CorePhysics", " DDG4ProductionCuts::setProdCuts: Problem with Region tags - no/more than one ProdCutsForGamma.");
  }
  temp =  map_.toDouble("ProdCutsForElectrons",lpart,electroncut);
  if (temp != 1){
    throw cms::Exception("SimG4CorePhysics", " DDG4ProductionCuts::setProdCuts: Problem with Region tags - no/more than one ProdCutsForElectrons.");
  }
  temp =  map_.toDouble("ProdCutsForPositrons",lpart,positroncut);
  if (temp != 1) {
    throw cms::Exception("SimG4CorePhysics", " DDG4ProductionCuts::setProdCuts: Problem with Region tags - no/more than one ProdCutsForPositrons.");
  }
  //
  // For the moment I assume all of the three are set
  //
  G4ProductionCuts * prodCuts = getProductionCuts(region);
  prodCuts->SetProductionCut( gammacut, idxG4GammaCut );
  prodCuts->SetProductionCut( electroncut, idxG4ElectronCut );
  prodCuts->SetProductionCut( positroncut, idxG4PositronCut );
  // For recoil use the same cut as for e-
  if(m_protonCut) { protoncut = electroncut; }
  prodCuts->SetProductionCut( protoncut, idxG4ProtonCut );
  if ( m_Verbosity > 0 ) {
    LogDebug("Physics") << "DDG4ProductionCuts : Setting cuts for " 
            << regionName << "\n    Electrons: " << electroncut
            << "\n    Positrons: " << positroncut
            << "\n    Gamma    : " << gammacut;
  }
}

G4Region * DDG4ProductionCuts::getRegion(const std::string & regName) {
  G4Region * reg =  G4RegionStore::GetInstance()->FindOrCreateRegion (regName);
  return reg;
}

G4ProductionCuts * DDG4ProductionCuts::getProductionCuts( G4Region* reg ) {

  G4ProductionCuts * prodCuts = reg->GetProductionCuts();
  if( !prodCuts ) {
    prodCuts = new G4ProductionCuts();
    reg->SetProductionCuts(prodCuts);
  }
  return prodCuts;
}