main.cpp

#include "src/Image.h"
#include "src/Scene.h"
#include "src/RayTracer.h"
#include "src/PointLight.h"
#include "src/AreaLight.h"

void TD1(int width, int height, std::string& filename){
    // Blue color
    Vec3<float> blue(0, 0, 1);
    // White color
    Vec3<float> white(1, 1, 1);

    Image img(width, height);
    img.fillBackgroundY(white, blue);
    img.savePPM(filename);
}

void TD2(int width, int height, std::string& filename) {
    // Blue color
    Vec3<float> blue(0, 0, 1);
    // White color
    Vec3<float> white(1, 1, 1);

    // Define our image
    Image img(width, height);
    img.fillBackgroundY(white, blue);
    // Define our Scene
    Scene scene;

    // Define a triangle model
    std::vector<Vec3<float>> triangle_vertices = {Vec3<float>(0, 0, -2),
                                                  Vec3<float>(-1, 0, -2),
                                                  Vec3<float>(0, -1, -2)};
    std::vector<Vec3<int>> triangle_indices = {Vec3<int>(0, 1, 2)};
    Model triangle(triangle_vertices, triangle_indices);
    // Add triangle to scene
    scene.add(triangle);

    // Define a sphere model
    Model sphere("../models/geometry/sphere2.off");
    // Add triangle to scene
    scene.add(sphere);

    RayTracer rayTracer;
    rayTracer.render(img, scene);
    img.savePPM(filename);
}

void TD3(int width, int height, std::string& filename) {
    // Blue color
    Vec3<float> blue(0, 0, 1);
    // White color
    Vec3<float> white(1, 1, 1);

    // Define our image
    Image img(width, height);
    img.fillBackgroundY(white, blue);
    // Define our Scene
    Scene scene;

    // Define a plane model
    std::vector<Vec3<float>> plane_vertices = {Vec3<float>(1.5, -0.5, -1.5),
                                                  Vec3<float>(-1.5, -0.5, -1.5),
                                                  Vec3<float>(-1.5, -0.5, -5.0),
                                                  Vec3<float>(1.5, -0.5, -5.0)};
    std::vector<Vec3<int>> plane_indices = {Vec3<int>(0, 2, 1), Vec3<int>(0, 3, 2)};
    Model plane(plane_vertices, plane_indices);
    plane.setMaterial(Material(Vec3<float>(1.0, 0.0, 0.2), 1.f));
    // Add plane to scene
    scene.add(plane);

    // Define a face model
    Model face("../models/face_lowres.off");
    face.setMaterial(Material(Vec3<float>(0.8, 0.4, 0), 1.f));
    // Add face to scene
    scene.add(face);

    // Define a light
    // Vec3<float> lightPos = Vec3<float>(1.f, 1.f, -2.f);
    Vec3<float> lightPos = Vec3<float>(2.f, 2.f, -2.25f);
    Vec3<float> lightColor = Vec3<float>(1.f, 1.f, 1.f);
    float lightIntensity = 1.f;
    // PointLight pointLight(lightPos, lightColor, lightIntensity);
    // scene.add(pointLight);

    Vec3<float> lightDir = Vec3<float>(-1.f, -1.f, 0.f);
    AreaLight areaLight(lightPos, lightColor, lightIntensity, lightDir, 0.4f);
    scene.add(areaLight);

    RayTracer rayTracer;
    rayTracer.enableShadow();
    rayTracer.enableAntiAliasing(4);
    rayTracer.printInfos();
    rayTracer.render(img, scene);
    img.savePPM(filename);
}

