gridtransform.hpp

/*
 * gridtransform.hpp
 *
 * Created on: 24 Sep 2012
 * Author: Holger Schmitz
 * Email: holger@notjustphysics.com
 *
 * Copyright 2012 Holger Schmitz
 *
 * This file is part of Schnek.
 *
 * Schnek is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Schnek is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Schnek.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

#ifndef SCHNEK_GRIDTRANSFORM_HPP_
#define SCHNEK_GRIDTRANSFORM_HPP_

#include "grid.hpp"
#include "range.hpp"

namespace schnek {

template<
  typename T,
  int rank,
  class BaseGrid,
  typename Transformation
>
class GridTransformStorage {
  public:
    typedef Array<int,rank> IndexType;
    typedef BaseGrid BaseGridType;
    typedef Transformation TransformationType;
  protected:
    BaseGridType *baseGrid;
    Transformation transformation;
  public:

    class const_storage_iterator {
      protected:
        typedef typename BaseGridType::const_storage_iterator BaseIter;
        BaseIter baseIter;
        const_storage_iterator(BaseIter baseIter_)
          : baseIter(baseIter) {}

        friend class GridTransformStorage;

      public:
        T operator*()
        { return transformation(*baseIter);}

        const_storage_iterator &operator++()
        {
          ++baseIter;
          return *this;
        }

        bool operator==(const const_storage_iterator &SI)
        { return baseIter==SI.baseIter; }

        bool operator!=(const const_storage_iterator &SI)
        { return baseIter!=SI.baseIter; }
    };

    GridTransformStorage();

    T&  get(const IndexType &index)
    {
      static T result;
      result = transformation(baseGrid->get(index));
      return result;
    }

    T get(const IndexType &index) const
    {
      return transformation(baseGrid->get(index));
    }
    const IndexType& getLo() const { return baseGrid->getLo(); }
    const IndexType& getHi() const { return baseGrid->getHi(); }
    const IndexType& getDims() const { return baseGrid->getDims(); }

    int getLo(int k) const { return baseGrid->getLo(k); }
    int getHi(int k) const { return baseGrid->getHi(k); }
    int getDims(int k) const { return baseGrid->getDims(k); }

    const_storage_iterator begin() { return const_storage_iterator(baseGrid->begin()); }
    const_storage_iterator end() { return const_storage_iterator(baseGrid->end()); }

    const_storage_iterator cbegin() const { return const_storage_iterator(baseGrid->cbegin()); }
    const_storage_iterator cend() const { return const_storage_iterator(baseGrid->cend()); }

    void setBaseGrid(BaseGridType &baseGrid_)
    {
      baseGrid = &baseGrid_;
    }

    Transformation getTransformation() const
    {
      return transformation;
    }

    void setTransformation(const Transformation &transformation)
    {
      this->transformation = transformation;
    }
};

template<
  class BaseGrid,
  typename Transformation,
  template<int> class CheckingPolicy = GridNoArgCheck
>
class GridTransform
  : public GridBase
    <
      typename Transformation::value_type,
      BaseGrid::Rank,
      CheckingPolicy<BaseGrid::Rank>,
      GridTransformStorage<
        typename BaseGrid::value_type,
        BaseGrid::Rank,
        BaseGrid,
        Transformation
      >
    >
{
  private:
    typedef GridBase
        <
          typename Transformation::value_type,
          BaseGrid::Rank,
          CheckingPolicy<BaseGrid::Rank>,
          GridTransformStorage<
            typename BaseGrid::value_type,
            BaseGrid::Rank,
            BaseGrid,
            Transformation
          >
        > ParentType;

  public:
    enum {Rank = BaseGrid::Rank};
    typedef typename Transformation::value_type value_type;
    typedef typename BaseGrid::IndexType IndexType;
    typedef BaseGrid BaseGridType;
    GridTransform();

    GridTransform(BaseGridType &baseGrid_);
};

template<typename SrcType, typename DestType>
class TypeCastTransform
{
  public:
    typedef DestType value_type;
  public:
    DestType operator()(SrcType x) {return (DestType)x; }
};

} // namespace schnek

#include "gridtransform.t"

#endif // SCHNEK_SUBGRID_HPP_