Inheritance diagram for DisplayGeom:

Public Member Functions
	DisplayGeom (const edm::ParameterSet &)

	~DisplayGeom () override

Public Member Functions inherited from edm::one::EDAnalyzer<>
	EDAnalyzer ()=default

	EDAnalyzer (const EDAnalyzer &)=delete

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () final

const EDAnalyzer &	operator= (const EDAnalyzer &)=delete

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsInputProcessBlocks () const final

bool	wantsProcessBlocks () const final

Public Member Functions inherited from edm::one::EDAnalyzerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzerBase ()

ModuleDescription const &	moduleDescription () const

bool	wantsStreamLuminosityBlocks () const

bool	wantsStreamRuns () const

	~EDAnalyzerBase () override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

std::vector< ESProxyIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const

std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector < ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::array< std::vector< ModuleDescription const * > , NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, ProductRegistry const &preg, std::map< std::string, ModuleDescription const > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Protected Member Functions
TEveGeoTopNode *	make_node (const TString &path, Int_t vis_level, Bool_t global_cs)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<Transition Tr = Transition::Event>
constexpr auto	esConsumes ()

template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag)

template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

void	resetItemsToGetFrom (BranchType iType)

Private Member Functions
void	analyze (const edm::Event &, const edm::EventSetup &) override

void	beginJob () override

void	endJob () override

void	remakeGeometry (const DisplayGeomRecord &dgRec)

Private Attributes
const edm::ESGetToken < TGeoManager, DisplayGeomRecord >	m_displayGeomToken

edm::Service< EveService >	m_eve

TEveElement *	m_geomList

edm::ESWatcher< DisplayGeomRecord >	m_geomWatcher

int	m_level

edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	m_magFieldToken

bool	m_MF

int	m_MF_component

bool	m_MF_isPickable

std::vector< double >	m_MF_plane_d0

std::vector< double >	m_MF_plane_d1

std::vector< double >	m_MF_plane_d2

int	m_MF_plane_draw_dir

int	m_MF_plane_N1

int	m_MF_plane_N2

Additional Inherited Members
Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::one::EDAnalyzerBase
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Detailed Description

Definition at line 53 of file DisplayGeom.cc.

Constructor & Destructor Documentation

DisplayGeom::DisplayGeom ( const edm::ParameterSet & iConfig )

explicit

Definition at line 93 of file DisplayGeom.cc.

References edm::EDConsumerBase::esConsumes(), edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), m_level, m_magFieldToken, m_MF, m_MF_component, m_MF_isPickable, m_MF_plane_d0, m_MF_plane_d1, m_MF_plane_d2, m_MF_plane_draw_dir, m_MF_plane_N1, m_MF_plane_N2, and AlCaHLTBitMon_QueryRunRegistry::string.

     : m_eve(),
       m_geomList(nullptr),
       m_MF_component(0),
       m_geomWatcher(this, &DisplayGeom::remakeGeometry),
       m_displayGeomToken(esConsumes()) {
   m_level = iConfig.getUntrackedParameter<int>("level", 2);  // Geometry level to visualize
 
   m_MF = iConfig.getUntrackedParameter<int>("MF", false);  // Show the MF geometry, instead of detector geometry
 
   std::string component = iConfig.getUntrackedParameter<std::string>("MF_component", "NONE");
   boost::algorithm::to_upper(component);
 
   if (component == "NONE") {
     m_MF_component = -1;
   } else if (component == "ABSBZ") {
     m_MF_component = 1;
   } else if (component == "ABSBR") {
     m_MF_component = 2;
   } else if (component == "ABSBPHI") {
     m_MF_component = 3;
   } else if (component == "BR") {
     m_MF_component = 4;
   } else if (component == "BPHI") {
     m_MF_component = 5;
   } else {  // Anything else -> |B|
     m_MF_component = 0;
   }
 
   if (m_MF_component != -1) {
     m_magFieldToken = esConsumes();
   }
 
   if (m_MF_component == 0) {
     m_MF_plane_d0 = iConfig.getUntrackedParameter<std::vector<double> >("MF_plane_d0", std::vector<double>(3, 0.0));
     m_MF_plane_d1 = iConfig.getParameter<std::vector<double> >("MF_plane_d1");
     m_MF_plane_d2 = iConfig.getParameter<std::vector<double> >("MF_plane_d2");
 
     m_MF_plane_N1 = iConfig.getUntrackedParameter<UInt_t>("MF_plane_N", 100);
     m_MF_plane_N2 = iConfig.getUntrackedParameter<UInt_t>("MF_plane_N2", m_MF_plane_N1);
 
     m_MF_plane_draw_dir = iConfig.getUntrackedParameter<int>("MF_plane_draw_dir", true);
     m_MF_isPickable = iConfig.getUntrackedParameter<bool>("MF_pickable", true);
   }
 }

DisplayGeom::~DisplayGeom ( )

override

Definition at line 139 of file DisplayGeom.cc.

139 {}

