CMSSW_4_4_3_patch1/src/SimG4CMS/Calo/src/ECalSD.cc

Go to the documentation of this file.

// File: ECalSD.cc
// Description: Sensitive Detector class for electromagnetic calorimeters
#include "SimG4CMS/Calo/interface/ECalSD.h"
#include "SimG4Core/Notification/interface/TrackInformation.h"
#include "Geometry/EcalCommonData/interface/EcalBarrelNumberingScheme.h"
#include "Geometry/EcalCommonData/interface/EcalBaseNumber.h"
#include "Geometry/EcalCommonData/interface/EcalEndcapNumberingScheme.h"
#include "Geometry/EcalCommonData/interface/EcalPreshowerNumberingScheme.h"
#include "Geometry/EcalCommonData/interface/ESTBNumberingScheme.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DetectorDescription/Core/interface/DDFilter.h"
#include "DetectorDescription/Core/interface/DDFilteredView.h"
#include "DetectorDescription/Core/interface/DDSolid.h"
#include "DetectorDescription/Core/interface/DDMaterial.h"
#include "DetectorDescription/Core/interface/DDValue.h"

#include "Geometry/EcalCommonData/interface/EcalBaseNumber.h"

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

#include<algorithm>

//#define DebugLog

ECalSD::ECalSD(G4String name, const DDCompactView & cpv,
               SensitiveDetectorCatalog & clg, 
               edm::ParameterSet const & p, const SimTrackManager* manager) : 
  CaloSD(name, cpv, clg, p, manager, 
         p.getParameter<edm::ParameterSet>("ECalSD").getParameter<int>("TimeSliceUnit"),
         p.getParameter<edm::ParameterSet>("ECalSD").getParameter<bool>("IgnoreTrackID")), 
  numberingScheme(0) {
  
  //   static SimpleConfigurable<bool>   on1(false,  "ECalSD:UseBirkLaw");
  //   static SimpleConfigurable<double> bk1(0.00463,"ECalSD:BirkC1");
  //   static SimpleConfigurable<double> bk2(-0.03,  "ECalSD:BirkC2");
  //   static SimpleConfigurable<double> bk3(1.0,    "ECalSD:BirkC3");
  // Values from NIM A484 (2002) 239-244: as implemented in Geant3
  //   useBirk          = on1.value();
  //   birk1            = bk1.value()*(g/(MeV*cm2));
  //   birk2            = bk2.value()*(g/(MeV*cm2))*(g/(MeV*cm2));

  edm::ParameterSet m_EC = p.getParameter<edm::ParameterSet>("ECalSD");
  useBirk      = m_EC.getParameter<bool>("UseBirkLaw");
  useBirkL3    = m_EC.getParameter<bool>("BirkL3Parametrization");
  birk1        = m_EC.getParameter<double>("BirkC1")*(g/(MeV*cm2));
  birk2        = m_EC.getParameter<double>("BirkC2");
  birk3        = m_EC.getParameter<double>("BirkC3");
  birkSlope    = m_EC.getParameter<double>("BirkSlope");
  birkCut      = m_EC.getParameter<double>("BirkCut");
  slopeLY      = m_EC.getParameter<double>("SlopeLightYield");
  storeTrack   = m_EC.getParameter<bool>("StoreSecondary");
  crystalMat   = m_EC.getUntrackedParameter<std::string>("XtalMat","E_PbWO4");
  bool isItTB  = m_EC.getUntrackedParameter<bool>("TestBeam", false);
  bool nullNS  = m_EC.getUntrackedParameter<bool>("NullNumbering", false);
  storeRL      = m_EC.getUntrackedParameter<bool>("StoreRadLength", false);
  
  //Material list for HB/HE/HO sensitive detectors
  std::string attribute = "ReadOutName";
  DDSpecificsFilter filter;
  DDValue           ddv(attribute,name,0);
  filter.setCriteria(ddv,DDSpecificsFilter::equals);
  DDFilteredView fv(cpv);
  fv.addFilter(filter);
  fv.firstChild();
  DDsvalues_type sv(fv.mergedSpecifics());
  // Use of Weight
  useWeight= true;
  std::vector<double>      tempD = getDDDArray("EnergyWeight",sv);
  if (tempD.size() > 0) { if (tempD[0] < 0.1) useWeight = false; }
  std::vector<std::string> tempS = getStringArray("Depth1Name",sv);
  if (tempS.size() > 0) depth1Name = tempS[0];
  else                  depth1Name = " ";
  tempS = getStringArray("Depth2Name",sv);
  if (tempS.size() > 0) depth2Name = tempS[0];
  else                  depth2Name = " ";

  EcalNumberingScheme* scheme=0;
  if (nullNS)                    scheme = 0;
  else if (name == "EcalHitsEB") scheme = dynamic_cast<EcalNumberingScheme*>(new EcalBarrelNumberingScheme());
  else if (name == "EcalHitsEE") scheme = dynamic_cast<EcalNumberingScheme*>(new EcalEndcapNumberingScheme());
  else if (name == "EcalHitsES") {
    if (isItTB) scheme = dynamic_cast<EcalNumberingScheme*>(new ESTBNumberingScheme());
    else        scheme = dynamic_cast<EcalNumberingScheme*>(new EcalPreshowerNumberingScheme());
    useWeight = false;
  } else {edm::LogWarning("EcalSim") << "ECalSD: ReadoutName not supported\n";}

  if (scheme)  setNumberingScheme(scheme);
#ifdef DebugLog
  LogDebug("EcalSim") << "Constructing a ECalSD  with name " << GetName();
#endif
  if (useWeight) {
    edm::LogInfo("EcalSim")  << "ECalSD:: Use of Birks law is set to      " 
                             << useBirk << "        with three constants kB = "
                             << birk1 << ", C1 = " << birk2 << ", C2 = " 
                             << birk3 <<"\n         Use of L3 parametrization "
                             << useBirkL3 << " with slope " << birkSlope
                             << " and cut off " << birkCut << "\n"
                             << "         Slope for Light yield is set to "
                             << slopeLY;
  } else {
    edm::LogInfo("EcalSim")  << "ECalSD:: energy deposit is not corrected "
                             << " by Birk or light yield curve";
  }

  edm::LogInfo("EcalSim") << "ECalSD:: Suppression Flag " << suppressHeavy
                          << " protons below " << kmaxProton << " MeV,"
                          << " neutrons below " << kmaxNeutron << " MeV and"
                          << " ions below " << kmaxIon << " MeV\n"
                          << "         Depth1 Name = " << depth1Name
                          << " and Depth2 Name = " << depth2Name;

  if (useWeight) initMap(name,cpv);

}

