cmtool::CBAlgoCenterOfMassSmall Class Reference

#include <CBAlgoCenterOfMassSmall.h>

Inheritance diagram for cmtool::CBAlgoCenterOfMassSmall:

Public Member Functions
	CBAlgoCenterOfMassSmall ()
	Default constructor. More...
virtual	~CBAlgoCenterOfMassSmall ()
	Default destructor. More...
virtual bool	Bool (const ::cluster::ClusterParamsAlg &cluster1, const ::cluster::ClusterParamsAlg &cluster2)
virtual void	Reset ()
	Function to reset the algorithm instance ... maybe implemented via child class. More...
virtual void	Report ()
	Function to report what's going on per merging iteration. More...
void	SetDebug (bool on)
	Function to set Debug mode of output. More...
void	SetMaxHitsSmallClus (size_t n)
	Function to set Max hits for small clsuters. More...
void	SetMaxDistance (double d)
	Function to set Max Distance for COM to be from start-end. More...
void	SetMaxCOMDistance (double d)
	Function to set Max Distance between COMs. More...
void	UseCOMInPoly (bool on)
	Use COM in Poly algo to decide merging. More...
void	UseCOMClose (bool on)
	Use COM in Poly algo to decide merging. More...
void	UseCOMNearClus (bool on)
	Use COM in Poly algo to decide merging. More...
void	SetLengthReach (double frac)
	Set Length Reach: How for out the cone extends as percent of cluster length. More...
double	ShortestDistanceSquared (double point_x, double point_y, double start_x, double start_y, double end_x, double end_y) const
Public Member Functions inherited from cmtool::CBoolAlgoBase
	CBoolAlgoBase ()
	Default constructor. More...
virtual	~CBoolAlgoBase ()
	Default destructor. More...
Public Member Functions inherited from cmtool::CMAlgoBase
	CMAlgoBase ()
virtual	~CMAlgoBase ()=default
virtual void	EventBegin (const std::vector< cluster::ClusterParamsAlg > &)
virtual void	EventEnd ()
virtual void	IterationBegin (const std::vector< cluster::ClusterParamsAlg > &)
virtual void	IterationEnd ()
void	SetAnaFile (TFile *fout)
	Setter function for an output plot TFile pointer. More...
virtual void	SetVerbose (bool doit=true)
	Setter function for verbosity. More...

Protected Attributes
bool	_debug
size_t	_maxHits
double	_MaxDist
double	_MaxCOMDistSquared
double	_lengthReach
bool	_COMinPolyAlg
	How four out - as percent of cluster length - cone will extend from start point. More...
bool	_COMsClose
bool	_COMNearClus
Protected Attributes inherited from cmtool::CMAlgoBase
TFile *	_fout
	TFile pointer to an output file. More...
bool	_verbose
	Boolean to choose verbose mode. Turned on if CMergeManager/CMatchManager's verbosity level is >= kPerMerging. More...

Detailed Description

Designed mainly for small clusters (< 10 hits): Find Cluster's center of mass weighing by Q Then see if COM is in polygon/cone of a big cluster If so -> merge

Definition at line 28 of file CBAlgoCenterOfMassSmall.h.

Constructor & Destructor Documentation

cmtool::CBAlgoCenterOfMassSmall::CBAlgoCenterOfMassSmall

(

)

Default constructor.

Definition at line 8 of file CBAlgoCenterOfMassSmall.cxx.

                                                   : CBoolAlgoBase()
  //----------------------------------------
  {

    SetDebug(false);
    SetMaxHitsSmallClus(30);
    SetMaxDistance(20.);
    SetMaxCOMDistance(25.);
    UseCOMInPoly(true);
    UseCOMClose(true);
    UseCOMNearClus(true);

  }

virtual cmtool::CBAlgoCenterOfMassSmall::~CBAlgoCenterOfMassSmall ( )

inlinevirtual

Default destructor.

Definition at line 36 of file CBAlgoCenterOfMassSmall.h.

{};

Member Function Documentation

bool cmtool::CBAlgoCenterOfMassSmall::Bool ( const ::cluster::ClusterParamsAlg &  cluster1, const ::cluster::ClusterParamsAlg &  cluster2  )

virtual

Core function: given the ClusterParamsAlg input, return whether a cluster should be merged or not.

Reimplemented from cmtool::CBoolAlgoBase.

