#include <nda/nda.hpp>
#include <nda/h5.hpp>

using namespace nda;

int main() {

  // Create array of shape (4,4,4)
  array<long, 3> A(4, 4, 4);

  // Create an array given its data
  array<long, 2> B{{1, 2}, {3, 4}, {5, 6}};

  // Assign
  A() = 0;
  A(0, 1, 2) = 40;

  // Access
  int a = A(0, 1, 2) + B(0, 1);

  // Slicing to a view of shape (3, 2)
  auto V = A(range(0, 3), range(0, 2), 0);

  // Lazy Arithmetic operations
  auto C = V + 2 * B;

  // Algorithms
  min_element(V);
  max_element(V);
  sum(V);

  // write to file
  {
    h5::file file("dat.h5", 'w');
    h5_write(file, "A", A);
  }

  // read from file
  array<long, 3> D;
  {
    h5::file file("dat.h5", 'r');
    h5_read(file, "A", D);
  }

}