Exoplanet.hpp

/*
 * Copyright (C) 2012 Alexander Wolf
 *
 * 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.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Suite 500, Boston, MA  02110-1335, USA.
 */

#ifndef _EXOPLANET_HPP_
#define _EXOPLANET_HPP_ 1

#include <QVariant>
#include <QString>
#include <QStringList>
#include <QFont>
#include <QList>
#include <QDateTime>

#include "StelObject.hpp"
#include "StelTextureTypes.hpp"
#include "StelFader.hpp"

typedef struct
{
    QString planetName;     
    QString planetProperName;   
    float mass;         
    float radius;           
    float period;           
    float semiAxis;         
    float eccentricity;     
    float inclination;      
    float angleDistance;        
    int discovered;         
    QString pclass;         
    int EqTemp;         
    int ESI;            
} exoplanetData;

class StelPainter;


class Exoplanet : public StelObject
{
    friend class Exoplanets;
public:
    static const QString EXOPLANET_TYPE;

    Exoplanet(const QVariantMap& map);
    ~Exoplanet();

    QVariantMap getMap(void) const;

    virtual QString getType(void) const
    {
        return EXOPLANET_TYPE;
    }

    virtual QString getID(void) const
    {
        return getDesignation();
    }

    virtual float getSelectPriority(const StelCore* core) const;

    virtual QString getInfoString(const StelCore* core, const InfoStringGroup& flags) const;
    virtual QVariantMap getInfoMap(const StelCore *core) const;
    virtual Vec3f getInfoColor(void) const;
    virtual Vec3d getJ2000EquatorialPos(const StelCore*) const
    {
        return XYZ;
    }
    virtual float getVMagnitude(const StelCore* core) const;
    virtual double getAngularSize(const StelCore* core) const;
    virtual QString getNameI18n(void) const;
    virtual QString getEnglishName(void) const;

    QString getDesignation(void) const;
    QStringList getExoplanetsEnglishNames(void) const;
    QStringList getExoplanetsNamesI18n(void) const;
    QStringList getExoplanetsDesignations(void) const;

    bool isDiscovered(const StelCore* core);

    void update(double deltaTime);

    int getCountExoplanets(void) const
    {
        return EPCount;
    }
    int getCountHabitableExoplanets(void) const
    {
        return PHEPCount;
    }

    QList<double> getData(int mode)
    {
        switch(mode)
        {
            case 1:
                return semiAxisList;
                break;
            case 2:
                return massList;
                break;
            case 3:
                return radiusList;
                break;
            case 4:
                return periodList;
                break;
            case 5:
                return angleDistanceList;
                break;
            case 6:
                return effectiveTempHostStarList;
                break;
            case 7:
                return yearDiscoveryList;
                break;
            case 8:
                return metallicityHostStarList;
                break;
            case 9:
                return vMagHostStarList;
                break;
            case 10:
                return raHostStarList;
                break;
            case 11:
                return decHostStarList;
                break;
            case 12:
                return distanceHostStarList;
                break;
            case 13:
                return massHostStarList;
                break;
            case 14:
                return radiusHostStarList;
                break;
            default:
                return eccentricityList;
        }
    }

private:

    QString getPlanetaryClassI18n(QString ptype) const;

    bool initialized;

    Vec3d XYZ;                         // holds J2000 position  

    static StelTextureSP markerTexture;
    static Vec3f habitableExoplanetMarkerColor;
    static Vec3f exoplanetMarkerColor;
    static bool distributionMode;
    static bool timelineMode;
    static bool habitableMode;
    static bool showDesignations;

    void draw(StelCore* core, StelPainter *painter);

    int EPCount;
    int PHEPCount;

    QString designation;            
    QString starProperName;         
    float RA;               
    float DE;               
    float distance;             
    QString stype;              
    float smass;                
    float smetal;               
    float Vmag;             
    float sradius;              
    int effectiveTemp;          
    bool hasHabitableExoplanets;        
    QList<exoplanetData> exoplanets;    

    QStringList englishNames, translatedNames, exoplanetDesignations;

    QList<double> eccentricityList, semiAxisList, massList, radiusList, periodList, angleDistanceList,
              effectiveTempHostStarList, yearDiscoveryList, metallicityHostStarList, vMagHostStarList,
              raHostStarList, decHostStarList, distanceHostStarList, massHostStarList, radiusHostStarList;

    LinearFader labelsFader;
};

#endif // _EXOPLANET_HPP_