AstroCalcDialog.hpp

/*
 * Stellarium
 * 
 * Copyright (C) 2015 Alexander Wolf
 * Copyright (C) 2016 Nick Fedoseev (visualization of ephemeris)
 * 
 * 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 _ASTROCALCDIALOG_HPP_
#define _ASTROCALCDIALOG_HPP_

#include <QObject>
#include <QTreeWidget>
#include <QTreeWidgetItem>
#include <QMap>
#include <QVector>
#include <QTimer>

#include "StelDialog.hpp"
#include "StelCore.hpp"
#include "Planet.hpp"
#include "SolarSystem.hpp"
#include "Nebula.hpp"
#include "NebulaMgr.hpp"
#include "StarMgr.hpp"

class Ui_astroCalcDialogForm;
class QListWidgetItem;

class AstroCalcDialog : public StelDialog
{
    Q_OBJECT

public:
    enum PlanetaryPositionsColumns {
        ColumnName,     
        ColumnRA,       
        ColumnDec,      
        ColumnMagnitude,    
        ColumnType,     
        ColumnCount     
    };

    enum EphemerisColumns {
        EphemerisDate,      
        EphemerisJD,        
        EphemerisRA,        
        EphemerisDec,       
        EphemerisMagnitude, 
        EphemerisCount      
    };

    enum PhenomenaColumns {
        PhenomenaType,      
        PhenomenaDate,      
        PhenomenaObject1,   
        PhenomenaObject2,   
        PhenomenaSeparation,    
        PhenomenaCount      
    };

    AstroCalcDialog(QObject* parent);
    virtual ~AstroCalcDialog();

    void styleChanged();

    static QVector<Vec3d> EphemerisListJ2000;
    static QVector<QString> EphemerisListDates;
    static int DisplayedPositionIndex;

public slots:
        void retranslate();

protected:
        virtual void createDialogContent();
        Ui_astroCalcDialogForm *ui;

private slots:
    void currentPlanetaryPositions();
    void selectCurrentPlanetaryPosition(const QModelIndex &modelIndex);
    void onChangedEphemerisPosition(const QModelIndex &modelIndex);

    void generateEphemeris();
    void cleanupEphemeris();
    void selectCurrentEphemeride(const QModelIndex &modelIndex);
    void saveEphemeris();

    void calculatePhenomena();
    void selectCurrentPhenomen(const QModelIndex &modelIndex);
    void savePhenomena();

    void drawAltVsTimeDiagram();
    void drawCurrentTimeDiagram();
    void mouseOverLine(QMouseEvent *event);

    void changePage(QListWidgetItem *current, QListWidgetItem *previous);

    void updateSolarSystemData();

private:
    class StelCore* core;
    class SolarSystem* solarSystem;
    class NebulaMgr* dsoMgr;
    class StarMgr* starMgr;
    class StelObjectMgr* objectMgr;
    QTimer *currentTimeLine;

    void setPlanetaryPositionsHeaderNames();
    void setEphemerisHeaderNames();
    void setPhenomenaHeaderNames();

    void initListPlanetaryPositions();
    void initListEphemeris();
    void initListPhenomena();

    void populateCelestialBodyList();
    void populateEphemerisTimeStepsList();
    void populateMajorPlanetList();
    void populateGroupCelestialBodyList();  
    void prepareAxesAndGraph();

    QMap<double, double> findClosestApproach(PlanetP& object1, PlanetP& object2, double startJD, double stopJD, double maxSeparation, bool opposition);
    double findDistance(double JD, PlanetP object1, PlanetP object2, bool opposition);
    bool findPrecise(QPair<double, double>* out, PlanetP object1, PlanetP object2, double JD, double step, int prevSign, bool opposition);
    void fillPhenomenaTable(const QMap<double, double> list, const PlanetP object1, const PlanetP object2, bool opposition);

    QMap<double, double> findClosestApproach(PlanetP& object1, NebulaP& object2, double startJD, double stopJD, double maxSeparation);
    double findDistance(double JD, PlanetP object1, NebulaP object2);
    bool findPrecise(QPair<double, double>* out, PlanetP object1, NebulaP object2, double JD, double step, int prevSign);
    void fillPhenomenaTable(const QMap<double, double> list, const PlanetP object1, const NebulaP object2);

    QMap<double, double> findClosestApproach(PlanetP& object1, StelObjectP& object2, double startJD, double stopJD, double maxSeparation);
    double findDistance(double JD, PlanetP object1, StelObjectP object2);
    bool findPrecise(QPair<double, double>* out, PlanetP object1, StelObjectP object2, double JD, double step, int prevSign);
    void fillPhenomenaTable(const QMap<double, double> list, const PlanetP object1, const StelObjectP object2);

    QString delimiter, acEndl;
    QStringList ephemerisHeader, phenomenaHeader, planetaryPositionsHeader;
    static float brightLimit;
    static float minY, maxY;

    void updateTabBarListWidgetWidth();
};

// Reimplements the QTreeWidgetItem class to fix the sorting bug
class ACTreeWidgetItem : public QTreeWidgetItem
{
public:
    ACTreeWidgetItem(QTreeWidget* parent)
        : QTreeWidgetItem(parent)
    {
    }

private:
    bool operator < (const QTreeWidgetItem &other) const
    {
        int column = treeWidget()->sortColumn();

        if (column == AstroCalcDialog::ColumnMagnitude)
        {
            return text(column).toFloat() < other.text(column).toFloat();
        }
        else
        {
            return text(column).toLower() < other.text(column).toLower();
        }
    }
};

#endif // _ASTROCALCDIALOG_HPP_