wbImport.cpp

#include	<wb.h>

static inline void wbImportCSV_setFile(wbImportCSV_t csv, const char * path) {
    if (csv != NULL) {
        if (wbImportCSV_getFile(csv) != NULL) {
            wbFile_delete(wbImportCSV_getFile(csv));
        }
        if (path != NULL) {
            wbImportCSV_getFile(csv) = wbFile_open(path, "r");
        } else {
            wbImportCSV_getFile(csv) = NULL;
        }
    }

    return ;
}

static inline wbImportCSV_t wbImportCSV_new(void) {
    wbImportCSV_t csv;

    csv = wbNew(struct st_wbImportCSV_t);

    wbImportCSV_setRowCount(csv, -1);
    wbImportCSV_setColumnCount(csv, -1);
    wbImportCSV_setData(csv, NULL);
    wbImportCSV_getFile(csv) = NULL;
    wbImportCSV_setSeperator(csv, '\0');

    return csv;
}

static inline void wbImportCSV_delete(wbImportCSV_t csv) {
    if (csv != NULL) {
        wbImportCSV_setFile(csv, NULL);
        if (wbImportCSV_getData(csv)) {
            wbDelete(wbImportCSV_getData(csv));
        }
        wbDelete(csv);
    }
}

static inline wbImportCSV_t wbImportCSV_findDimensions(wbImportCSV_t csv, int * resRows, int * resColumns) {
    int rows = 0, columns = -1;
    char * line;
    wbFile_t file;
    char seperator[2];

    if (csv == NULL) {
        return NULL;
    }

    if (wbImportCSV_getSeperator(csv) == '\0') {
        seperator[0] = ',';
    } else {
        seperator[0] = wbImportCSV_getSeperator(csv);
    }
    seperator[1] = '\0';

    file = wbImportCSV_getFile(csv);

    while ((line = wbFile_readLine(file)) != NULL) {
        int currColumn = 0;
        char * token = strtok(line, seperator);
        while (token != NULL) {
            token = strtok(NULL, seperator);
            currColumn++;
        }
        rows++;
        if (columns == -1) {
            columns = currColumn;
        }
        if (columns != currColumn) {
            wbLog(ERROR, "The csv file is not rectangular.");
        }
        wbAssert(columns == currColumn);
    }

    wbFile_rewind(file);

    *resRows = rows;
    *resColumns = columns;

    return csv;
}

static inline int * csv_readAsInteger(wbFile_t file, char sep, int rows, int columns) {
    int ii = 0;
    int * data;
    char * line;
    char seperator[1];

    if (file == NULL) {
        return NULL;
    }

    data = wbNewArray(int, rows * columns);

    if (sep == '\0') {
        seperator[0] = ',';
    } else {
        seperator[0] = sep;
    }

    while ((line = wbFile_readLine(file)) != NULL) {
        char * token = strtok(line, seperator);
        while (token != NULL) {
            int var;

            sscanf(token, "%d", &var);
            token = strtok(NULL, seperator);
            data[ii++] = var;
        }
    }

    return data;
}

static inline wbReal_t * csv_readAsReal(wbFile_t file, char sep, int rows, int columns) {
    int ii = 0;
    wbReal_t * data;
    char * line;
    wbReal_t var;
    char seperator[2];

    if (file == NULL) {
        return NULL;
    }

    data = wbNewArray(wbReal_t, rows * columns);

    if (sep == '\0') {
        seperator[0] = ',';
    } else {
        seperator[0] = sep;
    }
    seperator[1] = '\0';

    if (columns == 1) {
        while ((line = wbFile_readLine(file)) != NULL) {
            sscanf(line, "%f", &var);
            data[ii++] = var;
        }
    } else {
        while ((line = wbFile_readLine(file)) != NULL) {
            char * token = strtok(line, seperator);
            while (token != NULL) {
                sscanf(token, "%f", &var);
                token = strtok(NULL, seperator);
                data[ii++] = var;
            }
        }
    }

    return data;
}

