"""!

@brief CCORE Wrapper for oscillatory neural network based on Kuramoto model.

@authors Andrei Novikov (pyclustering@yandex.ru)
@date 2014-2020
@copyright BSD-3-Clause

"""


from pyclustering.core.wrapper import *
from pyclustering.core.pyclustering_package import pyclustering_package, package_extractor


def sync_create_network(num_osc, weight, frequency, type_conn, initial_phases):
    ccore = ccore_library.get()
    
    ccore.sync_create_network.restype = POINTER(c_void_p)
    pointer_network = ccore.sync_create_network(c_uint(num_osc), c_double(weight), c_double(frequency), c_uint(type_conn), c_uint(initial_phases))
    
    return pointer_network


def sync_get_size(pointer_network):
    ccore = ccore_library.get()
    ccore.sync_get_size.restype = c_size_t
    return ccore.sync_get_size(pointer_network)


def sync_destroy_network(pointer_network):
    ccore = ccore_library.get()
    ccore.sync_destroy_network(pointer_network)


def sync_simulate_static(pointer_network, steps, time, solution, collect_dynamic):
    ccore = ccore_library.get()
    ccore.sync_simulate_static.restype = POINTER(c_void_p)
    return ccore.sync_simulate_static(pointer_network, c_uint(steps), c_double(time), c_uint(solution), c_bool(collect_dynamic))


def sync_simulate_dynamic(pointer_network, order, solution, collect_dynamic, step, int_step, threshold_changes):
    ccore = ccore_library.get()
    ccore.sync_simulate_dynamic.restype = POINTER(c_void_p)
    return ccore.sync_simulate_dynamic(pointer_network, c_double(order), c_uint(solution), c_bool(collect_dynamic), c_double(step), c_double(int_step), c_double(threshold_changes))


def sync_order(pointer_network):
    ccore = ccore_library.get()
    ccore.sync_order.restype = c_double
    
    return ccore.sync_order(pointer_network)
    
    
def sync_local_order(pointer_network):
    ccore = ccore_library.get()
    ccore.sync_local_order.restype = c_double
    
    return ccore.sync_local_order(pointer_network)


def sync_connectivity_matrix(pointer_network):
    ccore = ccore_library.get()
    ccore.sync_connectivity_matrix.restype = POINTER(pyclustering_package)
    
    package = ccore.sync_connectivity_matrix(pointer_network)
    
    connectivity_matrix = package_extractor(package).extract()
    ccore.free_pyclustering_package(package)
    
    return connectivity_matrix


def sync_dynamic_get_size(pointer_dynamic):
    ccore = ccore_library.get()
    ccore.sync_dynamic_get_time.restype = c_size_t
    return ccore.sync_dynamic_get_size(pointer_dynamic)


def sync_dynamic_destroy(pointer_dynamic):
    ccore = ccore_library.get()
    ccore.sync_dynamic_destroy(pointer_dynamic)


def sync_dynamic_allocate_sync_ensembles(pointer_dynamic, tolerance, iteration):
    if iteration is None:
        iteration = sync_dynamic_get_size(pointer_dynamic) - 1
    
    ccore = ccore_library.get()
    
    ccore.sync_dynamic_allocate_sync_ensembles.restype = POINTER(pyclustering_package)
    package = ccore.sync_dynamic_allocate_sync_ensembles(pointer_dynamic, c_double(tolerance), c_size_t(iteration))
    
    result = package_extractor(package).extract()
    ccore.free_pyclustering_package(package)
    
    return result


def sync_dynamic_allocate_correlation_matrix(pointer_dynamic, iteration):
    analyse_iteration = iteration
    if analyse_iteration is None:
        analyse_iteration = sync_dynamic_get_size(pointer_dynamic) - 1
    
    ccore = ccore_library.get()
    
    ccore.sync_dynamic_allocate_correlation_matrix.restype = POINTER(pyclustering_package)
    package = ccore.sync_dynamic_allocate_correlation_matrix(pointer_dynamic, c_uint(analyse_iteration))
    
    result = package_extractor(package).extract()
    ccore.free_pyclustering_package(package)
    
    return result


def sync_dynamic_get_output(pointer_dynamic):
    ccore = ccore_library.get()
    
    ccore.sync_dynamic_get_output.restype = POINTER(pyclustering_package)
    package = ccore.sync_dynamic_get_output(pointer_dynamic)
    
    result = package_extractor(package).extract()
    ccore.free_pyclustering_package(package)
    
    return result


def sync_dynamic_get_time(pointer_dynamic):
    ccore = ccore_library.get()
    
    ccore.sync_dynamic_get_time.restype = POINTER(pyclustering_package)
    package = ccore.sync_dynamic_get_time(pointer_dynamic)
    
    result = package_extractor(package).extract()
    ccore.free_pyclustering_package(package)
    
    return result


def sync_dynamic_calculate_order(pointer_dynamic, start_iteration, stop_iteration):
    ccore = ccore_library.get()
    
    ccore.sync_dynamic_calculate_order.restype = POINTER(pyclustering_package)
    package = ccore.sync_dynamic_calculate_order(pointer_dynamic, start_iteration, stop_iteration)
    
    result = package_extractor(package).extract()
    ccore.free_pyclustering_package(package)
    
    return result


def sync_dynamic_calculate_local_order(pointer_dynamic, pointer_network, start_iteration, stop_iteration):
    ccore = ccore_library.get()

    ccore.sync_dynamic_calculate_local_order.restype = POINTER(pyclustering_package)
    package = ccore.sync_dynamic_calculate_local_order(pointer_dynamic, pointer_network, c_size_t(start_iteration), c_size_t(stop_iteration))
    
    result = package_extractor(package).extract()
    ccore.free_pyclustering_package(package)
    
    return result