1 files changed, 139 insertions, 0 deletions
diff --git a/source/blender/collada/LightExporter.cpp b/source/blender/collada/LightExporter.cpp
new file mode 100644
index 00000000000..5786c682d6a
--- /dev/null
+++ b/source/blender/collada/LightExporter.cpp
@@ -0,0 +1,139 @@
+/**
+ * $Id$
+ *
+ * ***** BEGIN GPL LICENSE BLOCK *****
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ *
+ * Contributor(s): Chingiz Dyussenov, Arystanbek Dyussenov, Jan Diederich, Tod Liverseed,
+ *                 Nathan Letwory
+ *
+ * ***** END GPL LICENSE BLOCK *****
+ */
+
+#include <string>
+
+#include "COLLADASWColor.h"
+#include "COLLADASWLight.h"
+
+#include "DNA_lamp_types.h"
+
+#include "BLI_math.h"
+
+#include "LightExporter.h"
+#include "collada_internal.h"
+
+template<class Functor>
+void forEachLampObjectInScene(Scene *sce, Functor &f)
+{
+	Base *base= (Base*) sce->base.first;
+	while(base) {
+		Object *ob = base->object;
+			
+		if (ob->type == OB_LAMP && ob->data) {
+			f(ob);
+		}
+		base= base->next;
+	}
+}
+
+LightsExporter::LightsExporter(COLLADASW::StreamWriter *sw): COLLADASW::LibraryLights(sw){}
+
+void LightsExporter::exportLights(Scene *sce)
+{
+	openLibrary();
+	
+	forEachLampObjectInScene(sce, *this);
+	
+	closeLibrary();
+}
+void LightsExporter::operator()(Object *ob)
+{
+	Lamp *la = (Lamp*)ob->data;
+	std::string la_id(get_light_id(ob));
+	std::string la_name(id_name(la));
+	COLLADASW::Color col(la->r, la->g, la->b);
+	float att1, att2;
+	float e, d, constatt, linatt, quadatt;
+	att1 = att2 = 0.0f;
+	
+	if(la->falloff_type==LA_FALLOFF_INVLINEAR) {
+		att1 = 1.0f;
+		att2 = 0.0f;
+	}
+	else if(la->falloff_type==LA_FALLOFF_INVSQUARE) {
+		att1 = 0.0f;
+		att2 = 1.0f;
+	}
+	else if(la->falloff_type==LA_FALLOFF_SLIDERS) {
+		att1 = la->att1;
+		att2 = la->att2;
+	}
+	
+	e = la->energy;
+	d = la->dist;
+	
+	constatt = linatt = quadatt = MAXFLOAT;
+	if(e > 0.0f) {
+		constatt = 1.0f/e;
+		linatt = att1/(d*e);
+		quadatt = att2/(d*d*(e*2));
+	}
+	
+	// sun
+	if (la->type == LA_SUN) {
+		COLLADASW::DirectionalLight cla(mSW, la_id, la_name, e);
+		cla.setColor(col);
+		cla.setConstantAttenuation(constatt);
+		addLight(cla);
+	}
+	// hemi
+	else if (la->type == LA_HEMI) {
+		COLLADASW::AmbientLight cla(mSW, la_id, la_name, e);
+		cla.setColor(col);
+		cla.setConstantAttenuation(constatt);
+		addLight(cla);
+	}
+	// spot
+	else if (la->type == LA_SPOT) {
+		COLLADASW::SpotLight cla(mSW, la_id, la_name, e);
+		cla.setColor(col);
+		cla.setFallOffAngle(la->spotsize);
+		cla.setFallOffExponent(la->spotblend);
+		cla.setConstantAttenuation(constatt);
+		cla.setLinearAttenuation(linatt);
+		cla.setQuadraticAttenuation(quadatt);
+		addLight(cla);
+	}
+	// lamp
+	else if (la->type == LA_LOCAL) {
+		COLLADASW::PointLight cla(mSW, la_id, la_name, e);
+		cla.setColor(col);
+		cla.setConstantAttenuation(constatt);
+		cla.setLinearAttenuation(linatt);
+		cla.setQuadraticAttenuation(quadatt);
+		addLight(cla);
+	}
+	// area lamp is not supported
+	// it will be exported as a local lamp
+	else {
+		COLLADASW::PointLight cla(mSW, la_id, la_name, e);
+		cla.setColor(col);
+		cla.setConstantAttenuation(constatt);
+		cla.setLinearAttenuation(linatt);
+		cla.setQuadraticAttenuation(quadatt);
+		addLight(cla);
+	}
+}