csvparser.h

Go to the documentation of this file.

 
#ifndef __CSV_PARSER_H__
#define __CSV_PARSER_H__
 
#include <stdbool.h>
#include <stddef.h>
 
// C++ compatibility
#ifdef __cplusplus
extern "C" {
#endif
 
#ifndef _GNU_SOURCE
#define _GNU_SOURCE 1
#endif
 
#ifndef CSV_ARENA_BLOCK_SIZE
#define CSV_ARENA_BLOCK_SIZE (1 << 20)
#endif
 
#ifndef MAX_FIELD_SIZE
// Maximum size of the csv line.
#define MAX_FIELD_SIZE 1024
#endif
 
// static assert to ensure MAX_FIELD_SIZE is reasonable for stack allocation
#if MAX_FIELD_SIZE > 4096
#error "MAX_FIELD_SIZE is too large for stack allocation. Please reduce it to 4096 or less."
#endif
 
typedef struct CsvReader CsvReader;
 
typedef struct {
    char** fields;  
    size_t count;   
} Row;
 
// Async callback for processed rows.
typedef void (*CsvRowCallback)(size_t row_index, Row* row);
 
CsvReader* csv_reader_new(const char* filename, size_t arena_memory);
 
Row** csv_reader_parse(CsvReader* reader);
 
void csv_reader_parse_async(CsvReader* reader, CsvRowCallback callback, size_t alloc_max);
 
size_t csv_reader_numrows(const CsvReader* reader);
 
void csv_reader_free(CsvReader* reader);
 
struct CsvReaderConfig {
    char delim;
    char quote;
    char comment;
    bool has_header;
    bool skip_header;
};
 
struct CsvWriterConfig {
    char delim;      // Delimiter character
    char quote;      // Quote character
    char newline;    // Newline character
    bool quote_all;  // Quote all fields
    bool flush;      // Flush the stream after writing each row
    bool append;     // Append to the file if it exists, otherwise create a new file
    bool first_row;  // Whether the current row is the first row written to the
                     // file
};
 
typedef struct CsvReaderConfig CsvReaderConfig;
typedef struct CsvWriterConfig CsvWriterConfig;
 
void csv_reader_setconfig(CsvReader* reader, CsvReaderConfig config);
CsvReaderConfig csv_reader_getconfig(CsvReader* reader);
 
typedef struct CsvWriter CsvWriter;
 
#define CsvWriterConfigure(writer, ...)                               \
    csvwriter_setconfig(writer, (CsvWriterConfig){.delim = ',',       \
                                                  .quote = '"',       \
                                                  .newline = '\n',    \
                                                  .quote_all = false, \
                                                  .flush = false,     \
                                                  .first_row = false, \
                                                  __VA_ARGS__})
 
// Create a new CSV writer associated with a filename.
CsvWriter* csvwriter_new(const char* filename);
 
// Set the configuration for the CSV writer.
bool csvwriter_write_row(CsvWriter* writer, const char** fields, size_t numfields);
 
// Free memory used by the CsvWriter and close the file stream.
void csvwriter_free(CsvWriter* writer);
 
#ifdef __cplusplus
}
#endif
 
#endif /* __CSV_PARSER_H__ */