#include <Fireworks/Core/interface/FWRPZViewGeometry.h>

Inheritance diagram for FWRPZViewGeometry:

Public Member Functions
	FWRPZViewGeometry (const fireworks::Context &context)

TEveElement *	getGeoElements (const FWViewType::EType id)

virtual	~FWRPZViewGeometry ()

Public Member Functions inherited from FWViewGeometryList
	FWViewGeometryList (const fireworks::Context &context, bool projected=true)

void	updateColors ()

void	updateTransparency (bool projectedType)

virtual	~FWViewGeometryList ()

Private Member Functions
void	estimateProjectionSize (const float *, float &, float &, float &, float &)

void	estimateProjectionSizeCSC (const FWGeometry::GeomDetInfo &info, float &, float &, float &, float &)

void	estimateProjectionSizeDT (const FWGeometry::GeomDetInfo &info, float &, float &, float &, float &)

	FWRPZViewGeometry (const FWRPZViewGeometry &)

TEveElement *	makeCaloOutlineRhoPhi ()

TEveElement *	makeCaloOutlineRhoZ ()

TEveElement *	makeMuonGeometryRhoPhi ()

TEveElement *	makeMuonGeometryRhoZ ()

TEveGeoShape *	makeShape (double, double, double, double)

const FWRPZViewGeometry &	operator= (const FWRPZViewGeometry &)

Private Attributes
TEveElementList *	m_rhoPhiGeo

TEveElementList *	m_rhoZGeo

Additional Inherited Members
Protected Member Functions inherited from FWViewGeometryList
void	addToCompound (TEveElement *el, FWGeomColorIndex idx, bool applyTransp=true) const

Protected Attributes inherited from FWViewGeometryList
TEveCompound *	m_colorComp [kFWGeomColorSize]

const fireworks::Context &	m_context

const FWGeometry *	m_geom

Detailed Description

Description: [one line class summary]

Usage: <usage>

Definition at line 33 of file FWRPZViewGeometry.h.

Constructor & Destructor Documentation

FWRPZViewGeometry::FWRPZViewGeometry ( const fireworks::Context & context )

Definition at line 51 of file FWRPZViewGeometry.cc.

                                                                    :
    FWViewGeometryList(context),
    m_rhoPhiGeo(0),
    m_rhoZGeo(0)
 {
    SetElementName("RPZGeomShared");
 }

FWRPZViewGeometry::~FWRPZViewGeometry ( )

virtual

Definition at line 64 of file FWRPZViewGeometry.cc.

References m_rhoPhiGeo, and m_rhoZGeo.

 {
    m_rhoPhiGeo->DecDenyDestroy();
    m_rhoZGeo->DecDenyDestroy();
 }

FWRPZViewGeometry::FWRPZViewGeometry ( const FWRPZViewGeometry & )

private

Member Function Documentation

void FWRPZViewGeometry::estimateProjectionSize	(	const float *	global,
		float &	min_rho,
		float &	max_rho,
		float &	min_z,
		float &	max_z
	)

private

Definition at line 378 of file FWRPZViewGeometry.cc.

References rho, and mathSSE::sqrt().

Referenced by estimateProjectionSizeCSC(), and estimateProjectionSizeDT().

 {
    double rho = sqrt(global[0] *global[0]+global[1] *global[1]);
    if ( min_rho > rho ) min_rho = rho;
    if ( max_rho < rho ) max_rho = rho;
    if ( min_z > global[2] ) min_z = global[2];
    if ( max_z < global[2] ) max_z = global[2];
 }

void FWRPZViewGeometry::estimateProjectionSizeCSC	(	const FWGeometry::GeomDetInfo &	info,
		float &	min_rho,
		float &	max_rho,
		float &	min_z,
		float &	max_z
	)

private

Definition at line 350 of file FWRPZViewGeometry.cc.

References estimateProjectionSize(), FWGeometry::localToGlobal(), FWViewGeometryList::m_geom, FWGeometry::GeomDetInfo::shape, and mathSSE::sqrt().

Referenced by makeMuonGeometryRhoZ().

 {
    float local[3], global[3];
 
    float dX = info.shape[2] - info.shape[1];
    float dY = info.shape[4];
    float ddY = sqrt( 4 * dY * dY + dX * dX ) * 0.5;
    float dZ = info.shape[3];
    
    local[0] = info.shape[2]; local[1] = ddY; local[2] = -dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = info.shape[1]; local[1] = -ddY; local[2] = -dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = info.shape[1]; local[1] = -ddY; local[2] = dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = info.shape[2]; local[1] = ddY; local[2] = dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 }

