CBAlgoMergeTinyWithBig.cxx

Go to the documentation of this file.

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

#include <cmath>

namespace cmtool {

  //-------------------------------------------------------
  CBAlgoMergeTinyWithBig::CBAlgoMergeTinyWithBig() : CBoolAlgoBase()
  //-------------------------------------------------------
  {
    SetMinHitsBig(50);
    SetMaxHitsBig(99999);
    SetMinHitsSmall(0);
    SetMaxHitsSmall(15);
    SetMinDistSquared(0);

    SetDebug(false);
  }

  //-----------------------------
  void CBAlgoMergeTinyWithBig::Reset()
  //-----------------------------
  {}

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

    if (_debug)
      std::cout << "MergeTinyWithBig. One cluster has " << cluster1.GetNHits()
                << " hits, the other has " << cluster2.GetNHits() << " hits." << std::endl;

    bool is_1_small = false;
    bool is_2_small = false;
    bool is_1_big = false;
    bool is_2_big = false;

    //if the first cluster counts as "small"
    if (cluster1.GetNHits() > _min_hits_small && cluster1.GetNHits() < _max_hits_small)
      is_1_small = true;
    //if the second cluster counts as "small"
    if (cluster2.GetNHits() > _min_hits_small && cluster2.GetNHits() < _max_hits_small)
      is_2_small = true;
    if (cluster1.GetNHits() > _min_hits_big && cluster1.GetNHits() < _max_hits_big) is_1_big = true;
    if (cluster2.GetNHits() > _min_hits_big && cluster2.GetNHits() < _max_hits_big) is_2_big = true;

    if (_debug)
      std::cout << "is_1_small, is_1_big, is_2_small, is_2_big are: " << is_1_small << ", "
                << is_1_big << ", " << is_2_small << ", " << is_2_big << std::endl;

    //if neither of the clusters is small don't merge
    if (!is_1_small && !is_2_small) return false;
    //if neither of the clusters is big, don't merge
    if (!is_1_big && !is_2_big) return false;
    //if both are small, don't merge
    if (is_1_small && is_2_small) return false;
    //if both are big, don't merge
    if (is_1_big && is_2_big) return false;

    if (_debug) std::cout << "Looks like one of them is big, and one is small." << std::endl;
    //god this code is ugly

    //now we know which one of them is big & the other is small.

    //loop over the points on the first polygon and calculate
    //distance to each point on the second polygon
    //if any two points are close enough to each other,
    //merge the two clusters

    unsigned int npoints1 = cluster1.GetParams().PolyObject.Size();
    unsigned int npoints2 = cluster2.GetParams().PolyObject.Size();
    //loop over points on first polygon
    for (unsigned int i = 0; i < npoints1; ++i) {
      float pt1w = cluster1.GetParams().PolyObject.Point(i).first;
      float pt1t = cluster1.GetParams().PolyObject.Point(i).second;
      //loop over points on second polygon
      for (unsigned int j = 0; j < npoints2; ++j) {
        float pt2w = cluster2.GetParams().PolyObject.Point(j).first;
        float pt2t = cluster2.GetParams().PolyObject.Point(j).second;
        double distsqrd = pow(pt2w - pt1w, 2) + pow(pt2t - pt1t, 2);

        if (_debug) { std::cout << "two polygon points dist2 is " << distsqrd << std::endl; }
        if (distsqrd < _dist_sqrd_cut) return true;
      }
    }

    return false;
  }

  //------------------------------
  void CBAlgoMergeTinyWithBig::Report()
  //------------------------------
  {}

}