diff options
Diffstat (limited to 'charf.c')
| -rw-r--r-- | charf.c | 39 |
1 files changed, 39 insertions, 0 deletions
@@ -4,6 +4,8 @@ //for multithreading #include <pthread.h> #include <stdbool.h> +//for reading from stdin +#include <string.h> #define DOUBLEMODE 0 #define DOUBLEERRORMODE 1 @@ -258,6 +260,34 @@ void* compute_chunk(void* arguments){ } } +#define OK 0 +#define NO_INPUT 1 +#define TOO_LONG 2 +static int getLine (char *prmpt, char *buff, size_t sz) { + int ch, extra; + + // Get line with buffer overrun protection. + if (prmpt != NULL) { + printf ("%s", prmpt); + fflush (stdout); + } + if (fgets (buff, sz, stdin) == NULL) + return NO_INPUT; + + // If it was too long, there'll be no newline. In that case, we flush + // to end of line so that excess doesn't affect the next call. + if (buff[strlen(buff)-1] != '\n') { + extra = 0; + while (((ch = getchar()) != '\n') && (ch != EOF)) + extra = 1; + return (extra == 1) ? TOO_LONG : OK; + } + + // Otherwise remove newline and give string back to caller. + buff[strlen(buff)-1] = '\0'; + return OK; +} + int main() { pthread_t threads[NUM_THREADS]; Args args[NUM_THREADS]; @@ -276,6 +306,15 @@ int main() { result_code = pthread_create(&threads[i], NULL, compute_chunk, args+i); } + size_t sz = 1024; + char prmpt[] = "To exit enter q,quit or exit.\nTo print current records to terminal enter r.\nTo print current records to a file enter p.\nTo do so with tex formatting enter t.\n"; + char buff[sz]; + bool cont = true; + do { + getLine(prmpt, buff, sz); + if( 0 == strcmp(buff,"q") || 0 == strcmp(buff,"quit") || 0 == strcmp(buff,"exit") ) exit(EXIT_SUCCESS); + } while(cont); + // wait for each thread to complete for (int i = 0; i < NUM_THREADS; i++) { result_code = pthread_join(threads[i], NULL); |