def isect_b_shader(cls, shader_name):
code = """
temp = ray.origin - sphere.origin
r_dir = ray.direction
a = dot(r_dir, r_dir)
b = dot(temp, r_dir) * 2.0
c = dot(temp, temp) - sphere.radius * sphere.radius
disc = b * b - 4.0 * a * c
if disc < 0.0:
return 0
e = sqrt(disc)
denom = 2.0 * a
t = (-1.0 * b - e) / denom
if t > 0.0005:
if t < min_dist:
return 1
t = (-1.0 * b + e) / denom
if t > 0.0005:
if t < min_dist:
return 1
return 0
"""
func_args = [StructArgPtr('ray', Ray.factory()),
StructArgPtr('sphere', Sphere.factory()),
FloatArg('min_dist', 0.0)]
shader = Shader(code=code, args=[], name=shader_name,
func_args=func_args, is_func=True)
return DependencyShader(shader)
def load(self, shader_name):
args = []
text = self._loader.load(shader_name, 'props.txt')
if text is not None:
args = parse_args(text)
w = Vec3Arg('w', self._w)
u = Vec3Arg('u', self._u)
v = Vec3Arg('v', self._v)
distance = FloatArg('distance', self._distance)
eye = Vec3Arg('eye', self._eye)
lookat = Vec3Arg('lookat', self._lookat)
args.extend([w, u, v, distance, eye, lookat])
code = self._loader.load(shader_name, 'code.py')
if code is None:
raise ValueError("code.py in %s shader dont exist!" % shader_name)
func_args = [
StructArgPtr('ray', Ray.factory()),
StructArgPtr('sample', Sample.factory())
]
self._shader_name = shader_name
self.shader = Shader(code=code,
args=args,
name='generate_ray',
func_args=func_args,
is_func=True)
def isect_shader(cls, shader_name):
code = """
temp = ray.origin - sphere.origin
r_dir = ray.direction
a = dot(r_dir, r_dir)
b = dot(temp, r_dir) * 2.0
c = dot(temp, temp) - sphere.radius * sphere.radius
disc = b * b - 4.0 * a * c
if disc < 0.0:
return 0
e = sqrt(disc)
denom = 2.0 * a
t = (-1.0 * b - e) / denom
if t > 0.0005:
if t < min_dist:
normal = (temp + r_dir * t) * (1.0 / sphere.radius)
hit = ray.origin + r_dir * t
hitpoint.t = t
hitpoint.normal = normal
hitpoint.hit = hit
hitpoint.mat_idx = sphere.mat_idx
hitpoint.light_id = sphere.light_id
hitpoint.u = 0.0
hitpoint.v = 0.0
return 1
t = (-1.0 * b + e) / denom
if t > 0.0005:
if t < min_dist:
normal = (temp + r_dir * t) * (1.0 / sphere.radius)
hit = ray.origin + r_dir * t
hitpoint.t = t
hitpoint.normal = normal
hitpoint.hit = hit
hitpoint.mat_idx = sphere.mat_idx
hitpoint.light_id = sphere.light_id
hitpoint.u = 0.0
hitpoint.v = 0.0
return 1
return 0
"""
func_args = [
StructArgPtr('ray', Ray.factory()),
StructArgPtr('sphere', Sphere.factory()),
StructArgPtr('hitpoint', HitPoint.factory()),
FloatArg('min_dist', 0.0)
]
shader = Shader(code=code,
args=[],
name=shader_name,
func_args=func_args,
is_func=True)
return DependencyShader(shader)
def isect_shader(cls, shader_name):
code = """
temp1 = dot(ray.direction, rectangle.normal)
if temp1 == 0.0:
return 0
tmp = rectangle.point - ray.origin
t = dot(tmp, rectangle.normal) / temp1
if t < 0.00001:
return 0
if t > min_dist:
return 0
p = ray.origin + ray.direction * t
d = p - rectangle.point
ddota = dot(d, rectangle.edge_a)
if ddota < 0.0:
return 0
if ddota > rectangle.edge_a_squared:
return 0
ddotb = dot(d, rectangle.edge_b)
if ddotb < 0.0:
return 0
if ddotb > rectangle.edge_b_squared:
return 0
hitpoint.t = t
hitpoint.normal = rectangle.normal
hitpoint.hit = p
hitpoint.mat_idx = rectangle.mat_idx
hitpoint.light_id = rectangle.light_id
hitpoint.u = 0.0
hitpoint.v = 0.0
return 1
"""
func_args = [
StructArgPtr('ray', Ray.factory()),
StructArgPtr('rectangle', Rectangle.factory()),
StructArgPtr('hitpoint', HitPoint.factory()),
FloatArg('min_dist', 0.0)
]
shader = Shader(code=code,
args=[],
name=shader_name,
func_args=func_args,
is_func=True)
return DependencyShader(shader)
def isect_shader(cls, shader_name):
code = """
temp = ray.origin - sphere.origin
r_dir = ray.direction
a = dot(r_dir, r_dir)
b = dot(temp, r_dir) * 2.0
c = dot(temp, temp) - sphere.radius * sphere.radius
disc = b * b - 4.0 * a * c
if disc < 0.0:
return 0
e = sqrt(disc)
denom = 2.0 * a
t = (-1.0 * b - e) / denom
if t > 0.0005:
if t < min_dist:
normal = (temp + r_dir * t) * (1.0 / sphere.radius)
hit = ray.origin + r_dir * t
hitpoint.t = t
hitpoint.normal = normal
hitpoint.hit = hit
hitpoint.mat_idx = sphere.mat_idx
hitpoint.light_id = sphere.light_id
hitpoint.u = 0.0
hitpoint.v = 0.0
return 1
t = (-1.0 * b + e) / denom
if t > 0.0005:
if t < min_dist:
normal = (temp + r_dir * t) * (1.0 / sphere.radius)
hit = ray.origin + r_dir * t
hitpoint.t = t
hitpoint.normal = normal
hitpoint.hit = hit
hitpoint.mat_idx = sphere.mat_idx
hitpoint.light_id = sphere.light_id
hitpoint.u = 0.0
hitpoint.v = 0.0
return 1
return 0
"""
func_args = [StructArgPtr('ray', Ray.factory()),
StructArgPtr('sphere', Sphere.factory()),
StructArgPtr('hitpoint', HitPoint.factory()),
FloatArg('min_dist', 0.0)]
shader = Shader(code=code, args=[], name=shader_name,
func_args=func_args, is_func=True)
return DependencyShader(shader)
def isect_shader(cls, shader_name):
code = """
temp1 = dot(ray.direction, rectangle.normal)
if temp1 == 0.0:
return 0
tmp = rectangle.point - ray.origin
t = dot(tmp, rectangle.normal) / temp1
if t < 0.00001:
return 0
if t > min_dist:
return 0
p = ray.origin + ray.direction * t
d = p - rectangle.point
ddota = dot(d, rectangle.edge_a)
if ddota < 0.0:
return 0
if ddota > rectangle.edge_a_squared:
return 0
ddotb = dot(d, rectangle.edge_b)
if ddotb < 0.0:
return 0
if ddotb > rectangle.edge_b_squared:
return 0
hitpoint.t = t
hitpoint.normal = rectangle.normal
hitpoint.hit = p
hitpoint.mat_idx = rectangle.mat_idx
hitpoint.light_id = rectangle.light_id
hitpoint.u = 0.0
hitpoint.v = 0.0
return 1
"""
func_args = [StructArgPtr('ray', Ray.factory()),
StructArgPtr('rectangle', Rectangle.factory()),
StructArgPtr('hitpoint', HitPoint.factory()),
FloatArg('min_dist', 0.0)]
shader = Shader(code=code, args=[], name=shader_name,
func_args=func_args, is_func=True)
return DependencyShader(shader)
def isect_b_shader(cls, shader_name):
label = 'ray_triangle_isect_b_%s' % id(cls)
tri_isect = ray_triangle_isect_shader(label, isect_bool=True)
code = """
return %s(ray, triangle.p0, triangle.p1, triangle.p2, min_dist)
""" % label
args = []
func_args = [StructArgPtr('ray', Ray.factory()),
StructArgPtr('triangle', FlatTriangle.factory()),
FloatArg('min_dist', 0.0)]
shader = Shader(code=code, args=args, name=shader_name,
func_args=func_args, is_func=True)
isect_shader = DependencyShader(shader, [tri_isect])
return isect_shader
def load(self, shader_name):
args = []
text = self._loader.load(shader_name, 'props.txt')
if text is not None:
args = parse_args(text)
w = Vec3Arg('w', self._w)
u = Vec3Arg('u', self._u)
v = Vec3Arg('v', self._v)
distance = FloatArg('distance', self._distance)
eye = Vec3Arg('eye', self._eye)
lookat = Vec3Arg('lookat', self._lookat)
args.extend([w, u, v, distance, eye, lookat])
code = self._loader.load(shader_name, 'code.py')
if code is None:
raise ValueError("code.py in %s shader dont exist!" % shader_name)
func_args = [StructArgPtr('ray', Ray.factory()),
StructArgPtr('sample', Sample.factory())]
self._shader_name = shader_name
self.shader = Shader(code=code, args=args, name='generate_ray',
func_args=func_args, is_func=True)
def isect_b_shader(cls, shader_name):
code = """
temp = ray.origin - sphere.origin
r_dir = ray.direction
a = dot(r_dir, r_dir)
b = dot(temp, r_dir) * 2.0
c = dot(temp, temp) - sphere.radius * sphere.radius
disc = b * b - 4.0 * a * c
if disc < 0.0:
return 0
e = sqrt(disc)
denom = 2.0 * a
t = (-1.0 * b - e) / denom
if t > 0.0005:
if t < min_dist:
return 1
t = (-1.0 * b + e) / denom
if t > 0.0005:
if t < min_dist:
return 1
return 0
"""
func_args = [
StructArgPtr('ray', Ray.factory()),
StructArgPtr('sphere', Sphere.factory()),
FloatArg('min_dist', 0.0)
]
shader = Shader(code=code,
args=[],
name=shader_name,
func_args=func_args,
is_func=True)
return DependencyShader(shader)
def ray_triangle_isect_shader(name, isect_bool=False):
code = """
origin = ray.origin
direction = ray.direction
a = p0[0] - p1[0]
b = p0[0] - p2[0]
c = direction[0]
d = p0[0] - origin[0]
e = p0[1] - p1[1]
f = p0[1] - p2[1]
g = direction[1]
h = p0[1]- origin[1]
i = p0[2] - p1[2]
j = p0[2] - p2[2]
k = direction[2]
l = p0[2] - origin[2]
m = f * k - g * j
n = h * k - g * l
p = f * l - h * j
q = g * i - e * k
s = e * j - f * i
temp3 = a * m + b * q + c * s
if temp3 == 0.0:
return 0
inv_denom = 1.0 / temp3
e1 = d * m - b * n - c * p
beta = e1 * inv_denom
if beta < 0.0:
return 0
r = e * l - h * i
e2 = a * n + d * q + c * r
gamma = e2 * inv_denom
if gamma < 0.0:
return 0
betagamma = beta + gamma
if betagamma > 1.0:
return 0
e3 = a * p - b * r + d * s
t = e3 * inv_denom
epsilon = 0.00001
if t < 0.00001:
return 0
if t > min_dist:
return 0
"""
if isect_bool:
code += """
return 1
"""
else:
code += """
hitpoint.t = t
hitpoint.hit = direction * t + origin
hitpoint.u = beta
hitpoint.v = gamma
return 1
"""
ray_a = StructArgPtr('ray', Ray.factory())
p0 = Vec3Arg('p0', Vector3(0.0, 0.0, 0.0))
p1 = Vec3Arg('p1', Vector3(0.0, 0.0, 0.0))
p2 = Vec3Arg('p2', Vector3(0.0, 0.0, 0.0))
dist = FloatArg('min_dist', 0.0)
hit_a = StructArgPtr('hitpoint', HitPoint.factory())
if isect_bool:
func_args = [ray_a, p0, p1, p2, dist]
else:
func_args = [ray_a, p0, p1, p2, dist, hit_a]
shader = Shader(code=code,
args=[],
name=name,
func_args=func_args,
is_func=True)
return shader
def ray_triangle_isect_shader(name, isect_bool=False):
code = """
origin = ray.origin
direction = ray.direction
a = p0[0] - p1[0]
b = p0[0] - p2[0]
c = direction[0]
d = p0[0] - origin[0]
e = p0[1] - p1[1]
f = p0[1] - p2[1]
g = direction[1]
h = p0[1]- origin[1]
i = p0[2] - p1[2]
j = p0[2] - p2[2]
k = direction[2]
l = p0[2] - origin[2]
m = f * k - g * j
n = h * k - g * l
p = f * l - h * j
q = g * i - e * k
s = e * j - f * i
temp3 = a * m + b * q + c * s
if temp3 == 0.0:
return 0
inv_denom = 1.0 / temp3
e1 = d * m - b * n - c * p
beta = e1 * inv_denom
if beta < 0.0:
return 0
r = e * l - h * i
e2 = a * n + d * q + c * r
gamma = e2 * inv_denom
if gamma < 0.0:
return 0
betagamma = beta + gamma
if betagamma > 1.0:
return 0
e3 = a * p - b * r + d * s
t = e3 * inv_denom
epsilon = 0.00001
if t < 0.00001:
return 0
if t > min_dist:
return 0
"""
if isect_bool:
code += """
return 1
"""
else:
code += """
hitpoint.t = t
hitpoint.hit = direction * t + origin
hitpoint.u = beta
hitpoint.v = gamma
return 1
"""
ray_a = StructArgPtr('ray', Ray.factory())
p0 = Vec3Arg('p0', Vector3(0.0, 0.0, 0.0))
p1 = Vec3Arg('p1', Vector3(0.0, 0.0, 0.0))
p2 = Vec3Arg('p2', Vector3(0.0, 0.0, 0.0))
dist = FloatArg('min_dist', 0.0)
hit_a = StructArgPtr('hitpoint', HitPoint.factory())
if isect_bool:
func_args = [ray_a, p0, p1, p2, dist]
else:
func_args = [ray_a, p0, p1, p2, dist, hit_a]
shader = Shader(code=code, args=[], name=name,
func_args=func_args, is_func=True)
return shader
