#include <gsl/gsl_math.h>

#include "bct_test.h"

DEFUN_DLD(breadth_cpp, args, , "Wrapper for C++ function.") {

if (args.length() != 2) {

return octave_value_list();

}

Matrix CIJ = args(0).matrix_value();

int source = args(1).int_value() - 1;

if (!error_state) {

gsl_matrix* CIJ_gsl = bct_test::to_gslm(CIJ);

gsl_vector* branch;

gsl_vector* distance = bct::breadth(CIJ_gsl, source, &branch);

for (int i = 0; i < (int)branch->size; i++) {

int value = (int)gsl_vector_get(branch, i);

if (gsl_isinf(gsl_vector_get(distance, i)) == 0 && value != -1) {

gsl_vector_set(branch, i, (double)value + 1.0);

}

}

octave_value_list ret;

ret(0) = octave_value(bct_test::from_gsl(distance));

ret(1) = octave_value(bct_test::from_gsl(branch));

gsl_matrix_free(CIJ_gsl);

gsl_vector_free(branch);

gsl_vector_free(distance);

return ret;

} else {

return octave_value_list();

}

}