eTower.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
//***************************************************************************
//                           eTower.h  -  description
//                              -------------------
//     begin                : 19 02 2019
//     email                : Alan.Watson@cern.ch, jacob.julian.kempster@cern.ch
//  ***************************************************************************/
 
#include "L1CaloFEXSim/eTower.h"
#include "L1CaloFEXSim/eFEXCompression.h"
#include <cmath>
 
 
namespace LVL1 {
 
  const size_t s_nLayers = 5;
  const int s_cells[s_nLayers] = {1,4,4,1,4};
  const int s_offsets[s_nLayers] = {0,1,5,9,10};
  const int s_eFEXOverflow = 0xffff;
 
  // default constructor
  eTower::eTower():
    m_eta(0.),
    m_phi(0.),
    m_tower_id(-9999999),
    m_posneg(0)
  {
    this->clear_scIDs();
    this->clearET();
  }
  
  eTower::eTower(float eta, float phi, int id_modifier, int posneg):
    m_eta(eta),
    m_phi(phi),
    m_tower_id(id_modifier + phi + (64 * eta)),
    m_posneg(posneg)
  {
    this->clear_scIDs();
    this->clearET();
  }
  
  void eTower::clearET()
  {
    m_et.clear();
    m_et.resize(14);
    m_et_float.clear();
    m_et_float.resize(14);
    for (unsigned int i=0; i<m_et.size(); i++){
      m_et[i] = 0;
      m_et_float[i] = 0.0;
    }
 
    m_etSplits.clear();
    m_etSplits.resize(14);
    for (unsigned int i=0; i<m_etSplits.size(); i++){
      m_etSplits[i] = 0;
    }
 
  }

  void eTower::clear_scIDs()
  {
    m_scID.clear();
    m_scID.resize(14);
    m_scID_split.clear();
    m_scID_split.resize(14);
    for (unsigned int i=0; i<m_scID.size(); i++){
      m_scID[i] = Identifier();
      m_scID_split[i] = Identifier();
    }
  }
 
 
  void eTower::setPosNeg(int posneg){
    
    m_posneg = posneg;
 
    return;
 
  }

  void eTower::addET(float et, int cell)
  {
    if (cell < 0  || cell > 13){ return; }
    
    m_et_float[cell] += et;
 
    return;
 
  }
  void eTower::setET(int cell, float et, int layer, bool ignoreDisable) {
    if (cell < 0  || cell > 13){ return; }
      
    addET(et, cell);
    
    //multi linear digitisation encoding ... except in tile (indicated by passing layer=5) .. just convert to 25 MeV steps
    m_et[cell] = (layer==5) ? std::round(m_et_float[cell]/25) : eFEXCompression::decode(std::round(m_et_float[cell]),layer, ignoreDisable);
  }

  void eTower::setSCID(Identifier ID, int cell, float et, int layer, bool doenergysplit)
  {
    
    if (cell < 0  || cell > 13){ return; }
 
    if(!doenergysplit){
      
      addET(et, cell);
      
      m_scID[cell] = ID;
    
      //multi linear digitisation encoding
      int outET = eFEXCompression::decode(std::round(m_et_float[cell]),layer);
      m_et[cell] = outET;
    }
    else{
 
      float et_half = et*0.5;
      addET(et_half, cell);
      addET(et_half, cell+1);
 
      m_etSplits[cell] = 1;
      m_etSplits[cell+1] = 1;
      
      if(cell==1){
          m_scID[12] = ID;
      }
      else if(cell==3){
          m_scID[13] = ID;
      }
      
      
      m_scID_split[cell] = ID;
      m_scID_split[cell+1] = ID;
      
      m_scID.push_back(ID);
 
      m_et[cell] = eFEXCompression::decode(std::round(m_et_float[cell]),layer);
      m_et[cell+1] = eFEXCompression::decode(std::round(m_et_float[cell+1]),layer);
    }
 
    return;
 
  }

  int eTower::iEta() const {
    const int index = (m_eta * m_posneg);
    return index;
  }
  
