/*
 * CINELERRA
 * Copyright (C) 1997-2015 Adam Williams <broadcast at earthling dot net>
 *
 * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include<stdio.h>
#include<stdint.h>
#include<math.h>
#include<string.h>

#include "arraylist.h"
#include "bccmodels.h"
#include "bccolors.h"
#include "clip.h"
#include "edlsession.h"
#include "filexml.h"
#include "sketcher.h"
#include "sketcherwindow.h"
#include "language.h"
#include "vframe.h"

void SketcherPoint::init(int id, int x, int y)
{
	this->id = id;
	this->x = x;    this->y = y;
}
SketcherPoint::SketcherPoint(int id)
{
	init(id, 0, 0);
}
SketcherPoint::SketcherPoint(int id, int x, int y)
{
	init(id, x, y);
}
SketcherPoint::~SketcherPoint()
{
}
SketcherPoint::SketcherPoint(SketcherPoint &pt)
{
	copy_from(pt);
}
int SketcherPoint::equivalent(SketcherPoint &that)
{
	return this->id == that.id &&
		this->x == that.x &&
		this->y == that.y ? 1 : 0;
}
void SketcherPoint::copy_from(SketcherPoint &that)
{
	this->id = that.id;
	this->x = that.x;    this->y = that.y;
}
void SketcherPoint::save_data(FileXML &output)
{
	char point[BCSTRLEN];
	sprintf(point,"/POINT_%d",id);
	output.tag.set_title(point+1);
	output.tag.set_property("X", x);
	output.tag.set_property("Y", y);
	output.append_tag();
	output.tag.set_title(point+0);
	output.append_tag();
	output.append_newline();
}
void SketcherPoint::read_data(FileXML &input)
{
	id = atoi(input.tag.get_title() + 6);
	x = input.tag.get_property("X", 0.f);
	y = input.tag.get_property("Y", 0.f);
}

void SketcherCurve::init(int id, int ty, int radius, int pen, int color)
{
	this->id = id;
	this->ty = ty;
	this->radius = radius;
	this->pen = pen;
	this->color = color;
}
SketcherCurve::SketcherCurve(int id)
{
	init(id, 0, 1, 0, BLACK);
}
SketcherCurve::SketcherCurve(int id, int ty,  int radius,int pen, int color)
{
	init(id, ty, radius, pen, color);
}
SketcherCurve::~SketcherCurve()
{
}
SketcherCurve::SketcherCurve(SketcherCurve &cv)
{
	copy_from(cv);
}
int SketcherCurve::equivalent(SketcherCurve &that)
{
	if( this->id != that.id ) return 0;
	if( this->ty != that.ty ) return 0;
	if( this->radius != that.radius ) return 0;
	if( this->pen != that.pen ) return 0;
	if( this->color != that.color ) return 0;
	int n = this->points.size();
	if( n != that.points.size() ) return 0;
	for( int i=0; i<n; ++i ) {
		if( !points[i]->equivalent(*that.points[i]) ) return 0;
	}
	return 1;
}
void SketcherCurve::copy_from(SketcherCurve &that)
{
	this->id = that.id;
	this->ty = that.ty;
	this->radius = that.radius;
	this->pen = that.pen;
	this->color = that.color;
	int m = points.size(), n = that.points.size();
	while( m > n ) points.remove_object_number(--m);
	while( m < n ) { points.append(new SketcherPoint());  ++m; }
	for( int i=0; i<n; ++i ) points[i]->copy_from(*that.points[i]);
}
void SketcherCurve::save_data(FileXML &output)
{
	this->ty = ty;
	this->pen = pen;  this->color = color;
	char curve[BCSTRLEN];
	sprintf(curve,"/CURVE_%d",id);
	output.tag.set_title(curve+1);
	output.tag.set_property("TYPE", ty);
	output.tag.set_property("RADIUS", radius);
	output.tag.set_property("PEN", pen);
	output.tag.set_property("COLOR", color);
	output.append_tag();
	output.append_newline();
	for( int i=0,n=points.size(); i<n; ++i )
		points[i]->save_data(output);
	output.tag.set_title(curve+0);
	output.append_tag();
	output.append_newline();
}
void SketcherCurve::read_data(FileXML &input)
{
	id = atoi(input.tag.get_title() + 6);
	ty = input.tag.get_property("TYPE", 0);
	radius = input.tag.get_property("RADIUS", 1.);
	pen = input.tag.get_property("PEN", 0);
	color = input.tag.get_property("COLOR", BLACK);
}

int Sketcher::new_curve(int ty, int radius, int pen, int color)
{
	SketcherCurves &curves = config.curves;
	int k = curves.size(), id = 1;
	for( int i=k; --i>=0; ) {
		int n = config.curves[i]->id;
		if( n >= id ) id = n + 1;
	}
	SketcherCurve *cv = new SketcherCurve(id, ty, radius, pen, color);
	curves.append(cv);
	config.cv_selected = k;
	return k;
}

int Sketcher::new_curve()
{
	return new_curve(0, 1, 0, BLACK);
}

int Sketcher::new_point(SketcherCurve *cv, int x, int y)
{
	int k = cv->points.size(), id = 1;
	for( int i=k; --i>=0; ) {
		int n = cv->points[i]->id;
		if( n >= id ) id = n + 1;
	}
	SketcherPoint *pt = new SketcherPoint(id, x, y);
	cv->points.append(pt);
	return k;
}

int Sketcher::new_point()
{
	int ci = config.cv_selected;
	if( ci < 0 || ci >= config.curves.size() )
		return -1;
	SketcherCurve *cv = config.curves[ci];
	EDLSession *session = get_edlsession();
	int x = !session ? 0.f : session->output_w / 2.f;
	int y = !session ? 0.f : session->output_h / 2.f;
	return new_point(cv, x, y);
}

REGISTER_PLUGIN(Sketcher)

SketcherConfig::SketcherConfig()
{
	drag = 1;
	cv_selected = 0;
	pt_selected = 0;
}
SketcherConfig::~SketcherConfig()
{
}

int SketcherConfig::equivalent(SketcherConfig &that)
{
	if( this->drag != that.drag ) return 0;
	if( this->cv_selected != that.cv_selected ) return 0;
	if( this->pt_selected != that.pt_selected ) return 0;
	if( this->curves.size() != that.curves.size() ) return 0;
	for( int i=0, n=curves.size(); i<n; ++i ) {
		if( !curves[i]->equivalent(*that.curves[i]) ) return 0;
	}
	return 1;
}

void SketcherConfig::copy_from(SketcherConfig &that)
{
	this->drag = that.drag;
	this->cv_selected = that.cv_selected;
	this->pt_selected = that.pt_selected;
	int m = curves.size(), n = that.curves.size();
	while( m > n ) curves.remove_object_number(--m);
	while( m < n ) { curves.append(new SketcherCurve());  ++m; }
	for( int i=0; i<n; ++i ) curves[i]->copy_from(*that.curves[i]);
}

void SketcherConfig::interpolate(SketcherConfig &prev, SketcherConfig &next,
		long prev_frame, long next_frame, long current_frame)
{
	this->cv_selected = prev.cv_selected;
	this->pt_selected = prev.pt_selected;
	this->drag = prev.drag;

	double next_scale = (double)(current_frame - prev_frame) / (next_frame - prev_frame);
	double prev_scale = (double)(next_frame - current_frame) / (next_frame - prev_frame);

	curves.remove_all_objects();
	int prev_cv_sz = prev.curves.size();
	int next_cv_sz = next.curves.size();
	for( int i=0; i<prev_cv_sz; ++i ) {
		SketcherCurve *pcv = prev.curves[i], *ncv = 0;
		SketcherCurve *cv = curves.append(new SketcherCurve());
		int k = next_cv_sz;  // associated by id in next
		while( --k >= 0 && pcv->id != (ncv=next.curves[k])->id );
		if( k >= 0 ) {
			cv->id = pcv->id;
			cv->ty = pcv->ty;
			cv->radius = pcv->radius;
			cv->pen = pcv->pen;
			cv->color = pcv->color;
			int prev_pt_sz = pcv->points.size(), next_pt_sz = ncv->points.size();
			for( int j=0; j<prev_pt_sz; ++j ) {
				SketcherPoint &pt = *pcv->points[j], *nt = 0;
				k = next_pt_sz;  // associated by id in next
				while( --k >= 0 && pt.id != (nt=ncv->points[k])->id );
				int x = pt.x, y = pt.y;
				if( k >= 0 ) {
					x = x * prev_scale + nt->x * next_scale;
					y = y * prev_scale + nt->y * next_scale;
				}
				cv->points.append(new SketcherPoint(pt.id, x, y));
			}
		}
		else
			cv->copy_from(*pcv);
	}
}

void SketcherConfig::limits()
{
}


Sketcher::Sketcher(PluginServer *server)
 : PluginVClient(server)
{
}

Sketcher::~Sketcher()
{
}

const char* Sketcher::plugin_title() { return N_("Sketcher"); }
int Sketcher::is_realtime() { return 1; }

NEW_WINDOW_MACRO(Sketcher, SketcherWindow);
LOAD_CONFIGURATION_MACRO(Sketcher, SketcherConfig)

void Sketcher::save_data(KeyFrame *keyframe)
{
	FileXML output;
// cause data to be stored directly in text
	output.set_shared_output(keyframe->xbuf);

	output.tag.set_title("SKETCHER");
	output.tag.set_property("DRAG", config.drag);
	output.tag.set_property("CURVE_SELECTED", config.cv_selected);
	output.tag.set_property("POINT_SELECTED", config.pt_selected);
	output.append_tag();
	output.append_newline();
	for( int i=0,n=config.curves.size(); i<n; ++i ) {
		config.curves[i]->save_data(output);
	}
	output.tag.set_title("/SKETCHER");
	output.append_tag();
	output.append_newline();
	output.terminate_string();
}

void Sketcher::read_data(KeyFrame *keyframe)
{
	FileXML input;
	input.set_shared_input(keyframe->xbuf);
	config.curves.remove_all_objects();
	int result = 0;
	SketcherCurve *cv = 0;

	while( !(result=input.read_tag()) ) {
		if( input.tag.title_is("SKETCHER") ) {
			config.drag = input.tag.get_property("DRAG", config.drag);
			config.cv_selected = input.tag.get_property("CV_SELECTED", 0);
			config.pt_selected = input.tag.get_property("PT_SELECTED", 0);
		}
		else if( !strncmp(input.tag.get_title(),"CURVE_",6) ) {
			cv = new SketcherCurve();
			cv->read_data(input);
			config.curves.append(cv);
		}
		else if( !strncmp(input.tag.get_title(),"/CURVE_",7) )
			cv = 0;
		else if( !strncmp(input.tag.get_title(),"POINT_",6) ) {
			if( cv ) {
				SketcherPoint *pt = new SketcherPoint();
				pt->read_data(input);
				cv->points.append(pt);
			}
			else
				printf("Sketcher::read_data: no curve for point\n");
		}
	}

	if( !config.curves.size() ) {
		new_curve(0, 1, 0, BLACK);
	}
	config.limits();
}

void Sketcher::update_gui()
{
	if( !thread ) return;
	thread->window->lock_window("Sketcher::update_gui");
	if( load_configuration() ) {
		SketcherWindow *window = (SketcherWindow*)thread->window;
		window->update_gui();
		window->flush();
	}
	thread->window->unlock_window();
}

void Sketcher::draw_point(VFrame *vfrm, SketcherPoint *pt, int color, int d)
{
	int r = d/2+1, x = pt->x, y = pt->y;
	vfrm->draw_smooth(x-r,y+0, x-r, y-r, x+0,y-r);
	vfrm->draw_smooth(x+0,y-r, x+r, y-r, x+r,y+0);
	vfrm->draw_smooth(x+r,y+0, x+r, y+r, x+0,y+r);
	vfrm->draw_smooth(x+0,y+r, x-r, y+r, x-r,y+0);
	vfrm->set_pixel_color(color);
	vfrm->draw_x(pt->x, pt->y, d);
}
void Sketcher::draw_point(VFrame *vfrm, SketcherPoint *pt, int color)
{
	draw_point(vfrm, pt, color, bmax(w,h)/200 + 2);
}


int SketcherPenSquare::draw_pixel(int x, int y)
{
	vfrm->draw_line(x-n, y, x+n, y);
	for( int i=-n; i<n; ++i )
		vfrm->draw_line(x-n, y+i, x+n, y+i);
	return 0;
}
int SketcherPenPlus::draw_pixel(int x, int y)
{
	if( n > 1 ) {
		vfrm->draw_line(x-n, y, x+n, y);
		vfrm->draw_line(x, y-n, x, y+n);
	}
	else
		vfrm->draw_pixel(x, y);
	return 0;
}
int SketcherPenSlant::draw_pixel(int x, int y)
{
	vfrm->draw_line(x-n,   y+n,   x+n,   y-n);
	vfrm->draw_line(x-n+1, y+n,   x+n+1, y-n);
	vfrm->draw_line(x-n,   y+n+1, x+n,   y-n+1);
	return 0;
}
int SketcherPenXlant::draw_pixel(int x, int y)
{
	vfrm->draw_line(x-n,   y+n,   x+n,   y-n);
	vfrm->draw_line(x-n+1, y+n,   x+n+1, y-n);
	vfrm->draw_line(x-n,   y+n+1, x+n,   y-n+1);
	vfrm->draw_line(x-n,   y-n,   x+n,   y+n);
	vfrm->draw_line(x-n+1, y-n,   x+n+1, y+n);
	vfrm->draw_line(x-n,   y-n+1, x+n,   y-n+1);
	return 0;
}


VFrame *SketcherCurve::new_vpen(VFrame *out)
{
	switch( pen ) {
	case PEN_SQUARE: return new SketcherPenSquare(out, radius);
	case PEN_PLUS:   return new SketcherPenPlus(out, radius);
	case PEN_SLANT:  return new SketcherPenSlant(out, radius);
	case PEN_XLANT:  return new SketcherPenXlant(out, radius);
	}
	return 0;
}

void SketcherCurve::draw_line(VFrame *out)
{
	SketcherPoint *pt0 = points[0];
	VFrame *vpen = new_vpen(out);
	out->set_pixel_color(color);
	int n = points.size();
	if( n >= 2 ) {
		for( int pi=1; pi<n; ++pi ) {
			SketcherPoint *pt1 = points[pi];
			vpen->draw_line(pt0->x, pt0->y, pt1->x, pt1->y);
			pt0 = pt1;
		}
	}
	else
		vpen->draw_pixel(pt0->x, pt0->y);
	delete vpen;
}

/*
# python
from sympy import *
var("x,y, ax,ay, bx,by, cx,cy, dx,dy")

var("abdx,abdy, acdx,acdy, bddx,bddy, cddx,cddy");
abdx = bx-ax;  abdy = by-ay;
acdx = cx-ax;  acdy = cy-ay;
bddx = dx-bx;  bddy = dy-by;
cddx = dx-cx;  cddy = dy-cy;

var("xc,yc, xd,yd, sx,sy");
xc = (bx+dx)/2;  yc = (by+dy)/2;
xd = cx-xc;      yd = cy-yc;
ax = xc-xd;      ay = yc-yd;
# line thru b with slope (c-a) intersects line thru c with slope (d-b)
sx = solve(((x - bx) * acdy/acdx + by) - ((x - cx) * bddy/bddx + cy),x)
sy = solve(((y - by) * acdx/acdy + bx) - ((y - cy) * bddx/bddy + cx),y)

var("zx,zy, zdx,zdy, sx,sy, px,py, qx,qy");
# point z = (b+c)/2
zx = (bx+cx)/2;  zy = (by+cy)/2;
zdx = (abdx+cddx)/2;  zdy = (abdy+cddy)/2;
# line thru z with slope (d-a) intersects line thru b with slope (c-a)
px = solve(((x-zx)*zdy/zdx + zy) - ((x-bx) * acdy/acdx + by),x);
py = solve(((y-zy)*zdx/zdy + zx) - ((y-by) * acdx/acdy + bx),y);
# line thru z with slope (c-a + d-b)/2 intersects line thru c with slope (d-b)
qx = solve(((x-zx)*zdy/zdx + zy) - ((x-cx) * bddy/bddx + cy),x);
qy = solve(((y-zy)*zdx/zdy + zx) - ((y-cy) * bddx/bddy + cx),y);
*/

