HWCaloEventDisplay.h

Go to the documentation of this file.

/*
 * HWCaloEventDisplay.h
 *
 *  Created on: 08 May 2019
 *      Author: Valerio Formato
 */

// HERD headers
#include "algorithms/display/HWBaseEventDisplay.h"
#include "algorithms/display/VPalette.h"
#include "dataobjects/CaloGeoParams.h"
#include "dataobjects/CaloHits.h"

// ROOT headers
#include "TEveGeoNode.h"

// c++ headers
#include <unordered_map>

#ifndef HWCaloEventDisplay_H_
#define HWCaloEventDisplay_H_

namespace Herd {
class HWCaloEventDisplay : public HWBaseEventDisplay {
public:
  HWCaloEventDisplay();
  ~HWCaloEventDisplay() = default;

  void GenerateHashTable(TEveGeoNode *node);
  void ImportCaloHits(const CaloHits &caloHits, const CaloGeoParams &caloParams);
  void SetRenderState(bool renderState);

  TEveElementList *CaloHits3D;
  TEveElementList *CaloHitsXZ;
  TEveElementList *CaloHitsYZ;
  TEveElementList *CaloHitsXY;

private:
  std::unordered_map<std::string, TEveGeoShape *> m_EGSMap;
  std::map<RefFrame::View, std::unique_ptr<VPalette>> m_palettes;
  std::unique_ptr<VPalette> m_3Dpalette;

  struct CaloProjHit {
    int iu;
    int iv;
    float eDep;
    float u;
    float v;
  };

public:
  using projHitIteratorBase = std::pair<std::pair<int, int>, CaloProjHit>;
  using CaloProjHits = std::map<std::pair<int, int>, CaloProjHit>;
  void Draw3DHits(const CaloHits &caloHits, const CaloGeoParams &caloParams);
  void DrawProjectedHits(const CaloProjHits &projHits, RefFrame::View view);

  CaloProjHits ProjectHits(const CaloHits &hits, const CaloGeoParams &geoParams, RefFrame::Axis dir1,
                           RefFrame::Axis dir2);
};
} // namespace Herd

#endif