kdtree.h

Go to the documentation of this file.

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

#ifndef NEIGHBORS_BUFFER_KD_TREE_KDTREE_H_
#define NEIGHBORS_BUFFER_KD_TREE_KDTREE_H_

#include "base.h"
#include "types.h"
#include "../../../include/util.h"

void kd_tree_build_tree(TREE_RECORD *tree_record, 
        TREE_PARAMETERS *params);

void kd_tree_find_best_split(int depth, 
        int left, 
        int right,
        TREE_RECORD *tree_record, 
        TREE_PARAMETERS *params,
        int *axis, 
        int *pivot_idx, 
        FLOAT_TYPE *splitting_value);

void kd_tree_build_recursive(TREE_RECORD *tree_record,
        TREE_PARAMETERS *params,
        INT_TYPE left,
        INT_TYPE right,
        INT_TYPE idx,
        INT_TYPE depth);

void kd_tree_generate_training_patterns_indices(void *XI, 
        FLOAT_TYPE * X, 
        INT_TYPE n,
        INT_TYPE dim);

INT_TYPE kd_tree_split_training_patterns_via_pivot(void *XI, 
        INT_TYPE left, 
        INT_TYPE right,
        INT_TYPE axis, 
        INT_TYPE dim);

#endif /* NEIGHBORS_BUFFER_KD_TREE_KDTREE_H_ */