diff options
Diffstat (limited to 'ste')
| -rw-r--r-- | ste/example/Makefile | 2 | ||||
| -rw-r--r-- | ste/ste.c | 19 |
2 files changed, 14 insertions, 7 deletions
diff --git a/ste/example/Makefile b/ste/example/Makefile index f8a6ef4..f9053d7 100644 --- a/ste/example/Makefile +++ b/ste/example/Makefile @@ -12,3 +12,5 @@ ste: ../ste.c main: main.c tmpl.c $(CC) -o $@ $< +clean: + $(RM) ste main tmpl.c @@ -21,29 +21,34 @@ void process_files(FILE* in, FILE* out, const char *macro) { int main(int argc, char **argv) { const char *macro = "OUT"; - char *filename = NULL; + char *filename = NULL, *outname = NULL; FILE *in, *out; if (argc < 2) { usage: - fprintf(stderr, "Usage: %s [-m name] <file.ste>\n", *argv); + fprintf(stderr, "Usage: %s [-m name] [-o out] <file.ste>\n", *argv); return 1; } for (int i = 0; i < argc; ++i) { if (!strcmp("-m", argv[i])) macro = argv[++i]; + else if (!strcmp("-o", argv[i])) outname = argv[++i]; else filename = argv[i]; } if (!filename || strlen(filename) <= 4) goto usage; in = fopen(filename, "r"); - for (char *fn = filename; fn[3]; ++fn) { - if (!strncmp(".ste", fn, 4)) { - *fn = '\0'; - break; + if (!outname) { + for (char *fn = filename; fn[3]; ++fn) { + if (!strncmp(".ste", fn, 4)) { + *fn = '\0'; + break; + } } + out = fopen(filename, "w"); + } else { + out = fopen(outname, "w"); } - out = fopen(filename, "w"); if (!in || !out) { perror("fopen"); goto usage; |