makestardb.cpp

来自「celestia源代码」· C++ 代码 · 共 244 行

// makestardb.cpp
//
// Copyright (C) 2004, Chris Laurel <[e-mail]>
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2
// of the License, or (at your option) any later version.
//
// Convert a file with ASCII star records to a Celestia star database

#include <iostream>
#include <fstream>
#include <iomanip>
#include <cctype>
#include <cassert>
#include <celutil/basictypes.h>
#include <celutil/bytes.h>
#include <celengine/astro.h>
#include <celengine/star.h>

using namespace std;


static string inputFilename;
static string outputFilename;
static bool useSphericalCoords = false;


void Usage()
{
    cerr << "Usage: makestardb [options] <input file> <output star database>\n";
    cerr << "  Options:\n";
    cerr << "    --spherical (or -s) : input file has spherical coords (RA/dec/distance\n";
}


bool parseCommandLine(int argc, char* argv[])
{
    int i = 1;
    int fileCount = 0;

    while (i < argc)
    {
        if (argv[i][0] == '-')
        {
            if (!strcmp(argv[i], "--spherical") || !strcmp(argv[i], "-s"))
            {
                useSphericalCoords = true;
            }
            else
            {
                cerr << "Unknown command line switch: " << argv[i] << '\n';
                return false;
            }
            i++;
        }
        else
        {
            if (fileCount == 0)
            {
                // input filename first
                inputFilename = string(argv[i]);
                fileCount++;
            }
            else if (fileCount == 1)
            {
                // output filename second
                outputFilename = string(argv[i]);
                fileCount++;
            }
            else
            {
                // more than two filenames on the command line is an error
                return false;
            }
            i++;
        }
    }

    return true;
}


static void writeUint(ostream& out, uint32 n)
{
    LE_TO_CPU_INT32(n, n);
    out.write(reinterpret_cast<char*>(&n), sizeof n);
}

static void writeFloat(ostream& out, float f)
{
    LE_TO_CPU_FLOAT(f, f);
    out.write(reinterpret_cast<char*>(&f), sizeof f);
}

static void writeUshort(ostream& out, uint16 n)
{
    LE_TO_CPU_INT16(n, n);
    out.write(reinterpret_cast<char*>(&n), sizeof n);
}

static void writeShort(ostream& out, int16 n)
{
    LE_TO_CPU_INT16(n, n);
    out.write(reinterpret_cast<char*>(&n), sizeof n);
}


bool WriteStarDatabase(istream& in, ostream& out, bool sphericalCoords)
{
    unsigned int record = 0;
    unsigned int nStarsInFile = 0;

    in >> nStarsInFile;
    if (!in.good())
    {
        cerr << "Error reading star count at beginning of input file.\n";
        return 1;
    }

    // Write the header
    out.write("CELSTARS", 8);

    // Write the version
    writeShort(out, 0x0100);

    writeUint(out, nStarsInFile);

    for (unsigned int record = 0; record < nStarsInFile; record++)
    {
        unsigned int catalogNumber;
        float x, y, z;
        float absMag;

        in >> catalogNumber;
        if (in.eof())
            return true;

        if (!in.good())
        {
            cerr << "Error parsing catalog number for record #" << record << '\n';
            return false;
        }

        if (sphericalCoords)
        {
            float RA, dec, distance;
            float appMag;

            in >> RA >> dec >> distance;
            if (!in.good())
            {
                cerr << "Error parsing position of star " << catalogNumber << '\n';
                return false;
            }

            in >> appMag;
            if (!in.good())
            {
                cerr << "Error parsing magnitude of star " << catalogNumber << '\n';
                return false;
            }

            Point3d pos =
                astro::equatorialToCelestialCart((double) RA * 24.0 / 360.0,
                                                 (double) dec,
                                                 (double) distance);
            x = (float) pos.x;
            y = (float) pos.y;
            z = (float) pos.z;
            absMag = (float) (appMag + 5 - 5 * log10(distance / 3.26));
        }
        else
        {
            in >> x >> y >> z;
            if (!in.good())
            {
                cerr << "Error parsing position of star " << catalogNumber << '\n';
                return false;
            }

            in >> absMag;
            if (!in.good())
            {
                cerr << "Error parsing magnitude of star " << catalogNumber << '\n';
                return false;
            }
        }

        string scString;
        in >> scString;
        StellarClass sc = StellarClass::parse(scString);
#if 0
        StarDetails* details = StarDetails::GetStarDetails(sc);
        if (details == NULL)
        {
            cerr << "Error parsing spectral type of star " << catalogNumber << '\n';
            return false;
        }

        // For spectral type parser debugging . . .
        cout << scString << ' ' << details->getSpectralType() << '\n';
#endif

        writeUint(out, catalogNumber);
        writeFloat(out, x);
        writeFloat(out, y);
        writeFloat(out, z);
        writeShort(out, (int16) (absMag * 256.0f));
        writeUshort(out, sc.pack());
    }

    return true;
}


int main(int argc, char* argv[])
{
    if (!parseCommandLine(argc, argv) || inputFilename.empty())
    {
        Usage();
        return 1;
    }

    ifstream inputFile(inputFilename.c_str(), ios::in);
    if (!inputFile.good())
    {
        cerr << "Error opening input file " << inputFilename << '\n';
        return 1;
    }

    ofstream stardbFile(outputFilename.c_str(), ios::out | ios::binary);
    if (!stardbFile.good())
    {
        cerr << "Error opening star database file " << outputFilename << '\n';
        return 1;
    }

    bool success = WriteStarDatabase(inputFile, stardbFile, useSphericalCoords);

    return success ? 0 : 1;
}