ndarray.h

Go to the documentation of this file.

// -*- C++ -*-
//
// This file is part of YODA -- Yet more Objects for Data Analysis
// Copyright (C) 2008-2023 The YODA collaboration (see AUTHORS for details)
//
#ifndef YODA_ARRAY_H
#define YODA_ARRAY_H
 
#include <vector>
#include <algorithm>
#include <sstream>
#include <array>
 
namespace YODA {
  namespace Utils {
 
 
    template <typename T, int N>
    class ndarray {
    public:
 
      using ArrayT = std::array<T,N>;
      using const_iterator = typename ArrayT::const_iterator;
 
 
 
      ndarray() {
        clear();
      }
 
      ndarray(const std::vector<T>& vec) {
        if (vec.size() != N) {
          std::stringstream msg;
          msg << "Value vector of wrong size supplied to a " << N << " dimensional array";
          throw RangeError(msg.str());
        }
        for (size_t i = 0; i < N; ++i) _val[i] = vec[i];
      }
 
 
      ndarray(const std::array<T,N> arr) {
        _val = arr;
      }
 
 
      // /// Conversion from C array
      // ndarray(const T arr[N]) {
      //   try {
      //     std::copy(arr, arr+N, _val);
      //   } catch (...) {
      //     std::stringstream msg;
      //     msg << "Value vector of wrong size supplied to a " << N << " dimensional array";
      //     throw RangeError(msg.str());
      //   }
      // }
 
 
 
 
 
      void clear() {
        for (size_t i = 0; i < N; ++i) _val[i] = T();
      }
 
      T& operator[](size_t i) {
        return _val[i];
      }
 
      const T& operator[](size_t i) const {
        return _val[i];
      }
 
      const_iterator begin() const { return _val.cbegin(); }
 
      const_iterator end() const { return _val.cend(); }
 
 
 
    private:
 
      std::array<T,N> _val;
 
    };
 
 
  }
}
 
#endif