Qjets Class Reference

#include <Qjets.h>

Public Member Functions
void	Cluster (fastjet::ClusterSequence &cs)
	Qjets (double zcut, double dcut_fctr, double exp_min, double exp_max, double rigidity, double truncation_fctr, CLHEP::HepRandomEngine *rnEngine)
void	SetRandSeed (unsigned int seed)

Private Member Functions
void	ComputeAllDistances (const std::vector< fastjet::PseudoJet > &inp)
void	computeDCut (fastjet::ClusterSequence &cs)
double	ComputeMinimumDistance ()
void	ComputeNewDistanceMeasures (fastjet::ClusterSequence &cs, unsigned int new_jet)
double	ComputeNormalization (double dmin)
double	d_ij (const fastjet::PseudoJet &v1, const fastjet::PseudoJet &v2) const
JetDistance	GetNextDistance ()
bool	JetsUnmerged (const JetDistance &jd) const
bool	JetUnmerged (int num) const
bool	Prune (JetDistance &jd, fastjet::ClusterSequence &cs)
double	Rand ()
bool	Same (const JetDistance &lhs, const JetDistance &rhs)

Private Attributes
double	_dcut
double	_dcut_fctr
std::priority_queue< JetDistance, std::vector< JetDistance >, JetDistanceCompare >	_distances
double	_exp_max
double	_exp_min
std::map< int, bool >	_merged_jets
bool	_rand_seed_set
double	_rigidity
CLHEP::HepRandomEngine *	_rnEngine
unsigned int	_seed
double	_truncation_fctr
double	_zcut
double	omega

Detailed Description

Definition at line 28 of file Qjets.h.

Constructor & Destructor Documentation

Qjets::Qjets ( double  zcut, double  dcut_fctr, double  exp_min, double  exp_max, double  rigidity, double  truncation_fctr, CLHEP::HepRandomEngine *  rnEngine  )

inline

Definition at line 53 of file Qjets.h.

References fwrapper::cs, and SurveyInfoScenario_cff::seed.

                                                                                                                                                 : _rand_seed_set(false),
    _zcut(zcut),
    _dcut(-1.),
    _dcut_fctr(dcut_fctr),
    _exp_min(exp_min),
    _exp_max(exp_max),
    _rigidity(rigidity),
    _truncation_fctr(truncation_fctr),
    _rnEngine(rnEngine)
    {};

Member Function Documentation

void Qjets::Cluster

(

fastjet::ClusterSequence &

cs

)

Definition at line 62 of file Qjets.cc.

References QjetsBaseExtras::_wij, and mps_fire::i.

Referenced by QjetsPlugin::run_clustering().

                                              {
  omega = 1.;
  QjetsBaseExtras * extras = new QjetsBaseExtras();
  computeDCut(cs);

  // Populate all the distances
  ComputeAllDistances(cs.jets());
  JetDistance jd = GetNextDistance();

  while(!_distances.empty() && jd.dij != -1.){
    if(!Prune(jd,cs)){
      //      _merged_jets.push_back(jd.j1);
      //      _merged_jets.push_back(jd.j2);
      _merged_jets[jd.j1] = true;
      _merged_jets[jd.j2] = true;

      int new_jet=-1;
      cs.plugin_record_ij_recombination(jd.j1, jd.j2, 1., new_jet);
      if(!JetUnmerged(new_jet))
    edm::LogError("QJets Clustering") << "Problem with FastJet::plugin_record_ij_recombination";
      ComputeNewDistanceMeasures(cs,new_jet);
    } else {
      double j1pt = cs.jets()[jd.j1].perp();
      double j2pt = cs.jets()[jd.j2].perp();
      if(j1pt>j2pt){
    //  _merged_jets.push_back(jd.j2);
    _merged_jets[jd.j2] = true;
    cs.plugin_record_iB_recombination(jd.j2, 1.);
      } else {
    //  _merged_jets.push_back(jd.j1);
    _merged_jets[jd.j1] = true;
    cs.plugin_record_iB_recombination(jd.j1, 1.);
      }
    }
    jd = GetNextDistance();
  } 

  extras->_wij = omega;
  cs.plugin_associate_extras(std::auto_ptr<fastjet::ClusterSequence::Extras>(extras));

  // merge remaining jets with beam
  int num_merged_final(0);
  for(unsigned int i = 0 ; i < cs.jets().size(); i++)
    if(JetUnmerged(i)){
      num_merged_final++;
      cs.plugin_record_iB_recombination(i,1.);
    }
  
  if(!(num_merged_final < 2))
    edm::LogError("QJets Clustering") << "More than 1 jet remains after reclustering";
}

