mirror of
https://github.com/game-stop/veejay.git
synced 2025-12-17 05:10:00 +01:00
117 lines
3.5 KiB
C
117 lines
3.5 KiB
C
/*
|
|
* Linux VeeJay
|
|
*
|
|
* Copyright(C)2010 Niels Elburg <nwelburg@gmail.com>
|
|
*
|
|
* This program 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.
|
|
*
|
|
* This program 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 this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307 , USA.
|
|
*/
|
|
|
|
// Uses: ctmf
|
|
/*
|
|
* ctmf.c - Constant-time median filtering
|
|
* Copyright (C) 2006 Simon Perreault
|
|
*
|
|
* Reference: S. Perreault and P. Hébert, "Median Filtering in Constant Time",
|
|
* IEEE Transactions on Image Processing, September 2007.
|
|
*
|
|
* This program has been obtained from http://nomis80.org/ctmf.html. No patent
|
|
* covers this program, although it is subject to the following license:
|
|
*/
|
|
|
|
#include <unistd.h>
|
|
#include "common.h"
|
|
#include <veejaycore/vjmem.h>
|
|
#include <ctmf/ctmf.h>
|
|
#include "median.h"
|
|
|
|
static long l2_cache_size_;
|
|
|
|
vj_effect *medianfilter_init(int w, int h)
|
|
{
|
|
vj_effect *ve = (vj_effect *) vj_calloc(sizeof(vj_effect));
|
|
ve->num_params = 1;
|
|
|
|
ve->defaults = (int *) vj_calloc(sizeof(int) * ve->num_params); /* default values */
|
|
ve->limits[0] = (int *) vj_calloc(sizeof(int) * ve->num_params); /* min */
|
|
ve->limits[1] = (int *) vj_calloc(sizeof(int) * ve->num_params); /* max */
|
|
ve->limits[0][0] = 0;
|
|
ve->limits[1][0] = 127;
|
|
ve->defaults[0] = 3;
|
|
ve->description = "Constant-time median filtering";
|
|
ve->sub_format = -1;
|
|
ve->extra_frame = 0;
|
|
ve->has_user = 0;
|
|
ve->param_description = vje_build_param_list( ve->num_params, "Value" );
|
|
|
|
l2_cache_size_ = sysconf( _SC_LEVEL2_CACHE_SIZE );
|
|
|
|
return ve;
|
|
}
|
|
|
|
typedef struct {
|
|
uint8_t *buffer[3];
|
|
} medianfilter_t;
|
|
|
|
void *medianfilter_malloc(int w, int h)
|
|
{
|
|
medianfilter_t *m = (medianfilter_t*) vj_calloc(sizeof(medianfilter_t));
|
|
if(!m) {
|
|
return NULL;
|
|
}
|
|
m->buffer[0] = (uint8_t*) vj_malloc( sizeof(uint8_t) * RUP8(w * h * 3));
|
|
if(!m->buffer[0]) {
|
|
free(m);
|
|
return NULL;
|
|
}
|
|
m->buffer[1] = m->buffer[0] + RUP8(w * h);
|
|
m->buffer[2] = m->buffer[1] + RUP8(w * h);
|
|
return (void*) m;
|
|
}
|
|
|
|
void medianfilter_free(void *ptr) {
|
|
medianfilter_t *m = (medianfilter_t*) ptr;
|
|
free(m->buffer[0]);
|
|
free(m);
|
|
}
|
|
|
|
void medianfilter_apply( void *ptr, VJFrame *frame, int *args ) {
|
|
int val = args[0];
|
|
|
|
if( val == 0 )
|
|
return;
|
|
|
|
medianfilter_t *m = (medianfilter_t*) ptr;
|
|
|
|
const unsigned int width = frame->width;
|
|
const unsigned int height = frame->height;
|
|
const int len = frame->len;
|
|
const int uv_len = frame->uv_len;
|
|
const int u_hei = frame->uv_height;
|
|
const int u_wid = frame->uv_width;
|
|
uint8_t *Y = frame->data[0];
|
|
uint8_t *Cb = frame->data[1];
|
|
uint8_t *Cr = frame->data[2];
|
|
|
|
uint8_t **buffer = m->buffer;
|
|
|
|
ctmf( Y, buffer[0], width,height, width, width, val,1,l2_cache_size_);
|
|
ctmf( Cb,buffer[1], u_wid, u_hei, u_wid, u_wid, val,1,l2_cache_size_);
|
|
ctmf( Cr,buffer[2], u_wid, u_hei, u_wid, u_wid, val,1,l2_cache_size_);
|
|
|
|
veejay_memcpy( Y, buffer[0], len);
|
|
veejay_memcpy( Cb,buffer[1], uv_len);
|
|
veejay_memcpy( Cr,buffer[2], uv_len);
|
|
}
|