diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.c | 61 | ||||
| -rw-r--r-- | src/matfile.c | 52 |
2 files changed, 113 insertions, 0 deletions
diff --git a/src/main.c b/src/main.c new file mode 100644 index 0000000..9b6d69d --- /dev/null +++ b/src/main.c @@ -0,0 +1,61 @@ +#include <stdio.h> +#include <assert.h> +#include <string.h> +#include "matfile.h" + +double floats[5] = {0.0, 1.0, 3.14, 2.72, 5.5}; + +int main(void) { + FILE *fp = fopen("test.mat", "w"); + if (fp == NULL) { + perror("fopen"); + return -1; + } + + // Write File Header + assert(write_header(fp)); + + // Setup Array Dimensions + int32_t dims[2] = {1, 5}; + mData_t dim_data; + dim_data.data = (void *)&dims; + dim_data.tag.type = miINT32; + dim_data.tag.size = 8; + + // Array Flags + mArrayFlags_t flags; + memset(&flags, 0, sizeof(flags)); + flags.global = 0; + flags.complex = 0; + flags.logical = 0; + flags.class = mxDOUBLE_CLASS; + + mData_t flags_data; + flags_data.tag.type = miUINT32; + flags_data.tag.size = 8; + flags_data.data = (void *)&flags; + + // Name + mSmallData_t name; + name.tag.type = miINT8; + name.tag.size = 1; + name.data = 'A'; + + mData_t data; + data.tag.type = miDOUBLE; + data.tag.size = sizeof(floats); + data.data = (void *)floats; + + mTag_t array; + array.type = miMATRIX; + array.size = 96; + + fwrite((void *)&array, 1, sizeof(array), fp); + write_data(fp, flags_data); + write_data(fp, dim_data); + write_small_data(fp, name); + write_data(fp, data); + + fclose(fp); + return 0; +} diff --git a/src/matfile.c b/src/matfile.c new file mode 100644 index 0000000..188e14f --- /dev/null +++ b/src/matfile.c @@ -0,0 +1,52 @@ +#include "matfile.h" +#include <string.h> +#include <time.h> +#include <stdlib.h> + +static const char *const default_description = "MATLAB 5.0 MAT-file, Created on: "; + +bool write_header(FILE *file) { + if (file == NULL) { + return false; + } + + mHeader_t header; + memset((void *)&header, 0, sizeof(header)); + memset((void *)header.description, ' ', sizeof(header.description)); + + time_t t = time(NULL); + const char *const strtime = ctime(&t); + + snprintf(header.description, sizeof(header.description), "%s%s", default_description, strtime); + + header.version = 0x0100; + header.endian_indicator = (('M' << 8) | 'I'); + + const size_t status = fwrite((const void *)&header, 1, sizeof(header), file); + + return (status == sizeof(header)); +} + +bool write_data(FILE *file, const mData_t data) { + if (data.data == NULL) { + return false; + } + + bool success = true; + success = (fwrite((const void *)&data.tag, 1, sizeof(data.tag), file) == sizeof(data.tag)) && success; + success = (fwrite(data.data, 1, data.tag.size, file) == data.tag.size) && success; + + const size_t padding = data.tag.size % 8; + const uint8_t padding_buffer[8] = {0}; + if (padding > 0 && success) { + success = (fwrite((const void *)padding_buffer, 1, padding, file) == padding) && success; + } + + return success; +} + +bool write_small_data(FILE *file, const mSmallData_t data) { + const size_t w = fwrite((const void *)&data, 1, sizeof(data), file); + + return (w == sizeof(data)); +}
\ No newline at end of file |
