data_view.hpp

Go to the documentation of this file.

/*
 * Copyright (C) 1998-2018 ALPS Collaboration. See COPYRIGHT.TXT
 * All rights reserved. Use is subject to license terms. See LICENSE.TXT
 * For use in publications, see ACKNOWLEDGE.TXT
 */

#ifndef ALPSCORE_GF_TENSORVIEW_H
#define ALPSCORE_GF_TENSORVIEW_H



#include <vector>
#include <array>
#include "data_storage.hpp"
#include "view.hpp"

namespace alps {
  namespace numerics {
    namespace detail {

      template<typename T>
      class data_view {
      private:
        view<T> data_slice_;
        size_t size_;
      public:
        data_view(data_storage<T> & storage) : data_slice_(storage), size_(storage.size()) {}
        data_view(const data_storage<T> & storage) : data_slice_(storage.data(), storage.size()), size_(storage.size()) {}
        template<typename S>
        data_view(const data_storage<S> & storage, size_t size, size_t offset = 0) : data_slice_(storage.data() + offset, size), size_(size) {}
        data_view(data_storage<T> & storage, size_t size, size_t offset = 0) : data_slice_(storage.data() + offset, size), size_(size) {}
        data_view(data_view<T> && storage, size_t size, size_t offset) : data_slice_(storage.data_slice_.data() + offset, size), size_(size) {}
        data_view(const data_view<T> & storage, size_t size, size_t offset) : data_slice_(storage.data_slice_.data() + offset, size), size_(size) {}
        data_view(T*data, size_t size) : data_slice_(data, size), size_(size){}
        data_view(const data_view<T> & storage) = default;

        template<typename S>
        data_view(const data_view<S>& storage, size_t size = 0, size_t offset = 0) :
          data_slice_(&storage.data()[offset], size), size_(size) {};


        data_view(data_view<T> && storage) = default;

        data_view<T>& operator=(const data_view<T>& rhs) = default;
        data_view<T>& operator=(data_view<T>&& rhs) = default;

        T& data(size_t i) {return data_slice_.data(i);};
        const T& data(size_t i) const {return data_slice_.data(i);};
        inline const T&  operator()(size_t i) const {return data_slice_.data(i);};
        inline T& operator()(size_t i) {return data_slice_.data(i);};
        size_t size() const {return size_;}
        T* data() {return data_slice_.data();}
        const T* data() const {return data_slice_.data();}

        template<typename T2>
        bool operator==(const data_view<T2>& r) const {
          return size() == r.size() && std::equal(r.data(), r.data() + r.size(), data());
        }

        template<typename T2>
        bool operator==(const data_storage<T2>& r) const {
          return size() == r.size() && std::equal(r.data(), r.data() + r.size(), data());
        }
      };
    }
  }
}
#endif //ALPSCORE_GF_TENSORVIEW_H