---

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>
namespace Atlfast{

  EPileupMap::EPileupMap(int luminosity){
    m_emPileupMap = new vector<EPileupDeposit*>;
    m_hadPileupMap = new 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();

    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/1000.0;

          if(intType == 1){
            m_emPileupMap->push_back(new EPileupDeposit((m_etaMap->find(eta))->second,(m_phiMap->find(phi))->second,et));
          }
          if(intType == 2){
            m_hadPileupMap->push_back(new EPileupDeposit((m_etaMap->find(eta))->second,(m_phiMap->find(phi))->second,et));
          }

        }
        
      }
    }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(pair<int,double>(i,((((i)-(EtaCells/2))/(EtaCells/EtaRange)) - (EtaRange/(EtaCells*2.0)))));
    }
    for (int i = 1 ; i <= PhiCells; ++i){
      m_phiMap->insert(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);
     }
    return;
  }

}

---