shithub: libvpx

ref: 9f33d7530ab42e21c2ec2151a5db94fc623823f1
dir: /tools/3D-Reconstruction/sketch_3D_reconstruction/Scene.pde/

View raw version
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");
  }
}