#include "../application/Controller.h"
#include "../application/AppView.h"
#include "../application/AppConfig.h"
#include "../application/AppCanvas.h"

#include <iostream>
#include <map>
#include <set>
using namespace std;

#ifdef __cplusplus
extern "C" {
#endif

#include "MEM_guardedalloc.h"

#include "DNA_camera_types.h"
#include "DNA_text_types.h"
#include "DNA_freestyle_types.h"

#include "BKE_global.h"
#include "BKE_library.h"
#include "BKE_linestyle.h"
#include "BKE_main.h"
#include "BKE_text.h"
#include "BLI_blenlib.h"
#include "BLI_math.h"
#include "BPY_extern.h"

#include "renderpipeline.h"
#include "pixelblending.h"

#include "../../FRS_freestyle.h"
#include "../../FRS_freestyle_config.h"

	// Freestyle configuration
	static short freestyle_is_initialized = 0;
	static Config::Path *pathconfig = NULL;
	static Controller *controller = NULL;
	static AppView *view = NULL;

	// camera information
	float freestyle_viewpoint[3];
	float freestyle_mv[4][4];
	float freestyle_proj[4][4];
	int freestyle_viewport[4];
	
	// current scene
	Scene *freestyle_scene;

	string default_module_path;

	//=======================================================
	//   Initialization 
	//=======================================================

	void FRS_initialize() {
		
		if( freestyle_is_initialized )
			return;

		pathconfig = new Config::Path;
		controller = new Controller();
		view = new AppView;
		controller->setView(view);
		controller->Clear();
		freestyle_scene = NULL;
			
		default_module_path = pathconfig->getProjectDir() + Config::DIR_SEP + "style_modules" + Config::DIR_SEP + "contour.py";
			
		freestyle_is_initialized = 1;
	}
	
	void FRS_set_context(bContext* C) {
		cout << "FRS_set_context: context 0x" << C << " scene 0x" << CTX_data_scene(C) << endl;
		controller->setContext(C);
	}

	void FRS_exit() {
		delete pathconfig;
		delete controller;
		delete view;
	}

	//=======================================================
	//   Rendering 
	//=======================================================

	static void init_view(Render* re){
		float ycor = ((float)re->r.yasp) / ((float)re->r.xasp);
		int width = re->r.xsch;
		int height = (int)(((float)re->r.ysch) * ycor);
		int xmin = re->r.border.xmin * width;
		int xmax = re->r.border.xmax * width;
		int ymin = re->r.border.ymin * height;
		int ymax = re->r.border.ymax * height;
		
		freestyle_viewport[0] = freestyle_viewport[1] = 0;
		freestyle_viewport[2] = width;
		freestyle_viewport[3] = height;
		
		view->setWidth( width );
		view->setHeight( height );
		view->setBorder( xmin, ymin, xmax, ymax );

		cout << "\n===  Dimensions of the 2D image coordinate system  ===" << endl;
		cout << "Width  : " << width << endl;
		cout << "Height : " << height << endl;
		if (re->r.mode & R_BORDER)
			cout << "Border : (" << xmin << ", " << ymin << ") - (" << xmax << ", " << ymax << ")" << endl;
	}

	static void init_camera(Render* re){
		// It is assumed that imported meshes are in the camera coordinate system.
		// Therefore, the view point (i.e., camera position) is at the origin, and
		// the the model-view matrix is simply the identity matrix.

		freestyle_viewpoint[0] = 0.0;
		freestyle_viewpoint[1] = 0.0;
		freestyle_viewpoint[2] = 0.0;
		
		for( int i = 0; i < 4; i++ )
		   for( int j = 0; j < 4; j++ )
			freestyle_mv[i][j] = (i == j) ? 1.0 : 0.0;
		
		for( int i = 0; i < 4; i++ )
		   for( int j = 0; j < 4; j++ )
			freestyle_proj[i][j] = re->winmat[i][j];

		//print_m4("mv", freestyle_mv);
		//print_m4("proj", freestyle_proj);
	}

	static Text *create_lineset_handler(char *layer_name, char *lineset_name)
	{
		Text *text = add_empty_text(lineset_name);
		write_text(text, "import parameter_editor; parameter_editor.process('");
		write_text(text, layer_name);
		write_text(text, "', '");
		write_text(text, lineset_name);
		write_text(text, "')\n");
		return text;
	}

	static void prepare(Render* re, SceneRenderLayer* srl ) {
				
		// load mesh
        re->i.infostr= "Freestyle: Mesh loading";
		re->stats_draw(re->sdh, &re->i);
        re->i.infostr= NULL;
		if( controller->LoadMesh(re, srl) ) // returns if scene cannot be loaded or if empty
			return;
        if( re->test_break(re->tbh) )
            return;
		
		// add style modules
		FreestyleConfig* config = &srl->freestyleConfig;
		
		cout << "\n===  Rendering options  ===" << endl;
		int layer_count = 0;
		
		switch (config->mode) {
		case FREESTYLE_CONTROL_SCRIPT_MODE:
			cout << "Modules :"<< endl;
			for (FreestyleModuleConfig* module_conf = (FreestyleModuleConfig *)config->modules.first; module_conf; module_conf = module_conf->next) {
				if( module_conf->is_displayed ) {
					cout << "  " << layer_count+1 << ": " << module_conf->module_path << endl;
					controller->InsertStyleModule( layer_count, module_conf->module_path );
					controller->toggleLayer(layer_count, true);
					layer_count++;
				}
			}
			cout << endl;
			break;
		case FREESTYLE_CONTROL_EDITOR_MODE:
			cout << "Linesets:"<< endl;
			for (FreestyleLineSet *lineset = (FreestyleLineSet *)config->linesets.first; lineset; lineset = lineset->next) {
				if (lineset->flags & FREESTYLE_LINESET_ENABLED) {
					cout << "  " << layer_count+1 << ": " << lineset->name << " - " << lineset->linestyle->id.name+2 << endl;
					Text *text = create_lineset_handler(srl->name, lineset->name);
					controller->InsertStyleModule( layer_count, lineset->name, text );
					controller->toggleLayer(layer_count, true);
					layer_count++;
				}
			}
			break;
		}
		
		// set parameters
		controller->setCreaseAngle( config->crease_angle );
		controller->setSphereRadius( config->sphere_radius );
		controller->setComputeRidgesAndValleysFlag( (config->flags & FREESTYLE_RIDGES_AND_VALLEYS_FLAG) ? true : false);
		controller->setComputeSuggestiveContoursFlag( (config->flags & FREESTYLE_SUGGESTIVE_CONTOURS_FLAG) ? true : false);
		controller->setComputeMaterialBoundariesFlag( (config->flags & FREESTYLE_MATERIAL_BOUNDARIES_FLAG) ? true : false);
		controller->setSuggestiveContourKrDerivativeEpsilon( config->dkr_epsilon ) ;

		cout << "Crease angle : " << controller->getCreaseAngle() << endl;
		cout << "Sphere radius : " << controller->getSphereRadius() << endl;
		cout << "Redges and valleys : " << (controller->getComputeRidgesAndValleysFlag() ? "enabled" : "disabled") << endl;
		cout << "Suggestive contours : " << (controller->getComputeSuggestiveContoursFlag() ? "enabled" : "disabled") << endl;
		cout << "Suggestive contour dkr epsilon : " << controller->getSuggestiveContourKrDerivativeEpsilon() << endl;
		cout << endl;

        // set diffuse and z depth passes
        RenderLayer *rl = RE_GetRenderLayer(re->result, srl->name);
		bool diffuse = false, z = false;
		for (RenderPass *rpass = (RenderPass *)rl->passes.first; rpass; rpass = rpass->next) {
			switch (rpass->passtype) {
			case SCE_PASS_DIFFUSE:
				controller->setPassDiffuse(rpass->rect, rpass->rectx, rpass->recty);
				diffuse = true;
				break;
			case SCE_PASS_Z:
				controller->setPassZ(rpass->rect, rpass->rectx, rpass->recty);
				z = true;
				break;
			}
		}
        cout << "Passes :" << endl;
        cout << "  Diffuse = " << (diffuse ? "enabled" : "disabled") << endl;
        cout << "  Z = " << (z ? "enabled" : "disabled") << endl;

		// compute view map
        re->i.infostr= "Freestyle: View map creation";
		re->stats_draw(re->sdh, &re->i);
        re->i.infostr= NULL;
		controller->ComputeViewMap();
	}
	
	void FRS_composite_result(Render* re, SceneRenderLayer* srl, Render* freestyle_render)
	{
		RenderLayer *rl;
	    float *src, *dest, *pixSrc, *pixDest;
		int x, y, rectx, recty;
		
		if( freestyle_render == NULL || freestyle_render->result == NULL )
			return;

		rl = render_get_active_layer( freestyle_render, freestyle_render->result );
	    if( !rl || rl->rectf == NULL) { cout << "Cannot find Freestyle result image" << endl; return; }
		src  = rl->rectf;
		//cout << "src: " << rl->rectx << " x " << rl->recty << endl;
		
		rl = RE_GetRenderLayer(re->result, srl->name);
	    if( !rl || rl->rectf == NULL) { cout << "No layer to composite to" << endl; return; }
		dest  = rl->rectf;
		//cout << "dest: " << rl->rectx << " x " << rl->recty << endl;

		rectx = re->rectx;
		recty = re->recty;
	    for( y = 0; y < recty; y++) {
	        for( x = 0; x < rectx; x++) {
	            pixSrc = src + 4 * (rectx * y + x);
	            if( pixSrc[3] > 0.0) {
					pixDest = dest + 4 * (rectx * y + x);
					addAlphaOverFloat(pixDest, pixSrc);
				}
			}
	    }
	}
	
	static int displayed_layer_count( SceneRenderLayer* srl ) {
		int count = 0;

		switch (srl->freestyleConfig.mode) {
		case FREESTYLE_CONTROL_SCRIPT_MODE:
			for (FreestyleModuleConfig* module = (FreestyleModuleConfig *)srl->freestyleConfig.modules.first; module; module = module->next) {
				if( module->is_displayed )
					count++;
			}
			break;
		case FREESTYLE_CONTROL_EDITOR_MODE:
			for (FreestyleLineSet *lineset = (FreestyleLineSet *)srl->freestyleConfig.linesets.first; lineset; lineset = lineset->next) {
				if (lineset->flags & FREESTYLE_LINESET_ENABLED)
					count++;
			}
			break;
		}
		return count;
	}

	int FRS_is_freestyle_enabled(SceneRenderLayer* srl) {
		return (!(srl->layflag & SCE_LAY_DISABLE) &&
			 	srl->layflag & SCE_LAY_FRS &&
				displayed_layer_count(srl) > 0);
	}
	
	void FRS_init_stroke_rendering(Render* re) {

		cout << "\n#===============================================================" << endl;
		cout << "#  Freestyle" << endl;
		cout << "#===============================================================" << endl;
		
		init_view(re);
		init_camera(re);

		controller->ResetRenderCount();
	}
	
	Render* FRS_do_stroke_rendering(Render* re, SceneRenderLayer *srl) {
		
		Render* freestyle_render = NULL;
		
		cout << "\n----------------------------------------------------------" << endl;
		cout << "|  " << (re->scene->id.name+2) << "|" << srl->name << endl;
		cout << "----------------------------------------------------------" << endl;
		
		// prepare Freestyle:
		//   - load mesh
		//   - add style modules
		//   - set parameters
		//   - compute view map
		prepare(re, srl);

        if( re->test_break(re->tbh) ) {
			controller->CloseFile();
            return NULL;
        }

		// render and composite Freestyle result
		if( controller->_ViewMap ) {
			
			// render strokes					
            re->i.infostr= "Freestyle: Stroke rendering";
            re->stats_draw(re->sdh, &re->i);
        	re->i.infostr= NULL;
			freestyle_scene = re->scene;
			controller->DrawStrokes();
			freestyle_render = controller->RenderStrokes(re);
			controller->CloseFile();
			freestyle_scene = NULL;
			
			// composite result
			FRS_composite_result(re, srl, freestyle_render);
			RE_FreeRenderResult(freestyle_render->result);
			freestyle_render->result = NULL;
		}

		return freestyle_render;
	}

	void FRS_finish_stroke_rendering(Render* re) {
		// clear canvas
		controller->Clear();
	}

	//=======================================================
	//   Freestyle Panel Configuration
	//=======================================================

	void FRS_add_freestyle_config( SceneRenderLayer* srl )
	{		
		FreestyleConfig* config = &srl->freestyleConfig;
		
		config->mode = FREESTYLE_CONTROL_SCRIPT_MODE;

		config->modules.first = config->modules.last = NULL;
		config->flags = 0;
		config->sphere_radius = 1.0f;
		config->dkr_epsilon = 0.001f;
		config->crease_angle = 134.43f;

		config->linesets.first = config->linesets.last = NULL;
	}
	
	void FRS_free_freestyle_config( SceneRenderLayer* srl )
	{		
		FreestyleLineSet *lineset;

		for(lineset=(FreestyleLineSet *)srl->freestyleConfig.linesets.first; lineset; lineset=lineset->next) {
			lineset->linestyle->id.us--;
			lineset->linestyle = NULL;
		}
		BLI_freelistN( &srl->freestyleConfig.linesets );
		BLI_freelistN( &srl->freestyleConfig.modules );
	}

	void FRS_add_module(FreestyleConfig *config)
	{
		FreestyleModuleConfig* module_conf = (FreestyleModuleConfig*) MEM_callocN( sizeof(FreestyleModuleConfig), "style module configuration");
		BLI_addtail(&config->modules, (void*) module_conf);
		
		strcpy( module_conf->module_path, default_module_path.c_str() );
		module_conf->is_displayed = 1;	
	}
	
	void FRS_delete_module(FreestyleConfig *config, FreestyleModuleConfig *module_conf)
	{
		BLI_freelinkN(&config->modules, module_conf);
	}
	
	void FRS_move_module_up(FreestyleConfig *config, FreestyleModuleConfig *module_conf)
	{
		BLI_remlink(&config->modules, module_conf);
		BLI_insertlinkbefore(&config->modules, module_conf->prev, module_conf);
	}
	
	void FRS_move_module_down(FreestyleConfig *config, FreestyleModuleConfig *module_conf)
	{			
		BLI_remlink(&config->modules, module_conf);
		BLI_insertlinkafter(&config->modules, module_conf->next, module_conf);
	}

	void FRS_add_lineset(FreestyleConfig *config)
	{
		int lineset_index = BLI_countlist(&config->linesets);

		FreestyleLineSet *lineset = (FreestyleLineSet *) MEM_callocN( sizeof(FreestyleLineSet), "Freestyle line set");
		BLI_addtail(&config->linesets, (void *) lineset);
		FRS_set_active_lineset_index(config, lineset_index);

		lineset->linestyle = FRS_new_linestyle("LineStyle", NULL);
		lineset->flags |= FREESTYLE_LINESET_ENABLED;
		lineset->selection = 0;
		lineset->qi = FREESTYLE_QI_VISIBLE;
		lineset->qi_start = 0;
		lineset->qi_end = 100;
		lineset->edge_types = 0;
		lineset->objects.first = lineset->objects.last = NULL;
		if (lineset_index > 0)
			sprintf(lineset->name, "LineSet %i", lineset_index+1);
		else
			strcpy(lineset->name, "LineSet");
		BLI_uniquename(&config->linesets, lineset, "FreestyleLineSet", '.', BLI_STRUCT_OFFSET(FreestyleLineSet, name), sizeof(lineset->name));
	}

	void FRS_delete_active_lineset(FreestyleConfig *config)
	{
		FreestyleLineSet *lineset = FRS_get_active_lineset(config);

		if (lineset) {
			lineset->linestyle->id.us--;
			lineset->linestyle = NULL;
			BLI_remlink(&config->linesets, lineset);
			MEM_freeN(lineset);
			FRS_set_active_lineset_index(config, 0);
		}
	}

	void FRS_move_active_lineset_up(FreestyleConfig *config)
	{
		FreestyleLineSet *lineset = FRS_get_active_lineset(config);

		if (lineset) {
			BLI_remlink(&config->linesets, lineset);
			BLI_insertlinkbefore(&config->linesets, lineset->prev, lineset);
		}
	}

	void FRS_move_active_lineset_down(FreestyleConfig *config)
	{
		FreestyleLineSet *lineset = FRS_get_active_lineset(config);

		if (lineset) {
			BLI_remlink(&config->linesets, lineset);
			BLI_insertlinkafter(&config->linesets, lineset->next, lineset);
		}
	}

	FreestyleLineSet *FRS_get_active_lineset(FreestyleConfig *config)
	{
		FreestyleLineSet *lineset;

		for(lineset=(FreestyleLineSet *)config->linesets.first; lineset; lineset=lineset->next)
			if(lineset->flags & FREESTYLE_LINESET_CURRENT)
				return lineset;
		return NULL;
	}

	short FRS_get_active_lineset_index(FreestyleConfig *config)
	{
		FreestyleLineSet *lineset;
		short i;

		for(lineset=(FreestyleLineSet *)config->linesets.first, i=0; lineset; lineset=lineset->next, i++)
			if(lineset->flags & FREESTYLE_LINESET_CURRENT)
				return i;
		return 0;
	}

	void FRS_set_active_lineset_index(FreestyleConfig *config, short index)
	{
		FreestyleLineSet *lineset;
		short i;

		for(lineset=(FreestyleLineSet *)config->linesets.first, i=0; lineset; lineset=lineset->next, i++) {
			if(i == index)
				lineset->flags |= FREESTYLE_LINESET_CURRENT;
			else
				lineset->flags &= ~FREESTYLE_LINESET_CURRENT;
		}
	}

#ifdef __cplusplus
}
#endif