SatellitesDialog.hpp

/*
 * Stellarium Satellites Plug-in GUI
 * 
 * Copyright (C) 2010 Matthew Gates
 * Copyright (C) 2015 Nick Fedoseev (Iridium flares)
 * 
 * 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 _SATELLITESDIALOG_HPP_
#define _SATELLITESDIALOG_HPP_

#include <QObject>
#include <QModelIndex>
#include <QTreeWidget>
#include <QTreeWidgetItem>
#include "StelDialog.hpp"
#include "Satellites.hpp"

class Ui_satellitesDialog;
class QCheckBox;
class QListWidgetItem;
class QSortFilterProxyModel;
class QStandardItemModel;
class QTimer;

class SatellitesImportDialog;
class SatellitesListFilterModel;

class SatellitesDialog : public StelDialog
{
    Q_OBJECT

public:
    enum IridiumFlaresColumns {
        IridiumFlaresDate,  
        IridiumFlaresMagnitude, 
        IridiumFlaresAltitude,  
        IridiumFlaresAzimuth,   
        IridiumFlaresSatellite, 
        IridiumFlaresCount  
    };

    SatellitesDialog();
    ~SatellitesDialog();

protected:
    void createDialogContent();

public slots:
    void retranslate();

private slots:
    void jumpToSourcesTab();
    void updateCountdown();
    void filterListByGroup(int index);
    void updateSatelliteData();
    void saveSatellites(void);
    void showUpdateState(Satellites::UpdateState state);
    void showUpdateCompleted(int updated, int total, int added, int missing);
    

    void saveEditedSource();
    void saveSourceList(void);
    void deleteSourceRow(void);
    void addSourceRow(void);
    void toggleCheckableSources();
    
    void restoreDefaults(void);
    void saveSettings(void);
    void addSatellites(const TleDataList& newSatellites);
    void removeSatellites();
    void setFlags();
    void handleGroupChanges(QListWidgetItem* item);
    void trackSatellite(const QModelIndex & index);
    void setOrbitParams(void);
    void updateTLEs(void);

    void predictIridiumFlares();
    void selectCurrentIridiumFlare(const QModelIndex &modelIndex);

private:
    void enableSatelliteDataForm(bool enabled);
    void populateAboutPage();
    void updateSettingsPage();
    void populateFilterMenu();
    void populateSourcesList();
    void addSpecialGroupItem();
    void setGroups();

    void setIridiumFlaresHeaderNames();

    void initListIridiumFlares();
    
    Ui_satellitesDialog* ui;
    bool satelliteModified;
    
    QTimer* updateTimer;
    
    SatellitesImportDialog* importWindow;
    
    SatellitesListFilterModel* filterModel;
    
    Qt::ItemDataRole checkStateRole;
};

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

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

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

#endif // _SATELLITESDIALOG_HPP_