EPileupMap.cxx

Go to the documentation of this file.

#include "AtlfastAlgs/EPileupMap.h"
#include "AtlfastEvent/EPileupDeposit.h"
#include "AtlfastUtils/NumberToString.h"
#include <iostream>
#include <fstream>

#include "CLHEP/Units/SystemOfUnits.h"
#include "PathResolver/PathResolver.h"

namespace Atlfast{

  EPileupMap::EPileupMap(int luminosity){
    m_emPileupMap = new std::vector<EPileupDeposit*>;
    m_hadPileupMap = new std::vector<EPileupDeposit*>;
    
    if (luminosity == 1){
      m_nFiles = 10401;
      m_fileDir = "allfiles_ll/";
    }else if (luminosity == 2){
      m_nFiles = 10830;
      m_fileDir = "allfiles_hl/";
    }
    this->makeCellMap();
  }
  EPileupMap::~EPileupMap(){
    this->emptyMap();
    delete m_emPileupMap;
    delete m_hadPileupMap;
    delete m_phiMap;
    delete m_etaMap;
  }
  
  
  void EPileupMap::fillMap(){
    this->emptyMap();
    this->setInputFile();

    std::ifstream input;
    input.open( m_pileupFile.c_str());
    if (input) {
      int code;
      double energy;
      while (input >> code && input >> energy){
        if (code > 0){ 
          std::string type;
          int intType = code/100000;
          int eta = (code - intType*100000)/100;
          int phi =  code - intType*100000 - eta*100;
          double et = energy*GeV;

          if(intType == 1){
            m_emPileupMap->
              push_back(
                        new EPileupDeposit(
                                           (m_etaMap->find(eta))->second,
                                           (m_phiMap->find(phi))->second,
                                           et)
                        );
          }//if(intType == 1)
          
          if(intType == 2){
            m_hadPileupMap->
              push_back(
                        new EPileupDeposit(
                                           (m_etaMap->find(eta))->second,
                                           (m_phiMap->find(phi))->second,
                                           et
                                           )
                        );
          }//if(intType == 2)
          
        }
        
      }
    }else{
      std::cout << "ERROR! COULD NOT OPEN EPILEUP FILE " 
                << m_pileupFile << std::endl;
    } 
    return;
  }
  void EPileupMap::emptyMap(){
    if (m_emPileupMap->size()){
      std::vector<EPileupDeposit*>::iterator itr=m_emPileupMap->begin();
      for (; itr != m_emPileupMap->end();++itr){
        delete (*itr);
      }
    m_emPileupMap->clear();
    }
    if (m_hadPileupMap->size()){
      std::vector<EPileupDeposit*>::iterator itr=m_hadPileupMap->begin();
      for (; itr != m_hadPileupMap->end();++itr){
        delete (*itr);
      }
    m_hadPileupMap->clear();
    }
    return;
  }
  void EPileupMap::makeCellMap(){
    m_phiMap = new std::map<int,double> ;
    m_etaMap = new std::map<int,double> ;
    int EtaCells = 100;
    double EtaRange = 10;
    int PhiCells = 64;
    double PhiRange = 2*3.1415;
    for (int i = 1 ; i <= EtaCells; ++i){
      m_etaMap->
        insert(std::pair<int,double>(i,
                                     (i-(EtaCells/2))/(EtaCells/EtaRange)-
                                      EtaRange/(EtaCells*2.0) 
                                     )
               );
    }
    for (int i = 1 ; i <= PhiCells; ++i){
      m_phiMap->
        insert(std::pair<int,double>(i,
                                     (i-(PhiCells/2))/(PhiCells/PhiRange)-
                                     PhiRange/(PhiCells*2.0)
                                     )
               ) ;
    }
    return;
  }


  void EPileupMap::setInputFile(){
    double max = RAND_MAX; 
    int i = (int)(rand()/max*m_nFiles);
    if(i < 10){
      m_pileupFile = m_fileDir + "part0000" + numberToString(i);
     } else if(i < 100){
       m_pileupFile = m_fileDir + "part000" + numberToString(i);
     } else if(i < 1000){
       m_pileupFile = m_fileDir + "part00" + numberToString(i);
     } else if(i < 10000){
       m_pileupFile = m_fileDir + "part0" + numberToString(i);
     } else {
       m_pileupFile = m_fileDir + "part" + numberToString(i);
     }

    m_pileupFile = PathResolver::find_file(m_pileupFile, "DATAPATH");

    return;
  }

}

---