OsgLight.h

// This file is a part of the OpenSurgSim project.
// Copyright 2013, SimQuest Solutions Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef SURGSIM_GRAPHICS_OSGLIGHT_H
#define SURGSIM_GRAPHICS_OSGLIGHT_H

#include <memory>
#include <string>
#include <unordered_map>

#include "SurgSim/Graphics/OsgRepresentation.h"
#include "SurgSim/Graphics/Light.h"

#include <osg/ref_ptr>

#if defined(_MSC_VER)
#pragma warning(push)
#pragma warning(disable:4250)
#endif

namespace osg
{

class Uniform;
class StateSet;
class Light;
class LightSource;

}

namespace SurgSim
{
namespace Graphics
{

class OsgGroup;

SURGSIM_STATIC_REGISTRATION(OsgLight);

class OsgLight : public OsgRepresentation, public Light
{
public:

    friend class OsgLightTests;

    explicit OsgLight(const std::string& name);
    virtual ~OsgLight();

    SURGSIM_CLASSNAME(SurgSim::Graphics::OsgLight);

    bool setGroup(std::shared_ptr<SurgSim::Graphics::Group> group) override;

    void setLightGroupReference(const std::string& name) override;

    std::string getLightGroupReference() override;

    std::shared_ptr<SurgSim::Graphics::Group> getGroup() override;

    void setDiffuseColor(const SurgSim::Math::Vector4d& color) override;

    SurgSim::Math::Vector4d getDiffuseColor() override;

    void setSpecularColor(const SurgSim::Math::Vector4d& color) override;

    SurgSim::Math::Vector4d getSpecularColor() override;

    void setConstantAttenuation(double val) override;

    double getConstantAttenuation() override;

    void setLinearAttenuation(double val) override;

    double getLinearAttenuation() override;

    void setQuadraticAttenuation(double val) override;

    double getQuadraticAttenuation() override;


private:
    void doUpdate(double dt) override;

    void apply(osg::ref_ptr<osg::StateSet> stateSet);

    void remove(osg::ref_ptr<osg::StateSet> stateSet);

    enum UniformType
    {
        POSITION = 0,
        DIFFUSE_COLOR,
        SPECULAR_COLOR,
        CONSTANT_ATTENUATION,
        LINEAR_ATTENUATION,
        QUADRATIC_ATTENUATION
    };

    std::shared_ptr<OsgGroup> m_group;

    std::unordered_map<int, osg::ref_ptr<osg::Uniform>> m_uniforms;

    SurgSim::Math::Vector4d m_diffuseColor;     
    SurgSim::Math::Vector4d m_specularColor;    

    double m_constantAttenuation;               
    double m_linearAttenuation;                 
    double m_quadraticAttenuation;              

    osg::ref_ptr<osg::Light> m_light;
    osg::ref_ptr<osg::LightSource> m_lightSource;

    std::string m_groupReference; 
};

#if defined(_MSC_VER)
#pragma warning(pop)
#endif

}; // Graphics
}; // SurgSim

#endif