StFmsClusterFinder.h

Go to the documentation of this file.

// $Id$
//
// $Log$
#ifndef STROOT_STFMSPOINTMAKER_STFMSCLUSTERFINDER_H_
#define STROOT_STFMSPOINTMAKER_STFMSCLUSTERFINDER_H_

#include <list>
#include <memory>  // For std::unique_ptr

#include "Rtypes.h"  // Provides ROOT ClassDef macro

class TObjArray;

namespace FMSCluster {  // $NMSPC
class StFmsTowerCluster;
class StFmsTower;
// Typedef a list of cluster unique_ptrs for convenience.
// Hide it from CINT as it can't handle parsing the header :(
#ifndef __CINT__
typedef std::list<std::unique_ptr<StFmsTowerCluster>> ClusterList;
#endif  // __CINT__

class StFmsClusterFinder {
 public:
  // Typedef StFmsTower pointer list for convenience
  typedef std::list<FMSCluster::StFmsTower*> TowerList;
  StFmsClusterFinder();
  // Use default copy constructor and assignment operator.
  ~StFmsClusterFinder();
  void calculateClusterMoments(StFmsTowerCluster* cluster) const;
  int categorise(StFmsTowerCluster* cluster);
  void setMomentEnergyCutoff(float cutoff = 0.5) { mEnergyCutoff = cutoff; }
  float momentEnergyCutoff() const { return mEnergyCutoff; }
#ifndef __CINT__  // Hide ClusterList from CINT

  int findClusters(TowerList* towers, ClusterList* clusters);
#endif  // __CINT__

 private:
  static const unsigned kMaxNClusters = 6;  
#ifndef __CINT__  // Hide ClusterList from CINT

  unsigned locateClusterSeeds(TowerList* towers, TowerList* neighbors,
                              ClusterList* clusters) const;
  unsigned associateTowersWithClusters(TowerList* neighbors,
                                       ClusterList* clusters,
                                       TObjArray* valleys) const;
  unsigned associateValleyTowersWithClusters(TowerList* neighbors,
                                             ClusterList* clusters,
                                             TObjArray* valleys) const;
  unsigned associateResidualTowersWithClusters(TowerList* neighbors,
                                               ClusterList* clusters) const;
  void associateSubThresholdTowersWithClusters(TowerList* towers,
                                               ClusterList* clusters) const;
#endif  // __CINT__
  Float_t mEnergyCutoff;  
  Int_t mNClusts;  
  ClassDef(StFmsClusterFinder, 0)
};  // class StFmsClusterFinder
}  // namespace FMSCluster
#endif  // STROOT_STFMSPOINTMAKER_STFMSCLUSTERFINDER_H_