/**
 * \file CbmCluster.h
 * \brief Base class for cluster objects.
 * \author Andrey Lebedev <andrey.lebedev@gsi.de>
 * \date 2012
 */

#ifndef CBMCLUSTER_H_
#define CBMCLUSTER_H_

#include "FairMultiLinkedData.h"

/**
 * \class CbmCluster
 * \brief Base class for cluster objects.
 * \author Andrey Lebedev <andrey.lebedev@gsi.de>
 * \date 2012
 */
class CbmCluster : public FairMultiLinkedData {
public:
	/**
	 * \brief Constructor.
	 */
	CbmCluster();

	/**
	 * \brief Destructor.
	 */
	virtual ~CbmCluster();

	/**
	 * \brief Add digi to cluster.
	 * \param[in] index Digi index in TClonesArray.
	 */
	void AddDigi(Int_t index) { fDigis.push_back(index); }

	/**
	 * \brief Add array of digi to cluster.
	 * \param[in] indices Array of digi indices in TClonesArray.
	 */
	void AddDigis(const std::vector<Int_t>& indices) { fDigis.insert(fDigis.end(), indices.begin(), indices.end()); }

	/**
	 * \brief Set array of digi to cluster. Overwrites existing array.
	 * \param[in] indices Array of digi indices in TClonesArray.
	 */
	void SetDigis(const std::vector<Int_t>& indices) { fDigis.assign(indices.begin(), indices.end()); }

	/**
	 * \brief Number of digis in cluster.
	 * \return Number of digis in cluster.
	 */
	Int_t GetNofDigis() const { return fDigis.size(); }

	/**
	 * \brief Get digi at position index.
	 * \param[in] index Position of digi in array.
	 * \return Digi index in TClonesArray.
	 */
	Int_t GetDigi(Int_t index) const { return fDigis[index]; }

	/**
	 * \brief Get array of digi indices.
	 * \return Array of digi indices in TClonesArray.
	 */
	const std::vector<Int_t>& GetDigis() const { return fDigis; }

	/**
	 * \brief Remove all digis.
	 */
	void Clear() { fDigis.clear(); }

private:
	std::vector<Int_t> fDigis; // Array of digi indices

	ClassDef(CbmCluster, 1);
};

#endif /* CBMCLUSTER_H_ */