BodyDiffusion.hpp

Go to the documentation of this file.

#ifndef ONKO3D_3_0_BODYDIFFUSION_HPP
#define ONKO3D_3_0_BODYDIFFUSION_HPP
 
#include "DiffusionGrid.hpp"
#include "../../../../src/core/BaseBody.hpp"
#include <vector>
#include <hash_map>
 
using namespace std;
 
namespace Diffusion {
 
    template<typename cell_t, class plugin_t>
    class BodyDiffusion : virtual BaseBody<plugin_t> {
 
    private:
        vector<double> consumptions; 
        DiffusionGrid * grid = nullptr;
        int nbMolecules = 0;
 
    public:
 
        inline BodyDiffusion() : consumptions()//, quantities()
        {}
 
        inline explicit BodyDiffusion(int size){
            initNbMolecules(size);
        }
 
        void initNbMolecules(int n){ 
            if(n > nbMolecules) {
                nbMolecules = n;
                consumptions.assign(n,0.);
            }
        }
 
        double getQuantity(MecaCell::Vec v, int mol) const{ 
             if(this->cellPlugin != nullptr) return grid->getMoleculeRealPos(v, grid->moleculesDict[mol]); 
             else return 0.;
        }
 
        double getConsumption(int mol) const{ 
            if(this->cellPlugin != nullptr) return consumptions[grid->moleculesDict[mol]]; 
            else return consumptions[mol];
        }
 
        double getConsumptionByIndex(int i) const{ 
            return consumptions[i]; 
        }
 
        inline void setConsumption(int mol, double value){ 
            if(this->cellPlugin != nullptr) consumptions[grid->moleculesDict[mol]] = value;
            else consumptions[mol] = value;
        }
 
        inline DiffusionGrid * getDiffusionGrid(){ return this->cellPlugin->pluginDiffusion.getGrid(); }
 
        void setDiffusionGrid(DiffusionGrid * g){
            if(grid == nullptr){
                vector<double> consCopy(consumptions);
                for(int i=0; i<consumptions.size(); ++i) consumptions[i] = consCopy[g->moleculesDict[i]];
            }
            grid = g; 
        }
 
        void onCellPluginLinking(){
            auto grid = getDiffusionGrid();
            vector<double> consCopy(consumptions);
            for(int i=0; i<consumptions.size(); ++i) consumptions[i] = consCopy[grid->moleculesDict[i]];
        }
 
    };
 
}
 
 
#endif //ONKO3D_3_0_BODYDIFFUSION_HPP