/*******************************************************
      Lightwave Object Loader for OSG

  Copyright (C) 2004 Marco Jez <marco.jez@poste.it>
  OpenSceneGraph is (C) 2004 Robert Osfield
********************************************************/

#include "VertexMap.h"

#include <osg/Notify>

using namespace lwosg;

osg::Vec4Array *VertexMap::asVec4Array(int num_vertices, const osg::Vec4 &default_value, const osg::Vec4 &modulator) const
{
	osg::ref_ptr<osg::Vec4Array> array = new osg::Vec4Array;
	array->assign(num_vertices, default_value);
	for (VertexMap::const_iterator i=begin(); i!=end(); ++i) {
		osg::Vec4 value = i->second;
		value.x() *= modulator.x();
		value.y() *= modulator.y();
		value.z() *= modulator.z();
		value.w() *= modulator.w();
		array->at(i->first) = value;
	}
	return array.take();
}

osg::Vec2Array *VertexMap::asVec2Array(int num_vertices, const osg::Vec2 &default_value, const osg::Vec2 &modulator) const
{
	osg::ref_ptr<osg::Vec2Array> array = new osg::Vec2Array;
	array->assign(num_vertices, default_value);
	for (VertexMap::const_iterator i=begin(); i!=end(); ++i) {
		osg::Vec4 value = i->second;
		value.x() *= modulator.x();
		value.y() *= modulator.y();
		array->at(i->first) = osg::Vec2(value.x(), value.y());
	}
	return array.take();
}

osg::Vec3Array *VertexMap::asVec3Array(int num_vertices, const osg::Vec3 &default_value, const osg::Vec3 &modulator) const
{
	osg::ref_ptr<osg::Vec3Array> array = new osg::Vec3Array;
	array->assign(num_vertices, default_value);
	for (VertexMap::const_iterator i=begin(); i!=end(); ++i) {
		osg::Vec4 value = i->second;
		value.x() *= modulator.x();
		value.y() *= modulator.y();
		value.z() *= modulator.z();
		array->at(i->first) = osg::Vec3(value.x(), value.y(), value.z());
	}
	return array.take();
}

VertexMap *VertexMap::remap(const std::vector<int> &remapping) const
{
	osg::ref_ptr<VertexMap> result = new VertexMap;

	for (VertexMap::const_iterator i=begin(); i!=end(); ++i) {
		if (i->first >= static_cast<int>(remapping.size())) {
			osg::notify(osg::WARN) << "Warning: lwosg::remap(): remapping index not found for vertex " << i->first << " (map size " << remapping.size() << ")" << std::endl;
		} else {
			int new_index = remapping[i->first];
			if (new_index != -1) {
				(*result.get())[new_index] = i->second;
			}
		}
	}

	return result.take();
}

VertexMap_map *VertexMap_map::remap(const std::vector<int> &remapping) const
{
	osg::ref_ptr<VertexMap_map> result = new VertexMap_map;
	for (VertexMap_map::const_iterator i=begin(); i!=end(); ++i) {
		(*result.get())[i->first] = i->second->remap(remapping);
	}
	return result.take();
}
