clean
This commit is contained in:
parent
10e9d6775e
commit
f7fb3365d7
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
*.csv
|
||||||
|
__pycache__/
|
47802
rainbow.nb
47802
rainbow.nb
File diff suppressed because it is too large
Load Diff
@ -23,11 +23,11 @@ points = []
|
|||||||
|
|
||||||
def init():
|
def init():
|
||||||
for x in np.linspace(-r + dx, r-dx, N):
|
for x in np.linspace(-r + dx, r-dx, N):
|
||||||
stack.append(Ray([x, 2*r], [0,-1], 1))
|
stack.append(Ray([x, 2*r], [0,-1], 10*np.abs(x)))
|
||||||
print(np.linspace(-r + dx, r-dx, N))
|
print(np.linspace(-r + dx, r-dx, N))
|
||||||
|
|
||||||
def reflection_and_refraction(ray:Ray, intersection_point, normal, n):
|
def reflection_and_refraction(ray:Ray, intersection_point, normal, n):
|
||||||
print("reflection/refraction at the point:", intersection_point)
|
# print("reflection/refraction at the point:", intersection_point)
|
||||||
k_n = np.dot(ray.direction, normal)
|
k_n = np.dot(ray.direction, normal)
|
||||||
if k_n > 0:
|
if k_n > 0:
|
||||||
n1,n2 = n,1
|
n1,n2 = n,1
|
||||||
@ -67,6 +67,7 @@ def trace(disk:Disk, stack:list, max_ray:int):
|
|||||||
points.append(np.concatenate((ray.origin, ray.origin+ray.direction,[ray.intensity])))
|
points.append(np.concatenate((ray.origin, ray.origin+ray.direction,[ray.intensity])))
|
||||||
if direction[1] > 0:
|
if direction[1] > 0:
|
||||||
result.append([np.arccos(direction[1]), ray.intensity*direction[1]])
|
result.append([np.arccos(direction[1]), ray.intensity*direction[1]])
|
||||||
|
print("ray trace finish, total ray count:",ray_count)
|
||||||
|
|
||||||
init()
|
init()
|
||||||
# stack.append(Ray([0.8, 2*r], [0,-1], 1))
|
# stack.append(Ray([0.8, 2*r], [0,-1], 1))
|
||||||
|
Loading…
Reference in New Issue
Block a user