55 lines
1.7 KiB
C++
55 lines
1.7 KiB
C++
|
#include <iostream>
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
#include "camera/perspectivecamera.h"
|
||
|
|
|
||
|
|
#include "scene/fastscene.h"
|
||
|
|
|
||
|
|
#include "renderer/kdtreerenderer.h"
|
||
|
|
#include "renderer/simplerenderer.h"
|
||
|
|
|
||
|
|
#include "shader/cooktorranceshader.h"
|
||
|
|
|
||
|
|
#include "light/ambientlight.h"
|
||
|
|
#include "light/pointlight.h"
|
||
|
|
#include "light/spotlight.h"
|
||
|
|
|
||
|
|
#include "primitive/objmodel.h"
|
||
|
|
|
||
|
|
int main() {
|
||
|
|
FastScene scene;
|
||
|
|
scene.setEnvironmentMap(std::make_shared<Texture>("data/space.png"));
|
||
|
|
|
||
|
|
// Set up the camera
|
||
|
|
PerspectiveCamera camera;
|
||
|
|
camera.setFovAngle(90.0f);
|
||
|
|
camera.setPosition(Vector3d(0.0f, -2.5f, 10.0f));
|
||
|
|
camera.setForwardDirection(Vector3d(0.0f, 0.6f, -1.0f));
|
||
|
|
camera.setUpDirection(Vector3d(0.0f, 1.0f, 0.0f));
|
||
|
|
|
||
|
|
// Add shaders
|
||
|
|
auto redCook = std::make_shared<CookTorranceShader>(Color(1.0f, 0.0f, 0.0f), Color(1.0f, 1.0f, 1.0f), 1.0f, 0.3f);
|
||
|
|
auto goldCook = std::make_shared<CookTorranceShader>(Color(0.83f, 0.69f, 0.22f), Color(1.0f, 1.0f, 0.0f), 1.2f, 0.2f);
|
||
|
|
|
||
|
|
scene.addObj("data/teapot_stadium.obj", Vector3d(1.0f, 1.0f, 1.0f) * 40.0f, Vector3d(-0.25f, -0.5f, -2.5f), goldCook);
|
||
|
|
scene.addObj("data/stadium.obj", Vector3d(1.0f, 1.0f, 1.0f) * 70.0f, Vector3d(-0.5f, -1.0f, -3.0f), redCook);
|
||
|
|
|
||
|
|
// Add lights
|
||
|
|
scene.add(std::make_shared<AmbientLight>(0.1f));
|
||
|
|
scene.add(std::make_shared<PointLight>(Vector3d(-0.25f, 20.0f, -3.0f), 60.0f));
|
||
|
|
scene.add(
|
||
|
|
std::make_shared<SpotLight>(Vector3d(-0.25f, 7.0f, -4.0f), Vector3d(0.0f, -1.0f, 0.0f), 10.0f, 30.0f, 25.0f));
|
||
|
|
|
||
|
|
// build the tree
|
||
|
|
scene.buildTree();
|
||
|
|
|
||
|
|
// Render the scene
|
||
|
|
SimpleRenderer renderer;
|
||
|
|
renderer.renderImage(scene, camera, 1920, 1080).save("result.png");
|
||
|
|
|
||
|
|
KDTreeRenderer kd_renderer;
|
||
|
|
kd_renderer.renderKDTree(scene, camera, 1920, 1080).save("result_kd.png");
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|