IsoHitClusteringAlgorithm.h

Go to the documentation of this file.

#ifndef ISOHITCLUSTERINGALGORITHM_H
#define ISOHITCLUSTERINGALGORITHM_H 1

#include "Pandora/Algorithm.h"
#include "APRILUtility/KDTreeLinkerAlgoT.h"

namespace april_content
{

class IsoHitClusteringAlgorithm : public pandora::Algorithm
{
public:
    class Factory : public pandora::AlgorithmFactory
    {
    public:
        pandora::Algorithm *CreateAlgorithm() const;
    };

    IsoHitClusteringAlgorithm();

private:
    pandora::StatusCode Run();
    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    void CreateCluster(const pandora::CaloHitList *const caloHitList) const;

    pandora::StatusCode GetIsoHits(pandora::CaloHitList& caloHitList);
  
    pandora::StatusCode MeanShiftClustering(pandora::CaloHitList& isoHitList, std::vector<pandora::CaloHitList>& clusterHitsCollection);

private:
    float                 m_kernelBandwidth;
    float                 m_clusterEpsilon;
};

//------------------------------------------------------------------------------------------------------------------------------------------

inline pandora::Algorithm *IsoHitClusteringAlgorithm::Factory::CreateAlgorithm() const
{
    return new IsoHitClusteringAlgorithm();
}

}

#endif  // ISOHITCLUSTERINGALGORITHM_H