static void smooth_sxy(
	float ax, float ay, float bx, float by,
	float cx, float cy, float dx, float dy,
	float &sx, float &sy)
{
	float acdx = cx-ax, acdy = cy-ay;
	float bddx = dx-bx, bddy = dy-by;
	float d = acdx*bddy - acdy*bddx;
	if( fabsf(d) < 1 ) d = 1;
	sx = (acdx*bddx*by - acdx*bddx*cy + acdx*bddy*cx - acdy*bddx*bx) / d;
	sy = (acdx*bddy*by - acdy*bddx*cy - acdy*bddy*bx + acdy*bddy*cx) / d;
	bclamp(sx, -32767.f, 32767.f);
	bclamp(sy, -32767.f, 32767.f);
}

static void smooth_pxy(
	float ax, float ay, float bx, float by,
	float cx, float cy, float dx, float dy,
	float &px, float &py)
{
	float abdx = bx - ax, abdy = by - ay;
	float acdx = cx - ax, acdy = cy - ay;
	float cddx = dx - cx, cddy = dy - cy;
	float d = (2*(abdx*acdy - abdy*acdx - acdx*cddy + acdy*cddx));
	if( fabsf(d) < 1 ) d = 1;
	px = (-abdx*acdx*by + abdx*acdx*cy + 2*abdx*acdy*bx - abdy*acdx*bx - abdy*acdx*cx -
		acdx*bx*cddy - acdx*by*cddx + acdx*cddx*cy - acdx*cddy*cx + 2*acdy*bx*cddx) / d;
	py = (abdx*acdy*by + abdx*acdy*cy - 2*abdy*acdx*by + abdy*acdy*bx - abdy*acdy*cx -
		 2*acdx*by*cddy + acdy*bx*cddy + acdy*by*cddx + acdy*cddx*cy - acdy*cddy*cx) / d;
	bclamp(px, -32767.f, 32767.f);
	bclamp(py, -32767.f, 32767.f);
}
static void smooth_qxy(
	float ax, float ay, float bx, float by,
	float cx, float cy, float dx, float dy,
	float &qx, float &qy)
{
	float abdx = bx - ax, abdy = by - ay;
	float bddx = dx - bx, bddy = dy - by;
	float cddx = dx - cx, cddy = dy - cy;
	float d = (2*(abdx*bddy - abdy*bddx - bddx*cddy + bddy*cddx));
	if( fabsf(d) < 1 ) d = 1;
	qx = (abdx*bddx*by - abdx*bddx*cy + 2*abdx*bddy*cx - abdy*bddx*bx - abdy*bddx*cx -
		bddx*bx*cddy + bddx*by*cddx - bddx*cddx*cy - bddx*cddy*cx + 2*bddy*cddx*cx) / d;
	qy = (abdx*bddy*by + abdx*bddy*cy - 2*abdy*bddx*cy - abdy*bddy*bx + abdy*bddy*cx -
		2*bddx*cddy*cy - bddy*bx*cddy + bddy*by*cddx + bddy*cddx*cy + bddy*cddy*cx) / d;
	bclamp(qx, -32767.f, 32767.f);
	bclamp(qy, -32767.f, 32767.f);
}