Definition at line 23 of file CBAlgoCenterOfMassSmall.cxx.

  {

    int Nhits1 = 0;
    int Nhits2 = 0;

    //Both clusters should have less hits than some threshold
    if ( (cluster1.GetHitVector().size() > _maxHits) or (cluster2.GetHitVector().size() > _maxHits) )
      return false;

    //Define COM values on w & t
    double COM_t_1  = 0;
    double COM_w_1  = 0;
    double Q_1      = 0;
    double start_w_1;
    double start_t_1;
    double end_w_1;
    double end_t_1;
    Polygon2D poly1;
    double COM_t_2  = 0;
    double COM_w_2  = 0;
    double Q_2      = 0;
    double start_w_2;
    double start_t_2;
    double end_w_2;
    double end_t_2;
    Polygon2D poly2;

    //Get Hit vector for cluster 1
    //std::vector<util::PxHit> hits1;
    std::vector<util::PxHit> hits1;
    hits1   = cluster1.GetHitVector();
    Nhits1  = hits1.size();
    poly1   = cluster1.GetParams().PolyObject;
    start_w_1 = cluster1.GetParams().start_point.w;
    start_t_1 = cluster1.GetParams().start_point.t;
    end_w_1   = cluster1.GetParams().end_point.w;
    end_t_1   = cluster1.GetParams().end_point.t;
    //Get Hit vector for cluster 2
    //std::vector<util::PxHit> hits2;
    std::vector<util::PxHit> hits2;
    hits2   = cluster2.GetHitVector();
    Nhits2 = hits2.size();
    poly2   = cluster2.GetParams().PolyObject;
    start_w_2 = cluster2.GetParams().start_point.w;
    start_t_2 = cluster2.GetParams().start_point.t;
    end_w_2   = cluster2.GetParams().end_point.w;
    end_t_2   = cluster2.GetParams().end_point.t;

    //Find COM for cluster 1
    for (auto& hit: hits1){
      COM_t_1 += hit.t * hit.charge;
      COM_w_1 += hit.w * hit.charge;
      Q_1     += hit.charge;
    }
    COM_t_1 /= Q_1;
    COM_w_1 /= Q_1;

    //Find COM for cluster 2
    for (auto& hit: hits2){
      COM_t_2 += hit.t * hit.charge;
      COM_w_2 += hit.w * hit.charge;
      Q_2     += hit.charge;
    }
    COM_t_2 /= Q_2;
    COM_w_2 /= Q_2;

    if (_debug) {
      std::cout << "Cluster 1: " << std::endl;
      std::cout << "N Hits: " << Nhits1 << std::endl;
      std::cout << "COM: (w,t) -> (" << COM_w_1 << ", " << COM_t_1 << ")" << std::endl;
      std::cout << "Cluster 2: " << std::endl;
      std::cout << "N Hits: " << Nhits2 << std::endl;
      std::cout << "COM: (w,t) -> (" << COM_w_2 << ", " << COM_t_2 << ")" << std::endl;
      std::cout << std::endl;
    }

    //Get COM
    std::pair<float,float> COM_1;
    COM_1 = std::make_pair( COM_w_1, COM_t_1 );
    std::pair<float,float> COM_2;
    COM_2 = std::make_pair( COM_w_2, COM_t_2 );

    //look for polygon overlap
    if ( ( ( poly2.PointInside(COM_1) ) and  _COMinPolyAlg ) or
         ( ( poly1.PointInside(COM_2) ) and  _COMinPolyAlg ) ){
      if (_verbose) { std::cout << "Polygon Overlap -> Merge!" << std::endl << std::endl;}
      return true;
    }

    //look for COM of 1 close to COM of 2
    double distCOMs = ( COM_w_1-COM_w_2 )*( COM_w_1-COM_w_2 ) +
      ( COM_t_1-COM_t_2 )*( COM_t_1-COM_t_2 );
    if ( _COMsClose and ( distCOMs < _MaxCOMDistSquared ) ){
      if (_verbose) { std::cout << "COMs close to each other -> Merge!" << std::endl << std::endl;}
      return true;
    }

    //look for COM close to start-end of other cluster
    if ( _COMNearClus and
         ( ( ShortestDistanceSquared( COM_w_1, COM_t_1, start_w_2, start_t_2, end_w_2, end_t_2 ) < _MaxDist ) or
           ( ShortestDistanceSquared( COM_w_2, COM_t_2, start_w_1, start_t_1, end_w_1, end_t_1 ) < _MaxDist ) ) ) {
      if (_verbose) { std::cout << "COM close to start-end -> Merge!" << std::endl; }
      return true;
    }

    return false;
  }

void cmtool::CBAlgoCenterOfMassSmall::Report ( )

virtual

Function to report what's going on per merging iteration.

Reimplemented from cmtool::CMAlgoBase.

Definition at line 135 of file CBAlgoCenterOfMassSmall.cxx.

  {

  }

virtual void cmtool::CBAlgoCenterOfMassSmall::Reset ( )

inlinevirtual

Function to reset the algorithm instance ... maybe implemented via child class.

Reimplemented from cmtool::CMAlgoBase.

Definition at line 46 of file CBAlgoCenterOfMassSmall.h.

{}

void cmtool::CBAlgoCenterOfMassSmall::SetDebug ( bool  on )

inline

Function to set Debug mode of output.

Definition at line 52 of file CBAlgoCenterOfMassSmall.h.

{ _debug = on; }

void cmtool::CBAlgoCenterOfMassSmall::SetLengthReach ( double  frac )

inline

Set Length Reach: How for out the cone extends as percent of cluster length.

Definition at line 73 of file CBAlgoCenterOfMassSmall.h.

{ _lengthReach = frac; }

void cmtool::CBAlgoCenterOfMassSmall::SetMaxCOMDistance ( double  d )

inline

Function to set Max Distance between COMs.

Definition at line 61 of file CBAlgoCenterOfMassSmall.h.

