/data/refman/pasoursint/CMSSW_4_1_8_patch9/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;
}

//______________________________________________________________________________
float
FWPFEcalRecHitRPProxyBuilder::calculateEt( const TEveVector &centre, float E )
{
   TEveVector vec = centre;
   float et;

   vec.Normalize();
   vec *= E;
   et = vec.Perp();

   return et;
}

//______________________________________________________________________________
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 );

      bool added = false;
      float E, et;
      float ecalR = m_pfUtils->getCaloR1();
      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 = 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 );