NumpyIO.cpp File Reference

#include "cloudViewer/t/io/NumpyIO.h"
#include <FileSystem.h>
#include <Logging.h>
#include <zlib.h>
#include <memory>
#include <numeric>
#include <regex>
#include <sstream>
#include <string>
#include <vector>
#include "cloudViewer/core/Blob.h"
#include "cloudViewer/core/Dispatch.h"
#include "cloudViewer/core/Dtype.h"
#include "cloudViewer/core/SizeVector.h"

Include dependency graph for NumpyIO.cpp:

Go to the source code of this file.

Classes
class	cloudViewer::t::io::CharVector
class	cloudViewer::t::io::NumpyArray

Namespaces
	cloudViewer
	Generic file read and write utility for python interface.
	cloudViewer::t
	cloudViewer::t::io

Functions
static char	cloudViewer::t::io::BigEndianChar ()
static char	cloudViewer::t::io::DtypeToChar (const core::Dtype &dtype)
static CharVector	cloudViewer::t::io::CreateNumpyHeader (const core::SizeVector &shape, const core::Dtype &dtype)
static std::tuple< core::SizeVector, char, int64_t, bool >	cloudViewer::t::io::ParsePropertyDict (const std::string &header)
static size_t	cloudViewer::t::io::ParseNpyPreamble (const char *preamble)
static std::tuple< core::SizeVector, char, int64_t, bool >	cloudViewer::t::io::ParseNpyHeaderFromFile (FILE *fp)
static std::tuple< core::SizeVector, char, int64_t, bool >	cloudViewer::t::io::ParseNpyHeaderFromBuffer (const char *buffer)
static std::tuple< size_t, size_t, size_t >	cloudViewer::t::io::ParseZipFooter (FILE *fp)
static void	cloudViewer::t::io::WriteNpzOneTensor (const std::string &file_name, const std::string &tensor_name, const core::Tensor &tensor, bool append)
static void	cloudViewer::t::io::WriteNpzEmpty (const std::string &file_name)
static NumpyArray	cloudViewer::t::io::CreateNumpyArrayFromFile (FILE *fp)
static NumpyArray	cloudViewer::t::io::CreateNumpyArrayFromCompressedFile (FILE *fp, uint32_t num_compressed_bytes, uint32_t num_uncompressed_bytes)
core::Tensor	cloudViewer::t::io::ReadNpy (const std::string &file_name)
void	cloudViewer::t::io::WriteNpy (const std::string &file_name, const core::Tensor &tensor)
std::unordered_map< std::string, core::Tensor >	cloudViewer::t::io::ReadNpz (const std::string &file_name)
void	cloudViewer::t::io::WriteNpz (const std::string &file_name, const std::unordered_map< std::string, core::Tensor > &tensor_map)