mirror of
http://CODE.RHODECODE.COM/u/O/O/O
synced 2024-11-29 05:08:45 -05:00
Delete NUMBA_FUNCTIONS.PY
This commit is contained in:
parent
a21168018a
commit
d92071a33a
@ -1,54 +0,0 @@
|
||||
# ##### BEGIN GPL LICENSE BLOCK #####
|
||||
#
|
||||
# This program is free software; you can redistribute it and/or
|
||||
# modify it under the terms of the GNU General Public License
|
||||
# as published by the Free Software Foundation; either version 2
|
||||
# of the License, or (at your option) any later version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program; if not, write to the Free Software Foundation,
|
||||
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||
#
|
||||
# ##### END GPL LICENSE BLOCK #####
|
||||
|
||||
import numpy as np
|
||||
try:
|
||||
from numba import jit
|
||||
print("Tissue: Numba module loaded succesfully")
|
||||
@jit
|
||||
def numba_reaction_diffusion(n_verts, n_edges, edge_verts, a, b, diff_a, diff_b, f, k, dt, time_steps):
|
||||
arr = np.arange(n_edges)*2
|
||||
id0 = edge_verts[arr] # first vertex indices for each edge
|
||||
id1 = edge_verts[arr+1] # second vertex indices for each edge
|
||||
for i in range(time_steps):
|
||||
lap_a = np.zeros(n_verts)
|
||||
lap_b = np.zeros(n_verts)
|
||||
lap_a0 = a[id1] - a[id0] # laplacian increment for first vertex of each edge
|
||||
lap_b0 = b[id1] - b[id0] # laplacian increment for first vertex of each edge
|
||||
|
||||
for i, j, la0, lb0 in zip(id0,id1,lap_a0,lap_b0):
|
||||
lap_a[i] += la0
|
||||
lap_b[i] += lb0
|
||||
lap_a[j] -= la0
|
||||
lap_b[j] -= lb0
|
||||
ab2 = a*b**2
|
||||
#a += eval("(diff_a*lap_a - ab2 + f*(1-a))*dt")
|
||||
#b += eval("(diff_b*lap_b + ab2 - (k+f)*b)*dt")
|
||||
a += (diff_a*lap_a - ab2 + f*(1-a))*dt
|
||||
b += (diff_b*lap_b + ab2 - (k+f)*b)*dt
|
||||
return a, b
|
||||
|
||||
@jit
|
||||
def numba_lerp2(v00, v10, v01, v11, vx, vy):
|
||||
co0 = v00 + (v10 - v00) * vx
|
||||
co1 = v01 + (v11 - v01) * vx
|
||||
co2 = co0 + (co1 - co0) * vy
|
||||
return co2
|
||||
except:
|
||||
print("Tissue: Numba not installed")
|
||||
pass
|
Loading…
Reference in New Issue
Block a user