LSS_FullConditional.hpp

Go to the documentation of this file.

#ifndef SPATIALSEIR_INCLUDEFILES
#define SPATIALSEIR_INCLUDEFILES

#include<cstring>
#include<vector>
#endif

#ifndef FULL_CONDITIONAL_INC
#define FULL_CONDITIONAL_INC

#define LSS_COMPARTMENT_FULL_CONDITIONAL_TYPE 1
#define LSS_INIT_COMPARTMENT_FULL_CONDITIONAL_TYPE 2
#define LSS_PARAMETER_FULL_CONDITIONAL_TYPE 3


namespace SpatialSEIR
{

    class ModelContext;
    class CompartmentalModelMatrix;
    class CovariateMatrix;
    class OCLProvider;
    class Sampler;

    struct priorControl
    {
        double betaPriorPrecision;
        double betaPrsPriorPrecision;
        double P_EI_priorAlpha;
        double P_EI_priorBeta;
        double P_IR_priorAlpha;
        double P_IR_priorBeta;
        double Phi_priorAlpha;
        double Phi_priorBeta;
    };

    struct sliceParameters
    {
        double* S_starWidth;
        double* E_starWidth;
        double* R_starWidth;
        double* S0Width;
        double* I0Width;
        double* betaWidth;
        double* betaPrsWidth;
        double* rhoWidth;
        double* gammaEiWidth;
        double* gammaIrWidth;
        double* phiWidth;
    };

    class InitData
    {
        public:
            InitData(int *_S0,
                        int *_E0,
                        int *_I0,
                        int *_R0, 
                        int *nLoc);
            InitData();
            void populate(int *_S0,
                          int *_E0,
                          int *_I0,
                          int *_R0,
                          int *nLoc);

            ~InitData();
            int *S0;
            int *E0;
            int *I0;
            int *R0;
            int *numLocations;
    };



    class FullConditional
    {
        public:
            //Template for shared methods
            virtual ~FullConditional(){}; 
            virtual void sample(int verbose) = 0;
            virtual long double getValue() = 0;
            virtual void setValue(long double value) = 0;
            virtual int calculateRelevantCompartments() = 0;
            virtual int calculateRelevantCompartments_OCL() = 0;
            virtual void updateSamplingParameters(double desiredRatio, double targetWidth, double proportionChange) = 0;
            virtual int getFullConditionalType() = 0;
            double acceptanceRatio();
            double* sliceWidth;
            void setSamplerType(int type);
            std::vector<Sampler*>* samplers;
            Sampler** currentSampler;
            int* samples;
            int* accepted;
    };

    class CompartmentFullConditional : public FullConditional 
    {
        public:
            //Template for shared methods
            virtual ~CompartmentFullConditional(){}; 
            virtual int evalCPU() = 0;
            virtual int evalCPU(int i, int j) = 0;
            virtual int evalOCL() = 0;
            virtual void sample(int verbose) = 0;
            virtual long double getValue() = 0;
            virtual void setValue(long double value) = 0;
            virtual int calculateRelevantCompartments() = 0;
            virtual int calculateRelevantCompartments(int i, int j) = 0;
            virtual int calculateRelevantCompartments_OCL() = 0;
            void updateSamplingParameters(double desiredRatio, double targetWidth, double proportionChange);

            int getFullConditionalType();

            double* steadyStateConstraintPrecision;
    };

    class ParameterFullConditional : public FullConditional
    {
        public:
            //Template for shared methods
            virtual ~ParameterFullConditional(){}; 
            virtual int evalCPU() = 0;
            virtual int evalOCL() = 0;
            virtual void sample(int verbose) = 0;
            virtual long double getValue() = 0;
            virtual void setValue(long double value) = 0;
            virtual int calculateRelevantCompartments() = 0;
            virtual int calculateRelevantCompartments_OCL() = 0;
            virtual double acceptanceRatio(int i);
            void updateSamplingParameters(double desiredRatio, double targetWidth, double proportionChange);

            int *varLen;

            int getFullConditionalType();
    };


    class InitCompartmentFullConditional : public FullConditional
    {
        public:

            virtual ~InitCompartmentFullConditional(){}; 
            virtual int evalCPU() = 0;
            virtual int evalOCL() = 0;
            virtual void sample(int verbose) = 0;
            virtual long double getValue() = 0;
            virtual void setValue(long double value) = 0;
            virtual int calculateRelevantCompartments() = 0;
            virtual int calculateRelevantCompartments_OCL() = 0;
            void updateSamplingParameters(double desiredRatio, double targetWidth, double proportionChange);


            int getFullConditionalType();
    };
}

#endif