ffmpeg/libavfilter/libmpcodecs/vf_hue.c

/*
 * This file is part of MPlayer.
 *
 * MPlayer 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.
 *
 * MPlayer 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 MPlayer; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#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"

struct vf_priv_s {
        uint8_t *buf[2];
        float hue;
        float saturation;
};

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 *vf, mp_image_t *mpi, double pts)
{
        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 && vf->priv->saturation == 1){
                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, pts);
}

static int control(struct vf_instance *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)/100.0;
                        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 *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_444P:
        case IMGFMT_422P:
        case IMGFMT_411P:
                return vf_next_query_format(vf, fmt);
        }
        return 0;
}

static void uninit(struct vf_instance *vf)
{
        free(vf->priv->buf[0]);
        free(vf->priv->buf[1]);
        free(vf->priv);
}

static int vf_open(vf_instance_t *vf, char *args)
{
        vf->control=control;
        vf->query_format=query_format;
        vf->put_image=put_image;
        vf->uninit=uninit;

    vf->priv = malloc(sizeof(struct vf_priv_s));
    memset(vf->priv, 0, sizeof(struct vf_priv_s));
    sscanf(args, "%f:%f", &vf->priv->hue, &vf->priv->saturation);
        vf->priv->hue *= M_PI / 180.0;

        process = process_C;
        return 1;
}

const vf_info_t vf_info_hue = {
        "hue changer",
        "hue",
        "Michael Niedermayer",
        "",
        vf_open,
};