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 <unordered_map>
 
using namespace std;
 
namespace Diffusion2D {
 
    template<typename cell_t, class plugin_t>
    class BodyDiffusion : public virtual BaseBody<plugin_t> {
 
    private:
        std::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 std::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();
            std::vector<double> consCopy(consumptions);
            for(int i = 0; i < consumptions.size(); ++i) consumptions[i] = consCopy[grid->moleculesDict[i]];
        }
 
    };
 
}
 
#endif //ONKO3D_3_0_BODYDIFFUSION_HPP