VectorComponentAccessor.hpp

// This file is part of the imaging2 class library.
//
// University of Innsbruck, Infmath Imaging, 2009.
// http://infmath.uibk.ac.at
//
// All rights reserved.


#ifndef VECTORCOMPONENTACCESSOR_H
#define VECTORCOMPONENTACCESSOR_H

#include <core/imaging2.hpp>

namespace imaging
{

  template <class vector_image_t>
  class VectorComponentAccessor : public ImageAccessorInterface<vector_image_t, typename vector_image_t::data_t::data_t>
  {
  public:
    typedef typename vector_image_t::data_t::data_t data_t;
    static const size_t dimension =
      ImageAccessorInterface<vector_image_t, data_t>::dimension;
  private:
    class data_reference
    {
      VectorComponentAccessor<vector_image_t> & _accessor_reference;
      const ublas::fixed_vector<size_t, dimension> & _index_reference;

    public:
      data_reference(VectorComponentAccessor<vector_image_t> & accessor, const ublas::fixed_vector<size_t, dimension> & index) : _accessor_reference(accessor), _index_reference(index) {}

      float_t operator=(const float_t & input)
      { _accessor_reference._image_reference[_index_reference](_accessor_reference._component) = input; return input; }

      operator float_t() const { return _accessor_reference._image_reference[_index_reference](_accessor_reference._component); }
    };
    size_t _component;

  public:
    VectorComponentAccessor(vector_image_t & image, size_t component) : ImageAccessorInterface<vector_image_t, data_t>(image), _component(component) {}

    data_t operator[](const ublas::fixed_vector<size_t, dimension> & index) const { return ImageAccessorInterface<vector_image_t, data_t>::_image_reference[index](_component); }

    data_reference operator[](const ublas::fixed_vector<size_t, dimension> & index) { return data_reference(*this, index); }
  };
}

#endif

---