Inheritance diagram for DisplayGeom:

Public Member Functions
	DisplayGeom (const edm::ParameterSet &)
	~DisplayGeom ()
Protected Member Functions
TEveGeoTopNode *	make_node (const TString &path, Int_t vis_level, Bool_t global_cs)
Private Member Functions
virtual void	analyze (const edm::Event &, const edm::EventSetup &)
virtual void	beginJob ()
virtual void	endJob ()
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

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.

{
}

Member Function Documentation

void DisplayGeom::analyze	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)		`[private, virtual]`

Implements edm::EDAnalyzer.

Definition at line 173 of file DisplayGeom.cc.

References 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(), siStripFEDMonitor_P5_cff::Min, AlCaHLTBitMon_ParallelJobs::p, PV3DBase< T, PVType, FrameType >::phi(), pos, lumiQueryAPI::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)
       {
          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 ) [private, virtual]

Reimplemented from edm::EDAnalyzer.

Definition at line 322 of file DisplayGeom.cc.

References 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 ) [private, virtual]

Reimplemented from edm::EDAnalyzer.

Definition at line 334 of file DisplayGeom.cc.

{
}

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< T >::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);
   }
}