nuclear@0: #include <stdio.h>
nuclear@0: #include <stdlib.h>
nuclear@0: #include <string.h>
nuclear@0: #include <limits.h>
nuclear@0: #include <dlfcn.h>
nuclear@0: #include <imago2.h>
nuclear@0: 
nuclear@0: int parse_opt(int argc, char **argv);
nuclear@0: 
nuclear@0: char *inpat, *outpat = "out%4d.png";
nuclear@0: int range_min, range_max = INT_MAX;
nuclear@0: float fps = 30;
nuclear@0: char *plugin_name;
nuclear@0: 
nuclear@0: void *dl;
nuclear@0: int (*init)(void);
nuclear@0: void (*shutdown)(void);
nuclear@0: int (*process)(struct img_pixmap*, int, float);
nuclear@0: 
nuclear@0: int main(int argc, char **argv)
nuclear@0: {
nuclear@0: 	char *fname;
nuclear@0: 	int i, frame_width = -1, frame_height;
nuclear@0: 
nuclear@0: 	if(parse_opt(argc, argv) == -1) {
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	if(!(dl = dlopen(plugin_name, RTLD_LAZY))) {
nuclear@0: 		fprintf(stderr, "failed to open post plugin: %s: %s\n", plugin_name, dlerror());
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 	init = dlsym(dl, "init");
nuclear@0: 	shutdown = dlsym(dl, "shutdown");
nuclear@0: 	if(!(process = dlsym(dl, "process"))) {
nuclear@0: 		fprintf(stderr, "invalid post plugin, failed to load process func: %s\n", dlerror());
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	if(init) {
nuclear@0: 		if(init() == -1) {
nuclear@0: 			fprintf(stderr, "init failed\n");
nuclear@0: 			return 1;
nuclear@0: 		}
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	fname = alloca(strlen(inpat) + 11);
nuclear@0: 
nuclear@0: 	for(i=range_min; i<range_max; i++) {
nuclear@0: 		struct img_pixmap frame;
nuclear@0: 		float tmsec = (float)i / fps;
nuclear@0: 
nuclear@0: 		sprintf(fname, inpat, i);
nuclear@0: 
nuclear@0: 		img_init(&frame);
nuclear@0: 		if(img_load(&frame, fname) == -1) {
nuclear@0: 			fprintf(stderr, "file not found: %s\n", fname);
nuclear@0: 			img_destroy(&frame);
nuclear@0: 			break;
nuclear@0: 		}
nuclear@0: 		if(frame_width == -1) {
nuclear@0: 			frame_width = frame.width;
nuclear@0: 			frame_height = frame.height;
nuclear@0: 		}
nuclear@0: 		if(frame_width != frame.width || frame_height != frame.height) {
nuclear@0: 			fprintf(stderr, "frame %d differs in dimensions! (%dx%d instead of %dx%d)\n", i, frame.width, frame.height, frame_width, frame_height);
nuclear@0: 			img_destroy(&frame);
nuclear@0: 			break;
nuclear@0: 		}
nuclear@0: 
nuclear@0: 		img_convert(&frame, IMG_FMT_RGB24);
nuclear@0: 		if(process(&frame, i, tmsec) == -1) {
nuclear@0: 			fprintf(stderr, "the post plugin signalled an error\n");
nuclear@0: 			img_destroy(&frame);
nuclear@0: 			break;
nuclear@0: 		}
nuclear@0: 
nuclear@0: 		sprintf(fname, outpat, i);
nuclear@0: 		if(img_save(&frame, fname) == -1) {
nuclear@0: 			fprintf(stderr, "failed to save frame %d, aborting\n", i);
nuclear@0: 			img_destroy(&frame);
nuclear@0: 			break;
nuclear@0: 		}
nuclear@0: 		img_destroy(&frame);
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	if(shutdown) {
nuclear@0: 		shutdown();
nuclear@0: 	}
nuclear@0: 	return 0;
nuclear@0: }
nuclear@0: 
nuclear@0: int parse_opt(int argc, char **argv)
nuclear@0: {
nuclear@0: 	int i;
nuclear@0: 	char *endp;
nuclear@0: 
nuclear@0: 	for(i=1; i<argc; i++) {
nuclear@0: 		if(argv[i][0] == '-') {
nuclear@0: 			if(argv[i][2] == 0) {
nuclear@0: 				switch(argv[i][1]) {
nuclear@0: 				case 'i':
nuclear@0: 					inpat = argv[++i];
nuclear@0: 					break;
nuclear@0: 
nuclear@0: 				case 'o':
nuclear@0: 					outpat = argv[++i];
nuclear@0: 					break;
nuclear@0: 
nuclear@0: 				case 'r':
nuclear@0: 					fps = strtod(argv[++i], &endp);
nuclear@0: 					if(endp == argv[i]) {
nuclear@0: 						fprintf(stderr, "-r must be followed by a number\n");
nuclear@0: 						return -1;
nuclear@0: 					}
nuclear@0: 					break;
nuclear@0: 
nuclear@0: 				case 's':
nuclear@0: 					range_min = strtol(argv[++i], &endp, 10);
nuclear@0: 					if(0) {
nuclear@0: 				case 'e':
nuclear@0: 						range_max = strtol(argv[++i], &endp, 10);
nuclear@0: 					}
nuclear@0: 					if(endp == argv[i]) {
nuclear@0: 						fprintf(stderr, "-%c must be followed by a number\n", argv[i - 1][1]);
nuclear@0: 						return -1;
nuclear@0: 					}
nuclear@0: 					break;
nuclear@0: 
nuclear@0: 				case 'p':
nuclear@0: 					plugin_name = argv[++i];
nuclear@0: 					break;
nuclear@0: 
nuclear@0: 				default:
nuclear@0: 					goto invalid;
nuclear@0: 				}
nuclear@0: 			} else {
nuclear@0: 				goto invalid;
nuclear@0: 			}
nuclear@0: 		} else {
nuclear@0: 			goto invalid;
nuclear@0: 		}
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	if(!inpat) {
nuclear@0: 		fprintf(stderr, "you must specify an input image sequence\n");
nuclear@0: 		return -1;
nuclear@0: 	}
nuclear@0: 	if(!plugin_name) {
nuclear@0: 		fprintf(stderr, "you must specify a post plugin\n");
nuclear@0: 		return -1;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	printf("%s -> [%s] -> %s\n", inpat, plugin_name, outpat);
nuclear@0: 	/*if(range_min > 0 || range_max < INT_MAX) {*/
nuclear@0: 		printf("frames: [%d, ", range_min);
nuclear@0: 		if(range_max < INT_MAX) {
nuclear@0: 			printf("%d]\n", range_max);
nuclear@0: 		} else {
nuclear@0: 			printf("inf]\n");
nuclear@0: 		}
nuclear@0: 	/*}*/
nuclear@0: 	printf("framerate: %f fps\n", fps);
nuclear@0: 
nuclear@0: 	return 0;
nuclear@0: 
nuclear@0: invalid:
nuclear@0: 	fprintf(stderr, "invalid argument: %s\n", argv[i]);
nuclear@0: 	return -1;
nuclear@0: }