Member Function Documentation

void DisplayGeom::analyze	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 165 of file DisplayGeom.cc.

References EveService::AddElement(), b, edm::ESWatcher< T >::check(), funct::cos(), d1, alignCSCRings::e, gpuVertexFinder::eps, edm::EventSetup::getData(), mps_fire::i, MagneticField::inTesla(), dqmiolumiharvest::j, eostools::ls(), m_eve, m_geomWatcher, m_magFieldToken, m_MF_component, m_MF_isPickable, m_MF_plane_d0, m_MF_plane_d1, m_MF_plane_d2, m_MF_plane_draw_dir, m_MF_plane_N1, m_MF_plane_N2, PV3DBase< T, PVType, FrameType >::mag(), siStripFEDMonitor_P5_cff::Min, AlCaHLTBitMon_ParallelJobs::p, PV3DBase< T, PVType, FrameType >::phi(), submitPVResolutionJobs::q, funct::sin(), Vector3DBase< T, FrameTag >::unit(), relativeConstraints::value, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

                                                                              {
   if (m_eve) {
     // Remake geometry if it has changed.
     m_geomWatcher.check(iSetup);
 
     if (m_MF_component != -1) {
       MagneticField const& field = iSetup.getData(m_magFieldToken);
 
       gStyle->SetPalette(1, nullptr);
 
       int minval = 0;
       int maxval = 4000;
       if (m_MF_component == 1) {  //AbsBZ
         minval = 0, maxval = 4000;
       } else if (m_MF_component == 2) {  //AbsBR
         minval = 0, maxval = 4000;
       } else if (m_MF_component == 3) {  //AbsBphi
         minval = 0, maxval = 1000;
       } else if (m_MF_component == 4) {  //BR
         minval = -4000, maxval = 4000;
       } else if (m_MF_component == 5) {  //Bphi
         minval = -1000, maxval = 1000;
       }
 
       TEveRGBAPalette* pal = new TEveRGBAPalette(minval, maxval);
 
       TEveStraightLineSet* ls = nullptr;
       if (m_MF_plane_draw_dir) {
         ls = new TEveStraightLineSet("MF_line_direction");
         ls->SetPickable(false);
         ls->SetLineColor(kGreen);
         ls->SetMarkerColor(kGreen);
         ls->SetMarkerStyle(1);
       }
 
       TEveQuadSet* q = new TEveQuadSet("MF_quad_values");
       q->Reset(TEveQuadSet::kQT_RectangleXY, kFALSE, 32);
       q->SetOwnIds(kTRUE);
       q->SetAlwaysSecSelect(true);
       q->SetPickable(m_MF_isPickable);
       q->SetPalette(pal);
 
       TEveVectorD v0(m_MF_plane_d0[0], m_MF_plane_d0[1], m_MF_plane_d0[2]);
       TEveVectorD v01(m_MF_plane_d1[0], m_MF_plane_d1[1], m_MF_plane_d1[2]);
       TEveVectorD v02(m_MF_plane_d2[0], m_MF_plane_d2[1], m_MF_plane_d2[2]);
 
       TEveVectorD b01 = (v01 - v0);
       TEveVectorD b02 = (v02 - v0);
       TEveVectorD b03 = b01.Cross(b02);
 
       TEveTrans trans;
       trans.SetBaseVec(1, b01.fX, b01.fY, b01.fZ);
       trans.SetBaseVec(2, b02.fX, b02.fY, b02.fZ);
       trans.SetBaseVec(3, b03.fX, b03.fY, b03.fZ);
       trans.SetPos(v0.Arr());
       trans.OrtoNorm3();
       q->SetTransMatrix(trans.Array());
 
       double w_step = b01.Mag() / m_MF_plane_N1;
       double h_step = b02.Mag() / m_MF_plane_N2;
 
       q->SetDefWidth(w_step);
       q->SetDefHeight(h_step);
       TEveVectorD d1;
       trans.GetBaseVec(1).GetXYZ(d1);
       d1 *= w_step;
       TEveVectorD d2;
       trans.GetBaseVec(2).GetXYZ(d2);
       d2 *= h_step;
 
       //d1.Print();
       d2.Dump();
       double line_step_size = TMath::Min(w_step, h_step);
 
       for (int i = 0; i < m_MF_plane_N1; i++) {
         for (int j = 0; j < m_MF_plane_N2; j++) {
           TEveVectorD p = d1 * Double_t(i) + d2 * Double_t(j) + v0;
           GlobalPoint pos(p.fX, p.fY, p.fZ);
           GlobalVector b = field.inTesla(pos) * 1000.;  // in mT
           float value = 0.;
           if (m_MF_component == 0) {  //BMOD
             value = b.mag();
           } else if (m_MF_component == 1) {  //BZ
             value = fabs(b.z());
           } else if (m_MF_component == 2) {  //ABSBR
             value = fabs(b.x() * cos(pos.phi()) + b.y() * sin(pos.phi()));
           } else if (m_MF_component == 3) {  //ABSBPHI
             value = fabs(-b.x() * sin(pos.phi()) + b.y() * cos(pos.phi()));
           } else if (m_MF_component == 2) {  //BR
             value = b.x() * cos(pos.phi()) + b.y() * sin(pos.phi());
           } else if (m_MF_component == 5) {  //BPHI
             value = -b.x() * sin(pos.phi()) + b.y() * cos(pos.phi());
           }
 
           q->AddQuad(w_step * i, h_step * j);
           q->QuadValue(value);
           if (m_MF_isPickable)
             q->QuadId(new TNamed(Form("Mag (%f, %f, %f) val = %f", b.x(), b.y(), b.z(), b.mag()), "Dong!"));
 
           if (ls) {
             if (b.mag() > 1e-6) {
               b.unit();
               b *= line_step_size;
               ls->AddLine(p.fX, p.fY, p.fZ, p.fX + b.x(), p.fY + b.y(), p.fZ + b.z());
             } else {
               ls->AddLine(p.fX, p.fY, p.fZ, p.fX + b.x(), p.fY + b.y(), p.fZ + b.z());
             }
 
             ls->AddMarker(ls->GetLinePlex().Size() - 1, 0);
             ls->AddMarker(i * m_MF_plane_N1 + j, 0);
           }
         }
       }
 
       TEveScene* eps = gEve->SpawnNewScene("FillStyleScene");
       gEve->GetDefaultViewer()->AddScene(eps);
       eps->GetGLScene()->SetStyle(TGLRnrCtx::kFill);
       eps->AddElement(q);
       if (ls)
         m_eve->AddElement(ls);
     } else {
       //     // Add a test obj
       //     if (!gRandom)
       //       gRandom = new TRandom(0);
       //     TRandom& r= *gRandom;
 
       //     Float_t s = 100;
 
       //     TEvePointSet* ps = new TEvePointSet();
       //     ps->SetOwnIds(kTRUE);
       //     for(Int_t i = 0; i< 100; i++)
       //       {
       //         ps->SetNextPoint(r.Uniform(-s,s), r.Uniform(-s,s), r.Uniform(-s,s));
       //         ps->SetPointId(new TNamed(Form("Point %d", i), ""));
       //       }
 
       //     ps->SetMarkerColor(TMath::Nint(r.Uniform(2, 9)));
       //     ps->SetMarkerSize(r.Uniform(1, 2));
       //     ps->SetMarkerStyle(4);
       //     m_eve->AddElement(ps);
     }
   }
 }

