|
41 | 41 | if ENABLE_RAYTRACER: |
42 | 42 | renderer = gs.renderers.RayTracer( |
43 | 43 | env_surface=gs.surfaces.Emission( |
44 | | - emissive_texture=gs.textures.ColorTexture(color=(0.2, 0.3, 0.5)), |
| 44 | + emissive_texture=gs.textures.ColorTexture( |
| 45 | + color=(0.2, 0.3, 0.5), |
| 46 | + ), |
45 | 47 | ), |
46 | 48 | env_radius=20.0, |
47 | 49 | ) |
|
50 | 52 | renderer = gs.renderers.Rasterizer() |
51 | 53 |
|
52 | 54 | scene = gs.Scene( |
53 | | - rigid_options=gs.options.RigidOptions( |
54 | | - enable_collision=True, |
55 | | - gravity=(0, 0, -9.8), |
56 | | - ), |
57 | 55 | renderer=renderer, |
58 | 56 | show_viewer=False, |
59 | 57 | ) |
60 | 58 |
|
61 | 59 | ########################## entities ########################## |
62 | 60 | plane = scene.add_entity( |
63 | 61 | morph=gs.morphs.Plane(), |
64 | | - surface=gs.surfaces.Rough(color=(0.4, 0.4, 0.4)), |
| 62 | + surface=gs.surfaces.Rough( |
| 63 | + color=(0.4, 0.4, 0.4), |
| 64 | + ), |
65 | 65 | ) |
66 | 66 |
|
67 | 67 | sphere = scene.add_entity( |
68 | | - morph=gs.morphs.Sphere(pos=(0.0, 0.0, 2.0), radius=0.5), |
69 | | - surface=gs.surfaces.Smooth(color=(1.0, 0.5, 0.5)), |
| 68 | + morph=gs.morphs.Sphere( |
| 69 | + radius=0.5, |
| 70 | + pos=(0.0, 0.0, 2.0), |
| 71 | + ), |
| 72 | + surface=gs.surfaces.Smooth( |
| 73 | + color=(1.0, 0.5, 0.5), |
| 74 | + ), |
70 | 75 | ) |
71 | 76 |
|
72 | 77 | box = scene.add_entity( |
73 | | - morph=gs.morphs.Box(pos=(1.0, 1.0, 1.0), size=(0.3, 0.3, 0.3)), |
74 | | - surface=gs.surfaces.Rough(color=(0.5, 1.0, 0.5)), |
| 78 | + morph=gs.morphs.Box( |
| 79 | + size=(0.3, 0.3, 0.3), |
| 80 | + pos=(1.0, 1.0, 1.0), |
| 81 | + ), |
| 82 | + surface=gs.surfaces.Rough( |
| 83 | + color=(0.5, 1.0, 0.5), |
| 84 | + ), |
75 | 85 | ) |
76 | 86 |
|
77 | 87 | ########################## Camera Configurations ########################## |
|
129 | 139 |
|
130 | 140 | # Create camera configurations for all backends |
131 | 141 | backends = [ |
132 | | - ("raster", RasterizerCameraOptions, True), # Always enabled |
133 | | - ("raytrace", RaytracerCameraOptions, ENABLE_RAYTRACER), |
134 | | - ("batch", BatchRendererCameraOptions, ENABLE_MADRONA), |
| 142 | + ("rasterizer", RasterizerCameraOptions, True), # Always enabled |
| 143 | + ("raytracer", RaytracerCameraOptions, ENABLE_RAYTRACER), |
| 144 | + ("batch_render", BatchRendererCameraOptions, ENABLE_MADRONA), |
135 | 145 | ] |
136 | 146 |
|
137 | 147 | backend_configs = {} |
|
168 | 178 | options_kwargs["offset_T"] = attachment["offset_T"] |
169 | 179 |
|
170 | 180 | # Add backend-specific parameters |
171 | | - if backend_name == "raster": |
| 181 | + if options_class is RasterizerCameraOptions: |
172 | 182 | options_kwargs.update({"near": CAMERA_COMMON_KWARGS["near"], "far": CAMERA_COMMON_KWARGS["far"]}) |
173 | | - elif backend_name == "raytrace": |
| 183 | + elif options_class is RaytracerCameraOptions: |
174 | 184 | options_kwargs.update( |
175 | 185 | { |
176 | 186 | "model": "pinhole", |
|
187 | 197 | "env_radius": 20.0, |
188 | 198 | } |
189 | 199 | ) |
190 | | - elif backend_name == "batch": |
| 200 | + elif options_class is BatchRendererCameraOptions: |
191 | 201 | options_kwargs.update({"use_rasterizer": True}) |
192 | 202 | if camera_config["lights"]: |
193 | 203 | adjusted_lights = [{**light, "directional": False} for light in camera_config["lights"]] |
194 | 204 | options_kwargs["lights"] = adjusted_lights |
195 | 205 |
|
196 | 206 | # Adjust lights for raytracer (different intensity/color) |
197 | | - if backend_name == "raytrace" and camera_config["lights"]: |
| 207 | + if options_class is RaytracerCameraOptions and camera_config["lights"]: |
198 | 208 | adjusted_lights = [ |
199 | 209 | {**light, "color": (10.0, 10.0, 10.0), "intensity": 1.0} for light in camera_config["lights"] |
200 | 210 | ] |
|
0 commit comments