static inline wbImportCSV_t wbImportCSV_read(wbImportCSV_t csv, wbBool asIntegerQ) {
    void * data;
    wbFile_t file;
    char seperator;
    int rows, columns;

    if (csv == NULL) {
        return NULL;
    }

    if (wbImportCSV_getRowCount(csv) == -1 || wbImportCSV_getColumnCount(csv) == -1) {
        if (wbImportCSV_findDimensions(csv, &rows, &columns) == NULL) {
            wbLog(ERROR, "Failed to figure out csv dimensions.");
            return NULL;
        }
        wbImportCSV_setRowCount(csv, rows);
        wbImportCSV_setColumnCount(csv, columns);
    }

    file = wbImportCSV_getFile(csv);
    seperator = wbImportCSV_getSeperator(csv);
    rows = wbImportCSV_getRowCount(csv);
    columns = wbImportCSV_getColumnCount(csv);

    if (wbImportCSV_getData(csv) != NULL) {
        wbDelete(wbImportCSV_getData(csv));
        wbImportCSV_setData(csv, NULL);
    }

    if (asIntegerQ) {
        data = csv_readAsInteger(file, seperator, rows, columns);
    } else {
        data = csv_readAsReal(file, seperator, rows, columns);
    }

    wbImportCSV_setData(csv, data);

    return csv;
}

static inline wbImportCSV_t wbImportCSV_readAsInteger(wbImportCSV_t csv) {
    return wbImportCSV_read(csv, wbTrue);
}

static inline wbImportCSV_t wbImportCSV_readAsReal(wbImportCSV_t csv) {
    return wbImportCSV_read(csv, wbFalse);
}

static inline void wbImportRaw_setFile(wbImportRaw_t raw, const char * path) {
    if (raw != NULL) {
        if (wbImportRaw_getFile(raw) != NULL) {
            wbFile_delete(wbImportRaw_getFile(raw));
        }
        if (path != NULL) {
            wbImportRaw_getFile(raw) = wbFile_open(path, "r");
        } else {
            wbImportRaw_getFile(raw) = NULL;
        }
    }

    return ;
}

static inline wbImportRaw_t wbImportRaw_new(void) {
    wbImportRaw_t raw;

    raw = wbNew(struct st_wbImportRaw_t);

    wbImportRaw_setRowCount(raw, -1);
    wbImportRaw_setColumnCount(raw, -1);
    wbImportRaw_setData(raw, NULL);
    wbImportRaw_getFile(raw) = NULL;

    return raw;
}

static inline void wbImportRaw_delete(wbImportRaw_t raw) {
    if (raw != NULL) {
        wbImportRaw_setFile(raw, NULL);
        if (wbImportRaw_getData(raw)) {
            wbDelete(wbImportRaw_getData(raw));
        }
        wbDelete(raw);
    }
}

static inline wbBool lineHasSpace(const char * line) {
    while (*line != '\0') {
        if (*line == ' ') {
            return wbTrue;
        }
        line++;
    }
    return wbFalse;
}

static inline char * lineStrip(const char * line) {
    char * sl = wbString_duplicate(line);
    char * iter = sl;
    size_t slen = strlen(line);

    iter += slen - 1;
    while (*iter == '\0' || *iter == '\r' || *iter == '\t' || *iter == '\n' || *iter == ' ') {
        *iter-- = '\0';
    }
    return sl;
}

static inline wbBool wbImportRaw_findDimensions(wbImportRaw_t raw) {
    if (raw != NULL) {
        int rows;
        int columns;
        char * line;
        wbFile_t file;
        char * strippedLine;

        file = wbImportRaw_getFile(raw);

        wbFile_rewind(file);

        line = wbFile_readLine(file);

        if (line == NULL) {
            return wbTrue;
        }

        strippedLine = lineStrip(line);

        if (lineHasSpace(strippedLine)) {
            sscanf(strippedLine, "%d %d", &rows, &columns);
        } else {
            columns = 1;
            sscanf(strippedLine, "%d", &rows);
        }

        wbImportRaw_setRowCount(raw, rows);
        wbImportRaw_setColumnCount(raw, columns);

        wbDelete(strippedLine);

        return wbFalse;
    }

    return wbTrue;
}