void DisplayGeom::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 310 of file DisplayGeom.cc.

References EveService::AddGlobalElement(), m_eve, and m_geomList.

                            {
   if (m_eve) {
     m_geomList = new TEveElementList("Display Geom");
     m_eve->AddGlobalElement(m_geomList);
     //      m_eve->getManager()->GetGlobalScene()->GetGLScene()->SetStyle(TGLRnrCtx::kWireFrame);
   }
 }

void DisplayGeom::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 319 of file DisplayGeom.cc.

319 {}

TEveGeoTopNode * DisplayGeom::make_node	(	const TString &	path,
		Int_t	vis_level,
		Bool_t	global_cs
	)

protected

Definition at line 145 of file DisplayGeom.cc.

References m_geomList.

Referenced by remakeGeometry().

                                                                                              {
   if (!gGeoManager->cd(path)) {
     Warning("make_node", "Path '%s' not found.", path.Data());
     return nullptr;
   }
 
   TEveGeoTopNode* tn = new TEveGeoTopNode(gGeoManager, gGeoManager->GetCurrentNode());
   tn->SetVisLevel(vis_level);
   if (global_cs) {
     tn->RefMainTrans().SetFrom(*gGeoManager->GetCurrentMatrix());
   }
   m_geomList->AddElement(tn);
   gEve->AddToListTree(tn, true);
   return tn;
 }

void DisplayGeom::remakeGeometry ( const DisplayGeomRecord & dgRec )

private

Definition at line 322 of file DisplayGeom.cc.

References edm::eventsetup::DependentRecordImplementation< RecordT, ListT >::get(), m_displayGeomToken, m_geomList, m_level, m_MF, and make_node().

                                                                {
   m_geomList->DestroyElements();
 
   TEveGeoManagerHolder _tgeo(const_cast<TGeoManager*>(&dgRec.get(m_displayGeomToken)));
 
   // To have a full one, all detectors in one top-node:
   // make_node("/cms:World_1/cms:CMSE_1", 4, kTRUE);
 
   if (m_MF) {
     make_node("/cms:World_1", m_level, kTRUE);
   } else {
     make_node("/cms:World_1/cms:CMSE_1/tracker:Tracker_1", m_level, kTRUE);
     make_node("/cms:World_1/cms:CMSE_1/caloBase:CALO_1", m_level, kTRUE);
     make_node("/cms:World_1/cms:CMSE_1/muonBase:MUON_1", m_level, kTRUE);
   }
 }