FWCaloTauProxyBuilder.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     Calo
// Class  :     FWCaloTauProxyBuilder
//
// Implementation:
//     <Notes on implementation>
//
// Original Author:
//         Created:  Sun Jan  6 23:57:00 EST 2008
// $Id: FWCaloTauProxyBuilder.cc,v 1.15 2010/09/16 15:42:20 yana 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 "Fireworks/Calo/interface/thetaBins.h"
#include "DataFormats/TauReco/interface/CaloTau.h"
#include "DataFormats/TauReco/interface/CaloTauFwd.h"

class FWViewContext;

class FWCaloTauProxyBuilder : public FWTauProxyBuilderBase
{
public:
   FWCaloTauProxyBuilder() {}
   virtual ~FWCaloTauProxyBuilder() {}
  
   REGISTER_PROXYBUILDER_METHODS();

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

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

};

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

      
   Int_t idx = 0;
   for( reco::CaloTauCollection::const_iterator it = caloTaus->begin(), itEnd = caloTaus->end(); it != itEnd; ++it, ++idx)
   {  
      TEveCompound* comp = createCompound();

      if (viewType == FWViewType::kLego)
      {
         fireworks::addCircle( (*it).eta(), (*it).phi(), 0.5, 20, comp, this );
      }
      else
      {
         const reco::CaloTauTagInfo *tauTagInfo = dynamic_cast<const reco::CaloTauTagInfo*>(((*it).caloTauTagInfoRef().get()));
         const reco::CaloJet *jet = dynamic_cast<const reco::CaloJet*>((tauTagInfo->calojetRef().get()));

         int min =  100;
         int max = -100;
         std::vector<double> phis;
         std::vector<CaloTowerPtr> towers = jet->getCaloConstituents();
         for( std::vector<CaloTowerPtr>::const_iterator tower = towers.begin(), towerEnd = towers.end();
              tower != towerEnd; ++tower )
         {
            unsigned int ieta = 41 + (*tower)->id().ieta();
            if( ieta > 40 ) --ieta;
            assert( ieta <= 82 );
     
            if( int(ieta) > max ) max = ieta;
            if( int(ieta) < min ) min = ieta;
            m_phis.push_back( (*tower)->phi() );
         }
         if( min > max ) {  
            min = 0; max = 0;
         }
         const std::vector<std::pair<double, double> > thetaBins = fireworks::thetaBins();
         m_minTheta = thetaBins[min].first;
         m_maxTheta = thetaBins[max].second;

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

}


REGISTER_FWPROXYBUILDER(FWCaloTauProxyBuilder, reco::CaloTauCollection, "CaloTau", FWViewType::kAll3DBits | FWViewType::kAllRPZBits);