MyParticleSystemModel.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2020 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/


#pragma once

#include "exampleEffectsPlugin.h"

#include <vrvCore/DtUniqueID.h>
#include <vrvOsg/DtOsgParticleSystemModel.h>

#include <osgEarth/VirtualProgram>
#include <osg/ref_ptr>

namespace osg
{
   class StateSet;
   class Uniform;
}

namespace makVrv
{
   class DtDe;

   class DT_DLL_EXAMPLEEFFECTSPLUGIN MyShaderManager
   {

   public:

      MyShaderManager(DtDe& de, osg::StateSet *ss);

      virtual ~MyShaderManager();

      //Function for changing the color that will be added to the fragments color
      virtual void changeColorOverride(int color, float deltaChange);

   protected:

      DtDe& myDe;

      //model state set to attach the custom shaders to
      osg::ref_ptr<osg::StateSet> myModelStateSet;

      //Uniforms used by the custom shaders
      osg::ref_ptr<osg::Uniform> myColorToAdd;

      //virtual program for the stateset
      osg::ref_ptr<osgEarth::VirtualProgram> myVirtualProgram;

   };

   class DT_DLL_EXAMPLEEFFECTSPLUGIN MyParticleSystemModel
      : public DtOsgParticleSystemModel
   {
   public:
      MyParticleSystemModel(DtDe& de, DtUniqueID& id);

      virtual ~MyParticleSystemModel();

      virtual void setModelInstance(DtModelInstance* modelInstance );

      virtual void changeColorOverride( int color, float deltaChange );

   protected:
      static MyShaderManager* theShaderManager;

   };
}