CMSSW_4_4_3_patch1/src/Fireworks/GenParticle/plugins/FWGenParticleProxyBuilder.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     Core
// Class  :     FWGenParticleProxyBuilder
//
//
// Original Author:
//         Created:  Thu Dec  6 18:01:21 PST 2007
// $Id: FWGenParticleProxyBuilder.cc,v 1.7 2010/11/11 20:25:28 amraktad Exp $
// 

#include "TDatabasePDG.h"
#include "TEveTrack.h"

#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/FWSimpleProxyBuilderTemplate.h"
#include "Fireworks/Candidates/interface/CandidateUtils.h"

#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"

class FWGenParticleProxyBuilder : public FWSimpleProxyBuilderTemplate<reco::GenParticle> {

public:
   FWGenParticleProxyBuilder() {}
   virtual ~FWGenParticleProxyBuilder() {}

   // ---------- member functions ---------------------------
   REGISTER_PROXYBUILDER_METHODS();

private:
   FWGenParticleProxyBuilder(const FWGenParticleProxyBuilder&); // stop default

   const FWGenParticleProxyBuilder& operator=(const FWGenParticleProxyBuilder&); // stop default
   
   void build(const reco::GenParticle& iData, unsigned int iIndex,TEveElement& oItemHolder, const FWViewContext*);

   // ---------- member data --------------------------------
   static TDatabasePDG* s_pdg;
};

//______________________________________________________________________________

TDatabasePDG* FWGenParticleProxyBuilder::s_pdg = 0;

void
FWGenParticleProxyBuilder::build(const reco::GenParticle& iData, unsigned int iIndex, TEveElement& oItemHolder, const FWViewContext*) 
{
   if (!s_pdg)
      s_pdg = new TDatabasePDG();
 
 
   TEveTrack* trk = fireworks::prepareCandidate( iData, context().getTrackPropagator() );    
   trk->MakeTrack();
   /*
     trk->SetTitle(s);
     char s[1024];
     TParticlePDG* pID = s_pdg->GetParticle(iData.pdgId());
     if ( pID )
     sprintf(s,"gen %s, Pt: %0.1f GeV", pID->GetName(), iData.pt());
     else
     sprintf(s,"gen pdg %d, Pt: %0.1f GeV", iData.pdgId(), iData.pt());
   */
   setupAddElement(trk, &oItemHolder);
}

REGISTER_FWPROXYBUILDER(FWGenParticleProxyBuilder, reco::GenParticle, "GenParticles", FWViewType::kAll3DBits | FWViewType::kAllRPZBits);