/* See LICENSE file for copyright and license details. */
#include "common.h"

USAGE("[-e] [theta0-stream]")

static int equal = 0;


#define PROCESS(TYPE, SUFFIX)\
	static void\
	process_##SUFFIX(struct stream *grad, struct stream *theta0)\
	{\
		size_t i, n, m = 0;\
		TYPE *theta0xyza;\
		TYPE x, theta0x = 0;\
		TYPE y, theta0y = 0;\
		TYPE z, theta0z = 0;\
		TYPE a, theta0a = 0;\
		echeck_dimensions(grad, WIDTH | HEIGHT, NULL);\
		do {\
			if (!m) {\
				m = grad->frame_size;\
				if (theta0) {\
					while (theta0->ptr < theta0->frame_size)\
						if (!eread_stream(theta0, theta0->frame_size - theta0->ptr))\
							return;\
					theta0xyza = (TYPE *)theta0->buf;\
					theta0x = (theta0xyza)[0];\
					theta0y = (theta0xyza)[1];\
					theta0z = (theta0xyza)[2];\
					theta0a = (theta0xyza)[3];\
					memmove(theta0->buf, theta0->buf + theta0->frame_size,\
						theta0->ptr -= theta0->frame_size);\
				}\
			}\
			n = MIN(grad->ptr, m) / grad->pixel_size;\
			if (equal) {\
				for (i = 0; i < n; i++) {\
					a = ((TYPE *)(grad->buf))[4 * i + 3];\
					a = (a ? sin(a + theta0y) / a : sin(a + theta0y)) / 2 + 0.5;\
					((TYPE *)(grad->buf))[4 * i + 0] = a;\
					((TYPE *)(grad->buf))[4 * i + 1] = a;\
					((TYPE *)(grad->buf))[4 * i + 2] = a;\
					((TYPE *)(grad->buf))[4 * i + 3] = a;\
				}\
			} else {\
				for (i = 0; i < n; i++) {\
					x = ((TYPE *)(grad->buf))[4 * i + 0];\
					y = ((TYPE *)(grad->buf))[4 * i + 1];\
					z = ((TYPE *)(grad->buf))[4 * i + 2];\
					a = ((TYPE *)(grad->buf))[4 * i + 3];\
					x = (x ? sin(x + theta0x) / x : sin(x + theta0x)) / 2 + 0.5;\
					y = (y ? sin(y + theta0y) / y : sin(y + theta0y)) / 2 + 0.5;\
					z = (z ? sin(z + theta0z) / z : sin(z + theta0z)) / 2 + 0.5;\
					a = (a ? sin(a + theta0a) / a : sin(a + theta0a)) / 2 + 0.5;\
					((TYPE *)(grad->buf))[4 * i + 0] = x;\
					((TYPE *)(grad->buf))[4 * i + 1] = y;\
					((TYPE *)(grad->buf))[4 * i + 2] = z;\
					((TYPE *)(grad->buf))[4 * i + 3] = a;\
				}\
			}\
			n *= grad->pixel_size;\
			m -= n;\
			ewriteall(STDOUT_FILENO, grad->buf, n, "<stdout>");\
			memmove(grad->buf, grad->buf + n, grad->ptr -= n);\
		} while (eread_stream(grad, SIZE_MAX));\
		if (grad->ptr)\
			eprintf("%s: incomplete frame\n", grad->file);\
	}

PROCESS(double, lf)
PROCESS(float, f)


int
main(int argc, char *argv[])
{
	struct stream stream, theta0;
	int have_theta0;
	void (*process)(struct stream *grad, struct stream *theta0);

	ARGBEGIN {
	case 'e':
		equal = 1;
		break;
	default:
		usage();
	} ARGEND;

	if (argc > 1)
		usage();

	eopen_stream(&stream, NULL);
	if ((have_theta0 = argc == 1)) {
		eopen_stream(&theta0, argv[0]);
		if (theta0.width != 1 || theta0.height != 1)
			eprintf("theta0-stream must be of dimension 1x1\n");
	}

	if (!strcmp(stream.pixfmt, "xyza"))
		process = process_lf;
	else if (!strcmp(stream.pixfmt, "xyza f"))
		process = process_f;
	else
		eprintf("pixel format %s is not supported, try xyza\n", stream.pixfmt);

	if (have_theta0 && strcmp(stream.pixfmt, theta0.pixfmt))
		eprintf("videos use incompatible pixel formats\n");

	fprint_stream_head(stdout, &stream);
	efflush(stdout, "<stdout>");
	process(&stream, have_theta0 ? &theta0 : NULL);
	return 0;
}