MaterialBudgetForward.cc

Go to the documentation of this file.

#include "Validation/Geometry/interface/MaterialBudgetForward.h"

#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

#include "SimG4Core/Notification/interface/BeginOfRun.h"
#include "SimG4Core/Notification/interface/BeginOfTrack.h"
#include "SimG4Core/Notification/interface/EndOfTrack.h"

#include "G4LogicalVolumeStore.hh"
#include "G4Step.hh"
#include "G4Track.hh"

#include <iostream>

const int MaterialBudgetForward::maxSet;

MaterialBudgetForward::MaterialBudgetForward(const edm::ParameterSet& p) {
  
  edm::ParameterSet m_p = p.getParameter<edm::ParameterSet>("MaterialBudgetForward");
  detTypes     = m_p.getParameter<std::vector<std::string> >("DetectorTypes");
  constituents = m_p.getParameter<std::vector<int> >("Constituents");
  stackOrder   = m_p.getParameter<std::vector<int> >("StackOrder");
  detNames     = m_p.getParameter<std::vector<std::string> >("DetectorNames");
  detLevels    = m_p.getParameter<std::vector<int> >("DetectorLevels");
  etaRegions   = m_p.getParameter<std::vector<double> >("EtaBoundaries");
  regionTypes  = m_p.getParameter<std::vector<int> >("RegionTypes");
  boundaries   = m_p.getParameter<std::vector<double> >("Boundaries");
  edm::LogInfo("MaterialBudget") << "MaterialBudgetForward initialized for "
                 << detTypes.size() << " detector types";
  unsigned int nc = 0;
  for (unsigned int ii=0; ii<detTypes.size(); ++ii) {
    edm::LogInfo("MaterialBudget") << "Type[" << ii << "] : " << detTypes[ii]
                   << " with " << constituents[ii] <<", order "
                   << stackOrder[ii] << " constituents --> ";
    for (int kk=0; kk<constituents[ii]; ++kk) {
      std::string name = "Unknown"; int level = -1;
      if (nc < detNames.size()) {
    name = detNames[nc]; level = detLevels[nc]; ++nc;
      }
      edm::LogInfo("MaterialBudget") << "    Constituent[" << kk << "]: "
                     << name << " at level " << level;
    }
  }
  edm::LogInfo("MaterialBudget") << "MaterialBudgetForward Stop condition for "
                 << etaRegions.size() << " eta regions";
  for (unsigned int ii=0; ii<etaRegions.size(); ++ii) {
    edm::LogInfo("MaterialBudget") << "Region[" << ii << "] : Eta < " 
                   << etaRegions[ii] << " boundary type "
                   << regionTypes[ii] << " limit "
                   << boundaries[ii];
  }
  book(m_p);
}

MaterialBudgetForward::~MaterialBudgetForward() {
}

void MaterialBudgetForward::update(const BeginOfRun* ) {

  const G4LogicalVolumeStore * lvs = G4LogicalVolumeStore::GetInstance();
  std::vector<G4LogicalVolume *>::const_iterator lvcite;

  unsigned int kount=detNames.size();
  for (unsigned int ii=0; ii<kount; ++ii) 
    logVolumes.push_back(0);

  for (lvcite = lvs->begin(); lvcite != lvs->end(); lvcite++) {
    for (unsigned int ii=0; ii<detNames.size(); ++ii) {
      if (strcmp(detNames[ii].c_str(),(*lvcite)->GetName().c_str()) == 0) {
    logVolumes[ii] = (*lvcite);
    kount--;
    break;
      }
    }
    if (kount <= 0) break;
  }
  edm::LogInfo("MaterialBudget") << "MaterialBudgetForward::Finds " 
                 << detNames.size()-kount << " out of "
                 << detNames.size() << " LV addresses";
  for (unsigned int ii=0; ii<detNames.size(); ++ii) {
    std::string name("Unknown");
    if (logVolumes[ii] != 0)  name = logVolumes[ii]->GetName();
    edm::LogInfo("MaterialBudget") << "LV[" << ii << "] : " << detNames[ii]
                   << " Address " << logVolumes[ii] << " | " 
                   << name;
  }

  for (unsigned int ii=0; ii<(detTypes.size()+1); ++ii) {
    stepLen.push_back(0); radLen.push_back(0); intLen.push_back(0);
  }
  stackOrder.push_back(0);
}

