TMBCellContainer.cpp

Go to the documentation of this file.

#include "tmb_tree/TMBCellContainer.hpp"

#include <algorithm>

namespace {

    struct compare_cell {
        compare_cell(Char_t ieta, UChar_t iphi, UChar_t ilyr)
            : _ieta(ieta), _iphi(iphi), _ilyr(ilyr) {}
        bool operator()(const TMBCaloCell& other)
        {
            return 
                _ieta == other.ieta() &&
                _iphi == other.iphi() &&
                _ilyr == other.layer();
        }
        Char_t  _ieta;
        UChar_t _iphi;
        UChar_t _ilyr;
    };

}

TMBCellContainer::TMBCellContainer()
{
}

TMBCellContainer::~TMBCellContainer()
{
}

UInt_t TMBCellContainer::NumCells() const
{
    return (UInt_t)_cells.size();
}

// Get a Calo cell based on index
const TMBCaloCell *TMBCellContainer::GetCell(Index_t index) const
{
    return (index < _cells.size()) ? &_cells[index] : 0;
}

// Get a Calo cell based on eta, phi, layer
const TMBCaloCell *TMBCellContainer::GetCell(Char_t ieta, UChar_t iphi, UChar_t ilayer)  const
{
    return GetCell(GetIndex(ieta, iphi, ilayer));
}

    
TMBCellContainer::Index_t TMBCellContainer::GetIndex(Char_t ieta, UChar_t iphi, UChar_t ilayer) const
{
    std::vector<TMBCaloCell>::const_iterator it = 
        std::find_if(_cells.begin(), _cells.end(), compare_cell(ieta, iphi, ilayer));
    if(it != _cells.end()) {
        return (Index_t) (it - _cells.begin());
    } else {
        return INVALID_CELL;
    }
}

TMBCellContainer::Index_t 
TMBCellContainer::AddCell(Char_t ieta, UChar_t iphi, UChar_t ilayer, Float_t energy, UChar_t flags)
{
    Index_t index = GetIndex(ieta, iphi, ilayer);
    if(index == INVALID_CELL) {
        _cells.push_back(TMBCaloCell(ieta, iphi, ilayer, energy, flags));
        return (Index_t )(_cells.size() - 1);
    } else {
        // check energy ?? would have to get cell for that...
        // epsilon(energy - _cells[index].E()) < 10e-6
        return index;
    }
}

void TMBCellContainer::ClearCells()
{
    _cells.clear();
}

Bool_t TMBCellContainer::IsFolder() const 
{
    return kTRUE;
}

ClassImp(TMBCellContainer);

---