TableView.h

Go to the documentation of this file.

#ifndef FWCore_SOA_TableView_h
#define FWCore_SOA_TableView_h
// -*- C++ -*-
//
// Package:     FWCore/SOA
// Class  :     TableView
//
//
// Original Author:  Chris Jones
//         Created:  Fri, 25 Aug 2017 19:31:50 GMT
//

// system include files
#include <tuple>
#include <array>

// user include files
#include "FWCore/SOA/interface/Table.h"
#include "FWCore/SOA/interface/ColumnValues.h"

// forward declarations

namespace edm {
  namespace soa {
    template <typename... Args>
    class TableView {
    public:
      using Layout = std::tuple<Args...>;
      static constexpr const size_t kNColumns = sizeof...(Args);
      using const_iterator = ConstTableItr<Args...>;

      template <typename... OArgs>
      TableView(Table<OArgs...> const& iTable) : m_size(iTable.size()) {
        fillArray(iTable, std::make_index_sequence<sizeof...(Args)>{});
      }
      TableView(unsigned int iSize, std::array<void*, sizeof...(Args)>& iArray) : m_size(iSize), m_values(iArray) {}

      TableView(unsigned int iSize, std::array<void const*, sizeof...(Args)>& iArray)
          : m_size(iSize), m_values(iArray) {}

      unsigned int size() const { return m_size; }

      template <typename U>
      typename U::type const& get(size_t iRow) const {
        return *(static_cast<typename U::type const*>(columnAddress<U>()) + iRow);
      }

      template <typename U>
      ColumnValues<typename U::type> column() const {
        return ColumnValues<typename U::type>{static_cast<typename U::type const*>(columnAddress<U>()), m_size};
      }

      const_iterator begin() const { return const_iterator{m_values}; }
      const_iterator end() const { return const_iterator{m_values, size()}; }

    private:
      std::array<void const*, sizeof...(Args)> m_values;
      unsigned int m_size;

      template <typename U>
      void const* columnAddress() const {
        return m_values[impl::GetIndex<0, U, Layout>::index];
      }

      template <typename T, size_t... I>
      void fillArray(T const& iTable, std::index_sequence<I...>) {
        ((m_values[I] = iTable.columnAddressWorkaround(
              static_cast<typename std::tuple_element<I, Layout>::type const*>(nullptr))),
         ...);
      }
    };

    template <typename T>
    struct ViewFromTable;

    template <typename... T>
    struct ViewFromTable<Table<T...>> {
      using type = TableView<T...>;
    };
    template <typename T>
    using ViewFromTable_t = typename ViewFromTable<T>::type;

  }  // namespace soa
}  // namespace edm

#endif