ECalSD::~ECalSD() {
  if (numberingScheme) delete numberingScheme;
}

double ECalSD::getEnergyDeposit(G4Step * aStep) {
  
  if (aStep == NULL) {
    return 0;
  } else {
    preStepPoint        = aStep->GetPreStepPoint();
    G4String nameVolume = preStepPoint->GetPhysicalVolume()->GetName();

    // take into account light collection curve for crystals
    double weight = 1.;
    if (suppressHeavy) {
      G4Track* theTrack = aStep->GetTrack();
      TrackInformation * trkInfo = (TrackInformation *)(theTrack->GetUserInformation());
      if (trkInfo) {
        int pdg = theTrack->GetDefinition()->GetPDGEncoding();
        if (!(trkInfo->isPrimary())) { // Only secondary particles
          double ke = theTrack->GetKineticEnergy()/MeV;
          if (((pdg/1000000000 == 1 && ((pdg/10000)%100) > 0 &&
                ((pdg/10)%100) > 0)) && (ke<kmaxIon)) weight = 0;
          if ((pdg == 2212) && (ke < kmaxProton))     weight = 0;
          if ((pdg == 2112) && (ke < kmaxNeutron))    weight = 0;
#ifdef DebugLog
          if (weight == 0) 
            LogDebug("EcalSim") << "Ignore Track " << theTrack->GetTrackID()
                                << " Type " << theTrack->GetDefinition()->GetParticleName()
                                << " Kinetic Energy " << ke << " MeV";
#endif
        }
      }
    }
    G4LogicalVolume* lv   = aStep->GetPreStepPoint()->GetTouchable()->GetVolume(0)->GetLogicalVolume();
    if (useWeight && std::count(noWeight.begin(),noWeight.end(),lv) == 0) {
      weight *= curve_LY(aStep);
      if (useBirk) {
        if (useBirkL3) weight *= getBirkL3(aStep);
        else           weight *= getAttenuation(aStep, birk1, birk2, birk3);
      }
    }
    double wt1    = getResponseWt(theTrack);
    double edep   = aStep->GetTotalEnergyDeposit() * weight * wt1;
#ifdef DebugLog
    LogDebug("EcalSim") << "ECalSD:: " << nameVolume
                        <<" Light Collection Efficiency " <<weight << ":" <<wt1
                        << " Weighted Energy Deposit " << edep/MeV << " MeV";
#endif
    return edep;
  } 
}

