Mandelbrot/main.cpp at master · Aaron-Fine/Mandelbrot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include <iostream>
#include <fstream>
#include <chrono>

#include "Utils.h"
#include "Mandelbrot.h"
#include "LinearColoring.h"
#include "BernsteinContinuous.h"
#include "Greyscale.h"

int main( int numArgs, char* args[] )
{
    std::string fileName;
    MandelConfig* config;
    Mandelbrot* mandelbrot;

    if ( numArgs < 2 )
    {
        config = new MandelConfig();
    } else
    {
        fileName = args[ 1 ];
        config = new MandelConfig( fileName );
    }

    std::cout << "Choose coloring scheme by entering integer associated with desired option:\n";
    std::cout << "0) Default\n";
    std::cout << "1) Linear (default coloring)\n";
    std::cout << "2) Linear (custom coloring)\n";
    std::cout << "3) Bernstein Continuous (default coloring)\n";
    std::cout << "4) Bernstein Continuous (custom coloring)\n";
    std::string choice;
    std::cin >> choice;

    switch ( convertStringToUnsignedInt( choice ))
    {
        case 1:
        {
            mandelbrot = new LinearColoring( *config, 255, 255, 102, 102, 0, 0 );
            break;
        }
        case 2:
        {
            std::string red1, green1, blue1;
            std::string red2, green2, blue2;

            std::cout << "Please enter RGB triplets between 0 and 255\nFirst Color: ";
            std::cin >> red1 >> green1 >> blue1;
            std::cout << "Second Color: ";
            std::cin >> red2 >> green2 >> blue2;
            mandelbrot = new LinearColoring( *config,
                                             convertStringToInt( red1 ),
                                             convertStringToInt( green1 ),
                                             convertStringToInt( blue1 ),
                                             convertStringToInt( red2 ),
                                             convertStringToInt( green2 ),
                                             convertStringToInt( blue2 ));
            break;
        }
        case 3:
        {
            mandelbrot = new BernsteinContinuous( *config );
            break;
        }
        case 4:
        {
            std::string red, green, blue;
            std::cout << "Please enter positive floats for the red, green, and blue color curve coefficents:\n";
            std::cout << "Red  :";
            std::cin >> red;
            std::cout << "Green:";
            std::cin >> green;
            std::cout << "Blue :";
            std::cin >> blue;

            mandelbrot = new BernsteinContinuous( *config,
                                                  convertStringToDouble( red ),
                                                  convertStringToDouble( green ),
                                                  convertStringToDouble( blue ));
            break;
        }
        default:
        {
            mandelbrot = new Greyscale( *config );
        }
    }

    mandelbrot->generate();
    mandelbrot->save();

//    mandelbrot->outputIterations("debug.csv");
//    mandelbrot->iterationHistogram("iterationHistogram.csv");


    return 0;
}