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 noexcept final

bool	wantsGlobalRuns () const noexcept final

bool	wantsInputProcessBlocks () const noexcept final

bool	wantsProcessBlocks () const noexcept 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 noexcept

bool	wantsStreamRuns () const noexcept

	~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

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

std::vector< ESResolverIndex > 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

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

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

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

void	updateLookup (eventsetup::ESRecordsToProductResolverIndices const &)

virtual	~EDConsumerBase () noexcept(false)

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &)

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

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 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 Types
typedef std::vector< std::string >	vstring

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

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

vstring	m_nodes

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

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Detailed Description

Definition at line 53 of file DisplayGeom.cc.

Member Typedef Documentation

◆ vstring

typedef std::vector<std::string> DisplayGeom::vstring

private

Definition at line 75 of file DisplayGeom.cc.

Constructor & Destructor Documentation

◆ DisplayGeom()

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

explicit

Definition at line 97 of file DisplayGeom.cc.

References edm::EDConsumerBase::esConsumes(), edm::ParameterSet::getUntrackedParameter(), 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, and AlCaHLTBitMon_QueryRunRegistry::string.

     : m_eve(),
       m_geomList(nullptr),
       m_level(iConfig.getUntrackedParameter<int>("level")),
       m_nodes(iConfig.getUntrackedParameter<vstring>("nodes")),
       m_MF_component(0),
       m_geomWatcher(this, &DisplayGeom::remakeGeometry),
       m_displayGeomToken(esConsumes()) {
   std::string component = iConfig.getUntrackedParameter<std::string>("MF_component");
   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();
 
     m_MF_plane_d0 = iConfig.getUntrackedParameter<std::vector<double> >("MF_plane_d0");
     m_MF_plane_d1 = iConfig.getUntrackedParameter<std::vector<double> >("MF_plane_d1");
     m_MF_plane_d2 = iConfig.getUntrackedParameter<std::vector<double> >("MF_plane_d2");
 
     m_MF_plane_N1 = iConfig.getUntrackedParameter<int>("MF_plane_N");
     m_MF_plane_N2 = iConfig.getUntrackedParameter<int>("MF_plane_N2");
     if (m_MF_plane_N2 < 0)
       m_MF_plane_N2 = m_MF_plane_N1;
 
     m_MF_plane_draw_dir = iConfig.getUntrackedParameter<int>("MF_plane_draw_dir");
     m_MF_isPickable = iConfig.getUntrackedParameter<bool>("MF_pickable");
   }
 }

◆ ~DisplayGeom()

DisplayGeom::~DisplayGeom ( )

override

Definition at line 141 of file DisplayGeom.cc.

141 {}

Member Function Documentation

◆ analyze()

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

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 167 of file DisplayGeom.cc.

References EveService::AddElement(), b, edm::ESWatcher< T >::check(), funct::cos(), d1, MillePedeFileConverter_cfg::e, HLT_2024v14_cff::eps, edm::EventSetup::getData(), EveService::getManager(), mps_fire::i, MagneticField::inTesla(), dqmiolumiharvest::j, relval_nano::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, METSkim_cff::Min, AlCaHLTBitMon_ParallelJobs::p, submitPVResolutionJobs::q, funct::sin(), and V0Monitor_cff::v0.

                                                                              {
   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 = -200, maxval = 200;
       }
 
       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("MF Map");
       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);
     }
   }
   m_eve->getManager()->FullRedraw3D(true, true);
 }

◆ beginJob()

void DisplayGeom::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 313 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);
   }
 }

◆ endJob()

void DisplayGeom::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 322 of file DisplayGeom.cc.

322 {}

◆ fillDescriptions()

void DisplayGeom::fillDescriptions ( edm::ConfigurationDescriptions & conf )

static

Definition at line 335 of file DisplayGeom.cc.

References edm::ConfigurationDescriptions::add(), submitPVResolutionJobs::desc, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                      {
   edm::ParameterSetDescription desc;
   desc.addUntracked<vstring>("nodes", vstring{"tracker:Tracker_1", "muonBase:MUON_1", "caloBase:CALO_1"})
       ->setComment("List of nodes to visualize");
   ;
   desc.addUntracked<int>("level", 4)->setComment("Levels into the geometry hierarchy visualized at startup");
   desc.addUntracked<std::string>("MF_component", "None")
       ->setComment("Component of the MF to show: 'None', 'B', 'AbsBZ', 'AbsBR', 'AbsBphi', 'BR', 'Bphi'");
   desc.addUntracked<std::vector<double> >("MF_plane_d0", std::vector<double>{0., -900., -2400.})
       ->setComment("1st corner of MF map");
   desc.addUntracked<std::vector<double> >("MF_plane_d1", std::vector<double>{0., -900., 2400.})
       ->setComment("2nd corner of MF map");
   desc.addUntracked<std::vector<double> >("MF_plane_d2", std::vector<double>{0., 900., -2400.})
       ->setComment("3rd corner of MF map");
   desc.addUntracked<int>("MF_plane_N", 200)->setComment("Number of bins for the MF map in 1st coord");
   desc.addUntracked<int>("MF_plane_N2", -1)->setComment("Number of bins for the MF map in 2nd coord");
   desc.addUntracked<int>("MF_plane_draw_dir", true)->setComment("Draw MF direction arrows (slow)");
   desc.addUntracked<bool>("MF_pickable", false)->setComment("MF values are pickable (slow)");
 
   conf.add("DisplayGeom", desc);
 }

◆ make_node()

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

protected

Definition at line 147 of file DisplayGeom.cc.

References m_geomList, and EnsembleCalibrationLA_cfg::path.

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;
 }

◆ remakeGeometry()

void DisplayGeom::remakeGeometry ( const DisplayGeomRecord & dgRec )

private

Definition at line 325 of file DisplayGeom.cc.

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

                                                                {
   m_geomList->DestroyElements();
 
   TEveGeoManagerHolder _tgeo(const_cast<TGeoManager*>(&dgRec.get(m_displayGeomToken)));
 
   for (std::string& aNode : m_nodes) {
     make_node(aNode, m_level, kTRUE);
   }
 }