void Qjets::ComputeAllDistances ( const std::vector< fastjet::PseudoJet > &  inp )

private

Definition at line 132 of file Qjets.cc.

References JetDistance::dij, mps_fire::i, JetDistance::j1, and JetDistance::j2.

                                                                  {
  for(unsigned int i = 0 ; i < inp.size()-1; i++){
    // jet-jet distances
    for(unsigned int j = i+1 ; j < inp.size(); j++){
      JetDistance jd;
      jd.j1 = i;
      jd.j2 = j;
      if(jd.j1 != jd.j2){
    jd.dij = d_ij(inp[i],inp[j]);
    _distances.push(jd);
      }
    }    
  }
}

void Qjets::computeDCut ( fastjet::ClusterSequence &  cs )

private

Definition at line 114 of file Qjets.cc.

                                                  {
  // assume all jets in cs form a single jet.  compute mass and pt
  fastjet::PseudoJet sum(0.,0.,0.,0.);
  for(vector<fastjet::PseudoJet>::const_iterator it = cs.jets().begin(); it != cs.jets().end(); it++)
    sum += (*it);
  _dcut = 2. * _dcut_fctr * sum.m()/sum.perp(); 
}

double Qjets::ComputeMinimumDistance ( )

private

void Qjets::ComputeNewDistanceMeasures ( fastjet::ClusterSequence &  cs, unsigned int  new_jet  )

private

Definition at line 147 of file Qjets.cc.

References JetDistance::dij, mps_fire::i, JetDistance::j1, and JetDistance::j2.

                                                                                       {
  // jet-jet distances
  for(unsigned int i = 0; i < cs.jets().size(); i++)
    if(JetUnmerged(i) && i != new_jet){
      JetDistance jd;
      jd.j1 = new_jet;
      jd.j2 = i;
      jd.dij = d_ij(cs.jets()[jd.j1], cs.jets()[jd.j2]);
      _distances.push(jd);
    }
}

double Qjets::ComputeNormalization ( double  dmin )

private

double Qjets::d_ij ( const fastjet::PseudoJet &  v1, const fastjet::PseudoJet &  v2  ) const

private

Definition at line 159 of file Qjets.cc.

References edm::isNotFinite(), hpstanc_transforms::max, min(), p1, p2, and funct::pow().

                                                                              {
  double p1 = v1.perp();
  double p2 = v2.perp();
  double ret = pow(min(p1,p2),_exp_min) * pow(max(p1,p2),_exp_max) * v1.squared_distance(v2);
  if(edm::isNotFinite(ret))
    edm::LogError("QJets Clustering") << "d_ij is not finite";
  return ret; 
}

JetDistance Qjets::GetNextDistance ( )

private

Definition at line 18 of file Qjets.cc.

