Track2D.h

Go to the documentation of this file.

/*
 * Track2D.h
 *
 *  Created on: 18 Dec 2018
 *      Author: Valerio Formato
 */

#ifndef HERD_TRACK2D_H_
#define HERD_TRACK2D_H_

// HerdSoftware headers
#include "dataobjects/FitClustersColl.h"
#include "dataobjects/FitGeoParamsColl.h"
#include "dataobjects/Line2D.h"
#include "dataobjects/ScdClustersColl.h"
#include "dataobjects/ScdGeoParamsColl.h"
#include "dataobjects/StkClustersColl.h"
#include "dataobjects/StkGeoParamsColl.h"

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

namespace Herd {

class Track2D {
public:
  Track2D() : m_line{} {};

  Track2D(Line2D line) : m_line(std::move(line)){};

  Track2D(const StkClustersColl &clusterColl) : m_stkClusterColl{clusterColl} {};

  Track2D(const ScdClustersColl &clusterColl) : m_scdClusterColl{clusterColl} {};

  Track2D(const FitClustersColl &clusterColl) : m_fitClusterColl{clusterColl} {};

  virtual ~Track2D() = default;

  void AddClusters(const StkClustersColl &clusterColl);

  void AddClusters(const ScdClustersColl &clusterColl);

  void AddClusters(const FitClustersColl &clusterColl);

  unsigned int NClusters() const;

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

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

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

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

  float ClusterResidual(const Cluster &cluster, const TrackingDetectorGeoParams &geoParams) const;

  SidesArray<std::vector<std::vector<float>>> GetStkResiduals(const StkGeoParamsColl &stkGeoParamsColl) const;

  SidesArray<std::vector<std::vector<float>>> GetScdResiduals(const ScdGeoParamsColl &scdGeoParamsColl) const;

  SidesArray<std::vector<std::vector<float>>> GetFitResiduals(const FitGeoParamsColl &fitGeoParamsColl) const;

private:
  Herd::Line2D m_line;
  StkClustersColl m_stkClusterColl;
  ScdClustersColl m_scdClusterColl;
  FitClustersColl m_fitClusterColl;

#ifdef HS_USE_ROOT
  ClassDef(Track2D, 1)
#endif
};

} // namespace Herd

#endif /* HERD_TRACK_H_ */