Program Listing for File TimeStepPBF.h

↰ Return to documentation for file (SPlisHSPlasH/PBF/TimeStepPBF.h)

#ifndef __TimeStepPBF_h__
#define __TimeStepPBF_h__

#include "SPlisHSPlasH/Common.h"
#include "SPlisHSPlasH/TimeStep.h"
#include "SimulationDataPBF.h"
#include "SPlisHSPlasH/SPHKernels.h"

namespace SPH
{
    class SimulationDataPBF;


    class TimeStepPBF : public TimeStep
    {
    protected:
        SimulationDataPBF m_simulationData;
        unsigned int m_iterations;
        Real m_maxError;
        unsigned int m_minIterations;
        unsigned int m_maxIterations;
        int m_velocityUpdateMethod;

        void pressureSolve();
        void pressureSolveIteration(const unsigned int fluidModelIndex, Real &avg_density_err);

        virtual void performNeighborhoodSearchSort();

        virtual void emittedParticles(FluidModel *model, const unsigned int startIndex);

        virtual void initParameters();

    public:
        static std::string METHOD_NAME;
        static int SOLVER_ITERATIONS;
        static int MIN_ITERATIONS;
        static int MAX_ITERATIONS;
        static int MAX_ERROR;
        static int VELOCITY_UPDATE_METHOD;
        static int ENUM_PBF_FIRST_ORDER;
        static int ENUM_PBF_SECOND_ORDER;

        TimeStepPBF();
        virtual ~TimeStepPBF(void);

        virtual void step();

        virtual void reset();
        virtual void resize();
        virtual std::string getMethodName() { return METHOD_NAME; }
        virtual int getNumIterations() { return m_iterations; }
    };
}

#endif