CBAlgoOutOfConeSeparate.cxx

Go to the documentation of this file.

#include "larreco/RecoAlg/CMTool/CMTAlgMerge/CBAlgoOutOfConeSeparate.h"

#include <cmath>

namespace cmtool {

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

    SetDebug(false);
    SetMaxAngleSep(20.);
    SetMinLength(15.);
    SetMinHits(20);
    SetStartAngleFalloff(2800); // in cm^2

  }

  //--------------------------------------------------------
  bool CBAlgoOutOfConeSeparate::Bool(const ::cluster::ClusterParamsAlg &cluster1,
                            const ::cluster::ClusterParamsAlg &cluster2)
  //--------------------------------------------------------
  {

    double angle1 = cluster1.GetParams().angle_2d;
    double angle2 = cluster2.GetParams().angle_2d;

    double w_start1 = cluster1.GetParams().start_point.w;
    double t_start1 = cluster1.GetParams().start_point.t;
    double w_start2 = cluster2.GetParams().start_point.w;
    double t_start2 = cluster2.GetParams().start_point.t;

    double len1 = cluster1.GetParams().length;
    double len2 = cluster2.GetParams().length;

    size_t hits1 = cluster1.GetHitVector().size();
    size_t hits2 = cluster2.GetHitVector().size();

    double startseparation = (w_start2-w_start1)*(w_start2-w_start1) + (t_start2-t_start1)*(t_start2-t_start1);
    //convert sepration to be instead of just angle -> angle/distance^n (n=1 for now)
    SetMaxAngleFar(_MaxAngle*(_FallOff/startseparation)); //distance^2 of 400 cm^2 taken as "standard"
    if ( _MaxAngleFar > 90. )
      _MaxAngleFar = 90.;

    //if either cluster has less than _minHits don't even try...
    if ( (hits1 < _minHits) or (hits2 < _minHits)
         or (angle1 < -360) or (angle2 < -360) )
      return false;

    if (_debug){
      std::cout << "Cluster 1:" << std::endl;
      std::cout << "\tStart: ( " << w_start1 << ", " << t_start1 << " )" << std::endl;
      std::cout << "\tAngle: " << angle1 << std::endl;
      std::cout << "\tLength: " << len1 << std::endl;
      std::cout << "\tN Hits: " << hits1 << std::endl;

      std::cout << "Cluster 2:" << std::endl;
      std::cout << "\tStart: ( " << w_start2 << ", " << t_start2 << " )" << std::endl;
      std::cout << "\tAngle: " << angle2 << std::endl;
      std::cout << "\tLength: " << len2 << std::endl;
      std::cout << "\tN Hits: " << hits2 << std::endl;

      std::cout << "Start Point Separation: " << startseparation << std::endl;
    }

    //cluster 1 needs to be long enough (i.e. good) and cluster 2 must have minimum number of hits
    double angle;
    double separation;

    if ( startseparation == 0 ) //do not prohibit merging if start point identical
      return false;

    //first calculate angle of line in 2D plane connectng the two start points
    if ( (t_start2-t_start1) == 0 )
      angle = 0.;
    else {
      double slope = (t_start2-t_start1)/(w_start2-w_start1);
      angle = atan(slope)*180./3.14;
    }

    separation = abs(angle-angle1);

    if (_debug){
      std::cout << "Angle S1--S2: " << angle << std::endl;
      std::cout << "Angle1--S2: " << separation << std::endl;
    }
    if ( ( ( ( (separation > _MaxAngle) and (separation < 180-_MaxAngle) ) or
             ( (separation > 180+_MaxAngle) and (separation< 360-_MaxAngle) ) )
           or ( ( (separation > _MaxAngleFar) and (separation < 180-_MaxAngleFar) ) or
                ( (separation > 180+_MaxAngleFar) and (separation< 360-_MaxAngleFar) ) ) )
         and (hits2 > _minHits)
         and (len1 > _MinLen)      ){
      if (_verbose) { std::cout << "Separate! cluster 1 BIG" << std::endl << std::endl; }
      return true;
    }

    //now change direction of angle so that it points from cluster 2 (big) to cluster 1 (small)
    angle += 180.;
    angle = (int)(angle) % 360;

    separation = abs(angle-angle2);
    //separation *= (400./startseparation); //distance^2 of 400 cm^2 taken as "standard"
    if (_debug){
      std::cout << "Angle S2--S1: " << angle << std::endl;
      std::cout << "Angle2--S1: " << separation << std::endl;
    }
    if ( ( ( ( (separation > _MaxAngle) and (separation < 180-_MaxAngle) ) or
             ( (separation > 180+_MaxAngle) and (separation< 360-_MaxAngle) ) )
           or ( ( (separation > _MaxAngleFar) and (separation < 180-_MaxAngleFar) ) or
                ( (separation > 180+_MaxAngleFar) and (separation< 360-_MaxAngleFar) ) ) )
         and (hits1 > _minHits)
         and (len2 > _MinLen)    ){
      if (_verbose) { std::cout << "Separate! cluster 2 BIG" << std::endl << std::endl; }
      return true;
    }

    return false;
  }

}