main.cpp

#include <vector>
#include <iostream>
#include "write_ppm.h"
#include <Eigen/Core>
#include "Object.h"
#include "ObjectList.h"
#include "Sphere.h"
#include <limits>
#include <Camera.h>
#include "raycolor.h"
#include "Material.h"
#include "random_scene.h"
#include "Texture.h"

#define STB_IMAGE_IMPLEMENTATION

int main(int argc, char * argv[])
{

  int width =  640; // 640
  int height = 360; // 360
  int numSample = 500;
  int recDepth = 50;

  std::vector<unsigned char> rgb_image(3*width*height);

  Eigen::Vector3d lookfrom = Eigen::Vector3d(-13, 2, 3);
  Eigen::Vector3d lookat = Eigen::Vector3d(0, 0, 0);
  Eigen::Vector3d vup = Eigen::Vector3d(0, 1, 0);
  double vfov = 30;
  double aspect = (double) width / height;
  double aperture = 0.1;
  double focus_dist = (lookfrom - lookat).norm();

  Camera cam(lookfrom, lookat, vup, vfov, aspect, aperture, focus_dist);

  Object * world = random_scene();

//  int nx, ny, nn;
//  unsigned char *tex_data = stbi_load("picture.jpg", &nx, &ny, &nn, 0);
//  Material * mat = new lambertian(new imageTexture(tex_data, nx, ny));

  for(int j=0;j<height;j++)
  {
    if (j % 10 == 0) {
      std::cout << "Rendering height " << (double) j * 100 / height << std::endl;
    }
    for(int i=0;i<width;i++)
    {
      Eigen::Vector3d rgb = Eigen::Vector3d(0, 0, 0);
      for (int s=0; s<numSample; s++) {
        double u = double(i) / double(width);
        double v = double(j) / double(height);

        Ray r = cam.get_ray(u, v);

        Eigen::Vector3d p = r.point_at_t(2.0);
        rgb += raycolor(r, world, recDepth);
      }

      rgb /= numSample;

      rgb = Eigen::Vector3d(sqrt(rgb[0]),sqrt(rgb[1]),sqrt(rgb[2]));


      auto clamp = [](double s){ return std::max(std::min(s,1.0),0.0);};
      rgb_image[0+3*(i+width*j)] = 255.0*clamp(rgb(0));
      rgb_image[1+3*(i+width*j)] = 255.0*clamp(rgb(1));
      rgb_image[2+3*(i+width*j)] = 255.0*clamp(rgb(2));
    }
  }

  write_ppm("rgb.ppm",rgb_image,width,height,3);
}