static int convex(float ax,float ay, float bx,float by,
		  float cx,float cy, float dx,float dy)
{
	float abdx = bx - ax, abdy = by - ay;
	float acdx = cx - ax, acdy = cy - ay;
	float bcdx = cx - bx, bcdy = cy - by;
	float bddx = dx - bx, bddy = dy - by;
	float abc = abdx*acdy - abdy*acdx;
	float bcd = bcdx*bddy - bcdy*bddx;
	float v = abc * bcd;
	return !v ? 0 : v>0 ? 1 : -1;
}

void SketcherCurve::draw_smooth(VFrame *out)
{
	VFrame *vpen = new_vpen(out);
	out->set_pixel_color(color);
	int n = points.size();
	if( !n ) return;
	if( n > 2 ) {
		SketcherPoint *pt0 = points[0], *pt1 = points[1], *pt2 = points[2];
		float bx = pt0->x, by = pt0->y;
		float cx = pt1->x, cy = pt1->y;
		float dx = pt2->x, dy = pt2->y;
		float xc = (bx+dx)/2.f, yc = (by+dy)/2.f;
		float xd = cx - xc, yd = cy - yc;
		float ax = xc - xd, ay = yc - yd;
		float sx, sy;
		for( int pi=0,n2=n-2; pi<n2; ++pi ) {
			float dx = points[pi+2]->x, dy = points[pi+2]->y;
			if( convex(ax,ay, bx,by, cx,cy, dx,dy) >= 0 ) {
				smooth_sxy(ax,ay, bx,by, cx,cy, dx,dy, sx, sy);
				vpen->draw_smooth(bx,by, sx,sy, cx,cy);
			}
			else {
				float zx = (bx+cx)/2.f, zy = (by+cy)/2.f;
				smooth_pxy(ax,ay, bx,by, cx,cy, dx,dy, sx,sy);
				vpen->draw_smooth(bx,by, sx,sy, zx,zy);
				smooth_qxy(ax,ay, bx,by, cx,cy, dx,dy, sx,sy);
				vpen->draw_smooth(zx,zy, sx,sy, cx,cy);
			}
			ax = bx;  ay = by;
			bx = cx;  by = cy;
			cx = dx;  cy = dy;
		}
		xc = (ax+cx)/2.f; yc = (ay+cy)/2.f;
		xd = bx - xc, yd = by - yc;
		dx = xc - xd, dy = yc - yd;
		smooth_sxy(ax, ay, bx, by, cx, cy, dx, dy, sx, sy);
		vpen->draw_smooth(bx, by, sx, sy, cx, cy);
	}
	else if( n == 2 ) {
		SketcherPoint *pt0 = points[0], *pt1 = points[1];
		vpen->draw_line(pt0->x, pt0->y, pt1->x, pt1->y);
	}
	else if( n > 0 ) {
		SketcherPoint *pt0 = points[0];
		vpen->draw_pixel(pt0->x, pt0->y);
	}
	delete vpen;
}

