mirror of http://CODE.RHODECODE.COM/u/O/O/O
0000OOOO0000
4 years ago
committed by
GitHub
1 changed files with 0 additions and 54 deletions
@ -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