这是一部分代码:
——time_iter.py
import taichi as ti
import math
#… #import all modules needed
import numpy as np
#… #初始化太极 ti.init()
from user_defined_functions import *
#…
#… #get nodelist (例如nodelist = [0 0 0 0 0 ])
addBC_Forc_user(nodelist) #这里会报错
#…
——user_defined_functions.py
import data
import taichi as ti
import numpy as np
pf = data.pointField
#… @ti.kernel
def addBC_Forc_user(nodelist:ti.types.ndarray()):
for i in nodelist:
pf[nodelist[i]].Forc += -pf[nodelist[i]].Disp * pf[nodelist[i]].Density*0.01