sosc_test/rbm/main.c

304 lines
8.3 KiB
C
Raw Permalink Normal View History

2020-12-10 20:42:39 +00:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#define MAX_LINE_LEN 2048
int end_chk;
typedef struct list_t list_t;
typedef struct list_t {
list_t* next;
float data[3];
uint32_t faces[3][3];
};
list_t* list_create() {
list_t* list = malloc(sizeof(list_t));
list->next = NULL;
return list;
}
void list_destroy(list_t* list) {
list_t* next = list->next;
free(list);
if(next != NULL)
list_destroy(next);
}
#define ISWP(X) swap(X, 4)
void swap(uint8_t* in, int length) {
2020-12-10 23:59:20 +00:00
if(end_chk)
return;
2020-12-10 20:42:39 +00:00
uint8_t tmp[8] = { 0 };
for(int i = 0; i < length; ++i)
tmp[i] = in[length - i - 1];
memcpy(in, tmp, length);
}
2020-12-10 23:59:20 +00:00
void write(FILE* fp, const uint8_t* data, int length) {
uint8_t tmp[8] = { 0 };
memcpy(tmp, data, length);
swap(tmp, length);
fwrite(tmp, 1, length, fp);
}
2020-12-10 20:42:39 +00:00
int main(int argc, char* argv[]) {
2020-12-10 23:59:20 +00:00
if(sizeof(float) != 4) {
printf("Operating system must define floats as four bytes.");
return -1;
}
2020-12-10 20:42:39 +00:00
if(argc < 2) {
printf("You must pass the OBJ file to be converted.");
return -1;
}
FILE* fpi = fopen(argv[1], "rb");
if(fpi == NULL) {
printf("File '%s' does not exist.", argv[1]);
return -1;
}
char* out;
if(argc >= 3)
out = argv[2];
else {
out = malloc(sizeof(char) * strlen(argv[1]) + 5);
strcpy(out, argv[1]);
char* ext = NULL;
if((ext = strrchr(out, '.')) == NULL)
ext = out + strlen(out);
strcpy(ext, ".rbm");
}
FILE* fpo = fopen(out, "wb");
if(fpo == NULL) {
printf("Could not write to file '%s'.", out);
fclose(fpi);
free(out);
return -1;
}
// 0 for big endian, 1 for little endian
uint16_t test = 0xB00B;
end_chk = ((uint8_t*)&test)[0] == 0x0B;
uint32_t vertex_cnt = 0,
normal_cnt = 0,
tex_cnt = 0,
face_cnt = 0;
// NOTE: to prevent writing special cases keep in mind
// the first node in these lists is unused and is just
// a blank root. do NOT include this node
list_t *verts = list_create(),
*normals = list_create(),
*texs = list_create(),
*faces = list_create();
list_t *verts_at = verts,
*normals_at = normals,
*texs_at = texs,
*faces_at = faces;
char* chk;
char line[MAX_LINE_LEN + 2];
for(;;) {
line[MAX_LINE_LEN] = '\n';
chk = fgets(line, MAX_LINE_LEN + 2, fpi);
if(feof(fpi) && chk == NULL)
break;
if(line[MAX_LINE_LEN] != '\n') {
printf("Lines cannot exceed %i characters.", MAX_LINE_LEN);
return -1;
}
2020-12-10 23:59:20 +00:00
list_t* next = list_create();
int used = 0;
if(strncmp("v ", line, 2) == 0) {
used = 1;
++vertex_cnt;
sscanf(line, "%*s %f %f %f",
&next->data[0],
&next->data[1],
&next->data[2]
);
verts_at->next = next;
verts_at = next;
} else if(strncmp("vn ", line, 3) == 0) {
used = 1;
++normal_cnt;
sscanf(line, "%*s %f %f %f",
&next->data[0],
&next->data[1],
&next->data[2]
);
2020-12-10 20:42:39 +00:00
2020-12-10 23:59:20 +00:00
normals_at->next = next;
normals_at = next;
} else if(strncmp("vt ", line, 3) == 0) {
used = 1;
++tex_cnt;
sscanf(line, "%*s %f %f",
&next->data[0],
&next->data[1]
);
texs_at->next = next;
texs_at = next;
} else if(strncmp("f ", line, 2) == 0) {
used = 1;
++face_cnt;
if(vertex_cnt > 0 && tex_cnt > 0 && normal_cnt > 0) {
sscanf(line, "%*s %u/%u/%u %u/%u/%u %u/%u/%u",
&next->faces[0][0], &next->faces[0][1], &next->faces[0][2],
&next->faces[1][0], &next->faces[1][1], &next->faces[1][2],
&next->faces[2][0], &next->faces[2][1], &next->faces[2][2]
);
} else if(vertex_cnt > 0 && tex_cnt > 0 && normal_cnt == 0) {
sscanf(line, "%*s %u/%u %u/%u %u/%u",
&next->faces[0][0], &next->faces[0][1],
&next->faces[1][0], &next->faces[1][1],
&next->faces[2][0], &next->faces[2][1]
);
} else if(vertex_cnt > 0 && tex_cnt == 0 && normal_cnt > 0) {
sscanf(line, "%*s %u//%u %u//%u %u//%u",
&next->faces[0][0], &next->faces[0][2],
&next->faces[1][0], &next->faces[1][2],
&next->faces[2][0], &next->faces[2][2]
);
} else if(vertex_cnt > 0 && tex_cnt == 0 && normal_cnt == 0) {
sscanf(line, "%*s %u %u %u",
&next->faces[0][0],
&next->faces[1][0],
&next->faces[2][0]
);
} else {
printf("OBJ file must at least include vertices!");
return -1;
}
faces_at->next = next;
faces_at = next;
}
if(!used)
free(next);
2020-12-10 20:42:39 +00:00
}
fclose(fpi);
2020-12-10 23:59:20 +00:00
if(face_cnt == 0) {
printf("OBJ file must have faces!");
return -1;
}
2020-12-10 23:59:20 +00:00
uint8_t magic[] = {0xDE, 0xAF, 0xB0, 0x0B};
fwrite(magic, 1, 4, fpo);
write(fpo, (uint8_t*)&vertex_cnt, 4);
write(fpo, (uint8_t*)&tex_cnt, 4);
write(fpo, (uint8_t*)&normal_cnt, 4);
write(fpo, (uint8_t*)&face_cnt, 4);
list_t* at = verts->next;
while(at != NULL) {
write(fpo, (uint8_t*)&(at->data[0]), 4);
write(fpo, (uint8_t*)&(at->data[1]), 4);
write(fpo, (uint8_t*)&(at->data[2]), 4);
at = at->next;
}
at = texs->next;
while(at != NULL) {
write(fpo, (uint8_t*)&(at->data[0]), 4);
write(fpo, (uint8_t*)&(at->data[1]), 4);
at = at->next;
}
at = normals->next;
while(at != NULL) {
write(fpo, (uint8_t*)&(at->data[0]), 4);
write(fpo, (uint8_t*)&(at->data[1]), 4);
write(fpo, (uint8_t*)&(at->data[2]), 4);
at = at->next;
}
int face_length = 4;
if(face_cnt <= 0xFFFF)
face_length = 2;
at = faces->next;
while(at != NULL) {
if(tex_cnt > 0 && normal_cnt > 0) {
write(fpo, (uint8_t*)&(at->faces[0][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[0][1]), face_length);
write(fpo, (uint8_t*)&(at->faces[0][2]), face_length);
write(fpo, (uint8_t*)&(at->faces[1][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[1][1]), face_length);
write(fpo, (uint8_t*)&(at->faces[1][2]), face_length);
write(fpo, (uint8_t*)&(at->faces[2][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[2][1]), face_length);
write(fpo, (uint8_t*)&(at->faces[2][2]), face_length);
} else if(tex_cnt > 0 && normal_cnt == 0) {
write(fpo, (uint8_t*)&(at->faces[0][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[0][1]), face_length);
write(fpo, (uint8_t*)&(at->faces[1][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[1][1]), face_length);
write(fpo, (uint8_t*)&(at->faces[2][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[2][1]), face_length);
} else if(tex_cnt == 0 && normal_cnt > 0) {
write(fpo, (uint8_t*)&(at->faces[0][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[0][2]), face_length);
write(fpo, (uint8_t*)&(at->faces[1][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[1][2]), face_length);
write(fpo, (uint8_t*)&(at->faces[2][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[2][2]), face_length);
} else {
write(fpo, (uint8_t*)&(at->faces[0][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[1][0]), face_length);
write(fpo, (uint8_t*)&(at->faces[2][0]), face_length);
}
at = at->next;
}
list_destroy(verts);
list_destroy(normals);
list_destroy(texs);
list_destroy(faces);
2020-12-10 20:42:39 +00:00
fclose(fpo);
free(out);
return 0;
}