HardCompound/src/main.c

#include <duckdb.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>

#include "array_utils.h"
#include "arrays.h"
#include "exit_code.h"
#include "lapinfo.h"

bool is_valid_path(const char *path) {
    if (path == NULL) {
        return false;
    }

    const char *dot = strrchr(path, '.'); // find the last dot which is for the extension
    if (!dot || strcmp(dot, ".duckdb") != 0) {
        return false;
    }

    struct stat path_stat;
    if (stat(path, &path_stat) != 0) {
        return false; // path does not exist
    }

    if (!S_ISREG(path_stat.st_mode)) {
        return false; // checks if it is a regular file
    }

    return true;
}

int main(int argc, char **argv) {
    duckdb_database db;
    duckdb_connection conn;
    duckdb_result lap_count_res;
    LapInfo *session_data = NULL;
    double *lap_timestamps = NULL;

    printf("\nHello from HardCompound!\n");

    if (argc < 2) {
        printf("You need to specify a duckdb file path! Terminating!\n");
        goto cleanup;
    }

    if (!is_valid_path(argv[1])) {
        printf("The file you gave is not valid. Exiting!");
        goto cleanup;
    }

    printf("First argument: %s\n", argv[1]);

    if (duckdb_open(argv[1], &db) == DuckDBError) {
        printf("Error opening duckdb file, terminating!\n");
        goto cleanup;
    }

    if (duckdb_connect(db, &conn) == DuckDBError) {
        printf("Error connecting to the duckdb database! Terminating!\n");
        goto cleanup;
    }

    // get how many laps is in the session and  allocate that many LapInfo-s
    int32_t lap_count = 0;
    duckdb_query(conn, "Select count(value) from Lap\n", &lap_count_res);

    duckdb_data_chunk data_chunk = duckdb_fetch_chunk(lap_count_res);
    if (!data_chunk) {
        printf("Error in retrieving laps");
        goto cleanup;
    }

    // set column
    duckdb_vector col1 = duckdb_data_chunk_get_vector(data_chunk, 0);
    int32_t *col1_data = (int32_t *)duckdb_vector_get_data(col1);
    uint64_t *col1_validity = duckdb_vector_get_validity(col1);
    idx_t row_count = duckdb_data_chunk_get_size(data_chunk); // number of rows from the data chunk

    for (idx_t row = 0; row < row_count; row++) {
        if (duckdb_validity_row_is_valid(col1_validity, 0)) {
            lap_count = col1_data[0];
        }
    }

    duckdb_destroy_data_chunk(&data_chunk);
    duckdb_destroy_result(&lap_count_res);

    printf("Lap count: %d\n", lap_count);
    session_data = calloc(lap_count, sizeof(LapInfo));
    lap_timestamps = calloc(lap_count, sizeof(double));

    if (!session_data) {
        printf("Session data failed to allocate!");
        goto cleanup;
    }

    if (!lap_timestamps) {
        printf("Lap timestaps array failed to init!");
        goto cleanup;
    }

    duckdb_result lap_timestamps_res;
    if (duckdb_query(conn, "Select ts as DOUBLE from Lap\n", &lap_timestamps_res) == DuckDBError) {
        goto cleanup;
    }

    int current_index = 0;
    while (true) {
        duckdb_data_chunk data_chunk_2 = duckdb_fetch_chunk(lap_timestamps_res);
        if (!data_chunk_2) {
            break;
        }

        idx_t current_chunk_size = duckdb_data_chunk_get_size(data_chunk_2);

        duckdb_vector col2 = duckdb_data_chunk_get_vector(data_chunk_2, 0);
        double *ts_data = (double *)duckdb_vector_get_data(col2);
        uint64_t *col2_validity = duckdb_vector_get_validity(col2);

        for (idx_t row = 0; row < current_chunk_size; row++) {
            if (!col2_validity || duckdb_validity_row_is_valid(col2_validity, row)) {
                if (current_index < lap_count) {
                    lap_timestamps[current_index++] = ts_data[row];
                }
            }
        }

        duckdb_destroy_data_chunk(&data_chunk_2);
    }

    duckdb_destroy_result(&lap_timestamps_res);

    for (int i = 0; i < lap_count; i++) {
        printf("%f, ", lap_timestamps[i]);
    }
    printf("\n");

    // throttle
    float_array throttle_data_all;
    float_array_init(&throttle_data_all);
    allocate_array_for_db_data(&throttle_data_all, "Throttle Pos", &conn);
    get_data_from_db("Select value as FLOAT from \"Throttle Pos\"\n", &conn, &throttle_data_all);
    for (int32_t i = 0; i < throttle_data_all.length; ++i) {
        // printf(" %f", float_array_get(&throttle_data_all, i));
    }

    // brake
    float_array brake_data_all;
    float_array_init(&brake_data_all);
    allocate_array_for_db_data(&brake_data_all, "Brake Pos", &conn);
    get_data_from_db("Select value as FLOAT from \"Brake Pos\"\n", &conn, &brake_data_all);
    for (int32_t i = 0; i < brake_data_all.length; ++i) {
        // printf(" %f", float_array_get(&brake_data_all, i));
    }

    // steering
    float_array steering_data_all;
    float_array_init(&steering_data_all);
    allocate_array_for_db_data(&steering_data_all, "Steering Pos", &conn);
    get_data_from_db("Select value as FLOAT from \"Steering Pos\"\n", &conn, &steering_data_all);
    for (int32_t i = 0; i < steering_data_all.length; ++i) {
        // printf(" %f", float_array_get(&steering_data_all, i));
    }
    // select avg(((value1+value2+value3+value4)/4)*3.6) as speed from "Wheel Speed"
    // speed
    float_array speed_data_all;
    float_array_init(&speed_data_all);
    allocate_array_for_db_data(&speed_data_all, "Wheel Speed", &conn);
    get_data_from_db("select ((value1+value2+value3+value4)/4)*3.6 as speed from \"Wheel Speed\"", &conn,
                     &speed_data_all);
    for (int32_t i = 0; i < speed_data_all.length; ++i) {
        // printf(" %f", float_array_get(&speed_data_all, i));
    }

    printf("\nShutting down HardCompound!\n");

    printf("Closing database and any connection.\n");
    goto cleanup;
exit:
    return EXIT_OK;
cleanup:
    free(NULL);
    free(session_data);
    free(lap_timestamps);
    float_array_free(&throttle_data_all);
    float_array_free(&brake_data_all);
    float_array_free(&steering_data_all);
    float_array_free(&speed_data_all);
    duckdb_disconnect(&conn);
    duckdb_close(&db);
    goto exit;
}