DtOsgFluidFrictionOperator.h

Go to the documentation of this file.

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


#ifndef DtOsgFluidFrictionOperator_H_
#define DtOsgFluidFrictionOperator_H_

#include <vrvOsg/vrvOsg.h> 

#include <osgParticle/Operator>
#include <osgParticle/FluidFrictionOperator>
#include <osgParticle/Particle>
#include <osgParticle/Program>

namespace makVrv
{
   class DT_DLL_VRVOSG DtOsgFluidFrictionOperator : public osgParticle::Operator
   {
   public:

      DtOsgFluidFrictionOperator();

      DtOsgFluidFrictionOperator(const DtOsgFluidFrictionOperator& copy, 
         const osg::CopyOp& copyop = osg::CopyOp::SHALLOW_COPY);

      static DtOsgFluidFrictionOperator* fromOsgFluidFrictionOperator(
         osgParticle::FluidFrictionOperator* osgOp);

      META_Object(makVrv, DtOsgFluidFrictionOperator);

      void setFluidDensity(float d);

      float getFluidDensity() const;

      void setFluidViscosity(float v);

      float getFluidViscosity() const;

      void setWind(const osg::Vec3& wind) { _wind = wind; }

      const osg::Vec3& getWind() const { return _wind; }

      void setUp(const osg::Vec3& up) {_up = up;}

      const osg::Vec3& getUp() const {return _up;}

      void setOverrideRadius(float r);

      float getOverrideRadius() const;        

      void setFluidToAir();

      void setFluidToWater();

      void operate(osgParticle::Particle* P, double dt);

      void beginOperate(osgParticle::Program* prg);

   protected:
      virtual ~DtOsgFluidFrictionOperator() {}
      DtOsgFluidFrictionOperator &operator=(const DtOsgFluidFrictionOperator &) { return *this; }

   private:
      float       viscosityCoefficient;
      float       densityCoefficient;
      float       _density;
      float       _viscosity;
      float       _ovr_rad;
      osg::Vec3   _wind;
      osg::Vec3   _up;
      osgParticle::Program*    _current_program;
   };
}

#endif