References JetDistance::dij, JetChargeProducer_cfi::exp, JetDistance::j1, JetDistance::j2, rand(), and mps_merge::weight.

                                  {
  vector< pair<JetDistance, double> > popped_distances;
  double norm(0.);
  JetDistance ret;
  ret.j1 = -1;
  ret.j2 = -1;
  ret.dij = -1.;
  bool dmin_set(false);
  double dmin(0.);

  while(!_distances.empty()){
    JetDistance dist = _distances.top();
    _distances.pop();
    if(JetsUnmerged(dist)){
      if(!dmin_set){
    dmin = dist.dij;
    dmin_set = true;
      }
      double weight = exp(-_rigidity * (dist.dij-dmin) /dmin);
      popped_distances.push_back(make_pair(dist,weight));        
      norm += weight; 
      if(weight/norm < _truncation_fctr)
    break;            
    }
  }
  
  double rand(Rand()), tot_weight(0.);
  for(vector<pair<JetDistance, double> >::iterator it = popped_distances.begin(); it != popped_distances.end(); it++){
    tot_weight += (*it).second/norm;
    if(tot_weight >= rand){
      ret = (*it).first;
      omega *= ((*it).second);
      break;
    }
  }
  
  // repopulate in reverse (maybe quicker?)
  for(vector<pair<JetDistance, double> >::reverse_iterator it = popped_distances.rbegin(); it != popped_distances.rend(); it++)
    if(JetsUnmerged((*it).first))
      _distances.push((*it).first);

  return ret;
}

bool Qjets::JetsUnmerged ( const JetDistance &  jd ) const

private

Definition at line 14 of file Qjets.cc.

References JetDistance::j1, and JetDistance::j2.

                                                   {
  return JetUnmerged(jd.j1) && JetUnmerged(jd.j2);
}

bool Qjets::JetUnmerged ( int  num ) const

private

Definition at line 10 of file Qjets.cc.

                                    {
  return _merged_jets.find(num) == _merged_jets.end();
}

bool Qjets::Prune ( JetDistance &  jd, fastjet::ClusterSequence &  cs  )

private

Definition at line 122 of file Qjets.cc.

References JetDistance::j1, JetDistance::j2, min(), CAHitQuadrupletGenerator_cfi::pt1, and CAHitQuadrupletGenerator_cfi::pt2.

                                                            {
  double pt1 = cs.jets()[jd.j1].perp();
  double pt2 = cs.jets()[jd.j2].perp();
  fastjet::PseudoJet sum_jet = cs.jets()[jd.j1]+cs.jets()[jd.j2];
  double sum_pt = sum_jet.perp();
  double z = min(pt1,pt2)/sum_pt;
  double d2 = cs.jets()[jd.j1].plain_distance(cs.jets()[jd.j2]);
  return (d2 > _dcut*_dcut) && (z < _zcut);
}

double Qjets::Rand ( )

private

Definition at line 168 of file Qjets.cc.

                  {
  double ret = 0.;
  //if(_rand_seed_set)
  //  ret = rand_r(&_seed)/(double)RAND_MAX;
  //else 
  //ret = rand()/(double)RAND_MAX;
  ret = _rnEngine->flat();
  return ret;
}

bool Qjets::Same ( const JetDistance &  lhs, const JetDistance &  rhs  )

private

void Qjets::SetRandSeed

(

unsigned int

seed

)

Definition at line 5 of file Qjets.cc.

References SurveyInfoScenario_cff::seed.

Referenced by QjetsPlugin::run_clustering().

                                        {
  _rand_seed_set = true;
  _seed = seed;
}

Member Data Documentation

double Qjets::_dcut

private

Definition at line 33 of file Qjets.h.

double Qjets::_dcut_fctr

private

Definition at line 33 of file Qjets.h.

std::priority_queue<JetDistance, std::vector<JetDistance>, JetDistanceCompare> Qjets::_distances

private

Definition at line 35 of file Qjets.h.

double Qjets::_exp_max

private

Definition at line 33 of file Qjets.h.

double Qjets::_exp_min

private

Definition at line 33 of file Qjets.h.

std::map<int,bool> Qjets::_merged_jets

private

Definition at line 34 of file Qjets.h.

bool Qjets::_rand_seed_set

private

Definition at line 31 of file Qjets.h.

double Qjets::_rigidity

private

Definition at line 33 of file Qjets.h.

CLHEP::HepRandomEngine* Qjets::_rnEngine

private

Definition at line 36 of file Qjets.h.

unsigned int Qjets::_seed

private

Definition at line 32 of file Qjets.h.

double Qjets::_truncation_fctr

private

Definition at line 33 of file Qjets.h.

double Qjets::_zcut

private

Definition at line 33 of file Qjets.h.

double Qjets::omega

private

Definition at line 30 of file Qjets.h.