void FWRPZViewGeometry::estimateProjectionSizeDT	(	const FWGeometry::GeomDetInfo &	info,
		float &	min_rho,
		float &	max_rho,
		float &	min_z,
		float &	max_z
	)

private

Definition at line 298 of file FWRPZViewGeometry.cc.

References estimateProjectionSize(), FWGeometry::localToGlobal(), FWViewGeometryList::m_geom, and FWGeometry::GeomDetInfo::shape.

Referenced by makeMuonGeometryRhoZ().

 {
    // we will test 5 points on both sides ( +/- z)
    float local[3], global[3];
 
    float dX = info.shape[1];
    float dY = info.shape[2];
    float dZ = info.shape[3];
 
    local[0] = 0; local[1] = 0; local[2] = dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = dX; local[1] = dY; local[2] = dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = -dX; local[1] = dY; local[2] = dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = dX; local[1] = -dY; local[2] = dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = -dX; local[1] = -dY; local[2] = dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = 0; local[1] = 0; local[2] = -dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = dX; local[1] = dY; local[2] = -dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = -dX; local[1] = dY; local[2] = -dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = dX; local[1] = -dY; local[2] = -dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 
    local[0] = -dX; local[1] = -dY; local[2] = -dZ;
    m_geom->localToGlobal( info, local, global );
    estimateProjectionSize( global, min_rho, max_rho, min_z, max_z );
 }

TEveElement * FWRPZViewGeometry::getGeoElements ( const FWViewType::EType id )

Definition at line 74 of file FWRPZViewGeometry.cc.

References FWViewType::kRhoZ, FWViewGeometryList::m_geom, m_rhoPhiGeo, m_rhoZGeo, makeCaloOutlineRhoPhi(), makeCaloOutlineRhoZ(), makeMuonGeometryRhoPhi(), and makeMuonGeometryRhoZ().

Referenced by FWRPZView::setContext().

 {
    if( !m_geom ) return ( new TEveCompound( "dummy" ));
 
    if (type == FWViewType::kRhoZ)
    {
       if ( !m_rhoZGeo)
       {
          m_rhoZGeo = new TEveElementList("Geometry RhoZ");
          
          m_rhoZGeo->IncDenyDestroy();
          m_rhoZGeo->AddElement(makeMuonGeometryRhoZ());
          m_rhoZGeo->AddElement(makeCaloOutlineRhoZ());
       }
       return m_rhoZGeo;
    }
    else 
    {
       if ( !m_rhoPhiGeo)
       {
          m_rhoPhiGeo = new TEveElementList("Geometry RhoPhi");
          m_rhoPhiGeo->IncDenyDestroy();
          m_rhoPhiGeo->AddElement(makeMuonGeometryRhoPhi());
          m_rhoPhiGeo->AddElement(makeCaloOutlineRhoPhi());
 
       }
       return m_rhoPhiGeo;
    }
    return 0;
 }

TEveElement * FWRPZViewGeometry::makeCaloOutlineRhoPhi ( )

private

Definition at line 132 of file FWRPZViewGeometry.cc.

References FWViewGeometryList::addToCompound(), fireworks::Context::caloR1(), fireworks::Context::colorManager(), funct::cos(), FWColorManager::geomColor(), i, kFWTrackerBarrelColorIndex, FWViewGeometryList::m_context, csvReporter::r, funct::sin(), and TwoPi.

Referenced by getGeoElements().

 { 
    TEveStraightLineSet* el = new TEveStraightLineSet( "TrackerRhoPhi" );
    addToCompound(el, kFWTrackerBarrelColorIndex, false);
 
    el->SetLineColor(m_context.colorManager()->geomColor(kFWTrackerBarrelColorIndex));
    const unsigned int nSegments = 100;
    const double r =  m_context.caloR1();
    for ( unsigned int i = 1; i <= nSegments; ++i )
       el->AddLine(r*sin(TMath::TwoPi()/nSegments*(i-1)), r*cos(TMath::TwoPi()/nSegments*(i-1)), 0,
                   r*sin(TMath::TwoPi()/nSegments*i), r*cos(TMath::TwoPi()/nSegments*i), 0);
 
    TEvePointSet* ref = new TEvePointSet("reference");
    ref->SetTitle("(0,0,0)");
    ref->SetMarkerStyle(4);
    ref->SetMarkerColor(kWhite);
    ref->SetNextPoint(0.,0.,0.);
    el->AddElement(ref);
 
    return el;
 }

