Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
zyxiaooo authored Nov 9, 2016
1 parent b08d1d2 commit dbeeb30
Showing 1 changed file with 107 additions and 108 deletions.
215 changes: 107 additions & 108 deletions ray_tracing/raytracer/raytracer5/main.cpp
View file
Original file line number Diff line number Diff line change
@@ -1,109 +1,108 @@

#include <iostream>
#include <vector>
#include <math.h>
#include <fstream>

#include "Vector3D.h"
#include "Colour.h"
#include "Sphere.h"

#define CANVAS_WIDTH 1000
#define CANVAS_HEIGHT 1000
#define CANVAS_DEPTH 700
#define MAX_RAY_DEPTH 1000
#define BIAS 1e-4
//view point(0,0,0), screen(1000,1000,700)
Colour trace(const Vector3D &rayOrig, const Vector3D &rayDir, const std::vector<Sphere> &spheres, int recursionDepth) {
Colour c(0, 0, 0);
const Sphere *intersectSphere = nullptr;
float closestIntersect = INFINITY;
for (int i = 0; i < spheres.size(); ++i) {//inspect every sphere and choose the closest intersection point.
float t0 = INFINITY;
if (spheres[i].intersect(rayOrig, rayDir, t0)) {
if (t0 < closestIntersect) {
intersectSphere = &spheres[i];
closestIntersect = t0;
}
}
}
if (intersectSphere) {//not NULL
//emmission(for light source)
c += intersectSphere->emissionColour;
//ambient reflection
c += intersectSphere->surfaceColour * 0.2;
Vector3D pIntersection = rayOrig + rayDir * closestIntersect;//P
Vector3D nIntersection = (pIntersection - intersectSphere->centre).norm();//N
//diffuse reflection
for (int i = 0; i < spheres.size(); ++i) {
if (spheres[i].emissionColour.red > 0 ||
spheres[i].emissionColour.green > 0 ||
spheres[i].emissionColour.blue > 0) {//light source
Vector3D lightRay = (spheres[i].centre - pIntersection).norm();
bool blocked = false;

for (int j = 0; j < spheres.size(); ++j) {
if (i != j) {
float t0 = 0;
if (spheres[j].intersect(pIntersection + nIntersection * BIAS, lightRay, t0)) {
blocked = true;
break;
}
}
}

if (!blocked) {
c += spheres[i].emissionColour * intersectSphere->surfaceColour * intersectSphere->kd *
std::max(float(0), nIntersection.dot(lightRay));
}
}
}

//specular reflection
if (intersectSphere->ks > 0 && recursionDepth < MAX_RAY_DEPTH) {//check max depth
Vector3D reflRay = (rayDir - nIntersection * 2 * rayDir.dot(nIntersection)).norm();
c += trace(pIntersection + nIntersection * BIAS, reflRay, spheres, recursionDepth + 1)
* intersectSphere->ks; //recursive(phong model)
}

}

return c;
}

void doit(const std::vector<Sphere> &spheres, char *filename) {
Colour *pixels = new Colour[CANVAS_WIDTH * CANVAS_HEIGHT], *pixel = pixels;

for (int i = 0; i < CANVAS_WIDTH; ++i) {
for (int j = 0; j < CANVAS_HEIGHT; ++j, ++pixel) {
float x = i - (CANVAS_WIDTH / 2 - 1);
float y = j - (CANVAS_HEIGHT / 2 - 1);
Vector3D primaryRay(x, y, CANVAS_DEPTH);
*pixel = trace(Vector3D(0, 0, 0), primaryRay.norm(), spheres, 0);
}
}

std::ofstream ofs(filename, std::ios::out | std::ios::binary);
ofs << "P6\n" << CANVAS_WIDTH << " " << CANVAS_HEIGHT << "\n255\n";
for (unsigned i = 0; i < CANVAS_WIDTH * CANVAS_HEIGHT; ++i) {
ofs << (unsigned char)(std::min(float(1), pixels[i].red) * 255) <<
(unsigned char)(std::min(float(1), pixels[i].green) * 255) <<
(unsigned char)(std::min(float(1), pixels[i].blue) * 255);
}
ofs.close();

delete[] pixels;
}

