numpy.h

Go to the documentation of this file.

// Copyright (c) 2021, Aalto University, developed by Erik Schultheis
// All rights reserved.
//
// SPDX-License-Identifier: MIT
 
#ifndef DISMEC_NUMPY_H
#define DISMEC_NUMPY_H
 
#include <iosfwd>
#include <string>
#include <string_view>
#include "matrix_types.h"
 
namespace dismec::io {
    bool is_npy(std::istream& target);
 
    void write_npy_header(std::streambuf& target, std::string_view description);
 
    std::string make_npy_description(std::string_view dtype_desc, bool column_major, std::size_t size);
 
    std::string make_npy_description(std::string_view dtype_desc, bool column_major, std::size_t rows, std::size_t cols);
 
    struct NpyHeaderData {
        std::string DataType;       
        bool ColumnMajor;           
        long Rows;                  
        long Cols;                  
    };
 
    NpyHeaderData parse_npy_header(std::streambuf& source);
 
    template<class S>
    const char* data_type_string();
 
    template<class Derived>
    std::string make_npy_description(const Eigen::DenseBase<Derived>& matrix) {
        return make_npy_description(data_type_string<typename Derived::Scalar>(), !Derived::IsRowMajor, matrix.rows(), matrix.cols());
    }
 
    types::DenseRowMajor<real_t> load_matrix_from_npy(std::istream& source);
    types::DenseRowMajor<real_t> load_matrix_from_npy(const std::string& path);
 
    void save_matrix_to_npy(std::ostream& source, const types::DenseRowMajor<real_t>&);
    void save_matrix_to_npy(const std::string& path, const types::DenseRowMajor<real_t>&);
}
 
#endif //DISMEC_NUMPY_H