Clust2D.cxx

Go to the documentation of this file.

// $Id: Clust2D.cxx,v 1.2 2009/08/17 13:11:50 jpaley Exp $
//
// A simple algorithm to form 2D clusters from rb::CalHits
//
// \author $Author: jpaley $
//

#include "Clusterer/Clust2D.h"
#include "RecoBase/CellHit.h"
#include "RecoBase/Cluster.h"
#include "Geometry/Geometry.h"
#include "JobControl/Exception.h"

#include <iostream>
#include <map>

using namespace clust;
using namespace recobase;
using namespace std;

MODULE_DECL(Clust2D); 

Clust2D::Clust2D(const char* version) :
  jobc::Module("clust::Clust2D") 
{
  // Set the configuration version tag
  this->SetCfgVersion(version); 
}

//......................................................................

void Clust2D::Update(const cfg::Config& c)
{
  fInFolder.clear();
  fOutFolder.clear();

  try { c("InFolder").Get(fInFolder); }
  catch (cfg::Exception e) {
    string s;
    c("InFolder").Get(s);
    fInFolder.push_back(s);
  }

  try { c("OutFolder").Get(fOutFolder); }
  catch (cfg::Exception e) {
    string s;
    c("OutFolder").Get(s);
    fOutFolder.push_back(s);
  }

  assert_jobc(fOutFolder.size() == fInFolder.size(),
"Clust2D::Update error: Num. Input Folders does not equal Num. Output Folders");

}

//......................................................................

Clust2D::~Clust2D()
{

}

//......................................................................

jobc::Result Clust2D::Reco(edm::EventHandle& evt)
{

  for (unsigned int iFld=0; iFld < fInFolder.size(); ++iFld) {

    vector<const CellHit*> cell(0);
    try { evt.Cal().Get(fInFolder[iFld].c_str(),cell); }
    catch (edm::Exception e) {
      continue;
    } 

    vector<Cluster*> clust(0);
    typedef map<int, vector<const CellHit*> > CellMap;
    typedef CellMap::iterator CMItr;

    CellMap mapCell;
    for (unsigned int i=0; i<cell.size(); ++i) {
      int plane = cell[i]->Plane();
      mapCell[plane].push_back(cell[i]);
    }

    CMItr itr    = mapCell.begin();
    CMItr itrEnd = mapCell.end();

    for (; itr != itrEnd; ++itr) {
      vector<const CellHit*>& chvec = itr->second;
      Cluster* cl = new Cluster();
      for (unsigned int j=0; j<chvec.size(); ++j) {
        recobase::CellHit* cell = new recobase::CellHit(*chvec[j]);
        cl->Add(cell);
      }
      clust.push_back(cl);
    }
    
    evt.Reco().MakeFolder(fOutFolder[iFld].c_str());

    evt.Reco().PutVector(clust,fOutFolder[iFld].c_str());

    // now clean up...
    for (unsigned int i=0; i<clust.size(); ++i) delete clust[i];
  }
  
  return jobc::kPassed; // kFailed if you want to fail the event
}

---