void MaterialBudgetForward::update(const BeginOfTrack* trk) {

  for (unsigned int ii=0; ii<(detTypes.size()+1); ++ii) {
    stepLen[ii] = 0; radLen[ii] = 0; intLen[ii] = 0;
  }

  const G4Track * aTrack = (*trk)(); // recover G4 pointer if wanted
  const G4ThreeVector& mom = aTrack->GetMomentum() ;
  if (mom.theta() != 0 ) {
    eta = mom.eta();
  } else {
    eta = -99;
  }
  phi = mom.phi();
  stepT = 0;

  double theEnergy = aTrack->GetTotalEnergy();
  int    theID     = (int)(aTrack->GetDefinition()->GetPDGEncoding());
  LogDebug("MaterialBudget") << "MaterialBudgetHcalHistos: Track "
                             << aTrack->GetTrackID() << " Code " << theID
                             << " Energy " <<theEnergy/CLHEP::GeV<<" GeV; Eta "
                             << eta << " Phi " << phi/CLHEP::deg << " PT "
                             << mom.perp()/CLHEP::GeV << " GeV *****";
}
 
void MaterialBudgetForward::update(const G4Step* aStep) {

  //---------- each step
  G4Material * material = aStep->GetPreStepPoint()->GetMaterial();
  double step    = aStep->GetStepLength();
  double radl    = material->GetRadlen();
  double intl    = material->GetNuclearInterLength();

  const G4VTouchable* touch = aStep->GetPreStepPoint()->GetTouchable();
  unsigned int indx = detTypes.size();
  unsigned int nc = 0;
  for (unsigned int ii=0; ii<detTypes.size(); ++ii) {
    for (int kk=0; kk<constituents[ii]; ++kk) {
      if (detLevels[nc+kk] <= (int)((touch->GetHistoryDepth())+1)) {
    int jj = (int)((touch->GetHistoryDepth())+1)-detLevels[nc+kk];
        if ((touch->GetVolume(jj)->GetLogicalVolume()) == logVolumes[nc+kk]) {
      indx = ii;
      break;
    }
      }
    }
    nc += (unsigned int)(constituents[ii]);
    if (indx == ii) break;
  }

  stepT         += step;
  stepLen[indx]  = stepT;
  radLen[indx]  += (step/radl);
  intLen[indx]  += (step/intl);
  LogDebug("MaterialBudget") << "MaterialBudgetForward::Step in "
                 << touch->GetVolume(0)->GetLogicalVolume()->GetName()
                 << " Index " << indx <<" Step " << step <<" RadL "
                 << step/radl << " IntL " << step/intl;

  //----- Stop tracking after selected position
  if (stopAfter(aStep)) {
    G4Track* track = aStep->GetTrack();
    track->SetTrackStatus( fStopAndKill );
  }
}


void MaterialBudgetForward::update(const EndOfTrack* trk) {

  for (unsigned int ii=0; ii<detTypes.size(); ++ii) {
    for (unsigned int jj=0; jj<=detTypes.size(); ++jj) {
      if (stackOrder[jj] == (int)(ii+1)) {
    for (unsigned int kk=0; kk<=detTypes.size(); ++kk) {
      if (stackOrder[kk] == (int)(ii)) {
        radLen[jj]  += radLen[kk];
        intLen[jj]  += intLen[kk];
        LogDebug("MaterialBudget") << "MaterialBudgetForward::Add " << kk
                       << ":" << stackOrder[kk] << " to "
                       << jj <<":" << stackOrder[jj] <<" RadL "
                       << radLen[kk] << " : " << radLen[jj]
                       << " IntL " << intLen[kk] << " : "
                       << intLen[jj] <<" StepL " << stepLen[kk]
                       << " : " << stepLen[jj];
        break;
      }
    }
    break;
      }
    }
  }

  for (unsigned int ii=0; ii<=detTypes.size(); ++ii) {
    me100[ii]->Fill(eta, radLen[ii]);
    me200[ii]->Fill(eta, intLen[ii]);
    me300[ii]->Fill(eta, stepLen[ii]);
    me400[ii]->Fill(eta);
    me500[ii]->Fill(eta, phi, radLen[ii]);
    me600[ii]->Fill(eta, phi, intLen[ii]);
    me700[ii]->Fill(eta, phi, stepLen[ii]);
    me800[ii]->Fill(eta, phi);
    
    std::string name("Unknown");
    if (ii < detTypes.size()) name = detTypes[ii];
    LogDebug("MaterialBudget") << "MaterialBudgetForward::Volume[" << ii
                   << "]: " << name << " == Step " << stepLen[ii] 
                   << " RadL " << radLen[ii] << " IntL " 
                   << intLen[ii];
  }
}