int ECalSD::getTrackID(G4Track* aTrack) {

  int  primaryID(0);
  bool flag(false);
  if (storeTrack) {
    G4LogicalVolume* lv  = preStepPoint->GetTouchable()->GetVolume(0)->GetLogicalVolume();
    if (std::count(useDepth1.begin(),useDepth1.end(),lv) != 0) {
      flag = true;
    } else if (std::count(useDepth2.begin(),useDepth2.end(),lv) != 0) {
      flag = true;
    }
  }
  if (flag) {
    forceSave = true;
    primaryID = aTrack->GetTrackID();
  } else {
    primaryID = CaloSD::getTrackID(aTrack);
  }
  return primaryID;
}

uint16_t ECalSD::getDepth(G4Step * aStep) {
  G4LogicalVolume* lv   = aStep->GetPreStepPoint()->GetTouchable()->GetVolume(0)->GetLogicalVolume();
  uint16_t ret = 0;
  if (std::count(useDepth1.begin(),useDepth1.end(),lv) != 0)      ret = 1;
  else if (std::count(useDepth2.begin(),useDepth2.end(),lv) != 0) ret = 2;
  else if (storeRL) ret = getRadiationLength(aStep);
#ifdef DebugLog
  LogDebug("EcalSim") << "Volume " << lv->GetName() << " Depth " << ret;
#endif
  return ret;
}

uint16_t ECalSD::getRadiationLength(G4Step * aStep) {
  
  uint16_t thisX0 = 0;
  if (aStep != NULL) {
    G4StepPoint* hitPoint = aStep->GetPreStepPoint();
    G4LogicalVolume* lv   = hitPoint->GetTouchable()->GetVolume(0)->GetLogicalVolume();
    
    if (useWeight) {
      G4ThreeVector  localPoint = setToLocal(hitPoint->GetPosition(),
                                             hitPoint->GetTouchable());
      double crlength = crystalLength(lv);
      double radl     = hitPoint->GetMaterial()->GetRadlen();
      double detz     = (float)(0.5*crlength + localPoint.z());
      thisX0 = (uint16_t)floor(detz/radl);   
    } 
  }
  return thisX0;
}

uint32_t ECalSD::setDetUnitId(G4Step * aStep) { 
  if (numberingScheme == 0) {
    return EBDetId(1,1)();
  } else {
    getBaseNumber(aStep);
    return numberingScheme->getUnitID(theBaseNumber);
  }
}

void ECalSD::setNumberingScheme(EcalNumberingScheme* scheme) {
  if (scheme != 0) {
    edm::LogInfo("EcalSim") << "EcalSD: updates numbering scheme for " 
                            << GetName() << "\n";
    if (numberingScheme) delete numberingScheme;
    numberingScheme = scheme;
  }
}

