#include "Rtypes.h"
#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 <map>
#include <vector>

namespace edm {
  class EventBase;
}

class FWGeometry;
class TEveCaloData;
class TEveCaloDataVec;
class TEveCaloLego;

// builder class for ecal detail view
class FWECALCaloDataDetailViewBuilder {
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.
  FWECALCaloDataDetailViewBuilder(const edm::EventBase *event,
                                  const FWGeometry *geom,
                                  float eta,
                                  float phi,
                                  int size = 50,
                                  Color_t defaultColor = kMagenta + 1)
      : m_event(event), m_geom(geom), m_eta(eta), m_phi(phi), m_size(size), m_defaultColor(defaultColor) {}

  // 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(const EcalRecHitCollection *hits, TEveCaloDataVec *data, bool xyEE);
  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

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

  // for keeping track of the colors to use for each slice
  std::vector<Color_t> m_colors;

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