Skip to content

Phase2-TB63Z Rearrange the header file and make first example for dd4hep (replacing #34752) #34791

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
cmsbuild merged 6 commits into from
Aug 9, 2021
Merged

Phase2-TB63Z Rearrange the header file and make first example for dd4hep (replacing #34752) #34791

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
678f521
Rearrange the header file and make first example for dd4hep (replacin…
Aug 5, 2021
6ed73a0
Put HGCPassive in src
Aug 5, 2021
8a26d82
Put HGCPassive in src
Aug 5, 2021
fa6db2b
Final changes to HGCPassive
Aug 5, 2021
c75e71f
get rid of a few includes
Aug 5, 2021
1d34db3
Code check
Aug 6, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
77 changes: 0 additions & 77 deletions SimG4CMS/HGCalTestBeam/CaloUtil.xml
View file
Open in desktop

This file was deleted.

39 changes: 39 additions & 0 deletions SimG4CMS/HGCalTestBeam/data/dd4hep/HGCalTB181Oct1.xml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
<?xml version="1.0"?>

<DDDefinition>
<debug>
<!--
<debug_shapes/>
<debug_rotations/>
<debug_volumes/>
<debug_placements/>
<debug_algorithms/>
<debug_constants/>
<debug_includes/>
<debug_materials/>
<debug_namespaces/>
<debug_visattr/>
-->
</debug>

<open_geometry/>
<close_geometry/>

<IncludeSection>
<Include ref='Geometry/CMSCommonData/data/materials/2021/v2/materials.xml'/>
<Include ref='SimG4CMS/CherenkovAnalysis/data/singleDREAM.xml'/>
<Include ref='Geometry/CMSCommonData/data/materials.xml'/>
<Include ref='Geometry/CMSCommonData/data/rotations.xml'/>
<Include ref='Geometry/HGCalCommonData/data/hgcalMaterial/v1/hgcalMaterial.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/Oct181/cms.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/Oct181/hgcal.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/Oct181/hgcalEE.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/Oct181/hgcalHE.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/ahcal.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/Oct181/hgcalBeam.xml'/>
<Include ref='Geometry/HGCalCommonData/data/hgcalwafer/v7/hgcalwafer.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/Oct181/hgcalsense.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/hgcProdCuts.xml'/>
<Include ref='Geometry/HGCalCommonData/data/TB181/Oct181/hgcalCons.xml'/>
</IncludeSection>
</DDDefinition>
1 change: 1 addition & 0 deletions SimG4CMS/HGCalTestBeam/plugins/BuildFile.xml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
<use name="SimG4Core/Watcher"/>
<use name="geant4core"/>
<use name="clhep"/>
<use name="dd4hep"/>
<use name="rootphysics"/>
<use name="root"/>
<use name="hepmc"/>
Expand Down
135 changes: 99 additions & 36 deletions SimG4CMS/HGCalTestBeam/plugins/HGCPassive.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,31 +4,88 @@
// Description: Main analysis class for HGCal Validation of G4 Hits
///////////////////////////////////////////////////////////////////////////////

#include "HGCPassive.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "CLHEP/Units/GlobalPhysicalConstants.h"
#include "CLHEP/Units/GlobalSystemOfUnits.h"
// to retreive hits
#include "SimDataFormats/CaloHit/interface/PassiveHit.h"

#include "SimG4Core/Notification/interface/BeginOfEvent.h"
#include "SimG4Core/Notification/interface/BeginOfRun.h"
#include "SimG4Core/Notification/interface/Observer.h"
#include "SimG4Core/Watcher/interface/SimProducer.h"

#include "G4LogicalVolumeStore.hh"
#include "G4PhysicalVolumeStore.hh"
#include "G4Step.hh"
#include "G4TransportationManager.hh"
#include "G4TouchableHistory.hh"
#include "G4Track.hh"
#include "DD4hep/Filter.h"

#include <array>
#include <cmath>
#include <iomanip>
#include <iostream>
#include <memory>
#include <utility>
#include <map>
#include <string>
#include <vector>

//#define EDM_ML_DEBUG

HGCPassive::HGCPassive(const edm::ParameterSet& p) : topPV_(nullptr), topLV_(nullptr), count_(0), init_(false) {
class HGCPassive : public SimProducer,
public Observer<const BeginOfRun *>,
public Observer<const BeginOfEvent *>,
public Observer<const G4Step *> {
public:
HGCPassive(const edm::ParameterSet &p);
HGCPassive(const HGCPassive &) = delete; // stop default
const HGCPassive &operator=(const HGCPassive &) = delete;
~HGCPassive() override;

void produce(edm::Event &, const edm::EventSetup &) override;

private:
// observer classes
void update(const BeginOfRun *run) override;
void update(const BeginOfEvent *evt) override;
void update(const G4Step *step) override;

// void endOfEvent(edm::PassiveHitContainer &HGCEEAbsE);
void endOfEvent(edm::PassiveHitContainer &hgcPH, unsigned int k);

typedef std::map<G4LogicalVolume *, std::pair<unsigned int, std::string>>::iterator volumeIterator;
G4VPhysicalVolume *getTopPV();
volumeIterator findLV(G4LogicalVolume *plv);
void storeInfo(
const volumeIterator itr, G4LogicalVolume *plv, unsigned int copy, double time, double energy, bool flag);

private:
std::vector<std::string> LVNames_;
G4VPhysicalVolume *topPV_;
G4LogicalVolume *topLV_;
std::map<G4LogicalVolume *, std::pair<unsigned int, std::string>> mapLV_;
std::string motherName_;
int addlevel_;

// some private members for ananlysis
unsigned int count_;
bool init_;
std::map<std::pair<G4LogicalVolume *, unsigned int>, std::array<double, 3>> store_;
};

HGCPassive::HGCPassive(const edm::ParameterSet &p) : topPV_(nullptr), topLV_(nullptr), count_(0), init_(false) {
edm::ParameterSet m_Passive = p.getParameter<edm::ParameterSet>("HGCPassive");
LVNames_ = m_Passive.getParameter<std::vector<std::string> >("LVNames");
LVNames_ = m_Passive.getParameter<std::vector<std::string>>("LVNames");
motherName_ = m_Passive.getParameter<std::string>("MotherName");
bool dd4hep = m_Passive.getParameter<bool>("IfDD4Hep");
addlevel_ = dd4hep ? 1 : 0;

#ifdef EDM_ML_DEBUG
edm::LogVerbatim("HGCSim") << "Name of the mother volume " << motherName_;
edm::LogVerbatim("HGCSim") << "Name of the mother volume " << motherName_ << " AddLevel " << addlevel_;
unsigned k(0);
#endif
for (const auto& name : LVNames_) {
for (const auto &name : LVNames_) {
produces<edm::PassiveHitContainer>(Form("%sPassiveHits", name.c_str()));
#ifdef EDM_ML_DEBUG
edm::LogVerbatim("HGCSim") << "Collection name[" << k << "] " << name;
Expand All @@ -39,29 +96,29 @@ HGCPassive::HGCPassive(const edm::ParameterSet& p) : topPV_(nullptr), topLV_(nul

HGCPassive::~HGCPassive() {}

void HGCPassive::produce(edm::Event& e, const edm::EventSetup&) {
void HGCPassive::produce(edm::Event &e, const edm::EventSetup &) {
for (unsigned int k = 0; k < LVNames_.size(); ++k) {
std::unique_ptr<edm::PassiveHitContainer> hgcPH(new edm::PassiveHitContainer);
endOfEvent(*hgcPH, k);
e.put(std::move(hgcPH), Form("%sPassiveHits", LVNames_[k].c_str()));
}
}

void HGCPassive::update(const BeginOfRun* run) {
void HGCPassive::update(const BeginOfRun *run) {
topPV_ = getTopPV();
if (topPV_ == nullptr) {
edm::LogWarning("HGCSim") << "Cannot find top level volume\n";
} else {
init_ = true;
const G4LogicalVolumeStore* lvs = G4LogicalVolumeStore::GetInstance();
const G4LogicalVolumeStore *lvs = G4LogicalVolumeStore::GetInstance();
for (auto lvcite : *lvs) {
findLV(lvcite);
}

#ifdef EDM_ML_DEBUG
edm::LogVerbatim("HGCSim") << "HGCPassive::Finds " << mapLV_.size() << " logical volumes";
unsigned int k(0);
for (const auto& lvs : mapLV_) {
for (const auto &lvs : mapLV_) {
edm::LogVerbatim("HGCSim") << "Entry[" << k << "] " << lvs.first << ": (" << (lvs.second).first << ", "
<< (lvs.second).second << ")";
++k;
Expand All @@ -71,7 +128,7 @@ void HGCPassive::update(const BeginOfRun* run) {
}

//=================================================================== per EVENT
void HGCPassive::update(const BeginOfEvent* evt) {
void HGCPassive::update(const BeginOfEvent *evt) {
int iev = (*evt)()->GetEventID();
edm::LogVerbatim("HGCSim") << "HGCPassive: =====> Begin event = " << iev << std::endl;

Expand All @@ -81,13 +138,14 @@ void HGCPassive::update(const BeginOfEvent* evt) {

// //=================================================================== each
// STEP
void HGCPassive::update(const G4Step* aStep) {
void HGCPassive::update(const G4Step *aStep) {
if (aStep != nullptr) {
G4VSensitiveDetector* curSD = aStep->GetPreStepPoint()->GetSensitiveDetector();
const G4VTouchable* touchable = aStep->GetPreStepPoint()->GetTouchable();
G4VSensitiveDetector *curSD = aStep->GetPreStepPoint()->GetSensitiveDetector();
const G4VTouchable *touchable = aStep->GetPreStepPoint()->GetTouchable();

int level = (touchable->GetHistoryDepth());
if (curSD == nullptr) {
G4LogicalVolume* plv = touchable->GetVolume()->GetLogicalVolume();
G4LogicalVolume *plv = touchable->GetVolume()->GetLogicalVolume();
auto it = (init_) ? mapLV_.find(plv) : findLV(plv);
double time = aStep->GetTrack()->GetGlobalTime();
double energy = (aStep->GetTotalEnergyDeposit()) / CLHEP::GeV;
Expand All @@ -96,8 +154,8 @@ void HGCPassive::update(const G4Step* aStep) {
if (((aStep->GetPostStepPoint() == nullptr) || (aStep->GetTrack()->GetNextVolume() == nullptr)) &&
(aStep->IsLastStepInVolume())) {
#ifdef EDM_ML_DEBUG
edm::LogVerbatim("HGCSim") << plv->GetName() << " F|L Step " << aStep->IsFirstStepInVolume() << ":"
<< aStep->IsLastStepInVolume() << " Position"
edm::LogVerbatim("HGCSim") << static_cast<std::string>(dd4hep::dd::noNamespace(plv->GetName())) << " F|L Step "
<< aStep->IsFirstStepInVolume() << ":" << aStep->IsLastStepInVolume() << " Position"
<< aStep->GetPreStepPoint()->GetPosition() << " Track "
<< aStep->GetTrack()->GetDefinition()->GetParticleName() << " at"
<< aStep->GetTrack()->GetPosition() << " Volume " << aStep->GetTrack()->GetVolume()
Expand All @@ -109,7 +167,9 @@ void HGCPassive::update(const G4Step* aStep) {
energy += (aStep->GetPreStepPoint()->GetKineticEnergy() / CLHEP::GeV);
} else {
time = (aStep->GetPostStepPoint()->GetGlobalTime());
copy = (unsigned int)(touchable->GetReplicaNumber(0) + 1000 * touchable->GetReplicaNumber(1));
copy = (level < 2)
? 0
: static_cast<unsigned int>(touchable->GetReplicaNumber(0) + 1000 * touchable->GetReplicaNumber(1));
}
if (it != mapLV_.end()) {
storeInfo(it, plv, copy, time, energy, true);
Expand All @@ -122,17 +182,17 @@ void HGCPassive::update(const G4Step* aStep) {
} // if (curSD==NULL)

// Now for the mother volumes
int level = (touchable->GetHistoryDepth());
if (level > 0) {
double energy = (aStep->GetTotalEnergyDeposit()) / CLHEP::GeV;
double time = (aStep->GetTrack()->GetGlobalTime());

for (int i = level; i > 0; --i) {
G4LogicalVolume* plv = touchable->GetVolume(i)->GetLogicalVolume();
G4LogicalVolume *plv = touchable->GetVolume(i)->GetLogicalVolume();
auto it = (init_) ? mapLV_.find(plv) : findLV(plv);
#ifdef EDM_ML_DEBUG
edm::LogVerbatim("HGCSim") << "Level: " << level << ":" << i << " " << plv->GetName() << " flag in the List "
<< (it != mapLV_.end());
edm::LogVerbatim("HGCSim") << "Level: " << level << ":" << i << " "
<< static_cast<std::string>(dd4hep::dd::noNamespace(plv->GetName()))
<< " flag in the List " << (it != mapLV_.end());
#endif
if (it != mapLV_.end()) {
unsigned int copy =
Expand All @@ -148,12 +208,12 @@ void HGCPassive::update(const G4Step* aStep) {

//================================================================ End of EVENT

void HGCPassive::endOfEvent(edm::PassiveHitContainer& hgcPH, unsigned int k) {
void HGCPassive::endOfEvent(edm::PassiveHitContainer &hgcPH, unsigned int k) {
#ifdef EDM_ML_DEBUG
unsigned int kount(0);
#endif
for (const auto& element : store_) {
G4LogicalVolume* lv = (element.first).first;
for (const auto &element : store_) {
G4LogicalVolume *lv = (element.first).first;
auto it = mapLV_.find(lv);
if (it != mapLV_.end()) {
if ((it->second).first == k) {
Expand All @@ -169,14 +229,14 @@ void HGCPassive::endOfEvent(edm::PassiveHitContainer& hgcPH, unsigned int k) {
}
}

G4VPhysicalVolume* HGCPassive::getTopPV() {
G4VPhysicalVolume *HGCPassive::getTopPV() {
return G4TransportationManager::GetTransportationManager()->GetNavigatorForTracking()->GetWorldVolume();
}

HGCPassive::volumeIterator HGCPassive::findLV(G4LogicalVolume* plv) {
HGCPassive::volumeIterator HGCPassive::findLV(G4LogicalVolume *plv) {
auto itr = mapLV_.find(plv);
if (itr == mapLV_.end()) {
std::string name = plv->GetName();
std::string name = static_cast<std::string>(dd4hep::dd::noNamespace(plv->GetName()));
for (unsigned int k = 0; k < LVNames_.size(); ++k) {
if (name.find(LVNames_[k]) != std::string::npos) {
mapLV_[plv] = std::pair<unsigned int, std::string>(k, name);
Expand All @@ -186,19 +246,19 @@ HGCPassive::volumeIterator HGCPassive::findLV(G4LogicalVolume* plv) {
}
}
if (topLV_ == nullptr) {
if (std::string(plv->GetName()) == motherName_)
if (static_cast<std::string>(dd4hep::dd::noNamespace(plv->GetName())) == motherName_)
topLV_ = plv;
}
return itr;
}

void HGCPassive::storeInfo(const HGCPassive::volumeIterator it,
G4LogicalVolume* plv,
G4LogicalVolume *plv,
unsigned int copy,
double time,
double energy,
bool flag) {
std::pair<G4LogicalVolume*, unsigned int> key(plv, copy);
std::pair<G4LogicalVolume *, unsigned int> key(plv, copy);
auto itr = store_.find(key);
double ee = (flag) ? energy : 0;
if (itr == store_.end()) {
Expand All @@ -215,4 +275,7 @@ void HGCPassive::storeInfo(const HGCPassive::volumeIterator it,
#endif
}

#include "SimG4Core/Watcher/interface/SimWatcherFactory.h"
#include "FWCore/PluginManager/interface/ModuleDef.h"

DEFINE_SIMWATCHER(HGCPassive);
Loading