From c298b3e6d02345b4ed897ec23923a79656da9b92 Mon Sep 17 00:00:00 2001
From: LittleTheFu <lttlstpd@live.com>
Date: Sun, 29 Sep 2024 15:30:14 +0800
Subject: [PATCH] save progress

---
 CMakeLists.txt    |   2 +
 camera/camera.cpp |   2 +-
 geometry/tri.cpp  | 134 ++++++++++++++++++++++++++++++++++++++++++++++
 geometry/tri.h    |  46 ++++++++++++++++
 light/light.cpp   |   2 +-
 main/main.cpp     |  11 ++--
 scene/scene.cpp   |  35 +++++++-----
 7 files changed, 212 insertions(+), 20 deletions(-)
 create mode 100644 geometry/tri.cpp
 create mode 100644 geometry/tri.h

diff --git a/CMakeLists.txt b/CMakeLists.txt
index d5461e29..e6c2f2c4 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -25,6 +25,7 @@ SET (RAY_TRACER_SOURCE
     geometry/mesh.cpp
     geometry/plane.cpp
     geometry/triangle.cpp
+    geometry/tri.cpp
     light/light.cpp
     # main/main.cpp
     material/dummyMaterial.cpp
@@ -75,6 +76,7 @@ SET (RAY_TRACER_HEADER
     geometry/geometry.h
     geometry/mesh.h
     geometry/plane.h
+    geometry/tri.h
     geometry/triangle.h
     light/light.h
     material/dummyMaterial.h
diff --git a/camera/camera.cpp b/camera/camera.cpp
index 20944262..e7b087da 100644
--- a/camera/camera.cpp
+++ b/camera/camera.cpp
@@ -6,7 +6,7 @@
 
 Camera::Camera()
 {
-    m_factor = 5;
+    m_factor = 3;
 
     m_Width = Common::default_screen_width * m_factor;
     m_Height = Common::default_screen_height * m_factor;
diff --git a/geometry/tri.cpp b/geometry/tri.cpp
new file mode 100644
index 00000000..c7061523
--- /dev/null
+++ b/geometry/tri.cpp
@@ -0,0 +1,134 @@
+#include "tri.h"
+#include "common.h"
+
+Tri::Tri()
+{
+    m_ab = m_b.pos - m_a.pos;
+    m_bc = m_c.pos - m_b.pos;
+    m_ca = m_a.pos - m_c.pos;
+}
+
+Tri::Tri(const TriVertex &a, const TriVertex &b, const TriVertex &c, Material *pMtrl)
+{
+    m_a = a;
+    m_b = b;
+    m_c = c;
+
+    m_ab = m_b.pos - m_a.pos;
+    m_bc = m_c.pos - m_b.pos;
+    m_ca = m_a.pos - m_c.pos;
+
+    this->m_pMtrl = pMtrl;
+
+    Vector3 pos(0,0,320);
+    init(Vector3::ZERO, pos);
+}
+
+bool Tri::hit(const Ray &ray, HitRecord &record, Light *pLight) const
+{
+    record.t = Common::FLOAT_MAX;
+
+    const Ray newRay = ray.genNewRay(m_transform);
+
+    bool reverse = false;
+    if (newRay.dir.isSameDir(getLocalNormal()))
+    {
+        reverse = true;
+    }
+
+    const float n = (-newRay.origin) * getLocalNormal(reverse);
+    const float d = newRay.dir * getLocalNormal(reverse);
+
+    record.t = n / d;
+    if (record.t < Common::FLOAT_SAMLL_NUMBER)
+    {
+        // std::cout << record.t << "," << n << "," << d << std::endl;
+        // std::cout<<"false"<<std::endl;
+        return false;
+    }
+
+    Vector3 localPoint = newRay.origin + record.t * newRay.dir;
+    if (!isAllFacePositive(localPoint))
+    {
+        // std::cout<<"false"<<std::endl;
+        return false;
+    }
+
+    record.mtrl = *m_pMtrl;
+    record.transform = m_transform;
+
+    record.point = m_transform.transformPoint(localPoint);
+    record.normal = m_transform.transformNormal(getLocalNormal(reverse));
+
+    // float w_a, w_b, w_c;
+    // getWeight(ap_ab, ap_bc, ap_ca, w_a, w_b, w_c);
+
+    // record.u = m_a.u * w_a + m_b.u * w_b + m_c.u * w_c;
+    // record.v = m_a.v * w_a + m_b.v * w_b + m_c.v * w_c;
+
+    if (m_pMtrl)
+    {
+        Vector3 r;
+        record.f = m_pMtrl->eval(record.u, record.v, -newRay.dir, r, record.reflectPdf);
+        record.dot = Common::clamp(std::abs(r * Common::LOCAL_NORMAL), Common::FLOAT_SAMLL_NUMBER, 1.0f);
+        record.reflect = m_transform.transformVector(r);
+        record.isMirror = m_pMtrl->isMirror();
+
+        if (record.isMirror)
+        {
+            record.dot = 1;
+        }
+    }
+
+    // std::cout<<"true"<<std::endl;
+    return true;
+}
+
+Vector3 Tri::getLocalNormal(bool reverse) const
+{
+    Vector3 n = m_ab.cross(-m_ca);
+
+    if(reverse)
+    {
+        n *= -1;
+    }
+
+    return n;
+}
+
+bool Tri::isAllFacePositive(const Vector3 &p) const
+{
+    Vector3 ap = p - m_a.pos;
+    Vector3 bp = p - m_b.pos;
+    Vector3 cp = p - m_c.pos;
+
+    Vector3 bc_bp = m_bc.cross(bp).dir();
+    Vector3 ca_cp = m_ca.cross(cp).dir();
+    Vector3 ab_ap = m_ab.cross(ap).dir();
+
+    if(bc_bp != ca_cp) return false;
+    if(ca_cp != ab_ap) return false;
+    if(ab_ap != bc_bp) return false;
+    
+    return true;
+}
+
+Vector3 Tri::dpdu(const Vector3 &point) const
+{
+    return Vector3();
+}
+
+Vector3 Tri::dpdv(const Vector3 &point) const
+{
+    return Vector3();
+}
+
+float Tri::u(const Vector3 &point) const
+{
+    return 0.0f;
+}
+
+float Tri::v(const Vector3 &point) const
+{
+    return 0.0f;
+}
\ No newline at end of file
diff --git a/geometry/tri.h b/geometry/tri.h
new file mode 100644
index 00000000..b901c971
--- /dev/null
+++ b/geometry/tri.h
@@ -0,0 +1,46 @@
+#ifndef _TRI_H_
+#define _TRI_H_
+
+#include "geometry.h"
+#include "vector.h"
+
+class TriVertex
+{
+public:
+    TriVertex() {};
+    TriVertex(float x, float y, float z) : pos(x, y, z) {};
+
+    Vector3 pos;
+};
+
+class Tri : public Geometry
+{
+public:
+    Tri();
+    Tri(const TriVertex &a, const TriVertex &b, const TriVertex &c, Material *pMtrl);
+
+    virtual bool hit(const Ray &ray, HitRecord &record, Light *pLight) const override;
+
+    Vector3 getLocalNormal(bool reverse = false) const;
+
+private:
+    bool isAllFacePositive(const Vector3 &p) const;
+
+private:
+    virtual Vector3 dpdu(const Vector3 &point) const override;
+    virtual Vector3 dpdv(const Vector3 &point) const override;
+
+    virtual float u(const Vector3 &point) const override;
+    virtual float v(const Vector3 &point) const override;
+
+private:
+    TriVertex m_a;
+    TriVertex m_b;
+    TriVertex m_c;
+
+    Vector3 m_ab;
+    Vector3 m_bc;
+    Vector3 m_ca;
+};
+
+#endif
\ No newline at end of file
diff --git a/light/light.cpp b/light/light.cpp
index 4ee3d6ad..dd2d40db 100644
--- a/light/light.cpp
+++ b/light/light.cpp
@@ -5,7 +5,7 @@
 
 Light::Light(const Vector3 &position)
 {
-    m_pGeometry = new Ball(Vector3::ZERO, position, 50, nullptr);
+    m_pGeometry = new Ball(Vector3::ZERO, position, 5, nullptr);
 }
 
 void Light::applyTransform(const Transform &t)
diff --git a/main/main.cpp b/main/main.cpp
index b24d41ec..6743ebdf 100644
--- a/main/main.cpp
+++ b/main/main.cpp
@@ -1,14 +1,15 @@
 #include "scene.h"
 #include "mesh.h"
 #include "common.h"
+#include "vector.h"
 
 int main()
 {
-    std::cout << "starting..." << std::endl;
-    Mesh* mesh = new Mesh(Common::BUNNY);
-    std::cout << "ending..." << std::endl;
-    // Scene scene;
-    // scene.run();
+    // std::cout << "starting..." << std::endl;
+    // Mesh* mesh = new Mesh(Common::BUNNY);
+    // std::cout << "ending..." << std::endl;
+    Scene scene;
+    scene.run();
 
     return 0;
 }
\ No newline at end of file
diff --git a/scene/scene.cpp b/scene/scene.cpp
index 6f2cc26c..20cd5e36 100644
--- a/scene/scene.cpp
+++ b/scene/scene.cpp
@@ -24,6 +24,7 @@
 #include "noiseTexture.h"
 #include "triangle.h"
 #include "mixMaterial.h"
+#include "tri.h"
 
 Scene::Scene()
 {
@@ -62,6 +63,11 @@ void Scene::constructScene()
 
     MixMaterial MtrlMix;
 
+    TriVertex v_a(20,0,0);
+    TriVertex v_b(0,0,0);
+    TriVertex v_c(0,20,0);
+    Tri *tri = new Tri(v_a, v_b, v_c, lambMtrlGreen);
+
     TriAngleVertex va = TriAngleVertex(-20, 20, 0, 0);
     TriAngleVertex vb = TriAngleVertex(0, -20, 0.5, 1);
     TriAngleVertex vc = TriAngleVertex(20, 20, 1, 0);
@@ -118,12 +124,12 @@ void Scene::constructScene()
     Vector3 squarePosition(70, -25, 290);
     Plane *squarePlane = new Plane(squareRotate, squarePosition, 20, lambMtrlLena);
 
-    frontPlane->setTag(Common::TAG_PLANE_FRONT);
-    backPlane->setTag(Common::TAG_PLANE_BACK);
-    topPlane->setTag(Common::TAG_PLANE_TOP);
-    bottomPlane->setTag(Common::TAG_PLANE_BOTTOM);
-    leftPlane->setTag(Common::TAG_PLANE_LEFT);
-    rightPlane->setTag(Common::TAG_PLANE_RIGHT);
+    // frontPlane->setTag(Common::TAG_PLANE_FRONT);
+    // backPlane->setTag(Common::TAG_PLANE_BACK);
+    // topPlane->setTag(Common::TAG_PLANE_TOP);
+    // bottomPlane->setTag(Common::TAG_PLANE_BOTTOM);
+    // leftPlane->setTag(Common::TAG_PLANE_LEFT);
+    // rightPlane->setTag(Common::TAG_PLANE_RIGHT);
 
     light->setTag(Common::TAG_LIGHT);
 
@@ -134,7 +140,10 @@ void Scene::constructScene()
 
     m_pObjectPool->add(light);
 
-    m_pObjectPool->add(triAngle);
+    m_pObjectPool->add(aquaBall);
+    m_pObjectPool->add(tri);
+
+    // m_pObjectPool->add(triAngle);
 
     // pool->add(redBall);
     // pool->add(yellowBall);
@@ -148,12 +157,12 @@ void Scene::constructScene()
 
     // pool->add(textureBall);
 
-    m_pObjectPool->add(frontPlane);
-    m_pObjectPool->add(backPlane);
-    m_pObjectPool->add(topPlane);
-    m_pObjectPool->add(bottomPlane);
-    m_pObjectPool->add(leftPlane);
-    m_pObjectPool->add(rightPlane);
+    // m_pObjectPool->add(frontPlane);
+    // m_pObjectPool->add(backPlane);
+    // m_pObjectPool->add(topPlane);
+    // m_pObjectPool->add(bottomPlane);
+    // m_pObjectPool->add(leftPlane);
+    // m_pObjectPool->add(rightPlane);
 }
 
 void Scene::preRender()