File: hhn_wrapper.py

package info (click to toggle)
python-pyclustering 0.10.1.2-2
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 11,128 kB
  • sloc: cpp: 38,888; python: 24,311; sh: 384; makefile: 105
file content (159 lines) | stat: -rwxr-xr-x 5,761 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
"""!

@brief CCORE Wrapper for Hodgkin-Huxley oscillatory network for image segmentation.

@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, package_builder;


class c_hhn_params(Structure):
    _fields_ = [
        ("nu",                  c_double),
        ("gNa",                 c_double),
        ("gK",                  c_double),
        ("gL",                  c_double),
        ("vNa",                 c_double),
        ("vK",                  c_double),
        ("vL",                  c_double),
        ("vRest",               c_double),
        ("Icn1",                c_double),
        ("Icn2",                c_double),
        ("Vsyninh",             c_double),
        ("Vsynexc",             c_double),
        ("alfa_inhibitory",     c_double),
        ("betta_inhibitory",    c_double),
        ("alfa_excitatory",     c_double),
        ("betta_excitatory",    c_double),
        ("w1",                  c_double),
        ("w2",                  c_double),
        ("w3",                  c_double),
        ("deltah",              c_double),
        ("threshold",           c_double),
        ("eps",                 c_double),
    ];


def hhn_create(size, params):
    c_params = c_hhn_params();

    c_params.nu                 = params.nu;
    c_params.gNa                = params.gNa;
    c_params.gK                 = params.gK;
    c_params.gL                 = params.gL;
    c_params.vNa                = params.vNa;
    c_params.vK                 = params.vK;
    c_params.vL                 = params.vL;
    c_params.vRest              = params.vRest;
    c_params.Icn1               = params.Icn1;
    c_params.Icn2               = params.Icn2;
    c_params.Vsyninh            = params.Vsyninh;
    c_params.Vsynexc            = params.Vsynexc;
    c_params.alfa_inhibitory    = params.alfa_inhibitory;
    c_params.betta_inhibitory   = params.betta_inhibitory;
    c_params.alfa_excitatory    = params.alfa_excitatory;
    c_params.betta_excitatory   = params.betta_excitatory;
    c_params.w1                 = params.w1;
    c_params.w2                 = params.w2;
    c_params.w3                 = params.w3;
    c_params.deltah             = params.deltah;
    c_params.threshold          = params.threshold;
    c_params.eps                = params.eps;

    ccore = ccore_library.get();

    ccore.hhn_create.restype = POINTER(c_void_p);
    hhn_network_pointer = ccore.hhn_create(c_size_t(size), pointer(c_params));
    return hhn_network_pointer;


def hhn_destroy(hhn_network_pointer):
    ccore = ccore_library.get();
    ccore.hhn_destroy(hhn_network_pointer);


def hhn_dynamic_create(collect_membrane, collect_active_cond_sodium, collect_inactive_cond_sodium, collect_active_cond_potassium):
    ccore = ccore_library.get();

    ccore.hhn_dynamic_create.restype = POINTER(c_void_p);
    hhn_dynamic_pointer = ccore.hhn_dynamic_create(c_bool(collect_membrane),
                                                   c_bool(collect_active_cond_sodium),
                                                   c_bool(collect_inactive_cond_sodium),
                                                   c_bool(collect_active_cond_potassium));
    return hhn_dynamic_pointer;


def hhn_dynamic_destroy(hhn_dynamic_pointer):
    ccore = ccore_library.get();
    ccore.hhn_dynamic_destroy(hhn_dynamic_pointer);


def hhn_simulate(hhn_network_pointer, steps, time, solution, stimulus, ccore_hhn_dynamic_pointer):
    ccore = ccore_library.get();

    c_stimulus = package_builder(stimulus, c_double).create();
    ccore.hhn_simulate(hhn_network_pointer,
                       c_size_t(steps),
                       c_double(time),
                       c_size_t(solution),
                       c_stimulus,
                       ccore_hhn_dynamic_pointer);


def hhn_dynamic_get_peripheral_evolution(ccore_hhn_dynamic_pointer, index_collection):
    ccore = ccore_library.get();

    ccore.hhn_dynamic_get_peripheral_evolution.restype = POINTER(pyclustering_package);
    dynamic_package = ccore.hhn_dynamic_get_peripheral_evolution(ccore_hhn_dynamic_pointer, c_size_t(index_collection));

    result = package_extractor(dynamic_package).extract();
    ccore.free_pyclustering_package(dynamic_package);

    return result;


def hhn_dynamic_get_central_evolution(ccore_hhn_dynamic_pointer, index_collection):
    ccore = ccore_library.get();

    ccore.hhn_dynamic_get_central_evolution.restype = POINTER(pyclustering_package);
    dynamic_package = ccore.hhn_dynamic_get_central_evolution(ccore_hhn_dynamic_pointer, c_size_t(index_collection));

    result = package_extractor(dynamic_package).extract();
    ccore.free_pyclustering_package(dynamic_package);

    return result;


def hhn_dynamic_get_time(ccore_hhn_dynamic_pointer):
    ccore = ccore_library.get();

    ccore.hhn_dynamic_get_time.restype = POINTER(pyclustering_package);
    dynamic_package = ccore.hhn_dynamic_get_time(ccore_hhn_dynamic_pointer);

    result = package_extractor(dynamic_package).extract();
    ccore.free_pyclustering_package(dynamic_package);

    return result;


def hhn_dynamic_write(ccore_hhn_dynamic_pointer, filename):
    ccore = ccore_library.get();

    byte_filename = filename.encode('utf-8');
    ccore.hhn_dynamic_write(ccore_hhn_dynamic_pointer, c_char_p(byte_filename));


def hhn_dynamic_read(filename):
    ccore = ccore_library.get();

    byte_filename = filename.encode('utf-8');

    ccore.hhn_dynamic_read.restype = POINTER(c_void_p);
    hhn_dynamic_pointer = ccore.hhn_dynamic_read(byte_filename);

    return hhn_dynamic_pointer;