st2csv.cpp

// test_rtcm.cpp : This file contains the 'main' function. Program execution begins and ends there.
//

#include <iostream>
#include <vector>
#include <map>
#include <string>
#include <algorithm>

#define MAXFIELD 100

#define PI          3.1415926535897932  /* pi */
#define D2R         (PI/180.0)          /* deg to rad */
#define R2D         (180.0/PI)          /* rad to deg */


static int parse_fields(char* const buffer, char** val, char key)
{
    char* p, * q;
    int n = 0;

    /* parse fields */
    for (p = buffer; *p && n < MAXFIELD; p = q + 1) {
        if (p == NULL) break;
        if ((q = strchr(p, key)) || (q = strchr(p, '\n')) || (q = strchr(p, '\r'))) {
            val[n++] = p; *q = '\0';
        }
        else break;
    }
    return n;
}

static void set_output_file_name(const char* fname, const char* key, char* outfname)
{
    char filename[255] = { 0 }, outfilename[255] = { 0 };
    strcpy(filename, fname);
    char* temp = strrchr(filename, '.');
    if (temp) temp[0] = '\0';
    sprintf(outfname, "%s-%s", filename, key);
}


static FILE* set_output_file(const char* fname, const char* key)
{
    char filename[255] = { 0 };
    set_output_file_name(fname, key, filename);
    return fopen(filename, "w");
}

struct mount_t
{
    std::string name;
    double lat;
    double lon;
    int freq;
    std::string sys;
    std::string rcv;
    std::string country;
    std::string antenna;
};

static void read_st(const char* fname, std::map<std::string, mount_t>& mMount)
{
    FILE* fLOG = fopen(fname, "rt");

    if (fLOG ==NULL) return;

    FILE* fOUT = set_output_file(fname, "-out.json");

    char buffer[255] = { 0 };
    char* val[MAXFIELD];

    int line_index = 0;
    int count = 0;

    while (fLOG && !feof(fLOG))
    {
        fgets(buffer, sizeof(buffer), fLOG);
        if (!(buffer[0] == 'S' && buffer[1] == 'T' && buffer[2] == 'R')) continue;
        int num = parse_fields(buffer, val, ';');
        if (num < 14) continue;

        std::string name = std::string(val[1]);
        int freq = atoi(val[5]);
        std::string sys = std::string(val[6]);
        std::string country = std::string(val[8]);
        double lat = atof(val[9]);
        double lon = atof(val[10]);
        std::string rcv = std::string(val[13]);

        mount_t mount;
        mount.freq = freq;
        mount.country = country;
        mount.sys = sys;
        mount.rcv = rcv;
        mount.lat = lat;
        mount.lon = lon;
        mount.name = name;

        mMount[name] = mount;

        if (fOUT)
        {
            if (count == 0)
            {
                fprintf(fOUT, "[");
            }
            else
            {
                fprintf(fOUT, ",");
            }
            std::string new_name;
            if (name.length() > 4)
                new_name = name.substr(name.length() - 4);
            else
            {
                for (int i = name.length(); i < 4; ++i)
                    new_name += "0";
                new_name += name;
            }
            transform(new_name.begin(), new_name.end(), new_name.begin(), ::toupper);
            std::string new_country;
            if (country.length() > 3)
                new_country = country.substr(country.length() - 3);
            else
            {
                for (int i = country.length(); i < 3; ++i)
                    new_country += "0";
                new_country += country;
            }
            fprintf(fOUT, "{\"no\":%i,\"host\":\"ntrip.info\",\"port\":\"2101\",\"mountpoint\": \"%s\",\"username\": \"user\",\"password\": \"password\",\"name\": \"00%s000%s\",\"type\": \"http\"}", count, name.c_str(), new_country.c_str(), new_name.c_str());
        }

        ++count;

    }
    if (fLOG) fclose(fLOG);
    if (fOUT)
    {
        if (count > 0)
        {
            fprintf(fOUT, "]");
        }
        fclose(fOUT);
    }
    return;
}

