array.h (Latest change: Author:Lakshman Anumolu <acrlakshman@yahoo.co.in>, 2019-07-22 23:59:29 -0500, [commit: bbc6432])

Go to the documentation of this file.

// Copyright 2019 Lakshman Anumolu, Raunak Bardia.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// 1. Redistributions of source code must retain the above copyright notice,
// this list of conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice,
// this list of conditions and the following disclaimer in the documentation
// and/or other materials provided with the distribution.
//
// 3. Neither the name of the copyright holder nor the names of its contributors
// may be used to endorse or promote products derived from this software without
// specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#pragma once

#include <iostream>
#include <vector>

#include "grid.h"
#include "vec3.h"

namespace GALS
{
namespace CPU
{
template <typename T_GRID, typename T_ARRAY>
class Array
{
 public:
  using value_type = T_ARRAY;

  Array(const Grid<typename T_GRID::value_type, T_GRID::dim> &grid);

  ~Array();

  const std::size_t size() const;

  const T_GRID &grid() const;

  const Vec3<int> numCells() const;

  const std::vector<T_ARRAY> &data() const { return m_data; }

  const T_ARRAY &operator[](const std::size_t idx) const;

  T_ARRAY &operator[](const std::size_t idx);

  const T_ARRAY &operator()(const int i, const int j, const int k) const;

  T_ARRAY &operator()(const int i, const int j, const int k);

  const T_ARRAY &operator()(const Vec3<int> node_id) const;

  T_ARRAY &operator()(const Vec3<int> node_id);

  void operator=(const Array<T_GRID, T_ARRAY> &array);

  friend std::ostream &operator<<(std::ostream &out, const Array<T_GRID, T_ARRAY> &array)
  {
    const auto &grid = array.grid();
    const auto num_cells = array.numCells();

    for (int i = 0; i < num_cells[0]; ++i)
      for (int j = 0; j < num_cells[1]; ++j)
        for (int k = 0; k < num_cells[2]; ++k) out << grid(i, j, k) << "\t" << array(i, j, k) << "\n";

    return out;
  }

 private:
  const T_GRID &m_grid;
  const int m_nx, m_ny, m_nz, m_pad;
  std::vector<T_ARRAY> m_data;
};

}  // namespace CPU
}  // namespace GALS