AsterismMgr.hpp

/*
 * Stellarium
 * Copyright (C) 2017 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 _ASTERISMMGR_HPP_
#define _ASTERISMMGR_HPP_

#include "StelObjectType.hpp"
#include "StelObjectModule.hpp"
#include "StelProjectorType.hpp"

#include <vector>
#include <QString>
#include <QStringList>
#include <QFont>

class StelToneReproducer;
class StarMgr;
class Asterism;
class StelProjector;
class StelPainter;

class AsterismMgr : public StelObjectModule
{
    Q_OBJECT
    Q_PROPERTY(float fontSize
           READ getFontSize
           WRITE setFontSize
           NOTIFY fontSizeChanged)
    Q_PROPERTY(Vec3f linesColor
           READ getLinesColor
           WRITE setLinesColor
           NOTIFY linesColorChanged)
    Q_PROPERTY(bool linesDisplayed
           READ getFlagLines
           WRITE setFlagLines
           NOTIFY linesDisplayedChanged)
    Q_PROPERTY(Vec3f rayHelpersColor
           READ getRayHelpersColor
           WRITE setRayHelpersColor
           NOTIFY rayHelpersColorChanged)
    Q_PROPERTY(bool rayHelpersDisplayed
           READ getFlagRayHelpers
           WRITE setFlagRayHelpers
           NOTIFY rayHelpersDisplayedChanged)
    Q_PROPERTY(Vec3f namesColor
           READ getLabelsColor
           WRITE setLabelsColor
           NOTIFY namesColorChanged)
    Q_PROPERTY(bool namesDisplayed
           READ getFlagLabels
           WRITE setFlagLabels
           NOTIFY namesDisplayedChanged)
    Q_PROPERTY(int asterismLineThickness
           READ getAsterismLineThickness
           WRITE setAsterismLineThickness
           NOTIFY asterismLineThicknessChanged)
    Q_PROPERTY(int rayHelperThickness
           READ getRayHelperThickness
           WRITE setRayHelperThickness
           NOTIFY rayHelperThicknessChanged)

public:
    AsterismMgr(StarMgr *stars);
    virtual ~AsterismMgr();

    // Methods defined in the StelModule class
    virtual void init();

    virtual void draw(StelCore* core);

    virtual void update(double deltaTime);

    virtual double getCallOrder(StelModuleActionName actionName) const;

    // Methods defined in StelObjectManager class
    virtual QList<StelObjectP> searchAround(const Vec3d& v, double limitFov, const StelCore* core) const;

    virtual StelObjectP searchByNameI18n(const QString& nameI18n) const;

    virtual StelObjectP searchByName(const QString& name) const;

    virtual StelObjectP searchByID(const QString &id) const;

    virtual QStringList listMatchingObjects(const QString& objPrefix, int maxNbItem=5, bool useStartOfWords=false, bool inEnglish=false) const;
    virtual QStringList listAllObjects(bool inEnglish) const;
    virtual QString getName() const { return "Asterisms"; }
    virtual QString getStelObjectType() const;

    // Properties setters and getters
public slots:   
    void setLinesColor(const Vec3f& color);
    Vec3f getLinesColor() const;

    void setFlagLines(const bool displayed);
    bool getFlagLines(void) const;

    void setRayHelpersColor(const Vec3f& color);
    Vec3f getRayHelpersColor() const;

    void setFlagRayHelpers(const bool displayed);
    bool getFlagRayHelpers(void) const;

    void setLabelsColor(const Vec3f& color);
    Vec3f getLabelsColor() const;

    void setFlagLabels(const bool displayed);
    bool getFlagLabels(void) const;

    void setFontSize(const float newFontSize);
    float getFontSize() const;

    void setAsterismLineThickness(const int thickness);
    int getAsterismLineThickness() const { return asterismLineThickness; }

    void setRayHelperThickness(const int thickness);
    int getRayHelperThickness() const { return rayHelperThickness; }

signals:
    void fontSizeChanged(const float newSize) const;
    void linesColorChanged(const Vec3f & color) const;
    void linesDisplayedChanged(const bool displayed) const;
    void namesColorChanged(const Vec3f & color) const;
    void namesDisplayedChanged(const bool displayed) const;
    void asterismLineThicknessChanged(int thickness) const;
    void rayHelpersColorChanged(const Vec3f & color) const;
    void rayHelpersDisplayedChanged(const bool displayed) const;
    void rayHelperThicknessChanged(int thickness) const;

private slots:
    void updateSkyCulture(const QString& skyCultureDir);

    void updateI18n();

private:
    void loadNames(const QString& namesFile);

    void loadLines(const QString& fileName);

    void drawLines(StelPainter& sPainter, const StelCore* core) const;
    void drawRayHelpers(StelPainter& sPainter, const StelCore* core) const;
    void drawNames(StelPainter& sPainter) const;

    Asterism* isStarIn(const StelObject *s) const;
    Asterism* findFromAbbreviation(const QString& abbreviation) const;

    //Constellation* isStarIn(const StelObject *s) const;
    //Constellation* findFromAbbreviation(const QString& abbreviation) const;
    std::vector<Asterism*> asterisms;
    QFont asterFont;
    StarMgr* hipStarMgr;

    QString lastLoadedSkyCulture;   // Store the last loaded sky culture directory name
    QString currentSkyCultureID;

    bool linesDisplayed;
    bool rayHelpersDisplayed;
    bool namesDisplayed;
    bool hasAsterism;

    // Store the thickness of lines of the asterisms
    int asterismLineThickness;
    int rayHelperThickness;
};

#endif // _ASTERISMMGR_HPP_