PluginDiffusion.hpp

Go to the documentation of this file.

#ifndef ONKO3D_3_0_PLUGINDIFFUSION_HPP
#define ONKO3D_3_0_PLUGINDIFFUSION_HPP
 
#include <vector>
#include <mecacell/mecacell.h>
#include <math.h>
#include "DiffusionGrid.hpp"
 
namespace Diffusion {
 
    template<typename cell_t>
    class PluginDiffusion {
 
    private:
        DiffusionGrid grid; 
    public:
 
        PluginDiffusion() = default;
 
        inline PluginDiffusion(double dx, double accuracy): grid(dx, accuracy) {}
 
        inline void setDx(double dx) { grid.setDx(dx); }
 
        inline void setAccuracy(double a) { grid.setAccuracy(a); }
 
        inline DiffusionGrid* getGrid() { return &grid; }
 
        inline void addMolecule(int n, Molecule m) { grid.addMolecule(n, m); }
 
        template<typename world_t>
        void onAddCell(world_t *w) {
            for (cell_t *c : w->newCells){
                c->getBody().setDiffusionGrid(&grid); // Transmits the diffusion grid
            }
        }
 
        template<typename world_t>
        void preBehaviorUpdate(world_t *w) {
            grid.updateBoundary(w);
            grid.computeMolecules(w);
        }
    };
}
 
#endif //ONKO3D_3_0_PLUGINDIFFUSION_HPP