MockMatrix.hpp

Go to the documentation of this file.

#ifndef KNNCOLLE_MOCK_MATRIX_HPP
#define KNNCOLLE_MOCK_MATRIX_HPP
 
namespace knncolle {
 
class MockMatrix {
public:
    MockMatrix(int num_dim, int num_obs, const double* data) : my_num_dim(num_dim), my_num_obs(num_obs), my_data(data), my_long_num_dim(num_dim) {}
public:
    typedef double data_type;
 
    typedef int index_type;
 
    typedef int dimension_type;
 
private:
    dimension_type my_num_dim;
    index_type my_num_obs;
    const data_type* my_data;
    size_t my_long_num_dim;
 
public:
    index_type num_observations() const {
        return my_num_obs;
    }
 
    dimension_type num_dimensions() const {
        return my_num_dim;
    }
 
public:
    struct Workspace {
        size_t at = 0;
    };
 
    Workspace create_workspace() const {
        return Workspace();
    }
 
    const data_type* get_observation(Workspace& workspace) const {
        return my_data + (workspace.at++) * my_long_num_dim; // already size_t to avoid overflow during multiplication.
    } 
};
 
template<typename Dim_, typename Index_, typename Data_>
class SimpleMatrix {
public:
    SimpleMatrix(Dim_ num_dimensions, Index_ num_observations, const Data_* data) : 
        my_num_dim(num_dimensions), my_num_obs(num_observations), my_data(data), my_long_num_dim(num_dimensions) {}
 
private:
    Dim_ my_num_dim;
    Index_ my_num_obs;
    const Data_* my_data;
    size_t my_long_num_dim;
 
public:
    typedef Data_ data_type;
 
    typedef Index_ index_type;
 
    typedef Dim_ dimension_type;
 
    struct Workspace{
        size_t at = 0;
    };
 
public:
    Index_ num_observations() const {
        return my_num_obs;
    }
 
    dimension_type num_dimensions() const {
        return my_num_dim;
    }
 
    Workspace create_workspace() const {
        return Workspace();
    }
 
    const data_type* get_observation(Workspace& workspace) const {
        return my_data + (workspace.at++) * my_long_num_dim; // already size_t's to avoid overflow during multiplication.
    } 
};
 
}
 
#endif