DirectForcingSolver.h

Go to the documentation of this file.

#pragma once

#include "NSWithBody.h"


template <typename memoryType>
class DirectForcingSolver : public NSWithBody<memoryType>
{
protected:
    // for the 1D interpolations
    cusp::array1d<int, host_memory>
        tags,   
        tags2;  
    cusp::array1d<int, device_memory>
        tagsD,   
        tags2D;  
    vecH coeffs,   
         coeffs2;  
    vecD coeffsD,   
         coeffs2D;  
    
    vecH uv;  
    vecD uvD; 
    cusp::array1d<real, memoryType> pressure;  
    
    // tag forcing points
    void tagPoints();
    void tagPoints(real *bx, real *by, real *uB, real *vB);

    // assemble the LHS matrix of the velocity system
    virtual void generateA(real alpha);

    // assemble the Laplacian operator
    virtual void generateL();
    
    // assemble the divergence operator
    void generateQT(int *QTRows, int *QTCols, real *QTVals){} // check this!
    virtual void generateQT();
    
    // update the gradient operator
    void updateQ();

    // compute the Poisson matrix
    virtual void generateC();
    
    // assemble the RHS of the velocity system
    virtual void assembleRHS1();
    // update the RHS of the velocity system
    void updateRHS1();

    // update the solver
    virtual void updateSolverState();

    // projection velocity onto divergence-free space
    virtual void projectionStep();

    // write info about mass flux
    void writeMassFluxInfo();
    
public:
    // constructor
    DirectForcingSolver(parameterDB *pDB=NULL, domain *dInfo=NULL);

    // initialize the direct-forcing solver
    virtual void initialise();

    // write data
    virtual void writeData();
    
    virtual std::string name()
    {
        return "Direct Forcing";
    }

}; // DirectForcingSolver