/********************************************************
* ██████╗ ██████╗████████╗██╗
* ██╔════╝ ██╔════╝╚══██╔══╝██║
* ██║ ███╗██║ ██║ ██║
* ██║ ██║██║ ██║ ██║
* ╚██████╔╝╚██████╗ ██║ ███████╗
* ╚═════╝ ╚═════╝ ╚═╝ ╚══════╝
* Geophysical Computational Tools & Library (GCTL)
*
* Copyright (c) 2023 Yi Zhang (yizhang-geo@zju.edu.cn)
*
* GCTL is distributed under a dual licensing scheme. You can redistribute
* it and/or modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation, either version 2
* of the License, or (at your option) any later version. You should have
* received a copy of the GNU Lesser General Public License along with this
* program. If not, see .
*
* If the terms and conditions of the LGPL v.2. would prevent you from using
* the GCTL, please consider the option to obtain a commercial license for a
* fee. These licenses are offered by the GCTL's original author. As a rule,
* licenses are provided "as-is", unlimited in time for a one time fee. Please
* send corresponding requests to: yizhang-geo@zju.edu.cn. Please do not forget
* to include some description of your company and the realm of its activities.
* Also add information on how to contact you by electronic and paper mail.
******************************************************/
#ifndef _GCTL_DSV_IO_H
#define _GCTL_DSV_IO_H
#include "../core.h"
#include "../utility.h"
#include "../geometry.h"
namespace gctl
{
struct table_cell
{
std::string str_;
template T value()
{
// 如果输出也是字符串 就直接赋值即可（避免l_str中含有空格会出现bug）
if constexpr (std::is_same::value) return str_;
else
{
T out;
str2type(str_, out);
return out;
}
}
template void value(const T &in, int p = 6)
{
// 对于double类型 可以设置转换的有效数字位数（精度）
if constexpr (std::is_same::value)
{
std::stringstream ss;
ss.precision(p);
ss << in;
ss >> str_;
}
else if constexpr (std::is_same::value) str_ = in;
else str_ = std::to_string(in);
return;
}
};
/**
* @brief 表格的头信息类型
*
*/
enum table_headtype_e
{
NoHead, // 没有表头
BothHead, // 同时有行与列表头
ColumnHead, // 只有列表头
RowHead, // 只有行表头
};
/**
* @brief DSV文本读写类
*
* 可以处理的文本数据应该符合下述要求：
* 1. 以'#'开始的行均为注释行，标识符可由用户指定；
* 2. 以'#!'开始的行均为标记行，标识符可由用户指定；
* 3. 文本开始可以包含n行头信息；
* 4. 数据体为一个row*col大小的表格；
* 5. 数据体可以包含一列行名称与列名称。
*
*/
class dsv_io
{
protected:
std::string file_;
table_headtype_e thead_;
// 头信息行数 表格行数 表格列数
int head_num_, row_num_, col_num_;
// 注释行起始符 标记行起始符 分割符
char att_sym_, tag_sym_, deli_sym_;
// 头信息行 注释行 标记行
std::vector heads_, annotates_, tags_;
// 内容表格
std::vector > table_;
public:
/**
* @brief Construct a new text content object
*
*/
dsv_io();
/**
* @brief Destroy the text content object
*
*/
~dsv_io();
/**
* @brief Construct a new text descriptor object and load text file
*
* @param filename 文件名
* @param file_exten 文件扩展名
*/
dsv_io(std::string filename, std::string file_exten = ".txt", table_headtype_e t = NoHead);
/**
* @brief 设置列分隔符
*
* @param deli_sym 分隔符
*/
void set_delimeter(char deli_sym){deli_sym_ = deli_sym;}
/**
* @brief 设置头信息行数
*
* @param num 行数
*/
void set_head_number(char num){head_num_ = num;}
/**
* @brief 设置注释行符号
*
* @param att_sym 注释符号
*/
void set_annotation_symbol(char att_sym){att_sym_ = att_sym;}
/**
* @brief 设置标记行符号
*
* @param tag_sym 标记符号
*/
void set_tag_symbol(char tag_sym){tag_sym_ = tag_sym;}
/**
* @brief 返回头信息行数
*
* @return 行数
*/
int head_number(){return head_num_;}
/**
* @brief 返回行数
*
* @return 行数
*/
int row_number(){return row_num_;}
/**
* @brief 返回列数
*
* @return 列数
*/
int col_number(){return col_num_;}
/**
* @brief 返回头信息
*
* @return 头信息
*/
const std::vector& get_head_records(){return heads_;}
/**
* @brief 返回注释行
*
* @return 注释行
*/
const std::vector& get_annotoations(){return annotates_;}
/**
* @brief 返回标记行
*
* @return 标记
*/
const std::vector& get_tags(){return tags_;}
/**
* @brief 设置头信息
*
* @param heads 头信息
*/
void set_head_records(const std::vector &heads);
/**
* @brief 设置注释
*
* @param att 注释
*/
void set_annotoations(const std::vector &att);
/**
* @brief 设置标记
*
* @param tags 标记
*/
void set_tags(const std::vector &tags);
/**
* @brief 清理字符串向量对象
*
*/
void clear();
/**
* @brief 读入文本文件
*
* @param filename 文件名
* @param file_exten 文件扩展名
*/
void load_text(std::string filename, std::string file_exten = ".txt", table_headtype_e t = NoHead);
/**
* @brief 读入CSV文件
*
* @param filename 文件名
*/
void load_csv(std::string filename, table_headtype_e t = ColumnHead);
/**
* @brief 将内容写入文件
*
* @param filename 文件名
* @param file_exten 文件扩展名
*/
void save_text(std::string filename, std::string file_exten = ".txt");
/**
* @brief 将内容写入CSV文件
*
* @param filename 文件名（无后缀）
*/
void save_csv(std::string filename);
/**
* @brief 初始化表格
*
* @param row 行数
* @param col 列数
*/
void init_table(int row, int col, table_headtype_e t = ColumnHead);
/**
* @brief 返回表格信息
*
*/
void info();
/**
* @brief 返回名称为name的行或列的索引
*
* @param name 名称
* @param iter_row 搜索行名称（默认为搜索列名称）
* @return 索引
*/
int name_index(std::string name, bool iter_row = false);
/**
* @brief 填充表格
*
* @tparam T 数据类型
* @param data 矩阵数据
* @param p 浮点类数据保存时的有效数字位数
*/
template void fill_table(const matrix &data, int p = 6);
/**
* @brief 获取表格
*
* @tparam T 数据类型
* @param data 矩阵数据
*/
template void get_table(matrix &data);
/**
* @brief 填充列
*
* @tparam T 数据类型
* @param idx 列索引
* @param data 列数据
* @param p 浮点类数据保存时的有效数字位数
*/
template void fill_column(int idx, const array &data, int p = 6);
/**
* @brief 填充列
*
* @tparam T 数据类型
* @param name 列名称
* @param data 列数据
* @param p 浮点类数据保存时的有效数字位数
*/
template void fill_column(std::string name, const array &data, int p = 6);
/**
* @brief 填充行
*
* @tparam T 数据类型
* @param idx 行索引
* @param data 行数据
* @param p 浮点类数据保存时的有效数字位数
*/
template void fill_row(int idx, const array &data, int p = 6);
/**
* @brief 填充行
*
* @tparam T 数据类型
* @param name 行名称
* @param data 行数据
* @param p 浮点类数据保存时的有效数字位数
*/
template void fill_row(std::string name, const array &data, int p = 6);
/**
* @brief 获取列数据
*
* @tparam T 数据类型
* @param idx 列索引
* @param data 列数据
*/
template void get_column(int idx, array &data);
/**
* @brief 获取列数据
*
* @tparam T 数据类型
* @param name 列名称
* @param data 列数据
*/
template void get_column(std::string name, array &data);
/**
* @brief 获取行数据
*
* @tparam T 数据类型
* @param idx 行索引
* @param data 行数据
*/
template void get_row(int idx, array &data);
/**
* @brief 获取行数据
*
* @tparam T 数据类型
* @param name 行名称
* @param data 行数据
*/
template void get_row(std::string name, array &data);
/**
* @brief 获取表格单元数据
*
* @tparam T 数据类型
* @param r 行号
* @param c 列号
* @return T 单元数据
*/
template T cell(int r, int c){return table_[r][c].value();}
/**
* @brief 填充表格单元数据
*
* @tparam T 数据类型
* @param r 行号
* @param c 列号
* @param d 数据
* @param p 浮点类数据保存时的有效数字位数
*/
template void cell(int r, int c, T d, int p = 6){table_[r][c].value(d, p); return;}
};
template
void dsv_io::fill_table(const matrix &data, int p)
{
int st = 0;
if (thead_ == ColumnHead || thead_ == BothHead) st = 1;
int ft = 0;
if (thead_ == RowHead || thead_ == BothHead) ft = 1;
for (size_t i = 0; i < std::min(row_num_ - st, (int) data.row_size()); i++)
{
for (size_t j = 0; j < std::min(col_num_ - ft, (int) data.col_size()); j++)
{
table_[i + st][j + ft].value(data[i][j], p);
}
}
return;
}
template
void dsv_io::get_table(matrix &data)
{
int st = 0;
if (thead_ == ColumnHead || thead_ == BothHead) st = 1;
int ft = 0;
if (thead_ == RowHead || thead_ == BothHead) ft = 1;
data.resize(row_num_ - st, col_num_ - ft);
for (size_t i = st; i < row_num_; i++)
{
for (size_t j = ft; j < col_num_; j++)
{
data[i - st][j - ft] = table_[i][j].value();
}
}
return;
}
template
void dsv_io::fill_column(int idx, const array &data, int p)
{
if (idx >= col_num_)
{
throw std::runtime_error("[gctl::dsv_io] Invalid column index.");
}
int st = 0;
if (thead_ == ColumnHead || thead_ == BothHead) st = 1;
for (size_t i = 0; i < std::min(row_num_ - st, (int) data.size()); i++)
{
table_[i + st][idx].value(data[i], p);
}
return;
}
template
void dsv_io::fill_column(std::string name, const array &data, int p)
{
fill_column(name_index(name, false), data, p);
return;
}
template
void dsv_io::fill_row(int idx, const array &data, int p)
{
if (idx >= row_num_)
{
throw std::runtime_error("[gctl::dsv_io] Invalid row index.");
}
int st = 0;
if (thead_ == RowHead || thead_ == BothHead) st = 1;
for (size_t i = 0; i < std::min(col_num_ - st, (int) data.size()); i++)
{
table_[idx][i + st].value(data[i], p);
}
return;
}
template
void dsv_io::fill_row(std::string name, const array &data, int p)
{
fill_row(name_index(name, true), data, p);
return;
}
template
void dsv_io::get_column(int idx, array &data)
{
if (idx >= col_num_)
{
throw std::runtime_error("[gctl::dsv_io] Invalid column index.");
}
int st = 0;
if (thead_ == ColumnHead || thead_ == BothHead) st = 1;
data.resize(row_num_ - st);
for (size_t i = st; i < row_num_; i++)
{
data[i - st] = table_[i][idx].value();
}
return;
}
template
void dsv_io::get_column(std::string name, array &data)
{
get_column(name_index(name, false), data);
return;
}
template
void dsv_io::get_row(int idx, array &data)
{
if (idx >= row_num_)
{
throw std::runtime_error("[gctl::dsv_io] Invalid row index.");
}
int st = 0;
if (thead_ == RowHead || thead_ == BothHead) st = 1;
data.resize(col_num_ - st);
for (size_t i = st; i < col_num_; i++)
{
data[i - st] = table_[idx][i].value();
}
return;
}
template
void dsv_io::get_row(std::string name, array &data)
{
get_row(name_index(name, true), data);
return;
}
/**
* @brief 地理数据类型DSV文件读写类
*
*/
class geodsv_io : public dsv_io
{
public:
/**
* @brief Construct a new geodsv_io object
*
*/
geodsv_io();
/**
* @brief Destroy the geodsv_io object
*
*/
~geodsv_io();
/**
* @brief Construct a new text descriptor object and load text file
*
* @param filename 文件名
* @param file_exten 文件扩展名
*/
geodsv_io(std::string filename, std::string file_exten = ".txt", table_headtype_e t = NoHead);
/**
* @brief 填充二维坐标列
*
* @param xid x坐标列索引
* @param yid y坐标列索引
* @param data 返回的二维坐标数据
* @param p 填入的浮点数据有效位数（精度）
*/
void fill_column_point2dc(int xid, int yid, const array &data, int p = 6);
/**
* @brief 填充二维坐标列
*
* @param xname x坐标列名称
* @param yname y坐标列名称
* @param data 返回的二维坐标数据
* @param p 填入的浮点数据有效位数（精度）
*/
void fill_column_point2dc(std::string xname, std::string yname, const array &data, int p = 6);
/**
* @brief 填充三维坐标列
*
* @param xid x坐标列索引
* @param yid y坐标列索引
* @param zid z坐标列索引
* @param data 返回的三维坐标数据
* @param p 填入的浮点数据有效位数（精度）
*/
void fill_column_point3dc(int xid, int yid, int zid, const array &data, int p = 6);
/**
* @brief 填充三维坐标列
*
* @param xname x坐标列名称
* @param yname y坐标列名称
* @param zname z坐标列名称
* @param data 返回的三维坐标数据
* @param p 填入的浮点数据有效位数（精度）
*/
void fill_column_point3dc(std::string xname, std::string yname, std::string zname, const array &data, int p = 6);
/**
* @brief 读取二维坐标列
*
* @param xid x坐标列索引
* @param yid y坐标列索引
* @param data 返回的二维坐标数据
*/
void get_column_point2dc(int xid, int yid, array &data);
/**
* @brief 读取二维坐标列
*
* @param xname x坐标列名称
* @param yname y坐标列名称
* @param data 返回的二维坐标数据
*/
void get_column_point2dc(std::string xname, std::string yname, array &data);
/**
* @brief 读取三维坐标列
*
* @param xid x坐标列索引
* @param yid y坐标列索引
* @param zid z坐标列索引
* @param data 返回的三维坐标数据
*/
void get_column_point3dc(int xid, int yid, int zid, array &data);
/**
* @brief 读取三维坐标列
*
* @param xname x坐标列名称
* @param yname y坐标列名称
* @param zname z坐标列名称
* @param data 返回的三维坐标数据
*/
void get_column_point3dc(std::string xname, std::string yname, std::string zname, array &data);
};
}
#endif //_GCTL_DSV_IO_H