void ECalSD::initMap(G4String sd, const DDCompactView & cpv) {

  G4String attribute = "ReadOutName";
  DDSpecificsFilter filter;
  DDValue           ddv(attribute,sd,0);
  filter.setCriteria(ddv,DDSpecificsFilter::equals);
  DDFilteredView fv(cpv);
  fv.addFilter(filter);
  fv.firstChild();

  std::vector<G4LogicalVolume*> lvused;
  const G4LogicalVolumeStore *  lvs = G4LogicalVolumeStore::GetInstance();
  std::vector<G4LogicalVolume *>::const_iterator lvcite;
  bool dodet=true;
  std::string lvnamx, lvnamy, lvname;  
  while (dodet) {
    const std::string &matname = fv.logicalPart().material().name().name();
    lvname = fv.logicalPart().name().name();
    G4LogicalVolume* lv=0;
    for (lvcite = lvs->begin(); lvcite != lvs->end(); lvcite++) {
      if (!strcmp((*lvcite)->GetName().c_str(), lvname.c_str())) {
        lv = (*lvcite);
        break;
      }
    }
    if (depth1Name != " ") {
      if (strncmp(lvname.c_str(), depth1Name.c_str(), 4) == 0) {
        if (std::count(useDepth1.begin(),useDepth1.end(),lv) == 0) {
          useDepth1.push_back(lv);
#ifdef DebugLog
          LogDebug("EcalSim") << "ECalSD::initMap Logical Volume " << lvname
                              <<" in Depth 1 volume list";
#endif
        }
        lvnamx = lvname + "_refl";
        G4LogicalVolume* lvr = 0;
        for (lvcite = lvs->begin(); lvcite != lvs->end(); lvcite++) {
          if (!strcmp((*lvcite)->GetName().c_str(), lvnamx.c_str())) {
            lvr = (*lvcite);
            break;
          }
        }
        if (lvr != 0 && std::count(useDepth1.begin(),useDepth1.end(),lvr)==0) {
          useDepth1.push_back(lvr);
#ifdef DebugLog
          LogDebug("EcalSim") << "ECalSD::initMap Logical Volume " << lvnamx
                              <<" in Depth 1 volume list";
#endif
        }
      }
    }
    if (depth2Name != " ") {
      if (strncmp(lvname.c_str(), depth2Name.c_str(), 4) == 0) {
        if (std::count(useDepth2.begin(),useDepth2.end(),lv) == 0) {
          useDepth2.push_back(lv);
#ifdef DebugLog
          LogDebug("EcalSim") << "ECalSD::initMap Logical Volume " << lvname
                              <<" in Depth 2 volume list";
#endif
        }
        lvnamy = lvname + "_refl";
        G4LogicalVolume* lvr = 0;
        for (lvcite = lvs->begin(); lvcite != lvs->end(); lvcite++) {
          if (!strcmp((*lvcite)->GetName().c_str(), lvnamy.c_str())) {
            lvr = (*lvcite);
            break;
          }
        }
        if (lvr != 0 && std::count(useDepth2.begin(),useDepth2.end(),lvr)==0) {
          useDepth2.push_back(lvr);
#ifdef DebugLog
          LogDebug("EcalSim") << "ECalSD::initMap Logical Volume " << lvnamy
                              <<" in Depth 2 volume list";
#endif
        }
      }
    }
    if (lv != 0) {
      if (strcmp(crystalMat.c_str(), matname.c_str()) == 0) {
        if (std::count(lvused.begin(),lvused.end(),lv) == 0) {
          lvused.push_back(lv);
          const DDSolid & sol  = fv.logicalPart().solid();
          const std::vector<double> & paras = sol.parameters();
#ifdef DebugLog
          LogDebug("EcalSim") << "ECalSD::initMap (for " << sd << "): Solid " 
                              << lvname << " Shape " << sol.shape() 
                              << " Parameter 0 = "<< paras[0] 
                              << " Logical Volume " << lv;
#endif
          if (sol.shape() == ddtrap) {
            double dz = 2*paras[0];
            xtalLMap.insert(std::pair<G4LogicalVolume*,double>(lv,dz));
            lvname += "_refl";
            lv = 0;
            for (lvcite = lvs->begin(); lvcite != lvs->end(); lvcite++) {
              if (!strcmp((*lvcite)->GetName().c_str(), lvname.c_str())) {
                lv = (*lvcite);
                break;
              }
            } 
            if (lv != 0)
              xtalLMap.insert(std::pair<G4LogicalVolume*,double>(lv,dz));
          }
        }
      } else {
        if (std::count(noWeight.begin(),noWeight.end(),lv) == 0) {
          noWeight.push_back(lv);
#ifdef DebugLog
          LogDebug("EcalSim") << "ECalSD::initMap Logical Volume " << lvname
                              << " Material " << matname <<" in noWeight list";
#endif
        }
        lvname += "_refl";
        lv = 0;
        for (lvcite = lvs->begin(); lvcite != lvs->end(); lvcite++) {
          if (!strcmp((*lvcite)->GetName().c_str(), lvname.c_str())) {
            lv = (*lvcite);
            break;
          }
        }
        if (lv != 0 && std::count(noWeight.begin(),noWeight.end(),lv) == 0) {
          noWeight.push_back(lv);
#ifdef DebugLog
          LogDebug("EcalSim") << "ECalSD::initMap Logical Volume " << lvname
                              << " Material " << matname <<" in noWeight list";
#endif
        }
      }
    }
    dodet = fv.next();
  }
#ifdef DebugLog
  LogDebug("EcalSim") << "ECalSD: Length Table for " << attribute << " = " 
                      << sd << ":";   
  std::map<G4LogicalVolume*,double>::const_iterator ite = xtalLMap.begin();
  int i=0;
  for (; ite != xtalLMap.end(); ite++, i++) {
    G4String name = "Unknown";
    if (ite->first != 0) name = (ite->first)->GetName();
    LogDebug("EcalSim") << " " << i << " " << ite->first << " " << name 
                        << " L = " << ite->second;
  }
#endif
}

