FlatTable.cc

Go to the documentation of this file.

#include "DataFormats/NanoAOD/interface/FlatTable.h"

int nanoaod::FlatTable::columnIndex(const std::string & name) const {
    for (unsigned int i = 0, n = columns_.size(); i < n; ++i) {
        if (columns_[i].name == name) return i;
    }
    return -1;
}

void nanoaod::FlatTable::addExtension(const nanoaod::FlatTable & other) {
    if (extension() || !other.extension() || name() != other.name() || size() != other.size()) throw cms::Exception("LogicError", "Mismatch in adding extension");
    for (unsigned int i = 0, n = other.nColumns(); i < n; ++i) {
        switch(other.columnType(i)) {
            case FloatColumn:
                addColumn<float>(other.columnName(i), other.columnData<float>(i), other.columnDoc(i), other.columnType(i));
                break;
            case IntColumn:
                addColumn<int>(other.columnName(i), other.columnData<int>(i), other.columnDoc(i), other.columnType(i));
                break;
            case BoolColumn: // as UInt8
            case UInt8Column:
                addColumn<uint8_t>(other.columnName(i), other.columnData<uint8_t>(i), other.columnDoc(i), other.columnType(i));
                break;
        }
    }
}

double nanoaod::FlatTable::getAnyValue(unsigned int row, unsigned int column) const {
    if (column >= nColumns()) throw cms::Exception("LogicError","Invalid column");
    switch(columnType(column)) {
        case FloatColumn:  return *(beginData<float>(column)+row);
        case IntColumn:  return *(beginData<int>(column)+row);
        case BoolColumn:  return *(beginData<uint8_t>(column)+row);
        case UInt8Column:  return *(beginData<uint8_t>(column)+row);
    }   
    throw cms::Exception("LogicError", "Unsupported type");
}