GeometryProducer.cc

Go to the documentation of this file.

#include "FWCore/PluginManager/interface/PluginManager.h"

#include "SimG4Core/GeometryProducer/interface/GeometryProducer.h"

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/ServiceRegistry/interface/Service.h"

#include "SimG4Core/Watcher/interface/SimProducer.h"
#include "SimG4Core/Watcher/interface/SimWatcherFactory.h"
#include "SimG4Core/Geometry/interface/DDDWorld.h"
#include "SimG4Core/Geometry/interface/G4LogicalVolumeToDDLogicalPartMap.h"
#include "SimG4Core/Geometry/interface/SensitiveDetectorCatalog.h"
#include "SimG4Core/MagneticField/interface/FieldBuilder.h"
#include "SimG4Core/MagneticField/interface/Field.h"
#include "SimG4Core/Notification/interface/SimG4Exception.h"
#include "SimG4Core/Application/interface/SimTrackManager.h"

#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

#include "Geometry/Records/interface/IdealGeometryRecord.h"

#include "DetectorDescription/Core/interface/DDCompactView.h"

#include "G4RunManagerKernel.hh"
#include "G4TransportationManager.hh"

#include <iostream>

static
void createWatchers(const edm::ParameterSet& iP, SimActivityRegistry& iReg,
            std::vector<std::shared_ptr<SimWatcher> >& oWatchers,
            std::vector<std::shared_ptr<SimProducer> >& oProds)
{
    using namespace std;
    using namespace edm;
    vector<ParameterSet> watchers;
    try { watchers = iP.getParameter<vector<ParameterSet> >("Watchers"); } 
    catch(edm::Exception) {}
  
    for(vector<ParameterSet>::iterator itWatcher = watchers.begin();
    itWatcher != watchers.end(); ++itWatcher) 
    {
    std::auto_ptr<SimWatcherMakerBase> 
        maker(SimWatcherFactory::get()->create(itWatcher->getParameter<std::string> ("type")));
    if(maker.get()==0) { throw SimG4Exception("Unable to find the requested Watcher"); }
    
    std::shared_ptr<SimWatcher> watcherTemp;
    std::shared_ptr<SimProducer> producerTemp;
    maker->make(*itWatcher,iReg,watcherTemp,producerTemp);
    oWatchers.push_back(watcherTemp);
    if(producerTemp) oProds.push_back(producerTemp);
  }
}

GeometryProducer::GeometryProducer(edm::ParameterSet const & p) :
    m_kernel(0), 
    m_pUseMagneticField(p.getParameter<bool>("UseMagneticField")),
    m_pField(p.getParameter<edm::ParameterSet>("MagneticField")), 
    m_pUseSensitiveDetectors(p.getParameter<bool>("UseSensitiveDetectors")),
    m_attach(0), m_p(p)
{
    //Look for an outside SimActivityRegistry
    //this is used by the visualization code
    edm::Service<SimActivityRegistry> otherRegistry;
    if (otherRegistry) m_registry.connect(*otherRegistry);
    createWatchers(m_p, m_registry, m_watchers, m_producers);
    produces<int>();
}

GeometryProducer::~GeometryProducer() 
{ 
    if (m_attach!=0) delete m_attach;
    if (m_kernel!=0) delete m_kernel; 
}

void GeometryProducer::beginJob(){
}
 
void GeometryProducer::endJob()
{ std::cout << " GeometryProducer terminating " << std::endl; }
 
void GeometryProducer::produce(edm::Event & e, const edm::EventSetup & es)
{
    m_kernel = G4RunManagerKernel::GetRunManagerKernel();
    if (m_kernel==0) m_kernel = new G4RunManagerKernel();
    std::cout << " GeometryProducer initializing " << std::endl;
    // DDDWorld: get the DDCV from the ES and use it to build the World
    edm::ESTransientHandle<DDCompactView> pDD;
    es.get<IdealGeometryRecord>().get(pDD);
   
    G4LogicalVolumeToDDLogicalPartMap map_;
    SensitiveDetectorCatalog catalog_;
    const DDDWorld * world = new DDDWorld(&(*pDD), map_, catalog_, false);
    m_registry.dddWorldSignal_(world);

    if (m_pUseMagneticField)
    {
    // setup the magnetic field
    edm::ESHandle<MagneticField> pMF;
    es.get<IdealMagneticFieldRecord>().get(pMF);
    const GlobalPoint g(0.,0.,0.);
    std::cout << "B-field(T) at (0,0,0)(cm): " << pMF->inTesla(g) << std::endl;

    m_fieldBuilder = std::auto_ptr<sim::FieldBuilder>
      (new sim::FieldBuilder(&(*pMF), m_pField));
    // G4TransportationManager * tM = G4TransportationManager::GetTransportationManager();
    // m_fieldBuilder->configure("MagneticFieldType",tM->GetFieldManager(),tM->GetPropagatorInField());
    }

    if (m_pUseSensitiveDetectors)
    {
    std::cout << " instantiating sensitive detectors " << std::endl;
    // instantiate and attach the sensitive detectors
    m_trackManager = std::auto_ptr<SimTrackManager>(new SimTrackManager);
    if (m_attach==0) m_attach = new AttachSD;
    {
        std::pair< std::vector<SensitiveTkDetector*>,
        std::vector<SensitiveCaloDetector*> > 
          sensDets = m_attach->create(*world,(*pDD),catalog_,m_p,m_trackManager.get(),m_registry);
      
        m_sensTkDets.swap(sensDets.first);
        m_sensCaloDets.swap(sensDets.second);
    }

    std::cout << " Sensitive Detector building finished; found " << m_sensTkDets.size()
          << " Tk type Producers, and " << m_sensCaloDets.size() 
          << " Calo type producers " << std::endl;
    }
    for(Producers::iterator itProd = m_producers.begin();itProd != m_producers.end();
    ++itProd) { (*itProd)->produce(e,es); }
}

DEFINE_FWK_MODULE(GeometryProducer);