static inline wbImportRaw_t wbImportRaw_read(wbImportRaw_t raw, wbBool asIntegerQ) {
    void * data;
    wbFile_t file;
    char seperator;
    int rows, columns;

    if (raw == NULL) {
        return NULL;
    }

    if (wbImportRaw_getRowCount(raw) == -1 || wbImportRaw_getColumnCount(raw) == -1) {
        if (wbImportRaw_findDimensions(raw)) {
            wbLog(ERROR, "Failed to figure out raw dimensions.");
            return NULL;
        }
    }

    file = wbImportRaw_getFile(raw);
    seperator = ' ';
    rows = wbImportRaw_getRowCount(raw);
    columns = wbImportRaw_getColumnCount(raw);

    if (wbImportRaw_getData(raw) != NULL) {
        wbDelete(wbImportRaw_getData(raw));
        wbImportRaw_setData(raw, NULL);
    }

    if (asIntegerQ) {
        data = csv_readAsInteger(file, seperator, rows, columns);
    } else {
        data = csv_readAsReal(file, seperator, rows, columns);
    }

    wbImportRaw_setData(raw, data);

    return raw;
}

static inline wbImportRaw_t wbImportRaw_readAsInteger(wbImportRaw_t raw) {
    return wbImportRaw_read(raw, wbTrue);
}

static inline wbImportRaw_t wbImportRaw_readAsReal(wbImportRaw_t raw) {
    return wbImportRaw_read(raw, wbFalse);
}

static inline wbImport_t wbImport_open(const char * file, wbImportKind_t kind) {
    wbImport_t imp;

    if (file == NULL) {
        wbLog(ERROR, "Go NULL for file value.");
        wbExit();
    }

    if (!wbFile_existsQ(file)) {
        wbLog(ERROR, "File ", file, " does not exist.");
        wbExit();
    }

    wbImport_setKind(imp, kind);

    if (kind == wbImportKind_raw) {
        wbImportRaw_t raw = wbImportRaw_new();
        wbImportRaw_setFile(raw, file);
        wbImport_setRaw(imp, raw);
    } else if (kind == wbImportKind_tsv || kind == wbImportKind_csv) {
        wbImportCSV_t csv = wbImportCSV_new();
        if (kind == wbImportKind_csv) {
            wbImportCSV_setSeperator(csv, ',');
        } else {
            wbImportCSV_setSeperator(csv, '\t');
        }
        wbImportCSV_setFile(csv, file);
        wbImport_setCSV(imp, csv);
    } else if (kind == wbImportKind_ppm) {
        wbImage_t img = wbPPM_import(file);
        wbImport_setImage(imp, img);
    } else {
        wbLog(ERROR, "Invalid import type.");
        wbExit();
    }

    return imp;
}

static inline wbImport_t wbImport_open(const char * file, const char * type0) {
    wbImport_t imp;
    wbImportKind_t kind;
    char * type;

    type = wbString_toLower(type0);

    if (wbString_sameQ(type, "cvs")) {
        kind = wbImportKind_csv;
    } else if (wbString_sameQ(type, "tsv")) {
        kind = wbImportKind_tsv;
    } else if (wbString_sameQ(type, "raw") || wbString_sameQ(type, "dat")) {
        kind = wbImportKind_raw;
    } else if (wbString_sameQ(type, "ppm")) {
        kind = wbImportKind_ppm;
    } else {
        wbLog(ERROR, "Invalid import type ", type0);
        wbExit();
    }

    imp = wbImport_open(file, kind);

    wbDelete(type);

    return imp;
}

static inline void wbImport_close(wbImport_t imp) {
    wbImportKind_t kind;

    kind = wbImport_getKind(imp);
    if (kind == wbImportKind_tsv || kind == wbImportKind_csv) {
        wbImportCSV_t csv = wbImport_getCSV(imp);
        wbImportCSV_delete(csv);
        wbImport_setCSV(imp, NULL);
    } else if (kind == wbImportKind_raw) {
        wbImportRaw_t raw = wbImport_getRaw(imp);
        wbImportRaw_delete(raw);
        wbImport_setRaw(imp, NULL);
    } else if (kind == wbImportKind_ppm) {
    } else {
        wbLog(ERROR, "Invalid import type.");
        wbExit();
    }
    return ;
}

