#ifndef TRIANGLE_GROUP_MAIN_H
#define TRIANGLE_GROUP_MAIN_H

#include <gsl/gsl_linalg.h>

#include "triangle.h"
#include "linalg.h"
#include "initcairo.h"

#undef ERROR
#define ERROR(condition, msg, ...) if(condition){fprintf(stderr, msg, ##__VA_ARGS__); exit(1);}
#define LOOP(i) for(int i = 0; i < 3; i++)

#define NUM_GROUP_ELEMENTS 10000
#define NUM_GROUP_ELEMENTS_COMBINATORIAL 100000

// (0,1) -> 2, (1,2) -> 0, (2,0) -> 1
// (1,0) -> 5, (2,1) -> 3, (0,2) -> 4
#define ROTATION_LETTER(x,y) (((y)-(x)+3)%3 == 1 ? ((y)+1)%3 : ((x)+1)%3+3)

typedef struct {
	double x[3];
} vector_t;

typedef struct {
	double x;
	double y;
} point_t;

typedef struct {
	// infos about the screen to draw on
	cairo_t *cairo;
	DimensionsInfo *dim;

	// a priori parameter
	int p[3],k[3];
	double parameter;
	double parameter2;
	char *movie_filename;
	double movie_parameter_duration;
	double movie_parameter2_duration;
	int movie_n_frames;
	int n_group_elements;
	int n_group_elements_combinatorial;
	int show_boxes;
	int show_boxes2;
	int show_attractors;
	int show_reflectors;
	int show_rotated_reflectors;
	int show_limit;
	int show_dual_limit;
	int show_text;
	int mode;
	int use_rotation_basis;
	int limit_with_lines;
	int show_marking;
	point_t marking;
	int show_coxeter_orbit;
	int use_repelling;
	gsl_matrix *cartan, *cob;

    // precomputed and constant
	groupelement_t *group;

	// computed stuff
	double *limit_curve; // x, y, angle triples
	int limit_curve_count;

	// temporary; matrices can only be freed from the top, but that's enough for us
	workspace_t *ws;
} DrawingContext;

typedef enum {
	VT_POINT,
	VT_LINE
} vector_type_t;

// implemented in limit_set.c
void cartanMatrix(gsl_matrix *cartan, double a1, double a2, double a3, double s);
void initializeTriangleGenerators(gsl_matrix **gen, double a1, double a2, double a3, double s, double t, workspace_t *ws);
void initializeTriangleGeneratorsCurrent(gsl_matrix **gen, DrawingContext *ctx);
int computeLimitCurve(DrawingContext *ctx);

// implemented in draw.c
vector_t cross(vector_t a, vector_t b);
int fixedPoints(DrawingContext *ctx, const char *word, vector_t *out);
void drawPoint(DrawingContext *ctx, point_t p);
void drawSegment2d(DrawingContext *ctx, point_t a, point_t b);
point_t vectorToPoint(DrawingContext *ctx, vector_t v);
void drawVector(DrawingContext *ctx, vector_t v);
void drawCovector(DrawingContext *ctx, vector_t v);
void drawSegment(DrawingContext *ctx, vector_t a, vector_t b);
void drawPolygon(DrawingContext *ctx, int segments, int sides, ...);
void drawTriangle(DrawingContext *ctx, const char *word);
void drawBox(DrawingContext *ctx, const char *word1, const char *word2);
void drawBoxLines(DrawingContext *ctx, const char *word1, const char *word2);
void drawBoxStd(DrawingContext *ctx, const char *word, char base);
void drawReflectors(DrawingContext *ctx);
void drawAttractors(DrawingContext *ctx);
void drawBoxes(DrawingContext *ctx);
void drawBoxes2(DrawingContext *ctx);
void drawLimitCurve(DrawingContext *ctx);
void drawText(DrawingContext *ctx);
void draw(DrawingContext *ctx);

// implemented in main.c
void setupContext(DrawingContext *ctx, int argc, char *argv[]);
void destroyContext(DrawingContext *ctx);
void print(DrawingContext *screen);
int processEvent(GraphicsInfo *info, XEvent *ev);
void computeMatrix(DrawingContext *ctx, gsl_matrix *result, const char *type);
void computeRotationMatrixFrame(DrawingContext *ctx, gsl_matrix *result, const char *type);
void updateMatrices(DrawingContext *ctx);

static vector_t vectorFromGsl(gsl_vector *v)
{
	vector_t result;
	LOOP(i) result.x[i] = gsl_vector_get(v, i);
	return result;
}

static void vectorToGsl(vector_t v, gsl_vector *out)
{
	LOOP(i) gsl_vector_set(out, i, v.x[i]);
}

#endif