CbmRichRingFinderHough.cxx

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CbmRichRingFinderHough.cxx

// --------------------------------------------------------------------------------------
// CbmRichRingFinderHough source file
// Base class for ring finders based on on HT method
// Implementation: Semen Lebedev (s.lebedev@gsi.de)

#include "CbmRichRingFinderHough.h"

#include "tbb/task_scheduler_init.h"
#include "tbb/task.h"
#include "tbb/tick_count.h"
#include "tbb/parallel_invoke.h"


#include <iostream>
#include <cmath>
#include <set>
#include <algorithm>
#include <fstream>

using std::cout;
using std::endl;
using std::vector;

class FinderTask{
  CbmRichRingFinderHoughImpl* fHTImpl;
public:
  FinderTask(CbmRichRingFinderHoughImpl* HTImpl) {
    fHTImpl = HTImpl;
  }
  void operator()() const {
    fHTImpl->DoFind();
  }
};

// -----   Standard constructor   ------------------------------------------
CbmRichRingFinderHough::CbmRichRingFinderHough  ( )
{
    fRingCount = 0;
    fNEvent = 0;

  fHTImpl1 = new CbmRichRingFinderHoughSimd();
  fHTImpl1->Init();

  fHTImpl2 = new CbmRichRingFinderHoughSimd();
  fHTImpl2->Init();

  tbb::task_scheduler_init init;
}

CbmRichRingFinderHough::~CbmRichRingFinderHough()
{
  delete fHTImpl1;
  delete fHTImpl2;
}

int CbmRichRingFinderHough::DoFind(const vector<CbmRichHoughHit>& data)
{
  fNEvent++;
  std::vector<CbmRichHoughHit> UpH;
  std::vector<CbmRichHoughHit> DownH;
    fRingCount = 0;

  UpH.reserve(data.size()/2);
  DownH.reserve(data.size()/2);

  for(int iHit = 0; iHit < data.size(); iHit++) {
    CbmRichHoughHit hit = data[iHit];
    if (hit.fHit.fY >= 0)
      UpH.push_back(hit);
    else
      DownH.push_back(hit);
  }

  fHTImpl1->SetData(UpH);
  fHTImpl1->DoFind();
  int nofFoundRings = fHTImpl1->GetFoundRings().size();
  UpH.clear();

  fHTImpl1->SetData(DownH);
  fHTImpl1->DoFind();
  nofFoundRings += fHTImpl1->GetFoundRings().size();
  DownH.clear();

  cout << "NofFoundRings = " << nofFoundRings << endl;
}

int CbmRichRingFinderHough::DoFindParallel(const vector<CbmRichHoughHit>& data)
{
  fNEvent++;
  std::vector<CbmRichHoughHit> UpH;
  std::vector<CbmRichHoughHit> DownH;
    fRingCount = 0;

  UpH.reserve(data.size()/2);
  DownH.reserve(data.size()/2);

  for(int iHit = 0; iHit < data.size(); iHit++) {
    CbmRichHoughHit hit = data[iHit];
    if (hit.fHit.fY >= 0)
      UpH.push_back(hit);
    else
      DownH.push_back(hit);
  }

  fHTImpl1->SetData(UpH);
  fHTImpl2->SetData(DownH);

  //fHTImpl1->DoFind();
  //fHTImpl2->DoFind();

  FinderTask a(fHTImpl1);
  FinderTask b(fHTImpl2);
  tbb::parallel_invoke(a, b);

  int nofFoundRings = fHTImpl1->GetFoundRings().size();
  nofFoundRings += fHTImpl2->GetFoundRings().size();

  UpH.clear();
  DownH.clear();

  cout << "NofFoundRings = " << nofFoundRings << endl;
}