FWMETProxyBuilder.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     Calo
// Class  :     FWMETProxyBuilder
//
// Implementation:
//     <Notes on implementation>
//
// Original Author:
//         Created:  Sun Jan  6 23:57:00 EST 2008
//
 
// system include files
#include "TEveGeoNode.h"
#include "TEveScalableStraightLineSet.h"
#include "TGeoTube.h"
#include "TMath.h"
 
// user include files
#include "Fireworks/Core/interface/FWSimpleProxyBuilderTemplate.h"
#include "Fireworks/Core/interface/FWViewType.h"
#include "Fireworks/Core/interface/BuilderUtils.h"
#include "Fireworks/Core/interface/Context.h"
#include "Fireworks/Core/interface/FWViewEnergyScale.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Calo/interface/scaleMarker.h"
 
#include "DataFormats/METReco/interface/MET.h"
 
//
//   RPZ proxy builder with shared MET shape
//
 
class FWMETProxyBuilder : public FWSimpleProxyBuilderTemplate<reco::MET> {
public:
  FWMETProxyBuilder() {}
  ~FWMETProxyBuilder() override {}
 
  bool haveSingleProduct() const override { return false; }  // use buildViewType instead of buildView
 
  bool havePerViewProduct(FWViewType::EType) const override { return true; }  // used energy scaling
 
  void scaleProduct(TEveElementList* parent, FWViewType::EType, const FWViewContext* vc) override;
 
  void cleanLocal() override { m_lines.clear(); }
 
  REGISTER_PROXYBUILDER_METHODS();
 
private:
  FWMETProxyBuilder(const FWMETProxyBuilder&) = delete;                   // stop default
  const FWMETProxyBuilder& operator=(const FWMETProxyBuilder&) = delete;  // stop default
 
  using FWSimpleProxyBuilderTemplate<reco::MET>::buildViewType;
  void buildViewType(const reco::MET& iData,
                     unsigned int iIndex,
                     TEveElement& oItemHolder,
                     FWViewType::EType type,
                     const FWViewContext*) override;
 
  std::vector<fireworks::scaleMarker> m_lines;
};
 
void FWMETProxyBuilder::scaleProduct(TEveElementList* parent, FWViewType::EType type, const FWViewContext* vc) {
  typedef std::vector<fireworks::scaleMarker> Lines_t;
  FWViewEnergyScale* caloScale = vc->getEnergyScale();
 
  //   printf("MET %s %p -> %f\n",  item()->name().c_str(),  vc, caloScale->getScaleFactor3D() );
  for (Lines_t::iterator i = m_lines.begin(); i != m_lines.end(); ++i) {
    if (vc == (*i).m_vc) {
      // printf("lineset %s  %p val %f ...%f\n", item()->name().c_str(), (*i).m_ls , (*i).m_et, caloScale->getScaleFactor3D()*(*i).m_et);
      float value = caloScale->getPlotEt() ? (*i).m_et : (*i).m_energy;
 
      (*i).m_ls->SetScale(caloScale->getScaleFactor3D() * value);
 
      TEveProjectable* pable = static_cast<TEveProjectable*>((*i).m_ls);
      for (TEveProjectable::ProjList_i j = pable->BeginProjecteds(); j != pable->EndProjecteds(); ++j) {
        (*j)->UpdateProjection();
      }
    }
  }
}
 
void FWMETProxyBuilder::buildViewType(const reco::MET& met,
                                      unsigned int iIndex,
                                      TEveElement& oItemHolder,
                                      FWViewType::EType type,
                                      const FWViewContext* vc) {
  using namespace TMath;
  double phi = met.phi();
  double theta = met.theta();
  double size = 1.f;
 
  FWViewEnergyScale* caloScale = vc->getEnergyScale();
 
  TEveScalableStraightLineSet* marker = new TEveScalableStraightLineSet("MET marker");
  marker->SetLineWidth(2);
 
  if (type == FWViewType::kRhoZ) {
    // body
    double r0;
    if (TMath::Abs(met.eta()) < context().caloTransEta()) {
      r0 = context().caloR1() / sin(theta);
    } else {
      r0 = context().caloZ1() / fabs(cos(theta));
    }
    marker->SetScaleCenter(0., Sign(r0 * sin(theta), phi), r0 * cos(theta));
    double r1 = r0 + 1;
    marker->AddLine(0., Sign(r0 * sin(theta), phi), r0 * cos(theta), 0., Sign(r1 * sin(theta), phi), r1 * cos(theta));
 
    // arrow pointer
    double r2 = r1 - 0.1;
    double dy = 0.05 * size;
    marker->AddLine(0.,
                    Sign(r2 * sin(theta) + dy * cos(theta), phi),
                    r2 * cos(theta) - dy * sin(theta),
                    0.,
                    Sign(r1 * sin(theta), phi),
                    r1 * cos(theta));
    dy = -dy;
    marker->AddLine(0.,
                    Sign(r2 * sin(theta) + dy * cos(theta), phi),
                    r2 * cos(theta) - dy * sin(theta),
                    0.,
                    Sign(r1 * sin(theta), phi),
                    r1 * cos(theta));
 
    // segment
    fireworks::addRhoZEnergyProjection(
        this, &oItemHolder, context().caloR1() - 1, context().caloZ1() - 1, theta - 0.04, theta + 0.04, phi);
  } else {
    // body
    double r0 = context().caloR1();
    double r1 = r0 + 1;
    marker->SetScaleCenter(r0 * cos(phi), r0 * sin(phi), 0);
    marker->AddLine(r0 * cos(phi), r0 * sin(phi), 0, r1 * cos(phi), r1 * sin(phi), 0);
 
    // arrow pointer, xy  rotate offset point ..
    double r2 = r1 - 0.1;
    double dy = 0.05 * size;
 
    marker->AddLine(r2 * cos(phi) - dy * sin(phi), r2 * sin(phi) + dy * cos(phi), 0, r1 * cos(phi), r1 * sin(phi), 0);
    dy = -dy;
    marker->AddLine(r2 * cos(phi) - dy * sin(phi), r2 * sin(phi) + dy * cos(phi), 0, r1 * cos(phi), r1 * sin(phi), 0);
 
    // segment
    double min_phi = phi - M_PI / 36 / 2;
    double max_phi = phi + M_PI / 36 / 2;
    TEveGeoManagerHolder gmgr(TEveGeoShape::GetGeoMangeur());
    TEveGeoShape* element =
        fireworks::getShape("spread", new TGeoTubeSeg(r0 - 2, r0, 1, min_phi * 180 / M_PI, max_phi * 180 / M_PI), 0);
    element->SetPickable(kTRUE);
    setupAddElement(element, &oItemHolder);
  }
 
  marker->SetScale(caloScale->getScaleFactor3D() * (caloScale->getPlotEt() ? met.et() : met.energy()));
  setupAddElement(marker, &oItemHolder);
 
  // printf("add line %s  %f %f .... eta %f theta %f\n", item()->name().c_str(), met.et(), met.energy(), met.eta(), met.theta());
  m_lines.push_back(fireworks::scaleMarker(marker, met.et(), met.energy(), vc));  // register for scales
 
  context().voteMaxEtAndEnergy(met.et(), met.energy());
}
 
REGISTER_FWPROXYBUILDER(FWMETProxyBuilder, reco::MET, "recoMET", FWViewType::kAllRPZBits);