CBAlgoStartTrack.cxx

Go to the documentation of this file.

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

namespace cmtool {

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

    SetMinHits(10);
    SetMinWidth(1.00000);
    SetMinOpeningAngle(0.15000);
    SetMinEP(0.99000);
    SetDebug(false);

  }

  //-----------------------------
  void CBAlgoStartTrack::Reset()
  //-----------------------------
  {

  }

  //------------------------------------------------------------------------------------------
  //void CBAlgoStartTrack::EventBegin(const std::vector<cluster::ClusterParamsAlg> &clusters)
  //------------------------------------------------------------------------------------------
  //{
  //
  //}

  //-------------------------------
  //void CBAlgoStartTrack::EventEnd()
  //-------------------------------
  //{
  //
  //}

  //-----------------------------------------------------------------------------------------------
  //void CBAlgoStartTrack::IterationBegin(const std::vector<cluster::ClusterParamsAlg> &clusters)
  //-----------------------------------------------------------------------------------------------
  //{
  //
  //}

  //------------------------------------
  //void CBAlgoStartTrack::IterationEnd()
  //------------------------------------
  //{
  //
  //}

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

    if(cluster1.GetHitVector().size() < _min_hits ||
       cluster2.GetHitVector().size() < _min_hits )
      return false;

    bool FirstTrackSecondBlob = false;
    bool SecondTrackFirstBlob = false;

    if(IsStartTrack(cluster1) && IsOverlappingBlob(cluster2))
      FirstTrackSecondBlob = true;
    if(IsStartTrack(cluster2) && IsOverlappingBlob(cluster1))
      SecondTrackFirstBlob = true;

    if(_debug){
      if(FirstTrackSecondBlob){
        std::cout<<"*********************************************"<<std::endl;
        std::cout<<"First track second blob! First has params:"<<std::endl;
        std::cout<<"Angle 2D = "<<cluster1.GetParams().angle_2d<<std::endl;
        std::cout<<"Opening Angle = "<<cluster1.GetParams().opening_angle<<std::endl;
        std::cout<<"Sum charge = "<<cluster1.GetParams().sum_charge<<std::endl;
        std::cout<<"Length = "<<cluster1.GetParams().length<<std::endl;
        std::cout<<"Width = "<<cluster1.GetParams().width<<std::endl;
        std::cout<<"N_Hits = "<<cluster1.GetHitVector().size()<<std::endl;
        std::cout<<"eigenvalue_principal = "<<cluster1.GetParams().eigenvalue_principal<<std::endl;
        std::cout<<"Second has params:"<<std::endl;
        std::cout<<"Angle 2D = "<<cluster2.GetParams().angle_2d<<std::endl;
        std::cout<<"Opening Angle = "<<cluster2.GetParams().opening_angle<<std::endl;
        std::cout<<"Sum charge = "<<cluster2.GetParams().sum_charge<<std::endl;
        std::cout<<"Length = "<<cluster2.GetParams().length<<std::endl;
        std::cout<<"Width = "<<cluster2.GetParams().width<<std::endl;
        std::cout<<"N_Hits = "<<cluster2.GetHitVector().size()<<std::endl;
        std::cout<<"eigenvalue_principal = "<<cluster2.GetParams().eigenvalue_principal<<std::endl;

      }


    }

    //if cluster1 looks like a track and cluster2 looks like a blob
    if (FirstTrackSecondBlob)
      {
        std::pair<float,float> trackEndPoint =
          std::pair<float,float>( cluster1.GetParams().end_point.w, cluster1.GetParams().end_point.t );

        //if the tracklike cluster's endpoint is inside polygon of blob
        if(cluster2.GetParams().PolyObject.PointInside(trackEndPoint))
          return true;
        else
          return false;
      }

    //if cluster2 looks like a track and cluster1 looks like a blob
    else if (SecondTrackFirstBlob)
      {
        std::pair<float,float> trackEndPoint =
          std::pair<float,float>( cluster2.GetParams().end_point.w, cluster2.GetParams().end_point.t );

        //if the tracklike cluster's endpoint is inside polygon of blob
        if(cluster1.GetParams().PolyObject.PointInside(trackEndPoint))
          return true;
        else
          return false;
      }

    else
      return false;

    //the reason for not just using (one track one blob) and
    //polyoverlap for this is that
    //this was merging BNB events where the track is clearly in a different
    //direction than the blob

    //this will fail when the tracklike cluster is reconstructed backwards
    //(looks like it's a track going into a blob, rather than exiting)


  }

  //------------------------------
  void CBAlgoStartTrack::Report()
  //------------------------------
  {

  }


  bool CBAlgoStartTrack::IsStartTrack(const ::cluster::ClusterParamsAlg &cluster)
  {
    if(cluster.GetParams().eigenvalue_principal > _min_EP)
      return true;
    else
      return false;

  }

  bool CBAlgoStartTrack::IsOverlappingBlob(const ::cluster::ClusterParamsAlg &cluster)
  {
    if(cluster.GetParams().width > _min_width &&
       cluster.GetParams().opening_angle > _min_opening_angle &&
       cluster.GetParams().eigenvalue_principal < _min_EP)
      return true;
    else
      return false;
  }
}