HWTracks3DEventDisplay.h

Go to the documentation of this file.

/*
 * HWTracks3DEventDisplay.h
 *
 *  Created on: 17 Feb 2022
 *      Author: Valerio Formato
 */
 
#ifndef HWTracks3DEventDisplay_H_
#define HWTracks3DEventDisplay_H_
 
// HERD headers
#include "herd/analysis/algorithms/display/HWBaseEventDisplay.h"
#include "herd/analysis/dataobjects/ClustersColl.h"
#include "herd/analysis/dataobjects/Line.h"
#include "herd/analysis/dataobjects/Track.h"
 
// ROOT headers
#include "TEveArrow.h"
#include "TEvePointSet.h"
#include "TEveTrack.h"
#include "TEveTrackPropagator.h"
 
namespace Herd {
class HWTracks3DEventDisplay : public HWBaseEventDisplay {
public:
  HWTracks3DEventDisplay(std::string name = "Tracks", Color_t color = kYellow);
  virtual ~HWTracks3DEventDisplay() = default;
 
  void ImportTracks(const std::vector<Herd::Line> &lines);
  void ImportTracks(const std::vector<Herd::Track> &tracks);
  void SetRenderState(bool renderState);
 
  void SetDrawArrow(bool drawArrow) { m_drawArrow = drawArrow; }
 
  TEveTrackList *trackList = nullptr;
  std::vector<TEveArrow *> arrows{};
 
protected:
  Color_t m_color;
  bool m_drawArrow = false;
};
} // namespace Herd
 
#endif