sgp4/RunTest.cpp

#include "Julian.h"
#include "Tle.h"
#include "SGP4.h"
#include "Globals.h"
#include "Observer.h"
#include "CoordGeodetic.h"
#include "CoordTopographic.h"

#include <list>
#include <string>
#include <iomanip>
#include <iostream>
#include <fstream>
#include <vector>
#include <cstdlib>

void RunTle(Tle tle, double start, double end, double inc) {
    double current = start;
    SGP4 model;
    model.SetTle(tle);
    bool running = true;
    bool first_run = true;
    std::cout << "  " << std::setprecision(0) << tle.NoradNumber() << "  xx" << std::endl;
    while (running) {
        try {
            double val;
            Eci eci;
            if (first_run && current != 0.0) {
                /*
                 * make sure first run is always as zero
                 */
                val = 0.0;
            } else {
                /*
                 * otherwise run as normal
                 */
                val = current;
            }
            model.FindPosition(eci, val);

            Vector position = eci.GetPosition();
            Vector velocity = eci.GetVelocity();

            std::cout << std::setprecision(8) << std::fixed;
            std::cout.width(17);
            std::cout << val << " ";
            std::cout.width(16);
            std::cout << position.GetX() << " ";
            std::cout.width(16);
            std::cout << position.GetY() << " ";
            std::cout.width(16);
            std::cout << position.GetZ() << " ";
            std::cout << std::setprecision(9) << std::fixed;
            std::cout.width(14);
            std::cout << velocity.GetX() << " ";
            std::cout.width(14);
            std::cout << velocity.GetY() << " ";
            std::cout.width(14);
            std::cout << velocity.GetZ() << std::endl;

        } catch (std::exception* ex) {
            std::cout << ex->what() << std::endl;
            running = false;
        }
        if ((first_run && current == 0.0) || !first_run) {
            if (current == end)
                running = false;
            else if (current + inc > end)
                current = end;
            else
                current += inc;
        }
        first_run = false;

    }
}

void tokenize(const std::string& str, std::vector<std::string>& tokens) {

    const std::string& delimiters = " ";

    /*
     * skip delimiters at beginning
     */
    std::string::size_type last_pos = str.find_first_not_of(delimiters, 0);

    /*
     * find first non-delimiter
     */
    std::string::size_type pos = str.find_first_of(delimiters, last_pos);

    while (std::string::npos != pos || std::string::npos != last_pos) {
        /*
         * add found token to vector
         */
        tokens.push_back(str.substr(last_pos, pos - last_pos));
        /*
         * skip delimiters
         */
        last_pos = str.find_first_not_of(delimiters, pos);
        /*
         * find next non-delimiter
         */
        pos = str.find_first_of(delimiters, last_pos);
    }
}

void RunTest(const char* infile) {

    std::ifstream file;

    file.open(infile);

    if (!file.is_open()) {
        std::cerr << "Error opening file" << std::endl;
        return;
    }

    bool got_first_line = false;
    std::string line1;
    std::string line2;
    std::string parameters;

    while (!file.eof()) {
        std::string line;
        std::getline(file, line);

        /*
         * trim spaces
         */
        Tle::TrimLeft(line);
        Tle::TrimRight(line);

        /*
         * skip blank lines or lines starting with #
         */
        if (line.length() == 0 || line[0] == '#') {
            got_first_line = false;
            continue;
        }

        /*
         * find first line
         */
        if (!got_first_line) {

            if (Tle::IsValidLine(line, 1)) {
                /*
                 * store line and now read in second line
                 */
                got_first_line = true;
                line1 = line;

            } else {
                std::cerr << "Error: Badly formatted first line:" << std::endl;
                std::cerr << line << std::endl;
            }
        } else {
            /*
             * no second chances, second line should follow the first
             */
            got_first_line = false;
            /*
             * split line, first 69 is the second line of the tle
             * the rest is the test parameters, if there is any
             */
            line2 = line.substr(0, 69);
            double start = 0.0;
            double end = 1440.0;
            double inc = 120.0;
            if (line.length() > 69) {
                std::vector<std::string> tokens;
                parameters = line.substr(70, line.length() - 69);
                tokenize(parameters, tokens);
                if (tokens.size() >= 3) {
                    start = atof(tokens[0].c_str());
                    end = atof(tokens[1].c_str());
                    inc = atof(tokens[2].c_str());
                }
            }

            /*
             * following line must be the second line
             */
            if (Tle::IsValidLine(line2, 2)) {
                Tle tle("Test", line1, line2);
                RunTle(tle, 0.0, 1440.0, 120.0);
            } else {
                std::cerr << "Error: Badly formatted second line:" << std::endl;
                std::cerr << line2 << std::endl;
            }
        }
    }

    /*
     * close file
     */
    file.close();

    return;
}

int main() {

    const char* file_name = "SGP4-VER.TLE";

    RunTest(file_name);

    return 0;
}