运行环境
[Taichi] version 0.7.32, llvm 10.0.0, commit 6652f94f, win, python 3.8.1
[Taichi] Starting on arch=cuda
问题描述
在参照第3节课高级数据结构的方法对github上的Galaxy项目的data block进行了修改之后，发现会有报错，不能正常跑起来。报错信息如图：
代码链接
代码只对celestial_objects.py进行了修改，修改后的代码如下：
import taichi as ti
# constants
G = 1
PI = 3.1415926
@ti.data_oriented
class CelestialObject:
def __init__(self, N, mass) -> None:
self.n = N
self.m = mass
self.p = ti.Vector.field(2, ti.f32)
self.v = ti.Vector.field(2, ti.f32)
self.f = ti.Vector.field(2, ti.f32)
self.pos = ti.root.dense(ti.i, self.n).place(self.p)
self.vel = ti.root.dense(ti.i, self.n).place(self.v)
self.force = ti.root.dense(ti.i, self.n).place(self.f)
def display(self, gui, radius=2, color=0xffffff):
gui.circles(self.pos.to_numpy(), radius=radius, color=color)
@ti.func
def Pos(self):
return self.pos
@ti.func
def Mass(self):
return self.m
@ti.func
def Number(self):
return self.n
@ti.func
def clearForce(self):
for i in self.force:
self.force[i] = ti.Vector([0.0, 0.0])
@ti.kernel
def initialize(self, center_x: ti.f32, center_y: ti.f32, size: ti.f32, init_speed: ti.f32):
for i in range(self.n):
if self.n == 1:
self.pos[i] = ti.Vector([center_x, center_y])
self.vel[i] = ti.Vector([0.0, 0.0])
else:
theta, r = self.generateThetaAndR(i, self.n)
offset_dir = ti.Vector([ti.cos(theta), ti.sin(theta)])
center = ti.Vector([center_x, center_y])
self.pos[i] = center + r * offset_dir * size
self.vel[i] = ti.Vector([-offset_dir[1], offset_dir[0]]) * init_speed
@ti.kernel
def computeForce(self):
self.clearForce()
for i in range(self.n):
p = self.pos[i]
for j in range(self.n):
if j != i:
diff = self.pos[j] - p
r = diff.norm(1e-2)
self.force[i] += G * self.Mass() * self.Mass() * diff / r**3
@ti.kernel
def update(self, h: ti.f32):
for i in self.vel:
self.vel[i] += h * self.force[i] / self.Mass()
self.pos[i] += h * self.vel[i]
@ti.data_oriented
class Star(CelestialObject):
def __init__(self, N, mass) -> None:
super().__init__(N, mass)
pass
@staticmethod
@ti.func
def generateThetaAndR(i, n):
theta = 2*PI*i/ti.cast(n, ti.f32)
r = 1
return theta, r
@ti.data_oriented
class Planet(CelestialObject):
def __init__(self, N, mass) -> None:
super().__init__(N, mass)
pass
@staticmethod
@ti.func
def generateThetaAndR(i,n):
theta = 2 * PI * ti.random() # theta \in (0, 2PI)
r = (ti.sqrt(ti.random()) * 0.4 + 0.6) # r \in (0.6,1)
return theta, r
@ti.kernel
def computeForce(self, stars: ti.template()):
self.clearForce()
for i in range(self.n):
p = self.pos[i]
for j in range(self.n):
if i != j:
diff = self.pos[j] - p
r = diff.norm(1e-2)
self.force[i] += G * self.Mass() * self.Mass() * diff / r**3
for j in range(stars.Number()):
diff = stars.Pos()[j] - p
r = diff.norm(1e-2)
self.force[i] += G * self.Mass() * stars.Mass() * diff / r**3