DistanceMatrix.hpp

Go to the documentation of this file.

/*Copyright 2014, Grant Brown*/

#ifndef SPATIALSEIR_INCLUDEFILES
#define SPATIALSEIR_INCLUDEFILES

#include<math.h>
#include<cstring>
#include<vector>
#endif

#ifndef DISTANCE_MATRIX_INC
#define DISTANCE_MATRIX_INC

namespace SpatialSEIR
{
    struct distanceArgs
    {
        std::vector<double*> inData;
        int* dim; 
    };

    struct scaledDistanceArgs
    {
        double* phi;
        std::vector<double*> inData;
        int* dim;
        int mode;
    };

    // Begin with a naive implementation, this should eventually 
    // be updated to take advantage of the inherent sparsity of 
    // a distance matrix with a max range, as well as the structure 
    // given by symmetry.
    //
    // First priority: make it work
    class DistanceMatrix
    {
        public:
            // Methods 
            DistanceMatrix();

            int genFromDataStream(double *indata, int *dim); 

            // Scaled options for already allocated distance matrix
            int scaledInvFunc_CPU(double phi);
            int scaledInvFunc_OCL(double phi);

            // Make rows sum to one. 
            void makeRowStochastic();

            // Scaled options for unallocated distance matrix
            int scaledInvFunc_CPU(double phi, double *indata, int *dim);
            int scaledInvFunc_OCL(double phi, double *indata, int *dim);
            int gravityFunc_CPU(double phi, double *indata, int *dim);
            int gravityFunc_OCL(double phi, double *indata, int *dim);


            ~DistanceMatrix();
            // Attributes
            double *data;
            int *numLocations;

        private:
            int *hasAlloc;
    };
}

#endif