TEveElement * FWRPZViewGeometry::makeCaloOutlineRhoZ ( )

private

Definition at line 109 of file FWRPZViewGeometry.cc.

References FWViewGeometryList::addToCompound(), fireworks::Context::caloMaxEta(), fireworks::Context::caloR1(), fireworks::Context::caloR2(), fireworks::Context::caloZ1(), fireworks::Context::caloZ2(), funct::exp(), kFWTrackerBarrelColorIndex, FWViewGeometryList::m_context, and funct::tan().

Referenced by getGeoElements().

 {
    using namespace fireworks;
 
    float ri = m_context.caloZ2()*tan(2*atan(exp(-m_context.caloMaxEta())));
 
    TEveStraightLineSet* el = new TEveStraightLineSet( "TrackerRhoZoutline" );
    el->SetPickable(kFALSE);
    addToCompound(el, kFWTrackerBarrelColorIndex, false);
 
    el->AddLine(0,  m_context.caloR1(), -m_context.caloZ1(), 0,  m_context.caloR1(),  m_context.caloZ1());
    el->AddLine(0, -m_context.caloR1(),  m_context.caloZ1(), 0, -m_context.caloR1(), -m_context.caloZ1());
 
    el->AddLine(0, -m_context.caloR2(),   m_context.caloZ2(), 0,  -ri,   m_context.caloZ2());
    el->AddLine(0, ri,  m_context.caloZ2(), 0,  m_context.caloR2(), m_context.caloZ2());
 
    el->AddLine(0, -m_context.caloR2(),   -m_context.caloZ2(), 0,  -ri,   -m_context.caloZ2());
    el->AddLine(0, ri,  -m_context.caloZ2(), 0,  m_context.caloR2(), -m_context.caloZ2());
  
    return el;
 }

TEveElement * FWRPZViewGeometry::makeMuonGeometryRhoPhi ( void )

private

Definition at line 157 of file FWRPZViewGeometry.cc.

References FWViewGeometryList::addToCompound(), FWGeometry::getEveShape(), kFWMuonBarrelLineColorIndex, FWViewGeometryList::m_colorComp, and FWViewGeometryList::m_geom.

Referenced by getGeoElements().

 {
    Int_t iWheel = 0;
  
    // rho-phi view
    TEveCompound* container = new TEveCompound( "MuonRhoPhi" );
 
 
    for( Int_t iStation = 1; iStation <= 4; ++iStation )
    {
       for( Int_t iSector = 1 ; iSector <= 14; ++iSector )
       {
          if( iStation < 4 && iSector > 12 ) continue;
          DTChamberId id( iWheel, iStation, iSector );
      TEveGeoShape* shape = m_geom->getEveShape( id.rawId() );
      if( shape ) 
      {
         shape->SetMainColor(m_colorComp[kFWMuonBarrelLineColorIndex]->GetMainColor());
         addToCompound(shape, kFWMuonBarrelLineColorIndex);
         container->AddElement( shape );
      }
       }
    }
    return container;
 }

TEveElement * FWRPZViewGeometry::makeMuonGeometryRhoZ ( void )

private

Definition at line 186 of file FWRPZViewGeometry.cc.

References FWViewGeometryList::addToCompound(), estimateProjectionSizeCSC(), estimateProjectionSizeDT(), FWGeometry::find(), kFWMuonBarrelLineColorIndex, kFWMuonEndcapLineColorIndex, FWViewGeometryList::m_geom, makeShape(), DetId::rawId(), and launcher::step.

