FWLegoEvePFCandidate.cc

Go to the documentation of this file.

#include "TEveCaloData.h"

#include "Fireworks/Core/interface/Context.h"
#include "Fireworks/Core/interface/FWViewContext.h"
#include "Fireworks/Core/interface/FWViewEnergyScale.h"
#include "Fireworks/ParticleFlow/interface/FWLegoEvePFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "Fireworks/Core/interface/fwLog.h"


FWLegoEvePFCandidate::FWLegoEvePFCandidate(const reco::PFCandidate& iData, const FWViewContext* vc,  const fireworks::Context& context):
   m_energy(0.f),
   m_et(0.f)
{
   m_et =  iData.et();
   m_energy = iData.energy();

   float base = 0.001; // flour offset 1%

   // first vertical  line , which is et/energy
   FWViewEnergyScale* caloScale = vc->getEnergyScale();
   float val = caloScale->getPlotEt() ?  m_et : m_energy;
   AddLine(iData.eta(),iData.phi(), base, 
           iData.eta(),iData.phi(), base + val*caloScale->getScaleFactorLego());


   AddMarker(0, 1.f);
   SetMarkerStyle(3); 
   SetMarkerSize(0.01); 
   SetDepthTest(false);

   // circle pt
   const unsigned int nLineSegments = 20;
   float circleScalingFactor = 50;
   const double jetRadius = iData.pt()/circleScalingFactor;
  
   for ( unsigned int iphi = 0; iphi < nLineSegments; ++iphi ) {
      AddLine(iData.eta()+jetRadius*cos(2*M_PI/nLineSegments*iphi),
              iData.phi()+jetRadius*sin(2*M_PI/nLineSegments*iphi),
              base,
              iData.eta()+jetRadius*cos(2*M_PI/nLineSegments*(iphi+1)),
              iData.phi()+jetRadius*sin(2*M_PI/nLineSegments*(iphi+1)),
              base);
   }
}

void
FWLegoEvePFCandidate::updateScale(const FWViewContext* vc, const fireworks::Context& context)
{
   FWViewEnergyScale* caloScale = vc->getEnergyScale();
   float val = caloScale->getPlotEt() ?  m_et : m_energy;

   // printf("update scale %f \n", getScale(vc, context)); fflush(stdout);
   float scaleFac =  caloScale->getScaleFactorLego();
   // resize first line
   TEveChunkManager::iterator li(GetLinePlex());
   li.next();
   TEveStraightLineSet::Line_t& l = * (TEveStraightLineSet::Line_t*) li(); 
   l.fV2[2] = l.fV1[2] + val*scaleFac;

   // move end point
   TEveChunkManager::iterator mi(GetMarkerPlex());
   mi.next();
   TEveStraightLineSet::Marker_t& m = * (TEveStraightLineSet::Marker_t*) mi();
   m.fV[2] =  l.fV2[2];
}