140 lines
5.4 KiB
C++
140 lines
5.4 KiB
C++
// Copyright (C) 2018-2022 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#pragma once
|
|
#include <cnpy.h>
|
|
|
|
#include <samples/common.hpp>
|
|
#include <samples/slog.hpp>
|
|
|
|
/// @brief Interface to work with files like input and output
|
|
class BaseFile {
|
|
public:
|
|
virtual void load_file(const char* fileName,
|
|
uint32_t arrayIndex,
|
|
std::string& ptrName,
|
|
std::vector<uint8_t>& memory,
|
|
uint32_t* ptrNumRows,
|
|
uint32_t* ptrNumColumns,
|
|
uint32_t* ptrNumBytesPerElement) = 0;
|
|
|
|
virtual void save_file(const char* fileName,
|
|
bool shouldAppend,
|
|
std::string name,
|
|
void* ptrMemory,
|
|
uint32_t numRows,
|
|
uint32_t numColumns) = 0;
|
|
|
|
virtual void get_file_info(const char* fileName,
|
|
uint32_t numArrayToFindSize,
|
|
uint32_t* ptrNumArrays,
|
|
uint32_t* ptrNumMemoryBytes) = 0;
|
|
};
|
|
|
|
/// @brief Responsible to work with .ark files
|
|
class ArkFile : public BaseFile {
|
|
public:
|
|
/**
|
|
* @brief Get info from Kaldi ARK speech feature vector file
|
|
* @param fileName .ark file name
|
|
* @param numArrayToFindSize number speech feature vectors in the file
|
|
* @param ptrNumArrays pointer to specific number array
|
|
* @param ptrNumMemoryBytes pointer to specific number of memory bytes
|
|
* @return none.
|
|
*/
|
|
void get_file_info(const char* fileName,
|
|
uint32_t numArrayToFindSize,
|
|
uint32_t* ptrNumArrays,
|
|
uint32_t* ptrNumMemoryBytes) override;
|
|
|
|
/**
|
|
* @brief Load Kaldi ARK speech feature vector file
|
|
* @param fileName .ark file name
|
|
* @param arrayIndex number speech feature vector in the file
|
|
* @param ptrName reference to variable length name
|
|
* @param memory reference to speech feature vector to save
|
|
* @param ptrNumRows pointer to number of rows to read
|
|
* @param ptrNumColumns pointer to number of columns to read
|
|
* @param ptrNumBytesPerElement pointer to number bytes per element (size of float by default)
|
|
* @return none.
|
|
*/
|
|
void load_file(const char* fileName,
|
|
uint32_t arrayIndex,
|
|
std::string& ptrName,
|
|
std::vector<uint8_t>& memory,
|
|
uint32_t* ptrNumRows,
|
|
uint32_t* ptrNumColumns,
|
|
uint32_t* ptrNumBytesPerElement) override;
|
|
|
|
/**
|
|
* @brief Save Kaldi ARK speech feature vector file
|
|
* @param fileName .ark file name
|
|
* @param shouldAppend bool flag to rewrite or add to the end of file
|
|
* @param name reference to variable length name
|
|
* @param ptrMemory pointer to speech feature vector to save
|
|
* @param numRows number of rows
|
|
* @param numColumns number of columns
|
|
* @return none.
|
|
*/
|
|
void save_file(const char* fileName,
|
|
bool shouldAppend,
|
|
std::string name,
|
|
void* ptrMemory,
|
|
uint32_t numRows,
|
|
uint32_t numColumns) override;
|
|
};
|
|
|
|
/// @brief Responsible to work with .npz files
|
|
class NumpyFile : public BaseFile {
|
|
public:
|
|
/**
|
|
* @brief Get info from Numpy* uncompressed NPZ speech feature vector file
|
|
* @param fileName .npz file name
|
|
* @param numArrayToFindSize number speech feature vectors in the file
|
|
* @param ptrNumArrays pointer to specific number array
|
|
* @param ptrNumMemoryBytes pointer to specific number of memory bytes
|
|
* @return none.
|
|
*/
|
|
void get_file_info(const char* fileName,
|
|
uint32_t numArrayToFindSize,
|
|
uint32_t* ptrNumArrays,
|
|
uint32_t* ptrNumMemoryBytes) override;
|
|
|
|
/**
|
|
* @brief Load Numpy* uncompressed NPZ speech feature vector file
|
|
* @param fileName .npz file name
|
|
* @param arrayIndex number speech feature vector in the file
|
|
* @param ptrName reference to variable length name
|
|
* @param memory reference to speech feature vector to save
|
|
* @param ptrNumRows pointer to number of rows to read
|
|
* @param ptrNumColumns pointer to number of columns to read
|
|
* @param ptrNumBytesPerElement pointer to number bytes per element (size of float by default)
|
|
* @return none.
|
|
*/
|
|
void load_file(const char* fileName,
|
|
uint32_t arrayIndex,
|
|
std::string& ptrName,
|
|
std::vector<uint8_t>& memory,
|
|
uint32_t* ptrNumRows,
|
|
uint32_t* ptrNumColumns,
|
|
uint32_t* ptrNumBytesPerElement) override;
|
|
|
|
/**
|
|
* @brief Save Numpy* uncompressed NPZ speech feature vector file
|
|
* @param fileName .npz file name
|
|
* @param shouldAppend bool flag to rewrite or add to the end of file
|
|
* @param name reference to variable length name
|
|
* @param ptrMemory pointer to speech feature vector to save
|
|
* @param numRows number of rows
|
|
* @param numColumns number of columns
|
|
* @return none.
|
|
*/
|
|
void save_file(const char* fileName,
|
|
bool shouldAppend,
|
|
std::string name,
|
|
void* ptrMemory,
|
|
uint32_t numRows,
|
|
uint32_t numColumns) override;
|
|
};
|