CMSSW_4_4_3_patch1/src/Fireworks/Calo/plugins/FWPFTauProxyBuilder.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     Calo
// Class  :     FWPFTauProxyBuilder
//
// Implementation:
//     <Notes on implementation>
//
// Original Author:
//         Created:  Sun Jan  6 23:57:00 EST 2008
// $Id: FWPFTauProxyBuilder.cc,v 1.16 2010/11/05 10:54:33 amraktad Exp $
//

// system include files
#include "TEveCompound.h"
#include "TGeoTube.h"
#include "TEveGeoNode.h"
#include "TEveStraightLineSet.h"
#include "TEveTrack.h"

// user include files
#include "Fireworks/Calo/interface/FWTauProxyBuilderBase.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/FWViewType.h"
#include "Fireworks/Core/interface/BuilderUtils.h"


#include "DataFormats/TauReco/interface/PFTau.h"
#include "DataFormats/TrackReco/interface/Track.h"

class FWPFTauProxyBuilder : public FWTauProxyBuilderBase
{
public:
   FWPFTauProxyBuilder() {}
   virtual ~FWPFTauProxyBuilder() {}

   REGISTER_PROXYBUILDER_METHODS();

private:
   FWPFTauProxyBuilder( const FWPFTauProxyBuilder& );    // stop default
   const FWPFTauProxyBuilder& operator=( const FWPFTauProxyBuilder& );    // stop default

   virtual void buildViewType( const FWEventItem* iItem, TEveElementList* product, FWViewType::EType type , const FWViewContext*);
};

void
FWPFTauProxyBuilder::buildViewType( const FWEventItem* iItem, TEveElementList* product, FWViewType::EType viewType , const FWViewContext* vc)
{
   reco::PFTauCollection const * pfTaus = 0;
   iItem->get( pfTaus );
   if( pfTaus == 0 ) return;

   for( reco::PFTauCollection::const_iterator it = pfTaus->begin(), itEnd = pfTaus->end(); it != itEnd; ++it)
   { 
      TEveCompound* comp = createCompound();

      if (viewType == FWViewType::kLego)
      {
         fireworks::addCircle( (*it).eta(), (*it).phi(), 0.5, 20, comp, this );
      }
      else
      {
         // prepare phi-list and theta range
         const reco::PFTauTagInfo *tauTagInfo = dynamic_cast<const reco::PFTauTagInfo*>( (*it).pfTauTagInfoRef().get() );
         const reco::PFJet *jet = dynamic_cast<const reco::PFJet*>( tauTagInfo->pfjetRef().get() );
         m_minTheta =  100;
         m_maxTheta = -100;
         std::vector<double> phis;
         std::vector <const reco::Candidate*> candidates = jet->getJetConstituentsQuick();
         for( std::vector<const reco::Candidate*>::const_iterator candidate = candidates.begin(), candidateEnd = candidates.end();
              candidate != candidateEnd; ++candidate )
         {
            double itheta = (*candidate)->theta();
            if( itheta > m_maxTheta ) m_maxTheta = itheta;
            if( itheta < m_minTheta ) m_minTheta = itheta;

            m_phis.push_back( (*candidate)->phi() );
         }
         if( m_minTheta > m_maxTheta ) {        
            m_minTheta = 0;
            m_maxTheta = 0;
         }

         buildBaseTau(*it, *jet, comp, viewType, vc);
         m_phis.clear();
      }
      setupAddElement( comp, product );
   }
}

REGISTER_FWPROXYBUILDER( FWPFTauProxyBuilder, reco::PFTauCollection, "PFTau", FWViewType::kAll3DBits | FWViewType::kAllRPZBits | FWViewType::kAllLegoBits);