Track.h

Go to the documentation of this file.

/*
 * Track.h
 *
 *  Created on: 19 Oct 2018
 *      Author: Nicola Mori
 */

#ifndef HERD_TRACK_H_
#define HERD_TRACK_H_

// HerdSoftware headers
#include "dataobjects/CaloHits.h"
#include "dataobjects/FitClustersColl.h"
#include "dataobjects/FitGeoParamsColl.h"
#include "dataobjects/Line.h"
#include "dataobjects/Point.h"
#include "dataobjects/ScdClustersColl.h"
#include "dataobjects/ScdGeoParamsColl.h"
#include "dataobjects/StkClustersColl.h"
#include "dataobjects/StkGeoParamsColl.h"
#include "dataobjects/Track2D.h"

#ifdef HS_USE_ROOT
#include "Rtypes.h"
#endif

namespace Herd {

class Track {
public:
  Track() : m_line{}, DEFAULT_INIT(m_caloClustersIndex), DEFAULT_INIT(m_caloDataObjName) {}

  Track(Line line) : m_line(std::move(line)), DEFAULT_INIT(m_caloClustersIndex), DEFAULT_INIT(m_caloDataObjName){};

  Track(const Track2D &proj1, RefFrame::View view1, const Track2D &proj2, RefFrame::View view2);

  virtual ~Track() = default;

  const Herd::Line &GetLine() const { return m_line; }

  void SetLine(Herd::Line line) { m_line = std::move(line); }

  void SetCaloHits(CaloHits hits, std::string objName = "", size_t index = Herd::defaultValue<size_t>);

  void AddClusters(const StkClustersColl &clusterColl);

  void AddClusters(const ScdClustersColl &clusterColl);

  void AddClusters(const FitClustersColl &clusterColl);

  void SetCaloClusterIndex(size_t index) { m_caloClustersIndex = index; }

  void SetCaloDataObjName(std::string name) { m_caloDataObjName = std::move(name); }

  const StkClustersColl &GetStkClusters() const { return m_stkClusterColl; }

  const ScdClustersColl &GetScdClusters() const { return m_scdClusterColl; }

  const FitClustersColl &GetFitClusters() const { return m_fitClusterColl; }

  const Herd::CaloHits &CaloHits() const { return m_caloHits; }

  size_t CaloClusterIndex() const { return m_caloClustersIndex; }

  const std::string &CaloDataObjName() const { return m_caloDataObjName; }

private:
  Herd::CaloHits m_caloHits;
  StkClustersColl m_stkClusterColl;
  ScdClustersColl m_scdClusterColl;
  FitClustersColl m_fitClusterColl;

  Herd::Line m_line;

  size_t m_caloClustersIndex;
  std::string m_caloDataObjName;

#ifdef HS_USE_ROOT
  ClassDef(Track, 2)
#endif
};

} // namespace Herd

#endif /* HERD_TRACK_H_ */