static inline void * wbImport_read(wbImport_t imp, wbBool asIntegerQ) {
    void * data = NULL;
    wbImportKind_t kind;

    kind = wbImport_getKind(imp);
    if (kind == wbImportKind_tsv || kind == wbImportKind_csv) {
        wbImportCSV_t csv = wbImport_getCSV(imp);
        wbImportCSV_read(csv, asIntegerQ);
        data = wbImportCSV_getData(csv);
    } else if (kind == wbImportKind_raw) {
        wbImportRaw_t raw = wbImport_getRaw(imp);
        wbImportRaw_read(raw, asIntegerQ);
        data = wbImportRaw_getData(raw);
    } else {
        wbLog(ERROR, "Invalid import type.");
        wbExit();
    }
    return data;
}

static inline int * wbImport_readAsInteger(wbImport_t imp) {
    void * data = wbImport_read(imp, wbTrue);
    return (int*) data;
}

static inline wbReal_t * wbImport_readAsReal(wbImport_t imp) {
    void * data = wbImport_read(imp, wbFalse);
    return (wbReal_t*) data;
}


static wbImportKind_t _parseImportExtension(const char * file) {
    char * extension;
    wbImportKind_t kind;

    extension = wbFile_extension(file);

    if (wbString_sameQ(extension, "csv")) {
        kind = wbImportKind_csv;
    } else if (wbString_sameQ(extension, "tsv")) {
        kind = wbImportKind_tsv;
    } else if (wbString_sameQ(extension, "raw") ||
               wbString_sameQ(extension, "dat")) {
        kind = wbImportKind_raw;
    } else if (wbString_sameQ(extension, "ppm")) {
        kind = wbImportKind_ppm;
    } else {
        kind = wbImportKind_unknown;
        wbLog(ERROR, "File ", file, " does not have a compatible extension.");
    }

    wbDelete(extension);

    return kind;
}

static void * wbImport(const char * file, int * resRows, int * resColumns, const char * type) {
    void * data, * res;
    wbImport_t imp;
    size_t sz;
    int columns = 0, rows = 0;
    wbImportKind_t kind;

    if (file == NULL) {
        fprintf(stderr, "Failed to import file.\n");
        wbExit();
    }

    kind = _parseImportExtension(file);

    wbAssert(kind != wbImportKind_unknown);

    imp = wbImport_open(file, kind);
    if (wbString_sameQ(type, "Real")) {
        data = wbImport_readAsReal(imp);
        sz = sizeof(wbReal_t);
    } else {
        data = wbImport_readAsInteger(imp);
        sz = sizeof(int);
    }

    if (kind == wbImportKind_csv || kind == wbImportKind_tsv) {
        rows = wbImportCSV_getRowCount(wbImport_getCSV(imp));
        columns = wbImportCSV_getColumnCount(wbImport_getCSV(imp));
    } else if (kind == wbImportKind_raw) {
        rows = wbImportRaw_getRowCount(wbImport_getRaw(imp));
        columns = wbImportRaw_getColumnCount(wbImport_getRaw(imp));
    }

    if (resRows != NULL) {
        *resRows = rows;
    }

    if (resColumns != NULL) {
        *resColumns = columns;
    }

    res = wbMalloc(sz * rows * columns);
    memcpy(res, data, sz * rows * columns);

    wbImport_close(imp);

    return res;
}

void * wbImport(const char * file, int * rows, int * columns) {
    return wbImport(file, rows, columns, "Real");
}

void * wbImport(const char * file, int * rows) {
    return wbImport(file, rows, NULL);
}

wbImage_t wbImport(const char * file) {
    wbImage_t img;
    wbImport_t imp;
    wbImportKind_t kind;

    if (file == NULL) {
        fprintf(stderr, "Failed to import file.\n");
        wbExit();
    }

    kind = _parseImportExtension(file);

    wbAssert(kind == wbImportKind_ppm);

    imp = wbImport_open(file, kind);
    img = wbImport_getImage(imp);
    wbImport_close(imp);

    return img;
}