/data/git/CMSSW_5_3_11_patch5/src/Fireworks/ParticleFlow/plugins/FWPFEcalRecHitRPProxyBuilder.cc

Go to the documentation of this file.

#include "FWPFEcalRecHitRPProxyBuilder.h"

//______________________________________________________________________________
void
FWPFEcalRecHitRPProxyBuilder::scaleProduct( TEveElementList *parent, FWViewType::EType type, const FWViewContext *vc )
{
   typedef std::vector<FWPFRhoPhiRecHit*> rpRecHits;
   unsigned int index = 0;
   
   for( rpRecHits::iterator i = m_towers.begin(); i != m_towers.end(); ++i )
   {
      m_towers[index]->updateScale( vc );
      index++;
   }
}

//______________________________________________________________________________
void
FWPFEcalRecHitRPProxyBuilder::cleanLocal()
{
   typedef std::vector<FWPFRhoPhiRecHit*> rpRecHits;
   for( rpRecHits::iterator i = m_towers.begin(); i != m_towers.end(); ++i )
      (*i)->clean();

   m_towers.clear();
}

//______________________________________________________________________________
TEveVector
FWPFEcalRecHitRPProxyBuilder::calculateCentre( const float *vertices )
{
   TEveVector centre;
   
   for( unsigned int i = 0; i < 8; i++ )
   {
      int j = i * 3;
      centre.fX += vertices[j];
      centre.fY += vertices[j+1];            // Total x,y,z values
      centre.fZ += vertices[j+2];
   }

   centre *= 1.0f / 8.0f;                  // Actually calculate the centre point

   return centre;
}

//______________________________________________________________________________
void
FWPFEcalRecHitRPProxyBuilder::build( const FWEventItem *iItem, TEveElementList *product, const FWViewContext *vc )
{
   m_towers.clear(); // Bug fix required for when multiple RhoPhiPF views are active
   for( unsigned int index = 0; index < static_cast<unsigned int>( iItem->size() ); ++index )
   {
      TEveCompound *itemHolder = createCompound();
      product->AddElement( itemHolder );
      const FWEventItem::ModelInfo &info = item()->modelInfo( index );

      if( info.displayProperties().isVisible() )
      {
         bool added = false;
         float E, et;
         float ecalR = FWPFGeom::caloR1();
         Double_t lPhi, rPhi;
         const EcalRecHit &iData = modelData( index );
         const float *vertices = item()->getGeom()->getCorners( iData.detid() );

         TEveVector centre = calculateCentre( vertices );
         TEveVector lVec = TEveVector( vertices[0], vertices[1], 0 );   // Bottom left corner of tower
         TEveVector rVec = TEveVector( vertices[9], vertices[10], 0 );  // Bottom right corner of tower
         
         lPhi = lVec.Phi();
         rPhi = rVec.Phi();
         E = iData.energy();
         et = FWPFMaths::calculateEt( centre, E );

         for( unsigned int i = 0; i < m_towers.size(); i++ )
         {   // Small range to catch rounding inaccuracies etc.
            Double_t phi = m_towers[i]->getlPhi();
            if( ( lPhi == phi ) || ( ( lPhi < phi + 0.0005 ) && ( lPhi > phi - 0.0005 ) ) )
            {
               m_towers[i]->addChild( this, itemHolder, vc, E, et );
               context().voteMaxEtAndEnergy( et, E );
               added = true;
               break;
            }
         }
      
         if( !added )
         {
            rVec.fX = ecalR * cos( rPhi ); rVec.fY = ecalR * sin( rPhi );
            lVec.fX = ecalR * cos( lPhi ); lVec.fY = ecalR * sin( lPhi );
            std::vector<TEveVector> bCorners(2);
            bCorners[0] = lVec;
            bCorners[1] = rVec;

            FWPFRhoPhiRecHit *rh = new FWPFRhoPhiRecHit( this, itemHolder, vc, E, et, lPhi, rPhi, bCorners );
            context().voteMaxEtAndEnergy(et, E);
            m_towers.push_back( rh );
         }
      }
   }
}

//______________________________________________________________________________
REGISTER_FWPROXYBUILDER( FWPFEcalRecHitRPProxyBuilder, EcalRecHit, "Ecal RecHit", FWViewType::kRhoPhiPFBit );