Referenced by getGeoElements().

 {
    TEveElementList* container = new TEveElementList( "MuonRhoZ" );
 
    {
       TEveCompound* dtContainer = new TEveCompound( "DT" );
       for( Int_t iWheel = -2; iWheel <= 2; ++iWheel )
       {
          for( Int_t iStation = 1; iStation <= 4; ++iStation )
          {
             float min_rho(1000), max_rho(0), min_z(2000), max_z(-2000);
 
             // This will give us a quarter of DTs
             // which is enough for our projection
             for( Int_t iSector = 1; iSector <= 4; ++iSector )
             {
                DTChamberId id( iWheel, iStation, iSector );
                unsigned int rawid = id.rawId();
            FWGeometry::IdToInfoItr det = m_geom->find( rawid );
            estimateProjectionSizeDT( *det, min_rho, max_rho, min_z, max_z );
             }
             if ( min_rho > max_rho || min_z > max_z ) continue;
             TEveElement* se =  makeShape( min_rho, max_rho, min_z, max_z );
             addToCompound(se, kFWMuonBarrelLineColorIndex);
             dtContainer->AddElement(se);
             se =  makeShape( -max_rho, -min_rho, min_z, max_z );
             addToCompound(se, kFWMuonBarrelLineColorIndex);
             dtContainer->AddElement(se);
          }
       }
 
       container->AddElement( dtContainer );
    }
    {
       TEveCompound* cscContainer = new TEveCompound( "CSC" );
    
       Int_t maxChambers = 36;
       Int_t step = 9;
       Int_t iLayer = 0; // chamber
       for( Int_t iEndcap = 1; iEndcap <= 2; ++iEndcap ) // 1=forward (+Z), 2=backward(-Z)
       {
          // Actual CSC geometry:
          // Station 1 has 4 rings with 36 chambers in each
          // Station 2: ring 1 has 18 chambers, ring 2 has 36 chambers
          // Station 3: ring 1 has 18 chambers, ring 2 has 36 chambers
          // Station 4: ring 1 has 18 chambers
          for( Int_t iStation = 1; iStation <= 4; ++iStation )
          {
             for( Int_t iRing = 1; iRing <= 4; ++iRing )
             {
                if( iStation > 1 && iRing > 2 ) continue;
                if( iStation > 3 && iRing > 1 ) continue;
                float min_rho(1000), max_rho(0), min_z(2000), max_z(-2000);
                ( iRing == 1 && iStation > 1 ) ? ( maxChambers = 18 ) : ( maxChambers = 36 );
                ( iRing == 1 && iStation > 1 ) ? ( step = 5 ) : (  step = 18 );
         
                // Skip most of the chambers since they will project
                // the same way as the two top ones and the two bottom ones
                for( Int_t iChamber = step; iChamber <= maxChambers; iChamber += step )
                {
                   CSCDetId id( iEndcap, iStation, iRing, iChamber, iLayer );
           FWGeometry::IdToInfoItr det = m_geom->find( id.rawId() );
           estimateProjectionSizeCSC( *det, min_rho, max_rho, min_z, max_z );
 
           // and a chamber next to it
           ++iChamber;
           CSCDetId nextid( iEndcap, iStation, iRing, iChamber, iLayer );
           det = m_geom->find( nextid.rawId() );
           estimateProjectionSizeCSC( *det, min_rho, max_rho, min_z, max_z );
                }
                if ( min_rho > max_rho || min_z > max_z ) continue;
 
                TEveElement* se = makeShape( min_rho, max_rho, min_z, max_z);
                addToCompound(se, kFWMuonEndcapLineColorIndex);
                cscContainer->AddElement(se);
 
                se = makeShape( -max_rho, -min_rho, min_z, max_z );
                addToCompound(se, kFWMuonEndcapLineColorIndex);
                cscContainer->AddElement(se);
             }
          }
       }
       container->AddElement( cscContainer );
    }
    return container;
 }

TEveGeoShape * FWRPZViewGeometry::makeShape	(	double	min_rho,
		double	max_rho,
		double	min_z,
		double	max_z
	)

private

Definition at line 276 of file FWRPZViewGeometry.cc.

References matplotRender::t.

Referenced by makeMuonGeometryRhoZ().

 {
    TEveTrans t;
    t(1,1) = 1; t(1,2) = 0; t(1,3) = 0;
    t(2,1) = 0; t(2,2) = 1; t(2,3) = 0;
    t(3,1) = 0; t(3,2) = 0; t(3,3) = 1;
    t(1,4) = 0; t(2,4) = (min_rho+max_rho)/2; t(3,4) = (min_z+max_z)/2;
 
    TEveGeoShape* shape = new TEveGeoShape;
    shape->SetTransMatrix(t.Array());
 
    shape->SetRnrSelf(kTRUE);
    shape->SetRnrChildren(kTRUE);
    TGeoBBox* box = new TGeoBBox( 0, (max_rho-min_rho)/2, (max_z-min_z)/2 );
    shape->SetShape( box );
 
    return shape;
 }