PaintBehavior.h

// This file is a part of the OpenSurgSim project.
// Copyright 2016, 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_PAINTBEHAVIOR_H
#define SURGSIM_GRAPHICS_PAINTBEHAVIOR_H

#include "SurgSim/Collision/Representation.h"
#include "SurgSim/DataStructures/IndexedLocalCoordinate.h"
#include "SurgSim/Framework/Behavior.h"
#include "SurgSim/Framework/Component.h"
#include "SurgSim/Framework/Logger.h"
#include "SurgSim/Graphics/OsgMeshRepresentation.h"
#include "SurgSim/Graphics/OsgTexture2d.h"
#include "SurgSim/Math/Matrix.h"
#include "SurgSim/Math/Vector.h"

namespace SurgSim
{
namespace Graphics
{

SURGSIM_STATIC_REGISTRATION(PaintBehavior);

class PaintBehavior : public Framework::Behavior
{
public:
    explicit PaintBehavior(const std::string& name);

    SURGSIM_CLASSNAME(SurgSim::Graphics::PaintBehavior);

    void setRepresentation(std::shared_ptr<Framework::Component> representation);

    std::shared_ptr<Graphics::OsgMeshRepresentation> getRepresentation() const;

    void setTextureSize(int width, int height);

    void setColor(const Math::Vector4d& color);

    Math::Vector4d getColor() const;

    void setRadius(double radius);

    double getRadius() const;

    void setAntiAlias(bool antialias);

    bool getAntiAlias() const;

    void setCoordinates(const std::vector<DataStructures::IndexedLocalCoordinate>& coordinate);

    bool doInitialize() override;
    bool doWakeUp() override;

    void update(double dt) override;

private:

    void buildBrush(double radius);

    void buildAntiAliasedBrush(double radius);

    Math::Vector2d toPixel(const Math::Vector2d& uv);

    void paint(const Math::Vector2d& coordinates);

    std::shared_ptr<Graphics::OsgMeshRepresentation> m_representation;

    std::shared_ptr<Graphics::OsgTexture2d> m_texture;

    Math::Vector4d m_color;

    int m_width;

    int m_height;

    double m_radius;

    bool m_antialias;

    std::vector<DataStructures::IndexedLocalCoordinate> m_coordinates;

    int m_brushOffsetX;
    int m_brushOffsetY;

    Math::Matrix m_brush;

    boost::mutex m_mutex;
};

} // Graphics
} // SurgSim

#endif // SURGSIM_GRAPHICS_PAINTBEHAVIOR_H