int main(int argc, char const *argv[]) {
#include <iostream>
#include <vector>
#include <math.h>
#include <fstream>

#include "Vector3D.h"
#include "Colour.h"
#include "Sphere.h"

#define CANVAS_WIDTH 1000
#define CANVAS_HEIGHT 1000
#define CANVAS_DEPTH 700
#define MAX_RAY_DEPTH 1000
#define BIAS 1e-4
//view point(0,0,0), screen(1000,1000,700)
Colour trace(const Vector3D &rayOrig, const Vector3D &rayDir, const std::vector<Sphere> &spheres, int recursionDepth) {
Colour c(0, 0, 0);
const Sphere *intersectSphere = nullptr;
float closestIntersect = INFINITY;
for (int i = 0; i < spheres.size(); ++i) {//inspect every sphere and choose the closest intersection point.
float t0 = INFINITY;
if (spheres[i].intersect(rayOrig, rayDir, t0)) {
if (t0 < closestIntersect) {
intersectSphere = &spheres[i];
closestIntersect = t0;
}
}
}
if (intersectSphere) {//not NULL
//emmission(for light source)
c += intersectSphere->emissionColour;
//ambient reflection
c += intersectSphere->surfaceColour * 0.2;
Vector3D pIntersection = rayOrig + rayDir * closestIntersect;//P
Vector3D nIntersection = (pIntersection - intersectSphere->centre).norm();//N
//diffuse reflection
for (int i = 0; i < spheres.size(); ++i) {
if (spheres[i].emissionColour.red > 0 ||
spheres[i].emissionColour.green > 0 ||
spheres[i].emissionColour.blue > 0) {//light source
Vector3D lightRay = (spheres[i].centre - pIntersection).norm();
bool blocked = false;

for (int j = 0; j < spheres.size(); ++j) {
if (i != j) {
float t0 = 0;
if (spheres[j].intersect(pIntersection + nIntersection * BIAS, lightRay, t0)) {
blocked = true;
break;
}
}
}

if (!blocked) {
c += spheres[i].emissionColour * intersectSphere->surfaceColour * intersectSphere->kd *
std::max(float(0), nIntersection.dot(lightRay));
}
}
}

//specular reflection
if (intersectSphere->ks > 0 && recursionDepth < MAX_RAY_DEPTH) {//check max depth
Vector3D reflRay = (rayDir - nIntersection * 2 * rayDir.dot(nIntersection)).norm();
c += trace(pIntersection + nIntersection * BIAS, reflRay, spheres, recursionDepth + 1)
* intersectSphere->ks; //recursive(phong model)
}

}

return c;
}

void doit(const std::vector<Sphere> &spheres, char *filename) {
Colour *pixels = new Colour[CANVAS_WIDTH * CANVAS_HEIGHT], *pixel = pixels;

for (int i = 0; i < CANVAS_WIDTH; ++i) {
for (int j = 0; j < CANVAS_HEIGHT; ++j, ++pixel) {
float x = i - (CANVAS_WIDTH / 2 - 1);
float y = j - (CANVAS_HEIGHT / 2 - 1);
Vector3D primaryRay(x, y, CANVAS_DEPTH);
*pixel = trace(Vector3D(0, 0, 0), primaryRay.norm(), spheres, 0);
}
}

std::ofstream ofs(filename, std::ios::out | std::ios::binary);
ofs << "P6\n" << CANVAS_WIDTH << " " << CANVAS_HEIGHT << "\n255\n";
for (unsigned i = 0; i < CANVAS_WIDTH * CANVAS_HEIGHT; ++i) {
ofs << (unsigned char)(std::min(float(1), pixels[i].red) * 255) <<
(unsigned char)(std::min(float(1), pixels[i].green) * 255) <<
(unsigned char)(std::min(float(1), pixels[i].blue) * 255);
}
ofs.close();

delete[] pixels;
}

int main(int argc, char const *argv[]) {
std::vector<Sphere> spheres;
//scene1/////////////////////////////////////////////////////////
for (int i = -100; i <= 200; i += 200) {
for (int j = -100; j <= 200; j += 400) {
spheres.push_back(Sphere(Vector3D(i, j, 600), 90.0, Colour(1.0, ((float)i + 100)/350, ((float)j + 100)/350), 0.6, 0.4));
//spheres.push_back(Sphere(Vector3D(i, j, 750), 70.0, Colour(((float)i + 100)/350, 1.0, ((float)j + 100)/350), 0.6, 0.4));
spheres.push_back(Sphere(Vector3D(i, j+200, 900), 70.0, Colour(((float)i + 100)/350, ((float)j + 100)/400, 1.0), 0.6, 0.4));
}
//scene1/////////////////////////////////////////////////////////
for (int i = -100; i <= 200; i += 200) {
for (int j = -100; j <= 200; j += 400) {
spheres.push_back(Sphere(Vector3D(i, j, 600), 90.0, Colour(1.0, ((float)i + 100)/350, ((float)j + 100)/350), 0.6, 0.4));
//spheres.push_back(Sphere(Vector3D(i, j, 750), 70.0, Colour(((float)i + 100)/350, 1.0, ((float)j + 100)/350), 0.6, 0.4));
spheres.push_back(Sphere(Vector3D(i, j+200, 900), 70.0, Colour(((float)i + 100)/350, ((float)j + 100)/400, 1.0), 0.6, 0.4));
}
}//add sphere
spheres.push_back(Sphere(Vector3D(-200, -200, 0), 50, Colour(0, 0, 0), Colour(1, 1, 1)));//add light source
doit(spheres,"scene1.ppm");
Expand All @@ -119,6 +118,6 @@ int main(int argc, char const *argv[]) {
// add sphere object like: spheres.push_back(Sphere(Vector3D(12280, 0, 800), 12000, Colour(0.9, 0.9, 0.9), 0.8, 0.2));
//add light source like: spheres.push_back(Sphere(Vector3D(-300, -300, 200), 50, Colour(0, 0, 0), Colour(1, 1, 1)));
//doit(spheres,"scene3.ppm")
//spheres.clear();
return 0;
}
//spheres.clear();
return 0;
}

0 comments on commit dbeeb30

Please sign in to comment.