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 Diffusion2D {
 
 
    template<typename cell_t, class plugin_t>
    class BodyDiffusion : public virtual BaseBody<plugin_t> {
 
    private:
        vector<double> consumptions; 
        int nbMolecules = 0; 
    public:
 
        inline BodyDiffusion() : consumptions() {}
 
        inline explicit BodyDiffusion(int size){
            initNbMolecules(size);
        }
 
        void initNbMolecules(int n){ 
            if(n > nbMolecules) {
                nbMolecules = n;
                consumptions.assign(n,0.);
            }
        }
 
        double getQuantity(int x, int y, int mol) { 
            if(this->cellPlugin != nullptr) return getDiffusionGrid()->getMolecule(x,y,mol);
            else return 0.;
        }
 
        double getConsumption(int mol) { 
            if(this->cellPlugin != nullptr) return consumptions[getDiffusionGrid()->moleculesDict[mol]];
            else return consumptions[mol];
        }
 
        inline vector<double> getConsumptions() const{ return consumptions; }
 
        inline void setConsumption(int mol, double value){
            if(this->cellPlugin != nullptr) consumptions[getDiffusionGrid()->moleculesDict[mol]] = value;
            else consumptions[mol] = value;
        }
 
        inline DiffusionGrid * getDiffusionGrid() { return this->cellPlugin->pluginDiffusion.getDiffusionGrid(); }
 
        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