SimG4HcalHitJetFinder.cc

Go to the documentation of this file.

// File: SimG4HcalHitJetFinder.cc
// Description: Jet finder class for SimG4HcalValidation
#include "Validation/HcalHits/interface/SimG4HcalHitJetFinder.h"
#include "DataFormats/HcalDetId/interface/HcalSubdetector.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include <algorithm>
#include <iostream>
#include <cmath>

SimG4HcalHitJetFinder::SimG4HcalHitJetFinder(double cone): jetcone(cone) {}

SimG4HcalHitJetFinder::~SimG4HcalHitJetFinder() {
  edm::LogInfo("ValidHcal") << "SimG4HcalHitJetFinder:: Deleting";
}

void SimG4HcalHitJetFinder::setCone(double cone) { 
  jetcone = cone;
}   

void SimG4HcalHitJetFinder::setInput(std::vector<CaloHit>* hhit) { 
  input   = * hhit;
}

std::vector<SimG4HcalHitCluster>* SimG4HcalHitJetFinder::getClusters(bool hcal_only){
  clusvector.erase(clusvector.begin(),clusvector.end()); 
  if (input.size() == 0) { 
    return &clusvector;
  }

  std::vector<CaloHit>::iterator itr;
  for (itr = input.begin(); itr != input.end(); itr++) {
    LogDebug("ValidHcal") << "HcalHitJetFinder::getClusters_1 - input :  e " 
              << itr->e() << "  eta " << itr->eta() << "  phi " 
              << itr->phi() << "  subdet " << itr->det();
  }

  sort(input.begin(),input.end()); // sort input in descending order
 
  for (itr = input.begin(); itr != input.end(); itr++) {
    LogDebug("ValidHcal") << "HcalHitJetFinder::getClusters_2 - input :  e " 
              << itr->e() << "  eta " << itr->eta() << "  phi " 
              << itr->phi() << "  subdet " << itr->det();
  }

  std::vector<SimG4HcalHitCluster> temp;   // dummy container for clusters

  //  first input hit -> first cluster

  CaloHit             hit;
  SimG4HcalHitCluster cluster;

  std::vector<CaloHit>::iterator itr_hits;

  int j, first_seed = 0;
  for (j=0, itr_hits = input.begin(); itr_hits != input.end();
       j++, itr_hits++) {
    int h_type = itr_hits->det(); //if desired HCAL hits (only) clusterfinding
    if (((h_type == static_cast<int>(HcalBarrel) ||
      h_type == static_cast<int>(HcalEndcap) ||
      h_type == static_cast<int>(HcalForward)) && hcal_only) || 
    (!hcal_only)) {
      cluster += input[j];
      LogDebug("ValidHcal") << "HcalHitJetFinder:: First seed hit "
                << "..................\n" << (*itr_hits);
      first_seed = j;
      break;
    }
  }
  
  temp.push_back(cluster);
  
  std::vector<SimG4HcalHitCluster>::iterator itr_clus;

  for (j=0, itr_hits = input.begin(); itr_hits != input.end();
       j++, itr_hits++) {
    int h_type = itr_hits->det(); //if desired HCAL hits (only) clusterfinding
    if ((((h_type == static_cast<int>(HcalBarrel) ||
       h_type == static_cast<int>(HcalEndcap) ||
       h_type == static_cast<int>(HcalForward)) && hcal_only) || 
     (!hcal_only)) && (j != first_seed)) {
      LogDebug("ValidHcal") << "HcalHitJetFinder:: ........... Consider hit"
                << " ..................\n" << (*itr_hits);
      
      int incl = 0; // if the hit is included in one of clusters
      
      int iclus;  
      for (itr_clus = temp.begin(), iclus = 0; itr_clus != temp.end();
       itr_clus++, iclus++) { 
    
    LogDebug("ValidHcal") << "HcalHitJetFinder::=======> Cluster " 
                  << iclus << "\n" << (*itr_clus);
    
    double d = rDist(&(*itr_clus), &(*itr_hits));
    if (d < jetcone) {
      LogDebug("ValidHcal") << "HcalHitJetFinder:: -> associated ... ";
      temp[iclus] += *itr_hits;
      incl = 1;
      break;  
    }
      }
      
      // to here jumps "break"
      if (incl == 0) {
    SimG4HcalHitCluster cl;
    cl += *itr_hits;
    temp.push_back(cl);
    LogDebug("ValidHcal") << "HcalHitJetFinder:: ************ NEW CLUSTER"
                  << " !\n" << cl;
      }
    }
  }

  clusvector = temp;
  return &clusvector;
}

double SimG4HcalHitJetFinder::rDist(const SimG4HcalHitCluster* cluster, 
                    const CaloHit* hit) const {

  double etac = cluster->eta();
  double phic = cluster->phi();
  
  double etah = hit->eta();
  double phih = hit->phi();

  return rDist(etac, phic, etah, phih);
}


double SimG4HcalHitJetFinder::rDist(const double etac,const double phic,
                    const double etah,const double phih) const{

  double delta_eta = etac - etah;
  double delta_phi = phic - phih;

  if (phic < phih)      delta_phi = phih - phic;
  if (delta_phi > M_PI) delta_phi = 2*M_PI - delta_phi;

  double tmp = sqrt(delta_eta * delta_eta + delta_phi * delta_phi);

  LogDebug("ValidHcal") << "HcalHitJetFinder::rDist:\n"
            << " Clus. eta, phi = " << etac << " " << phic << "\n"
            << " hit   eta, phi = " << etah << " " << phih 
            << " rDist = " << tmp;

  return tmp;
}