FWPFLegoRecHit.h

Go to the documentation of this file.

#ifndef _FWPFLEGORECHIT_H_
#define _FWPFLEGORECHIT_H_

// -*- C++ -*-
//
// Package:     ParticleFlow
// Class  :     FWPFLegoRecHit
//
// Implementation:
//     <Notes on implementation>
//
// Original Author:  Simon Harris
//

// System include files
#include "TEveBox.h"
#include "TEveCompound.h"
#include "TEveStraightLineSet.h"

#include "TEveCaloData.h"
#include "TEveChunkManager.h"

// User include files
#include "Fireworks/Core/interface/FWViewContext.h"
#include "Fireworks/Core/interface/FWViewEnergyScale.h"
#include "Fireworks/Core/interface/Context.h"

// Forward declarations
class FWProxyBuilderBase;

//-----------------------------------------------------------------------------
// FWPFLegoRechHit
//-----------------------------------------------------------------------------
class FWPFLegoRecHit {
public:
  // ---------------- Constructor(s)/Destructor ----------------------
  FWPFLegoRecHit(const std::vector<TEveVector> &corners,
                 TEveElement *comp,
                 FWProxyBuilderBase *pb,
                 const FWViewContext *vc,
                 float e,
                 float et);
  virtual ~FWPFLegoRecHit() {}

  // --------------------- Member Functions --------------------------
  void updateScale(const FWViewContext *vc, float maxLogVal);
  void setSquareColor(Color_t c) {
    m_ls->SetMarkerColor(c);
    m_ls->SetLineColor(kBlack);
  }

  TEveBox *getTower() { return m_tower; }
  void setLine(int idx, float x1, float y1, float z1, float x2, float y2, float z2);
  void addLine(float x1, float y1, float z1, float x2, float y2, float z2);
  void addLine(const TEveVector &v1, const TEveVector &v2);
  float getEtEnergy(bool b) const { return b ? m_et : m_energy; }
  bool isTallest() const { return m_isTallest; }
  void setIsTallest(bool b);

  FWPFLegoRecHit(const FWPFLegoRecHit &) = delete;                   // Disable default
  const FWPFLegoRecHit &operator=(const FWPFLegoRecHit &) = delete;  // Disable default

private:
  // --------------------- Member Functions --------------------------
  void setupEveBox(std::vector<TEveVector> &corners, float scale);
  void buildTower(const std::vector<TEveVector> &corners, const FWViewContext *vc);
  void buildLineSet(const std::vector<TEveVector> &corners, const FWViewContext *vc);

  // ----------------------- Data Members ----------------------------
  TEveBox *m_tower;
  TEveStraightLineSet *m_ls;
  float m_energy;
  float m_et;
  bool m_isTallest;
};
#endif
//=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_=_