int read_csv(const char* fname, std::map<std::string, mount_t>& mMount)
{
    FILE* fLOG = fopen(fname, "rt");

    if (fLOG == NULL) return 0;

    FILE* fOUT = set_output_file(fname, "-out.json");

    char buffer[255] = { 0 };
    char* val[MAXFIELD];

    int line_index = 0;

    int count = 0;

    while (fLOG && !feof(fLOG) && fgets(buffer, sizeof(buffer), fLOG))
    {
        ++line_index;
        if (line_index < 2) continue;

        int num = parse_fields(buffer, val, ',');
        if (num < 2) continue;

        std::string country = std::string(val[0]);

        std::string name = std::string(val[1]);

        double lat = (num > 2) ? atof(val[2]) : 0;
        double lon = (num > 3) ? atof(val[3]) : 0;

        mount_t mount;
        mount.lat = lat;
        mount.lon = lon;
        mount.name = name;
        mount.country = country;

        if (num > 4) mount.rcv = std::string(val[4]);
        //if (num > 5) mount.antenna = std::string(val[5]);

        mMount[name] = mount;

        if (fOUT)
        {
            std::string new_name;
            if (name.length() > 4)
                new_name = name.substr(name.length() - 4);
            else
            {
                for (int i = name.length(); i < 4; ++i)
                    new_name += "0";
                new_name += name;
            }
            transform(new_name.begin(), new_name.end(), new_name.begin(), ::toupper);
            std::string new_country;
            if (country.length() > 3)
                new_country = country.substr(country.length() - 3);
            else
            {
                for (int i = country.length(); i < 3; ++i)
                    new_country += "0";
                new_country += country;
            }
            fprintf(fOUT, "{\"no\":%i,\"host\":\"ntrip.info\",\"port\":\"2101\",\"mountpoint\": \"%s\",\"username\": \"user\",\"password\": \"password\",\"name\": \"00%s000%s\,\"type\": \"http\"},", count, name.c_str(), new_country.c_str(), new_name.c_str());
        }
        ++count;
    }
    if (fLOG) fclose(fLOG);
    if (fOUT)
    {
        if (count > 0)
        {
            fprintf(fOUT, "]");
        }
        fclose(fOUT);
    }
    return (int)mMount.size();
}

static double lat2local(double lat, double* lat2north)
{
    double f_WGS84 = (1.0 / 298.257223563);
    double e2WGS84 = (2.0 * f_WGS84 - f_WGS84 * f_WGS84);
    double slat = sin(lat);
    double clat = cos(lat);
    double one_e2_slat2 = 1.0 - e2WGS84 * slat * slat;
    double Rn = 6378137.0 / sqrt(one_e2_slat2);
    double Rm = Rn * (1.0 - e2WGS84) / (one_e2_slat2);
    *lat2north = Rm;
    return Rn * clat;
}

