IndexProcessor.h

Go to the documentation of this file.

#ifndef _tls_ParametricMuonProfileNS_IndexProcessor_h_
#define _tls_ParametricMuonProfileNS_IndexProcessor_h_

#include <vector>
#include <iostream>

namespace ParametricMuonProfileNS {

  template<size_t ADimension>
  class IndexProcessor
  {
  public:
    IndexProcessor(const std::vector<size_t>& extent)
    {
      fStride[ADimension - 1] = extent[ADimension - 1];
      for (unsigned char i = 1; i < ADimension; ++i)
        fStride[ADimension - 1 - i] = extent[ADimension - 1 - i]
                                     *fStride[ADimension - i];
    }

    inline
    void
    Linearize(const size_t* index, size_t& j)
      const
    {
      j = index[ADimension - 1];
      for (size_t i = 0; i < ADimension - 1; ++i)
        j += index[i] * fStride[i + 1];
    }

    inline
    void
    Delinearize(const size_t j, size_t* index)
      const
    {
      size_t l = j;
      for (size_t i = 0; i < ADimension - 1; ++i)
      {
        index[i] = l / fStride[i + 1];
        l %= fStride[i + 1];
      }
      index[ADimension - 1] = l;
    }

  protected:
    size_t fStride[ADimension];
  };

}

#endif