void TD4(int width, int height, std::string& filename) {
    // Blue color
    Vec3<float> blue(0.15, 0.15, 0.15);
    // White color
    Vec3<float> white(0, 0, 0);

    // Define our image
    Image img(width, height);
    img.printInfos();
    img.fillBackgroundY(white, blue);
    // Define our Scene
    Scene scene;

    // Define Worley noise
    Worley worley(60, 2.f, 2.f, 2.f);

    // Define a plane model
    std::vector<Vec3<float>> plane_vertices = {Vec3<float>(1.5, -0.5, -1.5),
                                                  Vec3<float>(-1.5, -0.5, -1.5),
                                                  Vec3<float>(-1.5, -0.5, -5.0),
                                                  Vec3<float>(1.5, -0.5, -5.0)};
    std::vector<Vec3<int>> plane_indices = {Vec3<int>(0, 2, 1), Vec3<int>(0, 3, 2)};
    Model plane(plane_vertices, plane_indices);
    Material planeMaterial(Vec3<float>(0.6, 0.0, 0.0), 1.f, 1.f);
    // planeMaterial.useWorleyNoise(&worley);
    plane.setMaterial(planeMaterial);
    // Add plane to scene
    scene.add(plane);

    // Define a face model
    Model face("../models/face_lowres.off");
    Material faceMaterial(Vec3<float>(0.8, 0.6, 0.3), 0.8f, 0.40f);
    // faceMaterial.useWorleyNoise(&worley);
    face.setMaterial(faceMaterial);
    // Add face to scene
    scene.add(face);

    // Define a light
    Vec3<float> lightPos = Vec3<float>(2.f, 2.f, -2.25f);
    Vec3<float> lightColor = Vec3<float>(1.f, 1.f, 1.f);
    float lightIntensity = 1.f;
    // PointLight pointLight(lightPos, lightColor, lightIntensity);
    // scene.add(pointLight);

    Vec3<float> lightDir = Vec3<float>(-1.f, -1.f, 0.f);
    AreaLight areaLight(lightPos, lightColor, lightIntensity, lightDir, 0.4f);
    scene.add(areaLight);

    RayTracer rayTracer;
    rayTracer.enableShadow();
    rayTracer.enableAntiAliasing(4);
    rayTracer.printInfos();
    rayTracer.render(img, scene);
    img.savePPM(filename);
}

void TD5(int width, int height, std::string& filename) {
    // Blue color
    Vec3<float> blue(0.15, 0.15, 0.15);
    // White color
    Vec3<float> white(0, 0, 0);

    // Define our image
    Image img(width, height);
    img.printInfos();
    img.fillBackgroundY(white, blue);
    // Define our Scene
    Scene scene;

    // Define a plane model
    std::vector<Vec3<float>> plane_vertices = {Vec3<float>(1.5, -0.5, -1.5),
                                                  Vec3<float>(-1.5, -0.5, -1.5),
                                                  Vec3<float>(-1.5, -0.5, -5.0),
                                                  Vec3<float>(1.5, -0.5, -5.0)};
    std::vector<Vec3<int>> plane_indices = {Vec3<int>(0, 2, 1), Vec3<int>(0, 3, 2)};
    Model plane(plane_vertices, plane_indices);
    Material planeMaterial(Vec3<float>(0.6, 0.0, 0.0), 1.f, 1.f);
    plane.setMaterial(planeMaterial);
    // Add plane to scene
    scene.add(plane);

    // Define a face model
    Model face("../models/face.off");
    Material faceMaterial(Vec3<float>(0.8, 0.6, 0.3), 0.8f, 0.40f);
    face.setMaterial(faceMaterial);
    // Add face to scene
    scene.add(face);

    // Define a light
    Vec3<float> lightPos = Vec3<float>(2.f, 2.f, -2.25f);
    Vec3<float> lightColor = Vec3<float>(1.f, 1.f, 1.f);
    float lightIntensity = 1.f;
    // PointLight pointLight(lightPos, lightColor, lightIntensity);
    // scene.add(pointLight);

    Vec3<float> lightDir = Vec3<float>(-1.f, -1.f, 0.f);
    AreaLight areaLight(lightPos, lightColor, lightIntensity, lightDir, 0.4f);
    scene.add(areaLight);

    RayTracer rayTracer;
    rayTracer.enableShadow();
    rayTracer.enableAntiAliasing(4);
    rayTracer.enableBVH();
    rayTracer.enablePathTracing(3, 16);
    rayTracer.printInfos();
    rayTracer.render(img, scene);
    img.savePPM(filename);
}

