ValArray.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstScript/Value.hpp"
 
AST_NAMESPACE_BEGIN
 
 
template<typename T, size_t NDIM>
class ValArray: public Value
{
public:
    // AST_EXPR(ValArray)
    ValArray();
 
    ~ValArray();
 
    void resize(size_t size);
 
    size_t column() const { return dims_[0]; }
 
    size_t row() const { return dims_[1]; }
 
    size_t page() const { return dims_[2]; }
 
    size_t size() const;
 
    T at(size_t index) const { return data_[index]; }
 
    T& at(size_t index) { return data_[index]; }
    
protected:
    size_t dims_[NDIM];             
    T*     data_{nullptr};          
};
 
 
template<typename T, size_t NDIM>
ValArray<T, NDIM>::ValArray()
{
}
 
 
template<typename T, size_t NDIM>
ValArray<T, NDIM>::~ValArray()
{
    if(data_)
        delete[] data_;
}
 
template<typename T, size_t NDIM>
void ValArray<T, NDIM>::resize(size_t size)
{
    size_t old_size = this->size();
    // 判断是否需要扩充数组大小
    if(old_size >= size)
    {
        T* old_data = data_;
        data_ = new T[size];
        if(old_data)
        {
            size_t copy_size = old_size < size ? old_size : size;
            for(size_t i = 0; i < copy_size; ++i)
                data_[i] = old_data[i];
            delete[] old_data;
        }
    }
    dims_[0] = size;
    for(size_t i = 1; i < NDIM; ++i)
        dims_[i] = 1;
}
 
template <typename T, size_t NDIM>
inline size_t ValArray<T, NDIM>::size() const
{
    size_t size = 1;
    for (size_t i = 0; i < NDIM; ++i)
    {
        size *= dims_[i];
    }
    return size;
}
 
 
AST_NAMESPACE_END