(Slightly) More advanced detection of input file type

git-svn-id: svn://cook.msm.cam.ac.uk:745/diff-tomo/dtr@21 bf6ca9ba-c028-0410-8290-897cf20841d1

1 files changed, 17 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
index cc302bb..681165a 100644
--- a/src/main.c
+++ b/src/main.c
@@ -175,6 +175,21 @@ int main(int argc, char *argv[]) {
 	ctx = malloc(sizeof(ControlContext));
 	type = INPUT_NONE;
 	
+	if ( qdrp_is_qdrprc(filename) ) {
+		printf("QDRP input file detected.\n");
+		ctx->inputfiletype = INPUT_QDRP;
+	} else if ( mrc_is_mrcfile(filename) ) {
+		printf("MRC tomography file detected.\n");
+		ctx->inputfiletype = INPUT_MRC;
+	} else if ( cache_is_cachefile(filename) ) {
+		printf("reflect.cache detected.\n");
+		ctx->inputfiletype = INPUT_CACHE;
+	} else {
+		fprintf(stderr, "Unrecognised input file type\n");
+		return 1;
+	}
+	
+	#if 0
 	if ( strcmp(filename, "qdrp.rc") == 0 ) {
 		printf("QDRP input file detected.\n");
 		ctx->inputfiletype = INPUT_QDRP;
@@ -188,6 +203,8 @@ int main(int argc, char *argv[]) {
 		fprintf(stderr, "Unrecognised input file type\n");
 		return 1;
 	}
+	#endif
+	
 	ctx->filename = strdup(argv[1]);
 	ctx->max_d = 0;