void TD6(int width, int height, std::string& filename) {
    // Blue color
    Vec3<float> blue(0.15, 0.15, 0.15);
    // White color
    Vec3<float> white(0, 0, 0);

    // Define our image
    Image img(width, height);
    img.printInfos();
    img.fillBackgroundY(white, blue);
    // Define our Scene
    Scene scene;

    // Define a plane model
    std::vector<Vec3<float>> plane_vertices = {Vec3<float>(1.5, -0.5, -0.5),
                                                  Vec3<float>(-1.5, -0.5, -0.5),
                                                  Vec3<float>(-1.5, -0.5, -4.0),
                                                  Vec3<float>(1.5, -0.5, -4.0),
                                                  
                                                  Vec3<float>(-1.5, 1.5, -0.5),
                                                  Vec3<float>(-1.5, 1.5, -4.0),};
    std::vector<Vec3<int>> plane_indices = {Vec3<int>(0, 2, 1), Vec3<int>(0, 3, 2)};
                                            // Vec3<int>(2, 4, 1), Vec3<int>(2, 5, 4)};
    Model plane(plane_vertices, plane_indices);
    Material planeMaterial(Vec3<float>(0.6, 0.0, 0.0), 1.f, 0.4f);
    plane.setMaterial(planeMaterial);
    // Add plane to scene
    scene.add(plane);

    // Define a face1 model
    Model face1("../models/face.off");
    // Material face1Material(Vec3<float>(0.8, 0.6, 0.3), 0.8f, 0.40f);
    Material face1Material(Vec3<float>(0.15, 0.15, 0.15), 0.8f, 0.40f);
    face1.translate(Vec3<float>(0.0f, 0.f, -2.f));
    face1.setMaterial(face1Material);
    // Add face to scene
    scene.add(face1);

    // Define a face2 model
    Model face2("../models/face.off");
    Material face2Material(Vec3<float>(0.5, 0.9, 0.5), 0.8f, 0.40f);
    face2.translate(Vec3<float>(0.8f, 0.f, -1.5f));
    face2.setMaterial(face2Material);
    // Add face to scene
    // scene.add(face2);

    // Define a face3 model
    Model face3("../models/face.off");
    Material face3Material(Vec3<float>(0.3, 0.6, 0.8), 0.8f, 0.40f);
    face3.translate(Vec3<float>(-0.8f, 0.f, -2.5f));
    face3.setMaterial(face3Material);
    // Add face to scene
    // scene.add(face3);

    // Define an area light
    Vec3<float> lightPos = Vec3<float>(2.f, 2.f, -1.25f);
    Vec3<float> lightColor = Vec3<float>(1.f, 1.f, 1.f);
    float lightIntensity = 1.0f;
    Vec3<float> lightDir = Vec3<float>(-1.f, -1.f, 0.f);
    AreaLight areaLight(lightPos, lightColor, lightIntensity, lightDir, 0.4f);
    scene.add(areaLight);

    // Define a point light
    lightPos = Vec3<float>(0.f, 2.f, -1.25f);
    lightColor = Vec3<float>(1.f, 1.f, 1.f);
    lightIntensity = 0.2f;
    PointLight pointLight(lightPos, lightColor, lightIntensity);
    // scene.add(pointLight);

    RayTracer rayTracer;
    rayTracer.enableShadow();
    rayTracer.enableAntiAliasing(4);
    rayTracer.enableBVH();
    //rayTracer.enagleCosineWeighted();
    //rayTracer.enableLearningLT();
    rayTracer.enablePathTracing(1, 9);
    //rayTracer.enablePathTracing(3, 25);
    rayTracer.printInfos();
    rayTracer.render(img, scene);
    img.savePPM(filename);
}

void parseArguments(int argc, char *argv[], int& width, int& height, std::string& filename) {
    // Parse arguments
    for(int i = 1; i < argc; i++) {
        std::string currArg(argv[i]);
        if (currArg.compare("-width") == 0) {
            width = strtol(argv[i+1], NULL, 10);
            i++;
        } else if (currArg.compare("-height") == 0) {
            height = strtol(argv[i+1], NULL, 10);
            i++;
        } else if (currArg.compare("-output") == 0) {
            filename = std::string(argv[i+1]);
            i++;
        }
    }
}

int main(int argc, char *argv[]) {
    //int width = 900;
    //int height = 600;
    int width = 450;
    int height = 300;
    //int width = 300;
    //int height = 200;
    std::string filename = "default.ppm";
    parseArguments(argc, argv, width, height, filename);

    // TD1(width, height, filename);
    // TD2(width, height, filename);
    // TD3(width, height, filename);
    // TD4(width, height, filename);
    // TD5(width, height, filename);
    // TD6(width, height, filename);

    return 0;
}