KernelMatrix.h

#ifndef _theplu_yat_utility_kernel_matrix_
#define _theplu_yat_utility_kernel_matrix_

// $Id: KernelMatrix.h 3190 2014-03-30 07:47:40Z peter $

/*
  Copyright (C) 2014 Peter Johansson

  This file is part of the yat library, http://dev.thep.lu.se/yat

  The yat library 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.

  The yat library 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 yat. If not, see <http://www.gnu.org/licenses/>.
*/

#include "Container2DIterator.h"
#include "StrideIterator.h"

namespace theplu {
namespace yat {
namespace utility {

  template <class RandomAccessRange1, class RandomAccessRange2,
            class BinaryFunctor>
  class KernelMatrix
  {
  public:
    typedef typename BinaryFunctor::result_type value_type;

    typedef value_type const_reference;

    typedef Container2DIterator<const KernelMatrix, value_type, const_reference>
    const_iterator;

    typedef StrideIterator<const_iterator> const_column_iterator;

    typedef const_iterator const_row_iterator;

    KernelMatrix(const RandomAccessRange1& r1, const RandomAccessRange2& r2)
      : range1_(r1), range2_(r2) {}

    KernelMatrix(const RandomAccessRange1& r1, const RandomAccessRange2& r2,
                 const BinaryFunctor& bf)
      : range1_(r1), range2_(r2), func_(bf) {}

    const_iterator begin(void) const { return const_iterator(*this, 0, 0); }


    const_column_iterator begin_column(size_t column) const
    {
      return const_column_iterator(const_iterator(*this, 0, column),
                                   this->columns());
    }


    const_row_iterator begin_row(size_t row) const
    { return const_row_iterator(*this, row, 0); }


    const_iterator end(void) const
    { return const_iterator(*this, rows(), 0); }


    const_column_iterator end_column(size_t column) const
    {
      return const_column_iterator(const_iterator(*this, this->rows(), column),
                                   this->columns());
    }


    const_row_iterator end_row(size_t row) const
    { return const_row_iterator(*this, row, this->columns()); }


    typename RandomAccessRange2::size_type columns(void) const
    { return range2_.size(); }


    typename RandomAccessRange1::size_type rows(void) const
    { return range1_.size(); }


    const_reference operator()(size_t row, size_t column) const
    { return func_(range1_[row], range2_[column]); }

  private:
    RandomAccessRange1 range1_;
    RandomAccessRange2 range2_;
    BinaryFunctor func_;
  };

}}} // of namespace utility, yat, and theplu

#endif