/*
    Copyright (c) 2005-2024 Intel Corporation

    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.
    You may obtain a copy of the License at

        http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
*/

#define VIDEO_WINMAIN_ARGS

#include <iostream>

#include "oneapi/tbb/tick_count.h"
#include "oneapi/tbb/global_control.h"

#include "common/utility/utility.hpp"
#include "common/utility/get_default_num_threads.hpp"

#include "seismic_video.hpp"
#include "universe.hpp"

Universe u;

struct RunOptions {
    //! It is used for console mode for test with different number of threads and also has
    //! meaning for GUI: threads.first  - use separate event/updating loop thread (>0) or not (0).
    //!                  threads.second - initialization value for scheduler
    utility::thread_number_range threads;
    int numberOfFrames;
    int numberOfIterations;
    bool silent;
    bool parallel;
    RunOptions(utility::thread_number_range threads_,
               int number_of_frames_,
               int number_of_iterations_,
               bool silent_,
               bool parallel_)
            : threads(threads_),
              numberOfFrames(number_of_frames_),
              numberOfIterations(number_of_iterations_),
              silent(silent_),
              parallel(parallel_) {}
};

RunOptions ParseCommandLine(int argc, char *argv[]) {
    // zero number of threads means to run serial version
    utility::thread_number_range threads(
        utility::get_default_num_threads, 0, utility::get_default_num_threads());

    int numberOfFrames = 0;
    int numberOfIterations = 0;
    bool silent = false;
    bool serial = false;

    utility::parse_cli_arguments(
        argc,
        argv,
        utility::cli_argument_pack()
            //"-h" option for displaying help is present implicitly
            .positional_arg(threads, "n-of-threads", utility::thread_number_range_desc)
            .positional_arg(numberOfFrames,
                            "n-of-frames",
                            "number of frames the example processes internally (0 means unlimited)")
            .positional_arg(numberOfIterations,
                            "n-of-iterations",
                            "number of iterations the example runs internally")
            .arg(silent, "silent", "no output except elapsed time")
            .arg(serial, "serial", "in GUI mode start with serial version of algorithm"));
    return RunOptions(threads, numberOfFrames, numberOfIterations, silent, !serial);
}

int main(int argc, char *argv[]) {
    oneapi::tbb::tick_count mainStartTime = oneapi::tbb::tick_count::now();
    RunOptions options = ParseCommandLine(argc, argv);
    SeismicVideo video(u, options.numberOfFrames, options.threads.last, options.parallel);
    double rel_error;

    // video layer init
    if (video.init_window(u.UniverseWidth, u.UniverseHeight)) {
        video.calc_fps = true;
        video.threaded = options.threads.first > 0;
        // video is ok, init Universe
        u.InitializeUniverse(video);
        // main loop
        video.main_loop();
    }
    else if (video.init_console()) {
        // do console mode
        if (options.numberOfFrames == 0) {
            options.numberOfFrames = 1000;
            std::cout << "Substituting 1000 for unlimited frames because not running interactively"
                      << "\n";
        }
        // TODO : Extend utility::cli_argument_pack() to allow specifying the default value.
        if (options.numberOfIterations <= 0) {
            options.numberOfIterations = 10;
            std::cout << "Setting the number of iterations = 10 default"
                      << "\n";
        }
        for (int p = options.threads.first; p <= options.threads.last;
             p = options.threads.step(p)) {
            oneapi::tbb::tick_count xwayParallelismStartTime = oneapi::tbb::tick_count::now();
            u.InitializeUniverse(video);
            int numberOfFrames = options.numberOfFrames;
            assert(options.numberOfIterations > 0 && "Number of iterations cannot be <= 0");
            unsigned numberOfIterations = unsigned(options.numberOfIterations);

            if (p == 0) {
                //run a serial version
                for (int i = 0; i < numberOfFrames; ++i) {
                    u.SerialUpdateUniverse();
                }
            }
            else {
                oneapi::tbb::global_control c(oneapi::tbb::global_control::max_allowed_parallelism,
                                              p);
                utility::measurements mu(numberOfIterations);
                for (int iter = 0; iter < numberOfIterations; ++iter) {
                    mu.start();
                    for (int i = 0; i < numberOfFrames; ++i) {
                        u.ParallelUpdateUniverse();
                    }
                    mu.stop();
                }
                rel_error = mu.computeRelError();
            }

            if (!options.silent) {
                double fps =
                    options.numberOfFrames /
                    ((oneapi::tbb::tick_count::now() - xwayParallelismStartTime).seconds());
                std::cout << fps << " frame per sec with ";
                if (p == 0) {
                    std::cout << "serial code"
                              << "\n";
                }
                else {
                    std::cout << p << " way parallelism"
                              << "\n";
                }
            }
        }
    }
    video.terminate();
    utility::report_elapsed_time((oneapi::tbb::tick_count::now() - mainStartTime).seconds());
    utility::report_relative_error(rel_error);
    return 0;
}
