SoView2DShaderOverlay.h

Go to the documentation of this file.

/*************************************************************************************
**
** Copyright 2015, MeVis Medical Solutions AG
**
** The user may use this file in accordance with the license agreement provided with
** the Software or, alternatively, in accordance with the terms contained in a
** written agreement between the user and MeVis Medical Solutions AG.
**
** For further information use the contact form at https://www.mevislab.de/contact
**
**************************************************************************************/
 
#pragma once
 
#include "SoView2DExtension.h"
#include "SoView2DShaderState.h"
#include <Inventor/nodes/SoVertexShader.h>
#include <Inventor/nodes/SoFragmentShader.h>
#include <Inventor/nodes/SoShaderProgram.h>
 
#include "SoView2DShaderBuiltInIncludes.h"
#include "SoView2DShaderBuiltInFunctions.h"
 
class SoView2DShaderDiagnosis;
class SoView2DShaderOperation;
 
class SOVIEW2D_API SoView2DShaderOverlay : public SoView2DExtension
{
 
  SO_NODE_HEADER(SoView2DShaderOverlay);
 
public:
 
  static void   initClass();
 
  SoView2DShaderOverlay();
 
  SoSFEnum    blendMode;
 
  SoSFBool    disableBaseImageDrawing;
 
  void draw(View2DSliceList *dsl, View2DSlice* dslice, int slice) override;
 
  void overwriteDrawingSettings(SoView2DDrawingSettings& settings) override;
 
  void addOverlay(SoGLShaderProgram* program, SoView2DShaderOverlayInfo* overlay, float dx1, float dy1, float dx2, float dy2);
 
  void GLRender(SoGLRenderAction* action) override;
 
protected:
  struct ActionState {
    std::vector<SoView2DShaderCustomInclude*> _includes;
    std::vector<SoView2DShaderOperation*> _operations;
    std::vector<SoView2DShaderOverlayInfo*> _overlays;
    SoNodeList _shaderParameters;
    std::vector<size_t> _nodeIds;
    SoView2DShaderDiagnosis* _diagnosis;
  };
 
  ActionState _actionState;
 
  struct CachedShader {
    CachedShader();
 
    std::string _vertexShaderString;
    std::string _fragmentShaderString;
    SoView2DShaderState     _currentState;
    SoRef<SoVertexShader>   _vertexShader;
    SoRef<SoFragmentShader> _fragmentShader;
    SoRef<SoShaderProgram>  _shaderProgram;
  };
 
  void initPipeline();
 
  void fillDiagnosis(CachedShader& current, SoView2DShaderDiagnosis* diagnosisModule, SoState* state);
 
  void createShaders(std::string& vertexShader, std::string& fragmentShader, SoState* state, const SoView2DShaderState& currentState);
 
  void combineShaderString(const SoView2DShaderState& currentState, const ShaderPipelineCodeTarget& source, std::string& output);
 
  void applyFragmentOperations(const std::vector<SoView2DShaderOperation*>& operations);
  void addFragmentIncludes(const std::vector<SoView2DShaderCustomInclude*>& includes);
 
  void appendFunction(ShaderPipeline& pipeline, int step, SoView2DShaderBuiltInFunctions::SoView2DShaderFunction* function,
    const SoView2DShaderState& currentState);
 
  CachedShader _shader;
 
  SoView2DShaderBuiltInIncludes::IncludeContainer _builtInIncludes;
 
  ShaderPipeline _fragmentPipeline;
 
  SoView2DShaderBuiltInFunctions::SetupState _setupState;
  SoView2DShaderBuiltInFunctions::WriteFragment _writeFragment;
};