vf_hue

git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@11250 b3059339-0415-0410-9bf9-f77b7e298cf2

1 files changed, 195 insertions, 0 deletions

diff --git a/libmpcodecs/vf_hue.c b/libmpcodecs/vf_hue.c
new file mode 100644
index 0000000000..1e36854240
--- /dev/null
+++ b/libmpcodecs/vf_hue.c
@@ -0,0 +1,195 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <inttypes.h>
+#include <math.h>
+
+#include "../config.h"
+#include "../mp_msg.h"
+#include "../cpudetect.h"
+
+#include "img_format.h"
+#include "mp_image.h"
+#include "vf.h"
+
+#include "../libvo/video_out.h"
+
+#include "m_option.h"
+#include "m_struct.h"
+
+static struct vf_priv_s {
+	uint8_t *buf[2];
+	float hue;
+	float saturation;
+} vf_priv_dflt = {
+  {NULL, NULL},
+  0.0,
+  1.0,
+};
+
+static void process_C(uint8_t *udst, uint8_t *vdst, uint8_t *usrc, uint8_t *vsrc, int dststride, int srcstride,
+		    int w, int h, float hue, float sat)
+{
+	int i;
+	const int s= rint(sin(hue) * (1<<16) * sat);
+	const int c= rint(cos(hue) * (1<<16) * sat);
+
+	while (h--) {
+		for (i = 0; i<w; i++)
+		{
+			const int u= usrc[i] - 128;
+			const int v= vsrc[i] - 128;
+			int new_u= (c*u - s*v + (1<<15) + (128<<16))>>16;
+			int new_v= (s*u + c*v + (1<<15) + (128<<16))>>16;
+			if(new_u & 768) new_u= (-new_u)>>31;
+			if(new_v & 768) new_v= (-new_v)>>31;
+			udst[i]= new_u;
+			vdst[i]= new_v;
+		}
+		usrc += srcstride;
+		vsrc += srcstride;
+		udst += dststride;
+		vdst += dststride;
+	}
+}
+
+static void (*process)(uint8_t *udst, uint8_t *vdst, uint8_t *usrc, uint8_t *vsrc, int dststride, int srcstride,
+		    int w, int h, float hue, float sat);
+
+/* FIXME: add packed yuv version of process */
+
+static int put_image(struct vf_instance_s* vf, mp_image_t *mpi)
+{
+	mp_image_t *dmpi;
+
+	dmpi=vf_get_image(vf->next, mpi->imgfmt,
+			  MP_IMGTYPE_EXPORT, 0,
+			  mpi->w, mpi->h);
+	
+	dmpi->planes[0] = mpi->planes[0];
+	dmpi->stride[0] = mpi->stride[0];
+	dmpi->stride[1] = mpi->stride[1];
+	dmpi->stride[2] = mpi->stride[2];
+
+	if (!vf->priv->buf[0]){
+		vf->priv->buf[0] = malloc(mpi->stride[1]*mpi->h >> mpi->chroma_y_shift);
+		vf->priv->buf[1] = malloc(mpi->stride[2]*mpi->h >> mpi->chroma_y_shift);
+	}
+	
+	if (vf->priv->hue == 0){
+		dmpi->planes[1] = mpi->planes[1];
+		dmpi->planes[2] = mpi->planes[2];
+	}else {
+		dmpi->planes[1] = vf->priv->buf[0];
+		dmpi->planes[2] = vf->priv->buf[1];
+		process(dmpi->planes[1], dmpi->planes[2],
+			mpi->planes[1], mpi->planes[2],
+			dmpi->stride[1],mpi->stride[1],
+			mpi->w>> mpi->chroma_x_shift, mpi->h>> mpi->chroma_y_shift, 
+			vf->priv->hue, vf->priv->saturation);
+	}
+
+	return vf_next_put_image(vf,dmpi);
+}
+
+static int control(struct vf_instance_s* vf, int request, void* data)
+{
+	vf_equalizer_t *eq;
+
+	switch (request) {
+	case VFCTRL_SET_EQUALIZER:
+		eq = data;
+		if (!strcmp(eq->item,"hue")) {
+			vf->priv->hue = eq->value * M_PI / 100;
+			return CONTROL_TRUE;
+		} else if (!strcmp(eq->item,"saturation")) {
+			vf->priv->saturation = eq->value/100.0 + 100;
+			return CONTROL_TRUE;
+		}
+		break;
+	case VFCTRL_GET_EQUALIZER:
+		eq = data;
+		if (!strcmp(eq->item,"hue")) {
+			eq->value = rint(vf->priv->hue *100 / M_PI);
+			return CONTROL_TRUE;
+		}else if (!strcmp(eq->item,"saturation")) {
+			eq->value = rint(vf->priv->saturation*100 - 100);
+			return CONTROL_TRUE;
+		}
+		break;
+	}
+	return vf_next_control(vf, request, data);
+}
+
+static int query_format(struct vf_instance_s* vf, unsigned int fmt)
+{
+	switch (fmt) {
+	case IMGFMT_YVU9:
+	case IMGFMT_IF09:
+	case IMGFMT_YV12:
+	case IMGFMT_I420:
+	case IMGFMT_IYUV:
+	case IMGFMT_CLPL:
+	case IMGFMT_Y800:
+	case IMGFMT_Y8:
+	case IMGFMT_NV12:
+	case IMGFMT_444P:
+	case IMGFMT_422P:
+	case IMGFMT_411P:
+		return vf_next_query_format(vf, fmt);
+	}
+	return 0;
+}
+
+static void uninit(struct vf_instance_s* vf)
+{
+	if (vf->priv->buf[0]) free(vf->priv->buf[0]);
+	if (vf->priv->buf[1]) free(vf->priv->buf[1]);
+	free(vf->priv);
+}
+
+static int open(vf_instance_t *vf, char* args)
+{
+	vf->control=control;
+	vf->query_format=query_format;
+	vf->put_image=put_image;
+	vf->uninit=uninit;
+	
+	if(!vf->priv) {
+	vf->priv = malloc(sizeof(struct vf_priv_s));
+	memset(vf->priv, 0, sizeof(struct vf_priv_s));
+	}
+	if (args) sscanf(args, "%f:%f", &vf->priv->hue, &vf->priv->saturation);
+        vf->priv->hue *= M_PI / 180.0;
+
+	process = process_C;
+#ifdef HAVE_MMXX
+	if(gCpuCaps.hasMMX) process = process_MMX;
+#endif
+	
+	return 1;
+}
+
+#define ST_OFF(f) M_ST_OFF(struct vf_priv_s,f)
+static m_option_t vf_opts_fields[] = {
+  {"hue", ST_OFF(hue), CONF_TYPE_FLOAT, M_OPT_RANGE,-180.0 ,180.0, NULL},
+  {"saturation", ST_OFF(saturation), CONF_TYPE_FLOAT, M_OPT_RANGE,-10.0 ,10.0, NULL},
+  { NULL, NULL, 0, 0, 0, 0,  NULL }
+};
+
+static m_struct_t vf_opts = {
+  "hue",
+  sizeof(struct vf_priv_s),
+  &vf_priv_dflt,
+  vf_opts_fields
+};
+
+vf_info_t vf_info_hue = {
+	"hue changer",
+	"hue",
+	"Michael Niedermayer",
+	"",
+	open,
+	&vf_opts
+};
+