double ECalSD::curve_LY(G4Step* aStep) {

  G4StepPoint*     stepPoint = aStep->GetPreStepPoint();
  G4LogicalVolume* lv        = stepPoint->GetTouchable()->GetVolume(0)->GetLogicalVolume();

  double weight = 1.;
  G4ThreeVector  localPoint = setToLocal(stepPoint->GetPosition(),
                                         stepPoint->GetTouchable());
  double crlength = crystalLength(lv);
  double dapd = 0.5 * crlength - localPoint.z();
  if (dapd >= -0.1 || dapd <= crlength+0.1) {
    if (dapd <= 100.)
      weight = 1.0 + slopeLY - dapd * 0.01 * slopeLY;
  } else {
    edm::LogWarning("EcalSim") << "ECalSD: light coll curve : wrong distance "
                               << "to APD " << dapd << " crlength = " 
                               << crlength <<" crystal name = " <<lv->GetName()
                               << " z of localPoint = " << localPoint.z() 
                               << " take weight = " << weight;
  }
#ifdef DebugLog
  LogDebug("EcalSim") << "ECalSD, light coll curve : " << dapd 
                      << " crlength = " << crlength
                      << " crystal name = " << lv->GetName()
                      << " z of localPoint = " << localPoint.z() 
                      << " take weight = " << weight;
#endif
  return weight;
}

double ECalSD::crystalLength(G4LogicalVolume* lv) {

  double length= 230.;
  std::map<G4LogicalVolume*,double>::const_iterator ite = xtalLMap.find(lv);
  if (ite != xtalLMap.end()) length = ite->second;
  return length;
}

void ECalSD::getBaseNumber(const G4Step* aStep) {

  theBaseNumber.reset();
  const G4VTouchable* touch = aStep->GetPreStepPoint()->GetTouchable();
  int theSize = touch->GetHistoryDepth()+1;
  if ( theBaseNumber.getCapacity() < theSize ) theBaseNumber.setSize(theSize);
  //Get name and copy numbers
  if ( theSize > 1 ) {
    for (int ii = 0; ii < theSize ; ii++) {
      theBaseNumber.addLevel(touch->GetVolume(ii)->GetName(),touch->GetReplicaNumber(ii));
#ifdef DebugLog
      LogDebug("EcalSim") << "ECalSD::getBaseNumber(): Adding level " << ii
                          << ": " << touch->GetVolume(ii)->GetName() << "["
                          << touch->GetReplicaNumber(ii) << "]";
#endif
    }
  }

}

double ECalSD::getBirkL3(G4Step* aStep) {

  double weight = 1.;
  double charge = aStep->GetPreStepPoint()->GetCharge();

  if (charge != 0. && aStep->GetStepLength() > 0) {
    G4Material* mat = aStep->GetPreStepPoint()->GetMaterial();
    double density = mat->GetDensity();
    double dedx    = aStep->GetTotalEnergyDeposit()/aStep->GetStepLength();
    double rkb     = birk1/density;
    if (dedx > 0) {
      weight         = 1. - birkSlope*log(rkb*dedx);
      if (weight < birkCut) weight = birkCut;
      else if (weight > 1.) weight = 1.;
    }
#ifdef DebugLog
    LogDebug("EcalSim") << "ECalSD::getBirkL3 in " << mat->GetName()
                        << " Charge " << charge << " dE/dx " << dedx
                        << " Birk Const " << rkb << " Weight = " << weight 
                        << " dE " << aStep->GetTotalEnergyDeposit();
#endif
  }
  return weight;

}

std::vector<double> ECalSD::getDDDArray(const std::string & str,
                                        const DDsvalues_type & sv) {

#ifdef DebugLog
  LogDebug("EcalSim") << "ECalSD:getDDDArray called for " << str;
#endif
  DDValue value(str);
  if (DDfetch(&sv,value)) {
#ifdef DebugLog
    LogDebug("EcalSim") << value;
#endif
    const std::vector<double> & fvec = value.doubles();
    return fvec;
  } else {
    std::vector<double> fvec;
    return fvec;
  }
}

std::vector<std::string> ECalSD::getStringArray(const std::string & str,
                                                const DDsvalues_type & sv) {

#ifdef DebugLog
  LogDebug("EcalSim") << "ECalSD:getStringArray called for " << str;
#endif
  DDValue value(str);
  if (DDfetch(&sv,value)) {
#ifdef DebugLog
    LogDebug("EcalSim") << value;
#endif
    const std::vector<std::string> & fvec = value.strings();
    return fvec;
  } else {
    std::vector<std::string> fvec;
    return fvec;
  }
}