-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_fig3_glossy_reflection.py
176 lines (139 loc) · 6.06 KB
/
test_fig3_glossy_reflection.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
import numpy as np
import os
import mitsuba
import enoki as ek
mts_variant = 'rgb'
mitsuba.set_variant('gpu_autodiff_' + mts_variant)
from mitsuba.core import Transform4f, Bitmap
from mitsuba.core.xml import load_string
from mitsuba.python.util import traverse
from utils import test_finite_difference
test_name = "test_fig3_glossy_reflection"
def make_scene(integrator, spp, translateX, rotateZ, scale):
return load_string("""
<?xml version="1.0"?>
<scene version="2.0.0">
{integrator}
<sensor type="perspective">
<string name="fov_axis" value="smaller"/>
<float name="near_clip" value="0.1"/>
<float name="far_clip" value="2800"/>
<float name="focus_distance" value="10"/>
<transform name="to_world">
<lookat origin="0, 0, 10" target="0, 0, 0" up="0, 1, 0"/>
</transform>
<float name="fov" value="25.00"/>
<sampler type="independent">
<integer name="sample_count" value="{spp}"/>
</sampler>
<film type="hdrfilm">
<integer name="width" value="128"/>
<integer name="height" value="128"/>
<rfilter type="box"/>
</film>
</sensor>
<shape type="obj">
<transform name="to_world">
<rotate x="1" angle="180"/>
<translate x="0" y="0" z="150"/>
<rotate x="1.0" y="0.0" z="0.0" angle="22"/>
</transform>
<string name="filename" value="data/meshes/xy_plane.obj"/>
<emitter type="smootharea" id="smooth_area_light">
<spectrum name="radiance" value="8000"/>
</emitter>
</shape>
<shape type="obj" id="object">
<string name="filename" value="data/meshes/smooth_empty_cube.obj"/>
<transform name="to_world">
<scale value="3.0"/>
<rotate x="0.0" y="0.0" z="1.0" angle="40"/>
<translate x="0.0" y="2.2" z="-1.0"/>
<rotate x="0.0" y="0.0" z="1.0" angle="{rotateZ}"/>
<translate x="{translateX}" y="0.0" z="0.0"/>
<scale value="{scale}"/>
</transform>
<bsdf type="diffuse">
<spectrum name="reflectance" value="0.5"/>
</bsdf>
</shape>
<shape type="obj">
<string name="filename" value="data/meshes/xy_plane_rough.obj"/>
<bsdf type="roughconductor">
<float name="alpha" value="0.01"/>
</bsdf>
<transform name="to_world">
<scale x="6.0" y="6.0" z="6.0"/>
<translate x="0.0" y="0.0" z="0.0"/>
<rotate x="1.0" y="0.0" z="0.0" angle="-70"/>
</transform>
</shape>
</scene>
""".format(integrator=integrator, spp=spp,
translateX=translateX, rotateZ=rotateZ, scale=scale))
# Rotation (Z axis)
def make_scene_rotation(integrator, spp, param):
return make_scene(integrator, spp, 0, param, 1)
def get_diff_param_rotation(scene):
# Create a differentiable hyperparameter
diff_param = mitsuba.core.Float(0.0);
ek.set_requires_gradient(diff_param);
# Update vertices so that they depend on diff_param
properties = traverse(scene)
t = mitsuba.core.Transform4f.rotate(mitsuba.core.Vector3f(0.0,0.0,1.0), diff_param)
vertex_positions = properties['object.vertex_positions']
vertex_positions_t = t.transform_point(vertex_positions)
properties['object.vertex_positions'] = vertex_positions_t
# Update the scene
properties.update()
return diff_param
# Translation (X axis)
def make_scene_translation(integrator, spp, param):
return make_scene(integrator, spp, param, 0, 1)
def get_diff_param_translation(scene):
# Create a differentiable hyperparameter
diff_param = mitsuba.core.Float(0.0);
ek.set_requires_gradient(diff_param);
# Update vertices so that they depend on diff_param
properties = traverse(scene)
t = mitsuba.core.Transform4f.translate(mitsuba.core.Vector3f(1.0,0.0,0.0) * diff_param)
vertex_positions = properties['object.vertex_positions']
vertex_positions_t = t.transform_point(vertex_positions)
properties['object.vertex_positions'] = vertex_positions_t
# Update the scene
properties.update()
return diff_param
# Scale
def make_scene_scale(integrator, spp, param):
return make_scene(integrator, spp, 0, 0, param+1)
def get_diff_param_scale(scene):
# Create a differentiable hyperparameter
diff_param = mitsuba.core.Float(1.0);
ek.set_requires_gradient(diff_param);
# Update vertices so that they depend on diff_param
properties = traverse(scene)
vertex_positions = properties['object.vertex_positions']
vertex_positions_t = vertex_positions * diff_param
properties['object.vertex_positions'] = vertex_positions_t
# Update the scene
properties.update()
return diff_param
fd_spp = 256
fd_passes = 10
fd_integrator = """<integrator type="path">
<integer name="max_depth" value="3"/>
</integrator>"""
diff_spp = 16
diff_passes = 10
diff_integrator = """<integrator type="pathreparam">
<integer name="max_depth" value="3"/>
</integrator>"""
test_finite_difference(test_name + "/rotation", make_scene_rotation, get_diff_param_rotation,
diff_integrator, diff_spp, diff_passes,
fd_integrator, fd_spp, fd_passes, 0.1)
test_finite_difference(test_name + "/translation", make_scene_translation, get_diff_param_translation,
diff_integrator, diff_spp, diff_passes,
fd_integrator, fd_spp, fd_passes, 0.02)
test_finite_difference(test_name + "/scale", make_scene_scale, get_diff_param_scale,
diff_integrator, diff_spp, diff_passes,
fd_integrator, fd_spp, fd_passes, 0.005)