| class Scene { |
| PointCloud point_cloud; |
| ArrayList<Triangle> mesh; |
| BVH bvh; |
| MotionField motion_field; |
| Camera last_cam; |
| Camera current_cam; |
| int frame_count; |
| |
| Scene(Camera camera, PointCloud point_cloud, MotionField motion_field) { |
| this.point_cloud = point_cloud; |
| this.motion_field = motion_field; |
| mesh = new ArrayList<Triangle>(); |
| for (int v = 0; v < height - 1; v++) |
| for (int u = 0; u < width - 1; u++) { |
| PVector p1 = point_cloud.getPosition(v * width + u); |
| PVector p2 = point_cloud.getPosition(v * width + u + 1); |
| PVector p3 = point_cloud.getPosition((v + 1) * width + u + 1); |
| PVector p4 = point_cloud.getPosition((v + 1) * width + u); |
| color c1 = point_cloud.getColor(v * width + u); |
| color c2 = point_cloud.getColor(v * width + u + 1); |
| color c3 = point_cloud.getColor((v + 1) * width + u + 1); |
| color c4 = point_cloud.getColor((v + 1) * width + u); |
| mesh.add(new Triangle(p1, p2, p3, c1, c2, c3)); |
| mesh.add(new Triangle(p3, p4, p1, c3, c4, c1)); |
| } |
| bvh = new BVH(mesh); |
| last_cam = camera.copy(); |
| current_cam = camera; |
| frame_count = 0; |
| } |
| |
| void run() { |
| last_cam = current_cam.copy(); |
| current_cam.run(); |
| motion_field.update(last_cam, current_cam, point_cloud, bvh); |
| frame_count += 1; |
| } |
| |
| void render(boolean show_motion_field) { |
| // build mesh |
| current_cam.open(); |
| noStroke(); |
| for (int i = 0; i < mesh.size(); i++) { |
| Triangle t = mesh.get(i); |
| t.render(); |
| } |
| if (show_motion_field) { |
| current_cam.close(); |
| motion_field.render(); |
| } |
| } |
| |
| void save(String path) { saveFrame(path + "_" + str(frame_count) + ".png"); } |
| |
| void saveMotionField(String path) { |
| motion_field.save(path + "_" + str(frame_count) + ".txt"); |
| } |
| } |