/*
 * ecgen, tool for generating Elliptic curve domain parameters
 * Copyright (C) 2017 J08nY
 */
#include "curve.h"
#include "point.h"
#include "seed.h"
#include "util/memory.h"

curve_t *curve_new(void) { return try_calloc(sizeof(curve_t)); }

curve_t *curve_copy(const curve_t *src, curve_t *dest) {
	if (src->seed) dest->seed = seed_new_copy(src->seed);
	if (src->field) dest->field = gcopy(src->field);
	if (src->a) dest->a = gcopy(src->a);
	if (src->b) dest->b = gcopy(src->b);
	if (src->curve) dest->curve = gcopy(src->curve);
	if (src->order) dest->order = gcopy(src->order);
	if (src->generators) {
		dest->generators = points_new_copy(src->generators, src->ngens);
		dest->ngens = src->ngens;
	}
	if (src->points) {
		dest->points = points_new_copy(src->points, src->npoints);
		dest->npoints = src->npoints;
	}
	return dest;
}

curve_t *curve_new_copy(const curve_t *src) {
	curve_t *result = curve_new();
	return curve_copy(src, result);
}

curve_t *curve_clone(const curve_t *src, curve_t *dest) {
	if (src->seed) dest->seed = seed_new_clone(src->seed);
	if (src->field) dest->field = gclone(src->field);
	if (src->a) dest->a = gclone(src->a);
	if (src->b) dest->b = gclone(src->b);
	if (src->curve) dest->curve = gclone(src->curve);
	if (src->order) dest->order = gclone(src->order);
	if (src->generators) {
		dest->generators = points_new_clone(src->generators, src->ngens);
		dest->ngens = src->ngens;
	}
	if (src->points) {
		dest->points = points_new_clone(src->points, src->npoints);
		dest->npoints = src->npoints;
	}
	return dest;
}

curve_t *curve_new_clone(const curve_t *src) {
	curve_t *result = curve_new();
	return curve_clone(src, result);
}

void curve_free(curve_t **curve) {
	if (*curve) {
		seed_free(&(*curve)->seed);
		points_free_deep(&(*curve)->generators, (*curve)->ngens);
		points_free_deep(&(*curve)->points, (*curve)->npoints);

		if ((*curve)->curve) {
			// TODO, this is possibly dangerous...
			obj_free((*curve)->curve);
			if (isclone((*curve)->curve)) {
				gunclone((*curve)->curve);
			}
		}

		if ((*curve)->field && isclone((*curve)->field)) {
			gunclone((*curve)->field);
		}
		if ((*curve)->a && isclone((*curve)->a)) {
			gunclone((*curve)->a);
		}
		if ((*curve)->b && isclone((*curve)->b)) {
			gunclone((*curve)->b);
		}
		if ((*curve)->order && isclone((*curve)->order)) {
			gunclone((*curve)->order);
		}

		try_free(*curve);
		*curve = NULL;
	}
}

GENERATOR(curve_gen_any) {
	pari_sp ltop = avma;
	GEN v = gen_0;
	switch (typ(curve->field)) {
		case t_INT:
			v = gtovec0(gen_0, 2);
			gel(v, 1) = curve->a;
			gel(v, 2) = curve->b;
			break;
		case t_FFELT:
			v = gtovec0(gen_0, 5);
			gel(v, 1) = gen_1;
			gel(v, 2) = curve->a;
			gel(v, 4) = curve->b;
			break;
		default:
			pari_err_TYPE("curve_gen_any", curve->field);
	}
	GEN crv = ellinit(v, curve->field, -1);

	if (glength(crv) == 0) {
		avma = ltop;
		return -3;
	} else {
		curve->curve = gerepilecopy(ltop, crv);
		return 1;
	}
}

GENERATOR(curve_gen_nonzero) {
	pari_sp ltop = avma;
	int any = curve_gen_any(curve, cfg, args);
	if (any <= 0) {
		return any;
	}
	if (gequal0(ell_get_disc(curve->curve))) {
		avma = ltop;
		return -3;
	} else {
		return 1;
	}
}

UNROLL(curve_unroll) {
	if (curve->curve) {
		obj_free(curve->curve);
		curve->curve = NULL;
	}
	return -1;
}