code for assignment 3

[MicroTrace.git] / Scene.hxx

diff --git a/Scene.hxx b/Scene.hxx
index 18deb79..b92e9e5 100644 (file)
--- a/Scene.hxx
+++ b/Scene.hxx
@@ -5,6 +5,7 @@
 
 #include "Ray.hxx"
 #include "Camera.hxx"
+#include "Light.hxx"
 #include "Primitive.hxx"
 
 class Scene
@@ -15,7 +16,9 @@ public:
   
   // add another primitive to the scene
   void Add(Primitive* p);
- 
+  // add another light source to the scene
+  void Add(Light* l);
+
   // intersect the ray with all objects in the scene
   virtual bool Intersect(Ray& ray);
   // find occluder
@@ -35,5 +38,7 @@ private:
   
   // primitives
   std::vector<Primitive*> m_primitives;
+  // lights
+  std::vector<Light*> m_lights;
 };
 #endif