int diff_csv(const char* fname1, const char* fname2)
{
    std::map<std::string, mount_t> mMount1;
    std::map<std::string, mount_t> mMount2;
    read_csv(fname1, mMount1);
    read_csv(fname2, mMount2);
    for (std::map<std::string, mount_t>::iterator pM1 = mMount1.begin(); pM1 != mMount1.end(); ++pM1)
    {
        if (fabs(pM1->second.lat) < 0.001 && fabs(pM1->second.lon) < 0.001) continue;
        double bestD = -1;
        double l2n = 0;
        double l2e = lat2local(pM1->second.lat * 3.1415926 / 180.0, &l2n);
        std::string bestM;
        for (std::map<std::string, mount_t>::iterator pM2 = mMount2.begin(); pM2 != mMount2.end(); ++pM2)
        {
            if (fabs(pM2->second.lat) < 0.001 && fabs(pM2->second.lon) < 0.001) continue;
            double dB = pM1->second.lat - pM2->second.lat;
            double dL = pM1->second.lon - pM2->second.lon;
            double dN = dB * 3.1415926 / 180.0 * l2n;
            double dE = dL * 3.1415926 / 180.0 * l2e;
            double currD = sqrt(dN * dN + dE * dE);
            if (bestD < 0 || currD < bestD)
            {
                bestD = currD;
                bestM = pM2->first;
            }
        }
        if (bestD < 0)
        {

        }
        else
        {
            std::string name;
            if (pM1->first.length() > 4) 
                name = pM1->first.substr(pM1->first.length() - 4);
            else
            {
                for (int i = pM1->first.length(); i < 4; ++i)
                    name += "0";
                name += pM1->first;
            }
            transform(name.begin(), name.end(), name.begin(),::toupper);
            std::string country;
            if (pM1->second.country.length() > 3)
                country = pM1->second.country.substr(pM1->second.country.length() - 3);
            else
            {
                for (int i = pM1->second.country.length(); i < 3; ++i)
                    country += "0";
                country += pM1->second.country;
            }
            printf("%s,%s,%7.2f,%7.2f,%s,%7.2f,%s,name=00%s000%s&mountpoint=%s'\n", pM1->second.country.c_str(), pM1->first.c_str(), pM1->second.lat, pM1->second.lon, pM1->second.rcv.c_str(), bestD / 1000.0, bestM.c_str(), country.c_str(), name.c_str(), pM1->first.c_str());
        }
    }
    return 0;
}
int exclude_csv(const char* fname1, const char* fname2)
{
    std::map<std::string, mount_t> mMount1;
    std::map<std::string, mount_t> mMount2;
    read_csv(fname1, mMount1);
    read_csv(fname2, mMount2);
    FILE* fOUT = set_output_file(fname1, "-exclude.csv");
    for (std::map<std::string, mount_t>::iterator pM1 = mMount1.begin(); pM1 != mMount1.end(); ++pM1)
    {
        std::map<std::string, mount_t>::iterator pM2 = mMount2.find(pM1->first);
        if (fOUT) fprintf(fOUT,"%s,%8.3f,%8.3f,%s,%s,%s,%c\n", pM1->first.c_str(), pM1->second.lat, pM1->second.lon, pM1->second.country.c_str(), pM1->second.antenna.c_str(), pM1->second.rcv.c_str(), pM2 == mMount2.end() ? ' ' : '*');
    }
    if (fOUT) fclose(fOUT);
    return 0;
}
int main(int argc, const char* argv[])
{
    if (argc > 2)
    {
        if (argc > 3 && strstr(argv[1], "diffcsv"))
        {
            diff_csv(argv[2], argv[3]);
        }
        else if (argc > 3 && strstr(argv[1], "exclude"))
        {
            exclude_csv(argv[2], argv[3]);
        }
        else if (argc > 3 && strstr(argv[1], "diffblh"))
        {
            double blh1[3] = { atof(argv[2]) * D2R, atof(argv[3]) * D2R, atof(argv[4]) };
            double blh2[3] = { atof(argv[5]) * D2R, atof(argv[6]) * D2R, atof(argv[7]) };
            double l2n = 0;
            double l2e = lat2local(blh1[0], &l2n);

            double dB = blh2[0] - blh1[0];
            double dL = blh2[1] - blh1[1];
            double dN = dB * l2n;
            double dE = dL * l2e;
            double dU = blh2[2] - blh1[2];
            printf("%10.4f,%10.4f,%10.4f\n", dN, dE, dU);
        }
    }
    else  if (argc > 1)
    {
        std::map<std::string, mount_t> mMount;
        read_st(argv[1], mMount);

        if (mMount.size() > 0)
        {
            FILE* fOUT1 = set_output_file(argv[1], "-st.csv");
            if (fOUT1) fprintf(fOUT1, "country,name,lat,lon,receiver type,constellation\n");
            for (std::map<std::string, mount_t>::iterator pMount = mMount.begin(); pMount != mMount.end(); ++pMount)
            {
                if (fOUT1) fprintf(fOUT1, "%s,%s,%8.4f,%8.4f,%s,%s\n", pMount->second.country.c_str(), pMount->first.c_str(), pMount->second.lat, pMount->second.lon, pMount->second.rcv.c_str(), pMount->second.sys.c_str());
            }
            if (fOUT1) fclose(fOUT1);
        }
    }
    return 0;
}