tensor_shape.hpp

 
#ifndef EXATN_NUMERICS_TENSOR_SHAPE_HPP_
#define EXATN_NUMERICS_TENSOR_SHAPE_HPP_
 
#include "tensor_basic.hpp"
 
#include <iostream>
#include <fstream>
#include <type_traits>
#include <initializer_list>
#include <vector>
 
#include <cassert>
 
namespace exatn{
 
namespace numerics{
 
class TensorShape{
public:
 
 template<typename T>
 TensorShape(std::initializer_list<T> extents);
 template<typename T>
 TensorShape(const std::vector<T> & extents);
 TensorShape();
 
 TensorShape(const TensorShape & tens_shape) = default;
 TensorShape & operator=(const TensorShape & tens_shape) = default;
 TensorShape(TensorShape && tens_shape) noexcept = default;
 TensorShape & operator=(TensorShape && tens_shape) noexcept = default;
 virtual ~TensorShape() = default;
 
 void printIt() const;
 void printItFile(std::ofstream & output_file) const;
 
 unsigned int getRank() const;
 
 DimExtent getDimExtent(unsigned int dim_id) const;
 
 const std::vector<DimExtent> & getDimExtents() const;
 
 bool isCongruentTo(const TensorShape & another) const;
 
 void resetDimension(unsigned int dim_id, DimExtent extent);
 
 void deleteDimension(unsigned int dim_id);
 
 void appendDimension(DimExtent dim_extent);
 
private:
 
 std::vector<DimExtent> extents_; //tensor dimension extents
};
 
 
//TEMPLATES:
template<typename T>
TensorShape::TensorShape(std::initializer_list<T> extents):
extents_(extents.size())
{
 static_assert(std::is_integral<T>::value,"FATAL(TensorShape::TensorShape): TensorShape extent type must be integral!");
 
 //DEBUG:
 for(const auto & extent: extents){
  if(extent < 0) std::cout << "ERROR(TensorShape::TensorShape): Negative dimension extent passed!" << std::endl;
  assert(extent >= 0);
 }
 
 int i = 0;
 for(const auto & extent: extents) extents_[i++] = static_cast<DimExtent>(extent);
}
 
template<typename T>
TensorShape::TensorShape(const std::vector<T> & extents):
extents_(extents.size())
{
 static_assert(std::is_integral<T>::value,"FATAL(TensorShape::TensorShape): TensorShape extent type must be integral!");
 
 //DEBUG:
 for(const auto & extent: extents){
  if(extent < 0) std::cout << "ERROR(TensorShape::TensorShape): Negative dimension extent passed!" << std::endl;
  assert(extent >= 0);
 }
 
 int i = 0;
 for(const auto & extent: extents) extents_[i++] = static_cast<DimExtent>(extent);
}
 
} //namespace numerics
 
} //namespace exatn
 
#endif //EXATN_NUMERICS_TENSOR_SHAPE_HPP_