[math] Matrix file I/O

This commit is contained in:
Al
2015-11-30 12:52:57 -05:00
parent 5682c347ac
commit e62eb1e697
2 changed files with 61 additions and 0 deletions

View File

@@ -136,6 +136,64 @@ int matrix_dot_matrix(matrix_t *m1, matrix_t *m2, matrix_t *result) {
return 0;
}
matrix_t *matrix_read(FILE *f) {
matrix_t *mat = malloc(sizeof(matrix_t));
if (mat == NULL) return NULL;
mat->data = NULL;
if (!file_read_uint32(f, &mat->m) ||
!file_read_uint32(f, &mat->n)) {
goto exit_sparse_matrix_allocated;
}
size_t len_data = (size_t)mat->m * (size_t)mat->n;
double_array *data = double_array_new_size(len_data);
if (data == NULL) {
printf("data alloc\n");
goto exit_matrix_allocated;
}
for (size_t i = 0; i < len_data; i++) {
if (!file_read_double(f, data->a + i)) {
printf("data\n");
goto exit_sparse_matrix_allocated;
}
}
data->n = (size_t)len_data;
mat->data = data;
return mat;
exit_matrix_allocated:
matrix_destroy(mat);
return NULL;
}
bool matrix_write(matrix_t *self, FILE *f) {
if (self == NULL || self->data == NULL) {
return false;
}
if (!file_write_uint32(f, self->m) ||
!file_write_uint32(f, self->n)) {
return false;
}
uint64_t len_data = (uint64_t)self->m * (uint64_t)self->n;
for (int i = 0; i < len_data; i++) {
if (!file_write_double(f, self->data->a[i])) {
return false;
}
}
return true;
}
void matrix_destroy(matrix_t *self) {
if (self == NULL) return;

View File

@@ -35,6 +35,9 @@ void matrix_div(matrix_t *self, double value);
void matrix_dot_vector(matrix_t *self, double *vec, double *result);
int matrix_dot_matrix(matrix_t *m1, matrix_t *m2, matrix_t *result);
matrix_t *matrix_read(FILE *f);
bool matrix_write(matrix_t *self, FILE *f);
void matrix_destroy(matrix_t *self);
#endif