int Sketcher::process_realtime(VFrame *input, VFrame *output)
{
	this->input = input;  this->output = output;
	w = output->get_w();  h = output->get_h();
	if( input != output ) output->transfer_from(input);

	load_configuration();

	for( int ci=0, n=config.curves.size(); ci<n; ++ci ) {
		SketcherCurve *cv = config.curves[ci];
		int m = cv->points.size();
		if( !m ) continue;
		switch( cv->ty ) {
		case TYP_OFF:
			break;
		case TYP_SKETCHER:
			cv->draw_line(output);
			break;
		case TYP_SMOOTH:
			cv->draw_smooth(output);
			break;
		}
	}

	if( config.drag ) {
		for( int ci=0, n=config.curves.size(); ci<n; ++ci ) {
			SketcherCurve *cv = config.curves[ci];
			for( int pi=0,m=cv->points.size(); pi<m; ++pi )
				draw_point(output, cv->points[pi], cv->color);
		}
	}

	return 0;
}

void SketcherCurves::dump()
{
	for( int i=0; i<size(); ++i ) {
		SketcherCurve *cv = get(i);
		printf("Curve %d, id=%d, ty=%s, r=%d, pen=%s, color=%02x%02x%02x\n",
			i, cv->id, cv_type[cv->ty], cv->radius, cv_pen[cv->pen],
			(cv->color>>16)&0xff, (cv->color>>8)&0xff, (cv->color>>0)&0xff);
		cv->points.dump();
	}
}
void SketcherPoints::dump()
{
	for( int i=0; i<size(); ++i ) {
		SketcherPoint *pt = get(i);
		printf("  Pt %d, id=%d, x=%d, y=%d\n", i, pt->id, pt->x, pt->y);
	}
}