  int eTower::iPhi() const {
    return m_phi;
  }
  
  int eTower::getET(unsigned int layer,  int cell) const {
    
    if (layer >= s_nLayers || cell < 0 || cell >= s_cells[layer]) return 0;
    
    // Return ET
    return m_et[s_offsets[layer] + cell];
    
  }

  float eTower::getET_float(unsigned int layer, int cell) const {

    if (layer >= s_nLayers || cell < 0 || cell >= s_cells[layer]) return 0;
 
    // Return ET
    return m_et_float[s_offsets[layer] + cell];
 
  }

  int eTower::getTotalET() const{
    int tmp = 0;
    for (unsigned int i=0; i<m_et.size(); i++){
      tmp += m_et[i];
    }
    // Sum saturates at 16 bits
    return std::min(tmp,s_eFEXOverflow);
  }

  float eTower::getTotalET_float() const{
 
    float tmp = 0;
    for (unsigned int i=0; i<m_et_float.size(); i++){
      tmp += m_et_float[i];
    }
 
    return tmp;
 
  }
 
  
  std::vector<int> eTower::getLayerETvec(unsigned int layer) const {
    
    std::vector<int> cells;
    
    if (layer >= s_nLayers) return cells;
    
    for (int cell = 0; cell < s_cells[layer]; ++cell) cells.push_back(m_et[s_offsets[layer] + cell]);
    
    return cells;
  }
 

  std::vector<float> eTower::getLayerETvec_float(unsigned int layer) const {

    std::vector<float> cells;

    if (layer >= s_nLayers) return cells;

    for (int cell = 0; cell < s_cells[layer]; ++cell) cells.push_back(m_et_float[s_offsets[layer] + cell]);
 
    return cells;
  }
 

  int eTower::getLayerTotalET(unsigned int layer) const {
        
    if (layer == 0){
      return m_et[0];
    }
    else if (layer == 1){
      return (m_et[1] + m_et[2] + m_et[3] + m_et[4]);
    }
    else if (layer == 2){
      return (m_et[5] + m_et[6] + m_et[7] + m_et[8]);
    }
    else if (layer == 3){
      return m_et[9];
    }
    else if (layer == 4){
      return (m_et[10] + m_et[11]);
    }
 
    
    return 0;
    
  }

  float eTower::getLayerTotalET_float(unsigned int layer) const {
 
    if (layer == 0){
      return m_et_float[0];
    }
    else if (layer == 1){
      return (m_et_float[1] + m_et_float[2] + m_et_float[3] + m_et_float[4]);
    }
    else if (layer == 2){
      return (m_et_float[5] + m_et_float[6] + m_et_float[7] + m_et_float[8]);
    }
    else if (layer == 3){
      return m_et_float[9];
    }
    else if (layer == 4){
      return (m_et_float[10] + m_et_float[11]);
    }
 
 
    return 0;
 
  }
 
  std::vector<Identifier> eTower::getLayerSCIDs(unsigned int layer) const{
 
    std::vector<Identifier> cells_in_layer;
    
    if (layer == 0){
      cells_in_layer.push_back(m_scID[0]);
    }
    else if (layer == 1){
      cells_in_layer.push_back(m_scID[1]);
      cells_in_layer.push_back(m_scID[2]);
      cells_in_layer.push_back(m_scID[3]);
      cells_in_layer.push_back(m_scID[4]);
    }
    else if (layer == 2){
      cells_in_layer.push_back(m_scID[5]);
      cells_in_layer.push_back(m_scID[6]);
      cells_in_layer.push_back(m_scID[7]);
      cells_in_layer.push_back(m_scID[8]);
    }
    else if (layer == 3){
      cells_in_layer.push_back(m_scID[9]);
    }
    else if (layer == 4){
      cells_in_layer.push_back(m_scID[10]);
      cells_in_layer.push_back(m_scID[11]);
    }
 
    return cells_in_layer;
 
  }
 
  
} // end of namespace bracket