diff options
Diffstat (limited to 'src/fractals.c')
| -rw-r--r-- | src/fractals.c | 176 |
1 files changed, 100 insertions, 76 deletions
diff --git a/src/fractals.c b/src/fractals.c index 1c0c4ea..be8c917 100644 --- a/src/fractals.c +++ b/src/fractals.c @@ -3,29 +3,54 @@ #include <stdlib.h> #include <sys/ioctl.h> #include <getopt.h> +#include <string.h> +#include <time.h> #include "grids.h" #include "precision.h" #include "fractals.h" +#define NUM_RUNS 5 -void print_usage(FILE* file, char* program_name){ - fprintf(file, "Usage: %s [-v] [-i iterations] [-x x_res] [-y y_res] [-z magnification] [-l lower_left] [-u upper_right]\n", program_name); +void print_usage(FILE* file, const char* program_name){ + fprintf(file, "Usage: %s [-v] [-i iterations] [-x x_res] [-y y_res] [-z magnification] [-l lower_left] [-u upper_right] [-o output_grid] -f fractal\n", program_name); } void print_help(){ - //TODO: add help + printf("Options:\n" + " -i, --iterations <value> the number of iterations (default: 100)\n" + " -x, --x-res <value> the horizontal resolution (default: terminal width)\n" + " -y, --y-res <value> the vertical resolution (default: terminal height)\n" + " -l, --lower-left <value> Set the lower left corner of the fractal area (default: -2-2i)\n" + " -u, --upper-right <value> Set the upper right corner of the fractal area (default: 2+2i)\n" + " -z, --magnification <value> Set the magnification factor (default: 1)\n" + " -o, --output <filename> the output filename (default: fractal.grid)\n" + " -f, --fractal <type> the fractal type (default: mandelbrot)\n" + " supported fractals: mandelbrot, tricorn, multibrot, multicorn, burning_ship, julia\n" + " -p, --performance print performance info\n" + " -v, --verbose verbose output\n" + " -h, --help prints this help message\n"); } -void print_info(){ +void print_info(const char* program_name){ #ifdef EXTENDED_PRECISION printf("Compiled with long double float precision\n"); #endif #ifndef EXTENDED_PRECISION - printf("Compiled with double float precision\n"); + printf("%s complied with double float precision\n", program_name); #endif } +double time_fractal(fractal_generator generator, grid_t* grid, grid_gen_params* params){ + struct timespec start, end; + clock_gettime(CLOCK_MONOTONIC, &start); + for(size_t i = 0; i < NUM_RUNS; i++){ + generator(grid, params); + } + clock_gettime(CLOCK_MONOTONIC, &end); + return (end.tv_sec - start.tv_sec + (end.tv_nsec - start.tv_nsec) * 1.0e-9) / NUM_RUNS; +} + int main(const int argc, char *argv[]) { struct winsize w; ioctl(STDOUT_FILENO, TIOCGWINSZ, &w); @@ -34,20 +59,16 @@ int main(const int argc, char *argv[]) { size_t iterations = 100; size_t x_res = w.ws_col; size_t y_res = w.ws_row; - CBASE re_lower_left = -2; - CBASE im_lower_left = -2; - CBASE re_upper_right = 2; - CBASE im_upper_right = 2; + complex_t lower_left = { .re = -2, .im = -2}; + complex_t upper_right = { .re = 2, .im = 2}; CBASE magnification = 1; bool verbose = false; - //TODO: allocate adequate size buffer - bool output_to_file = true; + bool performance = false; + grid_gen_params* params; + char* fractal_name; + fractal_generator generator = mandelbrot_grid; char* output_filename = "fractal.grid"; - // TODO: have output format option - // TODO: have output file - - static struct option long_options[] = { {"iterations", required_argument, NULL, 'i'}, {"x-res", required_argument, NULL, 'x'}, @@ -57,13 +78,15 @@ int main(const int argc, char *argv[]) { {"magnification", required_argument, NULL, 'z'}, {"output", required_argument, NULL, 'o'}, {"verbose", no_argument, NULL, 'v'}, + {"performance", no_argument, NULL, 'p'}, {"help", no_argument, NULL, 'h'}, + {"fractal", required_argument, NULL, 'f'}, {0, 0, 0, 0} // Termination element }; //parse command line arguments int opt; - while((opt = getopt_long(argc, argv, "i:x:y:l:u:z:o:vh", long_options, NULL)) != -1){ + while((opt = getopt_long(argc, argv, "i:x:y:l:u:z:o:vphf:", long_options, NULL)) != -1){ switch(opt){ case 'i': iterations = strtoull(optarg, NULL, 10); @@ -75,15 +98,50 @@ int main(const int argc, char *argv[]) { y_res = strtoull(optarg, NULL, 10); break; case 'l': - sscanf(optarg, CFORMAT"+"CFORMAT"i", &re_lower_left, &im_lower_left); + sscanf(optarg, CFORMAT"+"CFORMAT"i", &lower_left.re, &lower_left.im); break; case 'u': - sscanf(optarg, CFORMAT"+"CFORMAT"i", &re_upper_right, &im_upper_right); + sscanf(optarg, CFORMAT"+"CFORMAT"i", &upper_right.re, &upper_right.im); break; case 'o': - //TODO: check if can write to location - output_to_file = true; - //TODO: + output_filename = optarg; + break; + case 'f': + if(strncmp(optarg, "mandelbrot", strlen("mandelbrot")) == 0) { + fractal_name = "mandelbrot"; + generator = mandelbrot_grid; + } + else if(strncmp(optarg, "tricorn", strlen("tricorn")) == 0) { + fractal_name = "tricorn"; + generator = tricorn_grid; + } + else if(strncmp(optarg, "multibrot", strlen("multibrot")) == 0) { + fractal_name = "multibrot"; + generator = multibrot_grid; + params = malloc(sizeof(grid_gen_params)); + params->degree = 3; + } + else if(strncmp(optarg, "multicorn", strlen("multicorn")) == 0) { + fractal_name = "multicorn"; + generator = multicorn_grid; + params = malloc(sizeof(grid_gen_params)); + params->degree = 3; + } + else if(strncmp(optarg, "burning_ship", strlen("burning_ship")) == 0) { + fractal_name = "burning ship"; + generator = burning_ship_grid; + } + else if(strncmp(optarg, "julia", strlen("julia")) == 0) { + fractal_name = "julia"; + generator = julia_grid; + params = malloc(sizeof(grid_gen_params)); + params->cr.radius = 100; + params->cr.constant = (complex_t){ .re = 0.285, .im = 0.01 }; + } + else { + fprintf(stderr, "Invalid fractal type: %s, see --help for a list of supported fractals\n", optarg); + return 1; + } break; case 'z': sscanf(optarg, CFORMAT, &magnification); @@ -95,8 +153,12 @@ int main(const int argc, char *argv[]) { case 'v': verbose = true; break; + case 'p': + performance = true; + break; case 'h': print_usage(stdout, argv[0]); + print_help(); return 0; default: print_usage(stderr, argv[0]); @@ -104,75 +166,37 @@ int main(const int argc, char *argv[]) { } } - //const CBASE complex lower_left = re_lower_left + im_lower_left * I; - const complex_t lower_left = { .re=re_lower_left, .im=im_lower_left}; - //const CBASE complex upper_right = re_upper_right + im_upper_right * I; - const complex_t upper_right = { .re=re_upper_right, .im=im_upper_right}; - grid_t* grid = create_grid(x_res, y_res, iterations, lower_left, upper_right); if(!grid) return 1; if(magnification != 1){ - if(verbose) printf("Magnification: "CFORMAT"\n", magnification); zoom_grid(grid, magnification); } - // params for different fractals - const double degree = 3.0; - // const CBASE complex constant = 0.285L + 0.01L*I; - const complex_t constant = { - .re = 0.285, - .im = 0.01 - }; - // const CBASE complex constant = -0.835L -0.321L* I; - const double radius = 100; - - // enum fractal f = BURNING_SHIP; - enum fractal f = JULIA; - switch(f){ - case MANDELBROT: - mandelbrot_grid(grid); - break; - case TRICORN: - tricorn_grid(grid); - break; - case MULTIBROT: - multibrot_grid(grid, degree); - break; - case MULTICORN: - multicorn_grid(grid, degree); - break; - case BURNING_SHIP: - burning_ship_grid(grid); - break; - case JULIA: - julia_grid(grid, constant, radius); - break; - default: - fprintf(stderr, "Unrecognized fractal\n"); - return 1; + // TODO: make additional param to fractals that take these in + + generator(grid, params); + + if(performance){ + double time = time_fractal(generator, grid, params); + printf("%s,%f\n", fractal_name, time); } - if(verbose)print_grid_info(grid); + if(verbose){ + print_info(argv[0]); + printf("Magnification:\t"CFORMAT"\n", magnification); + print_grid_info(grid); + } - //write grid to file - if(output_to_file){ - FILE* write_file = fopen("test.grid", "wb"); - int err = write_grid(write_file , grid); - if(err == GRID_WRITE_ERROR){ - fprintf(stderr, "Error writing occured while writting to file %s\n", output_filename); - } - fclose(write_file); + FILE* file = fopen(output_filename, "wb"); + if(write_grid(file, grid) == GRID_WRITE_ERROR){ + fprintf(stderr, "Error writing occured while writting to file %s\n", output_filename); } + fclose(file); + free(params); free_grid(grid); - // //attempt to read grid from file - // FILE* read_file = fopen("test2.grid", "rb"); - // grid_t* grid2 = read_grid(read_file); - // fclose(read_file); - // - // printf("Grids are %s equal\n", grid_equal(grid, grid2) ? "exactly" :"not exactly"); return 0; } |