void MaterialBudgetForward::book(const edm::ParameterSet& m_p) {

  // Book histograms
  edm::Service<TFileService> tfile;

  if ( !tfile.isAvailable() )
    throw cms::Exception("BadConfig") << "TFileService unavailable: "
                                      << "please add it to config file";

  int    binEta  = m_p.getUntrackedParameter<int>("NBinEta", 320);
  int    binPhi  = m_p.getUntrackedParameter<int>("NBinPhi", 180);
  double minEta  = m_p.getUntrackedParameter<double>("MinEta",-8.0);
  double maxEta  = m_p.getUntrackedParameter<double>("MaxEta", 8.0);
  double maxPhi  = CLHEP::pi;
  edm::LogInfo("MaterialBudget") << "MaterialBudgetForward: Booking user "
                                 << "histos === with " << binEta << " bins "
                                 << "in eta from " << minEta << " to "
                                 << maxEta << " and " << binPhi << " bins "
                                 << "in phi from " << -maxPhi << " to "
                                 << maxPhi;

  char  name[20], title[80];
  std::string named;
  for (int i=0; i<std::min((int)(detTypes.size()+1),maxSet); i++) {
    if (i >= (int)(detTypes.size())) named = "Unknown";
    else                             named = detTypes[i];
    sprintf(name, "%d", i+100);
    sprintf(title, "MB(X0) prof Eta in %s", named.c_str());
    me100[i] =  tfile->make<TProfile>(name, title, binEta, minEta, maxEta);
    sprintf(name, "%d", i+200);
    sprintf(title, "MB(L0) prof Eta in %s", named.c_str());
    me200[i] = tfile->make<TProfile>(name, title, binEta, minEta, maxEta);
    sprintf(name, "%d", i+300);
    sprintf(title, "MB(Step) prof Eta in %s", named.c_str());
    me300[i] = tfile->make<TProfile>(name, title, binEta, minEta, maxEta);
    sprintf(name, "%d", i+400);
    sprintf(title, "Eta in %s", named.c_str());
    me400[i] = tfile->make<TH1F>(name, title, binEta, minEta, maxEta);
    sprintf(name, "%d", i+500);
    sprintf(title, "MB(X0) prof Eta Phi in %s", named.c_str());
    me500[i] = tfile->make<TProfile2D>(name, title, binEta/2, minEta, maxEta,
                                       binPhi/2, -maxPhi, maxPhi);
    sprintf(name, "%d", i+600);
    sprintf(title, "MB(L0) prof Eta Phi in %s", named.c_str());
    me600[i]= tfile->make<TProfile2D>(name, title, binEta/2, minEta, maxEta,
                      binPhi/2, -maxPhi, maxPhi);
    sprintf(name, "%d", i+700);
    sprintf(title, "MB(Step) prof Eta Phi in %s", named.c_str());
    me700[i]= tfile->make<TProfile2D>(name, title, binEta/2, minEta, maxEta,
                      binPhi/2, -maxPhi, maxPhi);
    sprintf(name, "%d", i+800);
    sprintf(title, "Eta vs Phi in %s", named.c_str());
    me800[i]= tfile->make<TH2F>(name, title, binEta/2, minEta, maxEta,
                binPhi/2, -maxPhi, maxPhi);
  }

  edm::LogInfo("MaterialBudget") << "MaterialBudgetForward: Booking user "
                                 << "histos done ===";

}

bool MaterialBudgetForward::stopAfter(const G4Step* aStep) {

  G4ThreeVector hitPoint    = aStep->GetPreStepPoint()->GetPosition();
  if (aStep->GetPostStepPoint() != 0) 
    hitPoint = aStep->GetPostStepPoint()->GetPosition();
  double rr    = hitPoint.perp();
  double zz    = std::abs(hitPoint.z());

  bool   flag(false);
  for (unsigned int ii=0; ii<etaRegions.size(); ++ii) {
    //  LogDebug("MaterialBudget") << " MaterialBudgetForward::Eta " << eta << " in Region[" << ii << "] with " << etaRegions[ii] << " type "   << regionTypes[ii] << "|" << boundaries[ii];
    if (fabs(eta) < etaRegions[ii]) {
      if (regionTypes[ii] == 0) {
    if (rr >= boundaries[ii]-0.001) flag = true;
      } else {
    if (zz >= boundaries[ii]-0.001) flag = true;
      }
      if (flag)
    LogDebug("MaterialBudget") <<" MaterialBudgetForward::Stop after R = " 
                   << rr << " and Z = " << zz;
      break;
    }
  }
  //  LogDebug("MaterialBudget") <<" MaterialBudgetForward:: R = " << rr << " and Z = "  << zz << " Flag " << flag;
  return flag;
}