{ _MaxCOMDistSquared = d; }

void cmtool::CBAlgoCenterOfMassSmall::SetMaxDistance ( double  d )

inline

Function to set Max Distance for COM to be from start-end.

Definition at line 58 of file CBAlgoCenterOfMassSmall.h.

{ _MaxDist = d; }

void cmtool::CBAlgoCenterOfMassSmall::SetMaxHitsSmallClus ( size_t  n )

inline

Function to set Max hits for small clsuters.

Definition at line 55 of file CBAlgoCenterOfMassSmall.h.

{ _maxHits = n; }

double cmtool::CBAlgoCenterOfMassSmall::ShortestDistanceSquared	(	double	point_x,
		double	point_y,
		double	start_x,
		double	start_y,
		double	end_x,
		double	end_y
	)		const

Definition at line 141 of file CBAlgoCenterOfMassSmall.cxx.

                                                                                                {

    //This code finds the shortest distance between a point and a line segment.
    //code based off sample from
    //http://stackoverflow.com/questions/849211/shortest-distance-between-a-point-and-a-line-segment
    //note to self: rewrite this with TVector2 and compare time differences...
    //TVector2 code might be more understandable

    double distance_squared = -1;

    // Line segment: from ("V") = (start_x, start_y) to ("W")=(end_x, end_y)
    double length_squared = pow((end_x - start_x), 2) + pow((end_y - start_y), 2);

    // Treat the case start & end point overlaps
    if(length_squared < 0.1) {
      if(_verbose){
        std::cout << std::endl;
        std::cout << Form(" Provided very short line segment: (%g,%g) => (%g,%g)",
                          start_x,start_y,end_x,end_y) << std::endl;
        std::cout << " Likely this means one of two clusters have start & end point identical." << std::endl;
        std::cout << " Check the cluster output!" << std::endl;
        std::cout << std::endl;
        std::cout << Form(" At this time, the algorithm uses a point (%g,%g)",start_x,start_y) << std::endl;
        std::cout << " to represent this cluster's location." << std::endl;
        std::cout << std::endl;
      }

      return (pow((point_x - start_x),2) + pow((point_y - start_y),2));
    }

    //Find shortest distance between point ("P")=(point_x,point_y) to this line segment
    double t = ( (point_x - start_x)*(end_x - start_x) + (point_y - start_y)*(end_y - start_y) ) / length_squared;

    if(t<0.0) distance_squared = pow((point_x - start_x), 2) + pow((point_y - start_y), 2);

    else if (t>1.0) distance_squared = pow((point_x - end_x), 2) + pow(point_y - end_y, 2);

    else distance_squared = pow((point_x - (start_x + t*(end_x - start_x))), 2) + pow((point_y - (start_y + t*(end_y - start_y))),2);

    return distance_squared;

  }//end ShortestDistanceSquared function

void cmtool::CBAlgoCenterOfMassSmall::UseCOMClose ( bool  on )

inline

Use COM in Poly algo to decide merging.

Definition at line 67 of file CBAlgoCenterOfMassSmall.h.

{ _COMsClose = on; }

void cmtool::CBAlgoCenterOfMassSmall::UseCOMInPoly ( bool  on )

inline

Use COM in Poly algo to decide merging.

Definition at line 64 of file CBAlgoCenterOfMassSmall.h.

{ _COMinPolyAlg = on; }

void cmtool::CBAlgoCenterOfMassSmall::UseCOMNearClus ( bool  on )

inline

Use COM in Poly algo to decide merging.

Definition at line 70 of file CBAlgoCenterOfMassSmall.h.

{ _COMNearClus = on; }

Member Data Documentation

bool cmtool::CBAlgoCenterOfMassSmall::_COMinPolyAlg

protected

How four out - as percent of cluster length - cone will extend from start point.

Definition at line 87 of file CBAlgoCenterOfMassSmall.h.

bool cmtool::CBAlgoCenterOfMassSmall::_COMNearClus

protected

Definition at line 89 of file CBAlgoCenterOfMassSmall.h.

bool cmtool::CBAlgoCenterOfMassSmall::_COMsClose

protected

Definition at line 88 of file CBAlgoCenterOfMassSmall.h.

bool cmtool::CBAlgoCenterOfMassSmall::_debug

protected

Definition at line 82 of file CBAlgoCenterOfMassSmall.h.

double cmtool::CBAlgoCenterOfMassSmall::_lengthReach

protected

Definition at line 86 of file CBAlgoCenterOfMassSmall.h.

double cmtool::CBAlgoCenterOfMassSmall::_MaxCOMDistSquared

protected

Definition at line 85 of file CBAlgoCenterOfMassSmall.h.

double cmtool::CBAlgoCenterOfMassSmall::_MaxDist

protected

Definition at line 84 of file CBAlgoCenterOfMassSmall.h.

size_t cmtool::CBAlgoCenterOfMassSmall::_maxHits

protected

Definition at line 83 of file CBAlgoCenterOfMassSmall.h.

---

The documentation for this class was generated from the following files:

• CBAlgoCenterOfMassSmall.h
• CBAlgoCenterOfMassSmall.cxx