tensor.hpp

Go to the documentation of this file.

/* Copyright (c) 2023 Otto Link. Distributed under the terms of the GNU General
   Public License. The full license is in the file LICENSE, distributed with
   this software. */
 
#pragma once
#include <opencv2/core/mat.hpp>
 
#include "highmap/algebra.hpp"
#include "highmap/array.hpp"
 
namespace hmap
{
 
class Tensor
{
public:
  Vec3<int> shape;
 
  std::vector<float> vector;
 
  Tensor(Vec3<int> shape);
 
  Tensor(Vec2<int> shape_xy, int shape_z);
 
  Tensor(const std::string &fname, bool flip_j = false);
 
  float &operator()(int i, int j, int k);
 
  const float &operator()(int i, int j, int k) const;
 
  Array get_slice(int k) const;
 
  float max() const;
 
  float min() const;
 
  void remap(float vmin = 0.f, float vmax = 1.f);
 
  Tensor resample_to_shape_xy(Vec2<int> new_shape_xy);
 
  void set_slice(int k, const Array &slice);
 
  cv::Mat to_cv_mat();
 
  std::vector<uint8_t> to_img_8bit();
 
  void to_png(const std::string &fname, int depth = CV_8U);
};
 
} // namespace hmap