---

ClusterKtStrategy.cxx

Go to the documentation of this file.

// ================================================
//        ClusterKtStrategy class Implementation
//
//  + PreCluster helper class
//
// ================================================
//
// Namespace Atlfast
//
#include "AtlfastAlgs/ClusterKtStrategy.h"
#include "AtlfastEvent/Cell.h"
#include "AtlfastEvent/Cluster.h"

#include "KtJet/KtLorentzVector.h"
#include "KtJet/KtEvent.h"

// Gaudi includes
#include "GaudiKernel/MsgStream.h"

//Other
#include "CLHEP/Vector/LorentzVector.h"
#include <map>
#include <vector>


namespace Atlfast {
  void ClusterKtStrategy::makeClusters( 
                                         MsgStream& log, 
                                         const localCellCollection&  storedCells,
                                         localCellCollection& unusedCells,
                                         ClusterCollection* clusters) const {
    log << MSG::DEBUG << storedCells.size() << " cells to start with"<<endreq;
    
    localCellIterator itr = storedCells.begin();
    std::vector<KtJet::KtLorentzVector> jetvec;
    std::map<KtJet::KtLorentzVector,Cell*> CellMap;
    for(; itr != storedCells.end() ; ++itr){
      KtJet::KtLorentzVector tempVec = KtJet::KtLorentzVector((*itr)->momentum());
      jetvec.push_back(tempVec);
      CellMap[tempVec] = (*itr);
    }
    log << MSG::DEBUG << "Made jetvec"<<endreq;
    KtJet::KtEvent ev(jetvec,4,2,1,1.0);
    log << MSG::DEBUG << "Made KtEvent"<<endreq;
    vector<KtJet::KtLorentzVector> jets0 = ev.getJets();
    log << MSG::DEBUG << "Got events jets "<< jets0.size() << endreq;
    std::vector<KtJet::KtLorentzVector>::const_iterator iter = jets0.begin();
    for (; iter != jets0.end() ; ++iter){
      if((*iter).et() > m_minClusterET) { 
        std::vector<KtJet::KtLorentzVector> tompVec;
        tompVec = (*iter).copyConstituents();
        //Make a vector of associated cells
        std::vector<Cell*> tempCellVec;
        std::vector<KtJet::KtLorentzVector>::const_iterator iter2  = tompVec.begin();
        for (; iter2 != tompVec.end() ; ++iter2){
          if(CellMap.find((*iter2)) == CellMap.end()){
            log << MSG::ERROR <<  "Cannot find this key" << endreq;
          }else {
            tempCellVec.push_back(CellMap[(*iter2)]);
            CellMap.erase((*iter2));
            //      log << MSG::DEBUG << "Removed cell from map" << endreq;
          }
        }
        //Make a Atlfast::KtCluster
        Cluster* clu = new Cluster(*iter,tempCellVec.begin(),tempCellVec.end());
        log << MSG::DEBUG << "Made a KtCluster" << endreq;
        clusters->push_back(clu);
      }
    }
    log << MSG::DEBUG <<  "Number of clusters Made by Kt algo " << clusters->size() << endreq;
    std::map<KtJet::KtLorentzVector,Cell*>::iterator celit = CellMap.begin();
    for (; celit != CellMap.end() ; ++celit) unusedCells.push_back((*celit).second);
    log << MSG::DEBUG <<  "Size of unusedCells vector " << unusedCells.size() << endreq;
    return;
  }
  
} // end of namespace bracket

---