nortega/indivisible-legacy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Read and write primes to and from file.

Browse Source
This commit is contained in:
Nicolás A. Ortega 2016-12-26 15:56:56 +01:00
parent fa3f2dd2b1
commit 6c2f96416b
No known key found for this signature in database
GPG Key ID: 614272579C2070D1
1 changed files with 44 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
src/main.c
View File

@ -7,7 +7,7 @@
#include "list.h" #include "list.h"
#define VERSION "v0.7" #define VERSION "v1.0"
static bool run; static bool run;
@ -20,11 +20,12 @@ int main(int argc, char *argv[]) {
f_version = false, f_version = false,
f_quiet = false; f_quiet = false;
int base = 10; int base = 10;
char *file = NULL; char *ofile = NULL;
char *ifile = NULL;
// Parse commandline arguments // Parse commandline arguments
int c; int c;
while((c = getopt(argc, argv, "hvqb:f:")) != -1) { while((c = getopt(argc, argv, "hvqb:o:i:")) != -1) {
switch(c) { switch(c) {
case 'h': case 'h':
f_help = true; f_help = true;
@ -39,13 +40,16 @@ int main(int argc, char *argv[]) {
base = atoi(optarg); base = atoi(optarg);
if(base < 2 || base > 62) { if(base < 2 || base > 62) {
fprintf(stderr, fprintf(stderr,
"Invalid base `%d'. Base must be between 2 and 62\n", "Invalid base `%d'. Base must be between 2 and 62.\n",
base); base);
exit(1); exit(1);
} }
break; break;
case 'f': case 'o':
file = optarg; ofile = optarg;
break;
case 'i':
ifile = optarg;
break; break;
default: default:
printUsage(argv[0]); printUsage(argv[0]);
@ -60,7 +64,8 @@ int main(int argc, char *argv[]) {
puts(" -v print version number of program"); puts(" -v print version number of program");
puts(" -q quiet mode"); puts(" -q quiet mode");
puts(" -b <base> base in which to print primes between 2 and 62 (default 10)"); puts(" -b <base> base in which to print primes between 2 and 62 (default 10)");
puts(" -f <file> file to save primes to"); puts(" -o <file> file to save primes to");
puts(" -i <file> file to read primes from a previous session");
return 0; return 0;
} else if(f_version) { } else if(f_version) {
printf("Indivisible %s\n", VERSION); printf("Indivisible %s\n", VERSION);
@ -83,6 +88,7 @@ int main(int argc, char *argv[]) {
mpz_t num; mpz_t num;
mpz_init(num); mpz_init(num);
if(ifile == NULL) {
// Add 2, a known prime to this list // Add 2, a known prime to this list
mpz_set_ui(num, 2); mpz_set_ui(num, 2);
addToList(&primes, num); addToList(&primes, num);
@ -94,6 +100,22 @@ int main(int argc, char *argv[]) {
printf("\n"); printf("\n");
} }
mpz_add_ui(num, num, 1); mpz_add_ui(num, num, 1);
} else {
// Load primes from file
FILE *pFile = fopen(ifile, "r");
if(pFile == NULL) {
fprintf(stderr, "Failed to open primes file `%s'.\n", ifile);
goto releaseMemory;
}
while(mpz_inp_str(num, pFile, base) != 0) {
addToList(&primes, num);
}
if(fclose(pFile) != 0) {
fprintf(stderr, "Failed to close file `%s'.\n", ifile);
goto releaseMemory;
}
printf("Loaded %zu primes.\n", primes.end);
}
// Variable for half `num' // Variable for half `num'
mpz_t halfNum; mpz_t halfNum;
@ -132,17 +154,17 @@ nextPrime:
mpz_clear(halfNum); mpz_clear(halfNum);
mpz_clear(num); mpz_clear(num);
if(file != NULL) { if(ofile != NULL) {
FILE *outFile = fopen(file, "w+"); FILE *outFile = fopen(ofile, "w+");
if(outFile == NULL) { if(outFile == NULL) {
fprintf(stderr, "Failed create file `%s'.\n", file); fprintf(stderr, "Failed create file `%s'.\n", ofile);
goto releaseMemory; goto releaseMemory;
} }
printf("Writing primes to `%s'...\n", file); printf("Writing primes to `%s'...\n", ofile);
puts("0%"); puts("0%");
for(size_t i = 0; i < primes.end; ++i) { for(size_t i = 0; i < primes.end; ++i) {
if(mpz_out_str(outFile, base, primes.list[i]) == 0) { if(mpz_out_str(outFile, base, primes.list[i]) == 0) {
fprintf(stderr, "Error occurred while writing to file `%s'.\n", file); fprintf(stderr, "Error occurred while writing to file `%s'.\n", ofile);
goto releaseMemory; goto releaseMemory;
} }
fprintf(outFile, "\n"); fprintf(outFile, "\n");
@ -151,7 +173,7 @@ nextPrime:
else if(i == primes.end * 3 / 4) puts("75%"); else if(i == primes.end * 3 / 4) puts("75%");
} }
if(fclose(outFile) != 0) { if(fclose(outFile) != 0) {
fprintf(stderr, "Failed to close file `%s'.\n", file); fprintf(stderr, "Failed to close file `%s'.\n", ofile);
goto releaseMemory; goto releaseMemory;
} }
puts("100%"); puts("100%");
@ -168,7 +190,7 @@ releaseMemory:
} }
void printUsage(char *progName) { void printUsage(char *progName) {
printf("%s [options...]\n", progName); printf("%s [OPTIONS]\n", progName);
} }
void leave() { run = false; } void leave() { run = false; }