FWECALDetailViewBuilder.h

Go to the documentation of this file.

#ifndef Fireworks_Calo_FWECALDetailViewBuilder_h
#define Fireworks_Calo_FWECALDetailViewBuilder_h

#include "Rtypes.h"
#include <map>
#include <vector>
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/EgammaReco/interface/SuperClusterFwd.h"
#include "DataFormats/CaloRecHit/interface/CaloClusterFwd.h"
#include "Fireworks/Calo/interface/FWBoxRecHit.h"

namespace edm {
  class EventBase;
}

class FWGeometry;
class TEveCaloDataVec;
class TEveCaloLego;

// builder class for ecal detail view
class FWECALDetailViewBuilder {
public:
  // construct an ecal detail view builder
  // the arguments are the event, a pointer to geometry object,
  // the eta and phi position to show,
  // the half width of the region (in indices, e.g. iEta) and
  // the default color for the hits.
  FWECALDetailViewBuilder(const edm::EventBase *event,
                          const FWGeometry *geom,
                          float eta,
                          float phi,
                          int size = 50,
                          Color_t defaultColor = kMagenta + 1);

  // draw the ecal information with the preset colors
  // (if any colors have been preset)
  TEveCaloLego *build();

  TEveCaloData *buildCaloData(bool xyEE);

  // set colors of some predefined detids
  void setColor(Color_t color, const std::vector<DetId> &detIds);

  // show superclusters using two alternating colors
  // to make adjacent clusters visible
  void showSuperClusters(Color_t color1 = kGreen + 2, Color_t color2 = kTeal);

  // show a specific supercluster in a specific color
  void showSuperCluster(const reco::SuperCluster &cluster, Color_t color = kYellow);

  // add legends; returns final y
  double makeLegend(double x0 = 0.02,
                    double y0 = 0.95,
                    Color_t clustered1 = kGreen + 1,
                    Color_t clustered2 = kTeal,
                    Color_t supercluster = kYellow);

private:
  // fill data
  void fillData(TEveCaloDataVec *data);

  void fillEtaPhi(const EcalRecHitCollection *hits, TEveCaloDataVec *data);

  const edm::EventBase *m_event;  // the event
  const FWGeometry *m_geom;       // the geometry
  float m_eta;                    // eta position view centred on
  float m_phi;                    // phi position view centred on
  int m_size;                     // view half width in number of crystals
  Color_t m_defaultColor;         // default color for crystals

  std::vector<FWBoxRecHit *> m_boxes;

  // for keeping track of what det id goes in what slice
  std::map<DetId, int> m_detIdsToColor;

  TEveElement *m_towerList;

  // sorting function to sort super clusters by eta.
  static bool superClusterEtaLess(const reco::CaloCluster &lhs, const reco::CaloCluster &rhs) {
    return (lhs.eta() < rhs.eta());
  }

  float sizeRad() const;
};

#endif  // Fireworks_Calo_FWECALDetailViewBuilder_h