dfhack/library/modules/XlsxReader.cpp

116 lines
3.5 KiB
C++

/*
* Wrapper for xlsxio_read library functions.
*
* Sample usage:
*
* std::string filename = "sample_file.xlsx";
* xlsxioreader xlsxfile = XlsxReader::open_xlsx_file(filename);
* if (xlsxfile == NULL) {
* printf("cannot open file: '%s'", filename.c_str());
* return false;
* }
* auto sheetNames = XlsxReader::list_sheets(xlsxfile);
* for (auto sheetName = sheetNames.begin();
* sheetName != sheetNames.end();
* ++sheetName) {
* printf("reading sheet: %s\n", sheetName->c_str());
* xlsxioreadersheet xlsxsheet =
* XlsxReader::open_sheet(xlsxfile, *sheetName);
* if (xlsxsheet == NULL) {
* printf("cannot open sheet: '%s'", sheetName->c_str());
* continue;
* }
* std::string value;
* int row_num = 1;
* while (XlsxReader::GetNextRow(xlsxsheet)) {
* std::string s;
* printf("%d:\t", row_num);
* while (XlsxReader::GetNextCell(xlsxsheet, s)) {
* printf("%s\t", s.c_str());
* }
* printf("\n");
* ++row_num;
* }
* XlsxReader::close_sheet(xlsxsheet);
* }
* XlsxReader::close_xlsx_file(xlsxfile);
* return true;
*/
#include <xlsxio_read.h>
#include "modules/XlsxReader.h"
using namespace DFHack;
// returns NULL on error
DFHACK_EXPORT XlsxReader::xlsx_file_handle XlsxReader::open_xlsx_file(
std::string filename)
{
return xlsxioread_open(filename.c_str());
}
DFHACK_EXPORT void XlsxReader::close_xlsx_file(
XlsxReader::xlsx_file_handle file_handle)
{
xlsxioread_close((xlsxioreader)file_handle);
}
static int list_callback(const XLSXIOCHAR* name, void* cbdata)
{
auto sheetNames = (std::vector<std::string> *)cbdata;
sheetNames->push_back(name);
return 0;
}
DFHACK_EXPORT std::vector<std::string> XlsxReader::list_sheets(
XlsxReader::xlsx_file_handle file_handle)
{
auto sheetNames = std::vector<std::string>();
xlsxioread_list_sheets(
(xlsxioreader)file_handle, list_callback, &sheetNames);
return sheetNames;
}
// returns XLSXIOReaderSheet object or NULL on error
DFHACK_EXPORT XlsxReader::xlsx_sheet_handle XlsxReader::open_sheet(
XlsxReader::xlsx_file_handle file_handle, std::string sheet_name)
{
if (file_handle == NULL)
return NULL;
return xlsxioread_sheet_open(
(xlsxioreader)file_handle, sheet_name.c_str(), XLSXIOREAD_SKIP_NONE);
}
DFHACK_EXPORT void XlsxReader::close_sheet(
XlsxReader::xlsx_sheet_handle sheet_handle)
{
xlsxioread_sheet_close((xlsxioreadersheet)sheet_handle);
}
// start reading the next row of data; must be called before GetNextCell .
// returns false if there is no next row to get.
DFHACK_EXPORT bool XlsxReader::get_next_row(
XlsxReader::xlsx_sheet_handle sheet_handle)
{
return xlsxioread_sheet_next_row((xlsxioreadersheet)sheet_handle) != 0;
}
// fills the value param with the contents of the cell in the next column cell
// in the current row.
// returns false if there are no more cells in this row.
DFHACK_EXPORT bool XlsxReader::get_next_cell(
XlsxReader::xlsx_sheet_handle sheet_handle, std::string& value)
{
char* result;
if (!xlsxioread_sheet_next_cell_string((xlsxioreadersheet)sheet_handle,
&result)) {
value.clear();
return false;
}
value.assign(result);
free(result);
return true;
}