StelSkyDrawer.hpp

/*
 * Stellarium
 * Copyright (C) 2007 Fabien Chereau
 *
 * 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 _STELSKYDRAWER_HPP_
#define _STELSKYDRAWER_HPP_

#include "StelProjectorType.hpp"
#include "VecMath.hpp"
#include "RefractionExtinction.hpp"

#include "renderer/StelIndexBuffer.hpp"
#include "renderer/StelVertexBuffer.hpp"

#include <QObject>


class StelToneReproducer;
class StelCore;

class StelSkyDrawer : public QObject
{
    Q_OBJECT
public:
    StelSkyDrawer(StelCore* core, class StelRenderer* renderer);
    ~StelSkyDrawer();

    void init();

    void update(double deltaTime);

    void preDrawPointSource();

    void postDrawPointSource(StelProjectorP projector);


    bool pointSourceVisible(StelProjector* projector, const Vec3f& v, const float rcMag[2],
                            bool checkInScreen, Vec3f& outWindowCoordinates)
    {
        // If radius is negative
        if (rcMag[0] <= 0.0f){return false;}
        return checkInScreen ? projector->projectCheck(v, outWindowCoordinates) 
                             : projector->project(v, outWindowCoordinates);
    }

    void drawPointSource(const Vec3f& win, const float rcMag[2], unsigned int bV)
    {
        return drawPointSource(win, rcMag, colorTable[bV]);
    }

    void drawPointSource(const Vec3f& win, const float rcMag[2], const Vec3f& bcolor);

    void drawSunCorona(StelProjectorP projector, const Vec3d& v, float radius, float alpha);

    void postDrawSky3dModel(StelProjectorP projector, const Vec3d& v, float illuminatedArea, float mag, const Vec3f& color = Vec3f(1.f,1.f,1.f));

    bool computeRCMag(float mag, float rcMag[2]) const;

    void reportLuminanceInFov(float lum, bool fastAdaptation=false);

    void preDraw();

    static float surfacebrightnessToLuminance(float sb);
    static float luminanceToSurfacebrightness(float lum);

    static inline float indexToBV(unsigned char bV)
    {
        return (float)bV*(4.f/127.f)-0.5f;
    }

    static inline const Vec3f& indexToColor(unsigned char bV)
    {
        return colorTable[bV];
    }

public slots:
    void setRelativeStarScale(double b=1.0) {starRelativeScale=b;}
    float getRelativeStarScale() const {return starRelativeScale;}

    void setAbsoluteStarScale(double b=1.0) {starAbsoluteScaleF=b;}
    float getAbsoluteStarScale() const {return starAbsoluteScaleF;}

    void setTwinkleAmount(double b) {twinkleAmount=b;}
    float getTwinkleAmount() const {return twinkleAmount;}

    void setFlagTwinkle(bool b) {flagStarTwinkle=b;}
    bool getFlagTwinkle() const {return flagStarTwinkle;}

    void setDrawStarsAsPoints(bool b) {drawStarsAsPoints=b;}
    bool getDrawStarsAsPoints() const {return drawStarsAsPoints;}

    void setBortleScale(int index);
    int getBortleScale() const {return bortleScaleIndex;}

    float getLimitMagnitude() const {return limitMagnitude;}

    // FIXME: Exposed to scripts - make sure it synchs with the GUI. --BM
    void setFlagStarMagnitudeLimit(bool b) {flagStarMagnitudeLimit = b;}
    // FIXME: Exposed to scripts - make sure it synchs with the GUI. --BM
    void setFlagNebulaMagnitudeLimit(bool b) {flagNebulaMagnitudeLimit = b;}
    bool getFlagStarMagnitudeLimit() const {return flagStarMagnitudeLimit;}
    bool getFlagNebulaMagnitudeLimit() const {return flagNebulaMagnitudeLimit;}

    float getCustomStarMagnitudeLimit() const {return customStarMagLimit;}
    void setCustomStarMagnitudeLimit(double limit) {customStarMagLimit=limit;}
    float getCustomNebulaMagnitudeLimit() const {return customNebulaMagLimit;}
    void setCustomNebulaMagnitudeLimit(double limit) {customNebulaMagLimit=limit;}

    float getLimitLuminance() const {return limitLuminance;}

    void setFlagLuminanceAdaptation(bool b) {flagLuminanceAdaptation=b;}
    bool getFlagLuminanceAdaptation() const {return flagLuminanceAdaptation;}

    void setFlagHasAtmosphere(bool b) {flagHasAtmosphere=b;}
    bool getFlagHasAtmosphere() const {return flagHasAtmosphere;}

    void setExtinctionCoefficient(double extCoeff) {extinction.setExtinctionCoefficient(extCoeff);}
    double getExtinctionCoefficient() const {return extinction.getExtinctionCoefficient();}
    void setAtmosphereTemperature(double celsius) {refraction.setTemperature(celsius);}
    double getAtmosphereTemperature() const {return refraction.getTemperature();}
    void setAtmospherePressure(double mbar) {refraction.setPressure(mbar);}
    double getAtmospherePressure() const {return refraction.getPressure();}

    const Extinction& getExtinction() const {return extinction;}
    const Refraction& getRefraction() const {return refraction;}

    float getBig3dModelHaloRadius() const {return big3dModelHaloRadius;}
    void setBig3dModelHaloRadius(float r) {big3dModelHaloRadius=r;}
    
private:
    // Debug
    float reverseComputeRCMag(float rmag) const;

    float computeLimitMagnitude() const;

    float computeLimitLuminance() const;

    float getMaxAdaptFov(void) const {return maxAdaptFov;}
    void setMaxAdaptFov(float fov) {maxAdaptFov = (fov < 60.f) ? 60.f : fov;}

    float getMinAdaptFov(void) const {return minAdaptFov;}
    void setMinAdaptFov(float fov) {minAdaptFov = (fov > 60.f) ? 60.f : fov;}

    void setInputScale(float in) {inScale = in;}
    float getInputScale() const {return inScale;}

    inline float pointSourceMagToLuminance(float mag) const {return std::exp(pointSourceMagToLnLuminance(mag));}

    float pointSourceLuminanceToMag(float lum);

    float pointSourceMagToLnLuminance(float mag) const;

    float findWorldLumForMag(float mag, float targetRadius);

    StelCore* core;

    class StelRenderer* renderer;
    StelToneReproducer* eye;

    Extinction extinction;
    Refraction refraction;

    float maxAdaptFov, minAdaptFov, lnfovFactor;
    bool drawStarsAsPoints;
    bool flagStarTwinkle;
    float twinkleAmount;

    bool flagHasAtmosphere;

    bool flagStarMagnitudeLimit;
    bool flagNebulaMagnitudeLimit;

    float starRelativeScale;
    float starAbsoluteScaleF;

    float starLinearScale;  // optimization variable

    float limitMagnitude;

    float limitLuminance;

    float customStarMagLimit;
    float customNebulaMagLimit;

    class StelTextureNew* texHalo;

    void initColorTableFromConfigFile(class QSettings* conf);

    static Vec3f colorTable[128];

    int bortleScaleIndex;

    float inScale;

    struct ColoredVertex
    {
        Vec2f position;
        Vec3f color;
        ColoredVertex(Vec2f position, Vec3f color):position(position), color(color){}

        VERTEX_ATTRIBUTES(Vec2f Position, Vec3f Color);
    };

    struct ColoredTexturedVertex
    {
        Vec2f position;
        Vec3f color;
        Vec2f texCoord;
        ColoredTexturedVertex(Vec2f position, Vec3f color, Vec2f texCoord)
            :position(position), color(color), texCoord(texCoord){}

        VERTEX_ATTRIBUTES(Vec2f Position, Vec3f Color, Vec2f TexCoord);
    };

    StelVertexBuffer<ColoredVertex>* starPointBuffer;

    StelVertexBuffer<ColoredTexturedVertex>* starSpriteBuffer;

    StelVertexBuffer<ColoredTexturedVertex>* bigHaloBuffer;

    StelVertexBuffer<ColoredTexturedVertex>* sunHaloBuffer;

    StelVertexBuffer<ColoredTexturedVertex>* coronaBuffer;

    bool drawing;

    float maxLum;
    float oldLum;

    class StelTextureNew* texBigHalo;
    class StelTextureNew* texSunHalo;
    class StelTextureNew* texCorona;

    bool flagLuminanceAdaptation;

    float big3dModelHaloRadius;


    bool statisticsInitialized;

    int bigHaloStatID;
    int sunHaloStatID;
    int starStatID;
};

#endif // _STELSKYDRAWER_HPP_