HWTrackingDetectorEventDisplay.h

Go to the documentation of this file.

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

#ifndef HWTrackingDetectorEventDisplay_H_
#define HWTrackingDetectorEventDisplay_H_

// HERD headers
#include "algorithms/display/HWBaseEventDisplay.h"
#include "dataobjects/ClustersColl.h"
#include "dataobjects/Track2D.h"

// ROOT headers
#include "TEvePointSet.h"
#include "TEveStraightLineSet.h"
#include "TEveTrack.h"
#include "TEveTrackPropagator.h"

namespace Herd {
class HWTrackingDetectorEventDisplay : public HWBaseEventDisplay {
public:
  HWTrackingDetectorEventDisplay(std::string name = "Tracker");
  ~HWTrackingDetectorEventDisplay();

  template <class TrackingGeoParams>
  void ImportClusters(const ClustersColl &clColl, const SidesArray<std::vector<TrackingGeoParams>> &geoParamsColl);
  void SetRenderState(bool renderState);

  TEvePointSet *TrkHitsXZ;
  TEvePointSet *TrkHitsYZ;
  TEvePointSet *TrkHitsXY;

  TEveStraightLineSet *TrkHits3D;

private:
  template <class TrackingGeoParams>
  std::pair<float, float> GetLineRange(const Cluster &cl, const TrackingGeoParams &geoParams);
};
} // namespace Herd

#endif