base.h

Go to the documentation of this file.

/*
 * base.h
 *
 * Copyright (C) 2013-2016 Fabian Gieseke <fabian.gieseke@di.ku.dk>
 * License: GPL v2
 *
 */

#ifndef NEIGHBORS_KDTREE_INCLUDE_BASE_H_
#define NEIGHBORS_KDTREE_INCLUDE_BASE_H_

#include <stdlib.h>
#include <math.h>
#include <stdio.h>
#include <omp.h>

#include "global.h"
#include "util.h"
#include "kdtree.h"

void init_extern(int n_neighbors,
        int tree_depth,
        int num_threads,
        int splitting_type,
        int verbosity_level,
        KD_TREE_PARAMETERS *params);

void fit_extern(FLOAT_TYPE *Xtrain,
        int nXtrain,
        int dXtrain,
        KD_TREE_RECORD *kdtree_record,
        KD_TREE_PARAMETERS *params);

void neighbors_extern(FLOAT_TYPE * Xtest,
        int nXtest,
        int dXtest,
        FLOAT_TYPE * distances,
        int ndistances,
        int ddistances,
        int *indices,
        int nindices,
        int dindices,
        KD_TREE_RECORD *kdtree_record,
        KD_TREE_PARAMETERS *params);

void free_resources_extern(void);

#endif /* NEIGHBORS_KDTREE_INCLUDE_BASE_H_ */