ShaderNode.hh

/*===========================================================================*\
 *                                                                           *
 *                              OpenFlipper                                  *
 *           Copyright (c) 2001-2015, RWTH-Aachen University                 *
 *           Department of Computer Graphics and Multimedia                  *
 *                          All rights reserved.                             *
 *                            www.openflipper.org                            *
 *                                                                           *
 *---------------------------------------------------------------------------*
 * This file is part of OpenFlipper.                                         *
 *---------------------------------------------------------------------------*
 *                                                                           *
 * Redistribution and use in source and binary forms, with or without        *
 * modification, are permitted provided that the following conditions        *
 * are met:                                                                  *
 *                                                                           *
 * 1. Redistributions of source code must retain the above copyright notice, *
 *    this list of conditions and the following disclaimer.                  *
 *                                                                           *
 * 2. Redistributions in binary form must reproduce the above copyright      *
 *    notice, this list of conditions and the following disclaimer in the    *
 *    documentation and/or other materials provided with the distribution.   *
 *                                                                           *
 * 3. Neither the name of the copyright holder nor the names of its          *
 *    contributors may be used to endorse or promote products derived from   *
 *    this software without specific prior written permission.               *
 *                                                                           *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS       *
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED *
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A           *
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER *
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,  *
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,       *
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR        *
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF    *
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING      *
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS        *
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.              *
 *                                                                           *
\*===========================================================================*/






//=============================================================================
//
//  CLASS ShaderNode
//
//=============================================================================

#ifndef ACG_SHADER_NODE_HH
#define ACG_SHADER_NODE_HH


//== INCLUDES =================================================================

#include "BaseNode.hh"
#include <string>

#include "../ShaderUtils/GLSLShader.hh"
#include <ACG/Scenegraph/DrawModes.hh>

//== NAMESPACES ===============================================================

namespace ACG {
namespace SceneGraph {


//== CLASS DEFINITION =========================================================


class ACGDLLEXPORT ShaderNode : public BaseNode
{
public:

  ShaderNode( BaseNode*           _parent = 0,
              const std::string&  _name = "<ShaderNode>" );

  virtual ~ShaderNode();


  ACG_CLASSNAME(ShaderNode);

  void enter(GLState& /*_state*/, const DrawModes::DrawMode& _drawmode) override;
  void leave(GLState& /*_state*/, const DrawModes::DrawMode& _drawmode) override;

  void enterPick(GLState& /*_state*/, PickTarget _target, const DrawModes::DrawMode& _drawmode) override;
  void leavePick(GLState& /*_state*/, PickTarget _target, const DrawModes::DrawMode& _drawmode) override;


  void setShaderDir( std::string _shaderDir);

  std::string shaderDir() { return shaderDir_; };

  std::string vertexShaderName(DrawModes::DrawMode _drawmode, bool _pick = false);

  std::string fragmentShaderName(DrawModes::DrawMode _drawmode, bool _pick = false);
  
  void disableShader (DrawModes::DrawMode _drawmode);
    
  void setShader( DrawModes::DrawMode _drawmode ,
                  const std::string&  _vertexShader,
                  const std::string&  _fragmentShader,
                  std::string         _pickVertexShader = "",
                  std::string         _pickFragmentShader = "");

  GLSL::PtrProgram getShader( DrawModes::DrawMode _drawmode, bool _pick = false);

  bool hasShader( DrawModes::DrawMode _drawmode, bool _pick = false);

  DrawModes::DrawMode availableDrawModes() const override;

private :
  // Path to the shaders ( if set ). If empty shaders will not be used.
  std::string shaderDir_;

  class ShaderInfo {

    public :
      ShaderInfo() :
        vertexShader(0),
        vertexShaderFile(""),
        fragmentShader(0),
        fragmentShaderFile(""),
        program(0),
        initialized(false)
      {
      };

    GLSL::PtrVertexShader   vertexShader;
    std::string             vertexShaderFile;

    GLSL::PtrFragmentShader fragmentShader;
    std::string             fragmentShaderFile;

    GLSL::PtrProgram        program;

    bool                    initialized;
  };

  std::map< size_t, ShaderInfo> shaders;
  std::map< size_t, ShaderInfo> pickShaders;
};


//=============================================================================
} // namespace SceneGraph
} // namespace ACG
//=============================================================================
#endif // ACG_SHADER_NODE_HH defined
//=============================================================================