Inheritance diagram for DisplayGeom:

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

	~DisplayGeom ()

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

	EDAnalyzer ()

ModuleDescription const &	moduleDescription () const

std::string	workerType () const

virtual	~EDAnalyzer ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

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

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

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

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

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)

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 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)

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

virtual void	beginJob () override

virtual void	endJob () override

void	remakeGeometry (const DisplayGeomRecord &dgRec)

Private Attributes
edm::Service< EveService >	m_eve

TEveElement *	m_geomList

edm::ESWatcher< DisplayGeomRecord >	m_geomWatcher

int	m_level

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::EDAnalyzer
typedef EDAnalyzer	ModuleType

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

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::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), m_level, 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(0),
    m_MF_component(0),
    m_geomWatcher(this, &DisplayGeom::remakeGeometry)
 {
    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==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 ( )

Definition at line 138 of file DisplayGeom.cc.

139 {

140 }

Member Function Documentation

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

overrideprivatevirtual

Implements edm::EDAnalyzer.

Definition at line 173 of file DisplayGeom.cc.

References EveService::AddElement(), b, edm::ESWatcher< T >::check(), funct::cos(), alignCSCRings::e, edm::EventSetup::get(), i, j, python.rootplot.utilities::ls(), m_eve, m_geomWatcher, 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(), Min(), AlCaHLTBitMon_ParallelJobs::p, PV3DBase< T, PVType, FrameType >::phi(), lumiQueryAPI::q, funct::sin(), edmStreamStallGrapher::trans, 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)
        {
           edm::ESHandle<MagneticField> field;
          iSetup.get<IdealMagneticFieldRecord>().get( field );
 
      gStyle->SetPalette(1, 0);
 
      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 = 0;
      if (m_MF_plane_draw_dir) {
        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(1);
      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::EDAnalyzer.

Definition at line 322 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::EDAnalyzer.

Definition at line 334 of file DisplayGeom.cc.

334 {

335 }

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.

Referenced by remakeGeometry().

 {
    if (! gGeoManager->cd(path))
    {
       Warning("make_node", "Path '%s' not found.", path.Data());
       return 0;
    }
 
    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 338 of file DisplayGeom.cc.

References relativeConstraints::geom, edm::eventsetup::EventSetupRecord::get(), m_geomList, m_level, m_MF, make_node(), and edm::ESHandle< class >::product().

 {
    m_geomList->DestroyElements();
 
    edm::ESHandle<TGeoManager> geom;
    dgRec.get(geom);
    TEveGeoManagerHolder _tgeo(const_cast<TGeoManager*>(geom.product()));
 
    // 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);
    }
 }