Cluster.h

Go to the documentation of this file.

/*
 * Cluster.h
 *
 *  Created on: 5 Nov 2018
 *      Author: Nicola Mori
 */

#ifndef HERD_CLUSTER_H_
#define HERD_CLUSTER_H_

#include "dataobjects/Hit.h"
#include "dataobjects/TrackingDetectorGeoParams.h"

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

#include <string>
#include <vector>

namespace Herd {

class Cluster {
public:
  Cluster();

  Cluster(std::vector<Hit> hits, size_t seedIndex);

  virtual ~Cluster() = default;

  const std::vector<Hit> &Hits() const { return _hits; }

  void SetHits(std::vector<Hit> hits, size_t seedIndex);

  size_t SeedHitIndex() const { return _seedHitIndex; }

  float EDep() const { return _eDep; }

  Point COGPosition(const TrackingDetectorGeoParams &geoParams) const;

  // /**
  //  * @brief Get the center of gravity position of the fit cluster
  //  *
  //  * @param The geoparams associated to this cluster
  //  * @return The center of gravity
  //  */
  // Point COGPosition(const FitGeoParams &geoParams) const;

  bool operator==(const Cluster &right) const {
    return (Hits()[SeedHitIndex()].VolumeID() == right.Hits()[right.SeedHitIndex()].VolumeID());
  }

private:
  float _eDep;          
  size_t _seedHitIndex; 

  void _ComputeEdep(); 

protected:
  std::vector<Hit> _hits; 

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

} // namespace Herd

#endif /* HERD_CLUSTER_H_ */