REMORA
Energy Research and Forecasting: An Atmospheric Modeling Code
NDArray< DataType > Struct Template Reference

#include <NCFile.H>

Collaboration diagram for NDArray< DataType >:

Public Types

using DType = typename std::remove_const< DataType >::type
 

Public Member Functions

 NDArray (const std::string vname, const std::vector< size_t > &vshape)
 
 NDArray ()
 
 NDArray (const NDArray &array)
 
NDArrayoperator= (const NDArray &array)
 
 ~NDArray ()
 
decltype(auto) get_data ()
 
std::string get_vname ()
 
std::vector< size_t > get_vshape ()
 
size_t ndim ()
 
void set_vshape (std::vector< size_t > vshape)
 

Private Attributes

std::string name
 
std::vector< size_t > shape
 
std::atomic< size_t > ref_counted
 
bool owned
 
DTypedata
 

Member Typedef Documentation

◆ DType

template<typename DataType >
using NDArray< DataType >::DType = typename std::remove_const<DataType>::type

Constructor & Destructor Documentation

◆ NDArray() [1/3]

template<typename DataType >
NDArray< DataType >::NDArray ( const std::string  vname,
const std::vector< size_t > &  vshape 
)
inlineexplicit
36  : name{vname}, shape{vshape}, ref_counted{1}, owned{true} {
37  data = new DType [this->ndim()];
38  }
bool owned
Definition: NCFile.H:101
std::string name
Definition: NCFile.H:98
size_t ndim()
Definition: NCFile.H:85
std::atomic< size_t > ref_counted
Definition: NCFile.H:100
typename std::remove_const< DataType >::type DType
Definition: NCFile.H:32
std::vector< size_t > shape
Definition: NCFile.H:99
DType * data
Definition: NCFile.H:102
Here is the call graph for this function:

◆ NDArray() [2/3]

template<typename DataType >
NDArray< DataType >::NDArray ( )
inline
41 : name{"null"}, ref_counted{1}, owned{false}, data{nullptr} {}

◆ NDArray() [3/3]

template<typename DataType >
NDArray< DataType >::NDArray ( const NDArray< DataType > &  array)
inline
44  {
45  name = array.name;
46  shape = array.shape;
47  data = array.data;
48  owned = false;
49  ref_counted.fetch_add(1, std::memory_order_relaxed);
50  }

◆ ~NDArray()

template<typename DataType >
NDArray< DataType >::~NDArray ( )
inline
63  {
64  ref_counted.fetch_sub(1, std::memory_order_acq_rel);
65  if (ref_counted == 1 && owned) delete[] data;
66  }

Member Function Documentation

◆ get_data()

template<typename DataType >
decltype(auto) NDArray< DataType >::get_data ( )
inline
69  {
70  ref_counted.fetch_add(1, std::memory_order_relaxed);
71  return data;
72  }

◆ get_vname()

template<typename DataType >
std::string NDArray< DataType >::get_vname ( )
inline
75  {
76  return name;
77  }

◆ get_vshape()

template<typename DataType >
std::vector<size_t> NDArray< DataType >::get_vshape ( )
inline
80  {
81  return shape;
82  }

◆ ndim()

template<typename DataType >
size_t NDArray< DataType >::ndim ( )
inline
85  {
86  size_t num = 1;
87  int isize = static_cast<int>(shape.size());
88  for (auto i=0; i<isize; ++i) num *= shape[i];
89  return num;
90  }

Referenced by NDArray< DataType >::NDArray().

Here is the caller graph for this function:

◆ operator=()

template<typename DataType >
NDArray& NDArray< DataType >::operator= ( const NDArray< DataType > &  array)
inline
53  {
54  name = array.name;
55  shape = array.shape;
56  data = array.data;
57  owned = false;
58  ref_counted.fetch_add(1, std::memory_order_relaxed);
59  return *this;
60  }

◆ set_vshape()

template<typename DataType >
void NDArray< DataType >::set_vshape ( std::vector< size_t >  vshape)
inline
93  {
94  shape = vshape;
95  }

Member Data Documentation

◆ data

◆ name

template<typename DataType >
std::string NDArray< DataType >::name
private

◆ owned

template<typename DataType >
bool NDArray< DataType >::owned
private

◆ ref_counted

template<typename DataType >
std::atomic<size_t> NDArray< DataType >::ref_counted
mutableprivate

◆ shape


The documentation for this struct was generated from the following file: