/**
 * projectM -- Milkdrop-esque visualisation SDK
 * Copyright (C)2003-2004 projectM Team
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed i 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 * See 'LICENSE.txt' included within this release
 *
 */

#include <sstream>
#include "Common.hpp"
#include "fatal.h"

#include "CustomShape.hpp"
#include "Eval.hpp"
#include "Expr.hpp"
#include "InitCond.hpp"
#include "Param.hpp"
#include "PerFrameEqn.hpp"
#include "Preset.hpp"
#include <map>
#include "ParamUtils.hpp"
#include "InitCondUtils.hpp"
#include "wipemalloc.h"


CustomShape::CustomShape() : Shape()
{
	CustomShape(0);
};

CustomShape::CustomShape ( int id ) : Shape()
{

	Param * param;

	this->id = id;
	this->per_frame_count = 0;

	/* Start: Load custom shape parameters */
	param = Param::new_param_float ( "r", P_FLAG_NONE, &this->r, NULL, 1.0, 0.0, 0.5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "g", P_FLAG_NONE, &this->g, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "b", P_FLAG_NONE, &this->b, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "a", P_FLAG_NONE, &this->a, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "border_r", P_FLAG_NONE, &this->border_r, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "border_g", P_FLAG_NONE, &this->border_g, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "border_b", P_FLAG_NONE, &this->border_b, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "border_a", P_FLAG_NONE, &this->border_a, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "r2", P_FLAG_NONE, &this->r2, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "g2", P_FLAG_NONE, &this->g2, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "b2", P_FLAG_NONE, &this->b2, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "a2", P_FLAG_NONE, &this->a2, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "x", P_FLAG_NONE, &this->x, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "y", P_FLAG_NONE, &this->y, NULL, 1.0, 0.0, .5 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_bool ( "thickoutline", P_FLAG_NONE, &this->thickOutline, 1, 0, 0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_bool ( "enabled", P_FLAG_NONE, &this->enabled, 1, 0, 0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_int ( "sides", P_FLAG_NONE, &this->sides, 100, 3, 3 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_bool ( "additive", P_FLAG_NONE, &this->additive, 1, 0, 0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_bool ( "textured", P_FLAG_NONE, &this->textured, 1, 0, 0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "rad", P_FLAG_NONE, &this->radius, NULL, MAX_DOUBLE_SIZE, 0, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "ang", P_FLAG_NONE, &this->ang, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "tex_zoom", P_FLAG_NONE, &this->tex_zoom, NULL, MAX_DOUBLE_SIZE, .00000000001, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "tex_ang", P_FLAG_NONE, &this->tex_ang, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}

	param = Param::new_param_float ( "t1", P_FLAG_TVAR, &this->t1, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "t2", P_FLAG_TVAR, &this->t2, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "t3", P_FLAG_TVAR, &this->t3, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "t4", P_FLAG_TVAR, &this->t4, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "t5", P_FLAG_TVAR, &this->t5, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "t6", P_FLAG_TVAR, &this->t6, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "t7", P_FLAG_TVAR, &this->t7, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}
	param = Param::new_param_float ( "t8", P_FLAG_TVAR, &this->t8, NULL, MAX_DOUBLE_SIZE, -MAX_DOUBLE_SIZE, 0.0 );
	if ( !ParamUtils::insert( param, &this->param_tree ) )
	{
		abort();
	}

   for (unsigned int i = 1; i <= NUM_Q_VARIABLES;i++) {
	std::ostringstream os;
	os << "q" << i;
	param = Param::new_param_float ( os.str().c_str(), P_FLAG_QVAR, &this->q[i], NULL, MAX_DOUBLE_SIZE,
		 -MAX_DOUBLE_SIZE, 0.0 );
	if ( ParamUtils::insert ( param, &this->param_tree ) < 0 )
	{
		abort();
	}
  }

	param = Param::new_param_string ( "imageurl", P_FLAG_NONE, &this->imageUrl);
	if ( !ParamUtils::insert( param, &this->text_properties_tree ) )
	{
		abort();
	}

}

/* Frees a custom shape form object */
CustomShape::~CustomShape()
{

	traverseVector<TraverseFunctors::Delete<PerFrameEqn> > ( per_frame_eqn_tree );
	traverse<TraverseFunctors::Delete<InitCond> > ( init_cond_tree );
	traverse<TraverseFunctors::Delete<Param> > ( param_tree );
	traverse<TraverseFunctors::Delete<InitCond> > ( per_frame_init_eqn_tree );
	traverse<TraverseFunctors::Delete<Param> > ( text_properties_tree );

}

void CustomShape::loadUnspecInitConds()
{

	InitCondUtils::LoadUnspecInitCond fun ( this->init_cond_tree, this->per_frame_init_eqn_tree );
	traverse ( param_tree, fun );
}

void CustomShape::evalInitConds()
{
	// NOTE: This is verified to be same behavior as trunk
	for ( std::map<std::string, InitCond*>::iterator pos = per_frame_init_eqn_tree.begin(); 
		     pos != per_frame_init_eqn_tree.end();++pos )
		pos->second->evaluate();
}