raytrace/main.cpp at master · simply-nicky/raytrace · 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
#include <headers/trace.hpp>
#include <headers/make_trace.hpp>
#include <headers/include.hpp>

int main (int ac, char * av[])
{
    using namespace raytrace;
    try
    {
        auto check = [](const std::string & str)
        {
            return [&str](const std::vector<double> & vec)
            {
                for(const auto & x : vec)
                    if(x < 0.0)
                        throw std::invalid_argument(str + std::string(" must be a positive number."));
            };
        };
        po::options_description generic("Generic options");
        generic.add_options()
            ("help,h", "This usage information.")
            ("help-config-file", "Print config file manual.")
            ("rays,N", po::value<int>()->value_name(" ")->notifier([](const int n){if(n < 0) throw std::invalid_argument("The number of rays must be a positive number."); }), "Number of rays to trace.")
            ("verbose,v", po::value<int>()->value_name(" ")->default_value(0), "Verbosity of output.")
            ("write-mode", po::value<std::string>()->value_name(" ")->default_value("Trace")->notifier([](const std::string & str){if(str != "Trace" && str != "Detector") throw std::invalid_argument("Write mode must be Full or Detector."); }), "Ray-trace write mode:\nTrace - write ray-traces\nDetector - detector write points")
            ("config-filename", po::value<std::string>()->value_name(" ")->default_value(std::string("config.ini")), "Specify config filename.\nThe file must be in the same folder as the executable.")
            ;
        po::options_description config("Configuration");
        config.add_options()
            ("beam-mode", po::value<std::string>()->value_name(" ")->default_value("Plane")->notifier([](const std::string & str){if(str != "Spherical" && str != "Plane") throw std::invalid_argument("Beam mode must be Spherical or Plane."); }), "Incident beam mode: Spherical or Plane.")
            ("dimensions", po::value<std::string>()->value_name(" ")->default_value("3D")->notifier([](const std::string & str){if(str != "2D" && str != "3D") throw std::invalid_argument("Dimensions must be 2D or 3D."); }), "3D or 2D Ray-tracing.")
            ("inc-angle", po::value<float>()->value_name(" ")->default_value(0), "Incident beam angle.")
            ("rms-height",po::value<float>()->value_name(" ")->default_value(Constants::RMSHeight, "0.35 nm")->notifier([](const double val){if(val < 0.0) throw std::invalid_argument("RMSHeight must be a positive number."); }),"Root-mean-square height of surface roughness.")
            ("corr-length",po::value<float>()->value_name(" ")->default_value(Constants::CorrLength, "2.7 um")->notifier([](const double val){if(val < 0.0) throw std::invalid_argument("Correlation length must be a positive number."); }),"Correlation length of surface roughness.")
            ("imp-mode", "Ray tracing with imperfections, config file is required.")
            ;
        po::options_description config_file("Configuration files");
        config_file.add_options()
            ("Sphere.height", po::value<std::vector<double>>()->notifier(check("height"))->value_name(" "), "Imperfection sphere height.")
            ("Sphere.radius", po::value<std::vector<double>>()->notifier(check("radius"))->value_name(" "), "Imperfection sphere radius.")
            ("Sphere.location.x", po::value<std::vector<double>>()->value_name(" "), "Imperfection sphere location: x coordinate")
            ("Sphere.location.y", po::value<std::vector<double>>()->value_name(" "), "Imperfection sphere location: y coordinate")
            ;
        po::options_description all("Ray tracing on spherical surfaces.\nAllowed options");
        all.add(generic).add(config);
        po::variables_map vm;
        po::store(po::parse_command_line(ac, av, all), vm);
        if(vm.count("imp-mode"))
        {
            std::ifstream config_stream {vm["config-filename"].as<std::string>()};
            if(config_stream)
                po::store(po::parse_config_file(config_stream, config_file), vm);
        }
        po::notify(vm);

        if(ac == 1 || vm.count("help"))
            std::cout << all;
        else if(vm.count("help-config-file"))
            std::cout << config_file;
        else if(vm.count("rays"))
        {
            trace_parser tr (vm);
            tr.run();
        }
        else
            std::cout << "Number of rays is necessary.\n";
    }
    catch (const std::exception & e)
    {
        std::cerr << e.what() << std::endl;
    }
}