/data/refman/pasoursint/CMSSW_4_1_8_patch9/src/Fireworks/ParticleFlow/plugins/FWPFClusterLegoProxyBuilder.cc

Go to the documentation of this file.

#include "FWPFClusterLegoProxyBuilder.h"

//______________________________________________________________________________
float
FWPFClusterLegoProxyBuilder::calculateEt( const reco::PFCluster &iData, float E )
{
    float et = 0.f;
    TEveVector vec;
    
    vec.fX = iData.x();
    vec.fY = iData.y();         // Get the cluster centroid
    vec.fZ = iData.z();
    
    vec.Normalize();
    vec *= E;   
    et = vec.Perp();            // Get perpendicular vector

    return et;
}

//______________________________________________________________________________
void
FWPFClusterLegoProxyBuilder::localModelChanges( const FWModelId &iId, TEveElement *parent, FWViewType::EType viewType, const FWViewContext *vc )
{
    // Line set marker is not the same colour as line, fixed here
    if( ( parent )->HasChildren() )
    {
        for( TEveElement::List_i j = parent->BeginChildren(); j != parent->EndChildren(); j++ )
        {
            FWPFLegoCandidate *cluster = dynamic_cast<FWPFLegoCandidate*>( *j );
            const FWDisplayProperties &dp = FWProxyBuilderBase::item()->modelInfo( iId.index() ).displayProperties();
            cluster->SetMarkerColor( dp.color() );
            cluster->ElementChanged();
        }
    }
}

//______________________________________________________________________________
void
FWPFClusterLegoProxyBuilder::scaleProduct( TEveElementList* parent, FWViewType::EType type, const FWViewContext* vc )
{
   for (TEveElement::List_i i = parent->BeginChildren(); i!= parent->EndChildren(); ++i)
   {
      if ((*i)->HasChildren())
      {
         TEveElement* el = (*i)->FirstChild();  // there is only one child added in this proxy builder
         FWPFLegoCandidate *cluster = dynamic_cast<FWPFLegoCandidate*>( el );
         cluster->updateScale(vc, context());
      }
   }
}

//______________________________________________________________________________
void
FWPFClusterLegoProxyBuilder::sharedBuild( const reco::PFCluster &iData, TEveElement &oItemHolder, const FWViewContext *vc )
{
   float energy = iData.energy();
   float et = calculateEt( iData, energy );
   float pt = et;
   float eta = iData.eta();
   float phi = iData.phi();

   context().voteMaxEtAndEnergy( et, energy );

   FWPFLegoCandidate *cluster = new FWPFLegoCandidate( vc, FWProxyBuilderBase::context(), energy, et, pt, eta, phi );
   cluster->SetMarkerColor( FWProxyBuilderBase::item()->defaultDisplayProperties().color() );
   setupAddElement( cluster, &oItemHolder );
}

//______________________________ECAL____________________________________________
void
FWPFEcalClusterLegoProxyBuilder::build( const reco::PFCluster &iData, unsigned int iIndex, TEveElement &oItemHolder, const FWViewContext *vc )
{
   PFLayer::Layer layer = iData.layer();
   if( layer < 0 )
      sharedBuild( iData, oItemHolder, vc ); 
}

//______________________________HCAL____________________________________________
void
FWPFHcalClusterLegoProxyBuilder::build( const reco::PFCluster &iData, unsigned int iIndex, TEveElement &oItemHolder, const FWViewContext *vc )
{
   PFLayer::Layer layer = iData.layer();
   if( layer > 0 )
      sharedBuild( iData, oItemHolder, vc );
}

//______________________________________________________________________________
REGISTER_FWPROXYBUILDER( FWPFEcalClusterLegoProxyBuilder, reco::PFCluster, "PF Cluster", FWViewType::kLegoPFECALBit );
REGISTER_FWPROXYBUILDER( FWPFHcalClusterLegoProxyBuilder, reco::PFCluster, "PF Cluster", FWViewType::kLegoBit );