Libav
vf_unsharp.c
Go to the documentation of this file.
1 /*
2  * Original copyright (c) 2002 Remi Guyomarch <rguyom@pobox.com>
3  * Port copyright (c) 2010 Daniel G. Taylor <dan@programmer-art.org>
4  * Relicensed to the LGPL with permission from Remi Guyomarch.
5  *
6  * This file is part of Libav.
7  *
8  * Libav is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * Libav is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
39 #include "avfilter.h"
40 #include "formats.h"
41 #include "internal.h"
42 #include "video.h"
43 #include "libavutil/common.h"
44 #include "libavutil/mem.h"
45 #include "libavutil/opt.h"
46 #include "libavutil/pixdesc.h"
47 
48 #define MIN_SIZE 3
49 #define MAX_SIZE 13
50 
51 typedef struct FilterParam {
52  int msize_x;
53  int msize_y;
54  int amount;
55  int steps_x;
56  int steps_y;
57  int scalebits;
59  uint32_t *sc[(MAX_SIZE * MAX_SIZE) - 1];
60 } FilterParam;
61 
62 typedef struct UnsharpContext {
63  const AVClass *class;
64  int lmsize_x, lmsize_y, cmsize_x, cmsize_y;
65  float lamount, camount;
68  int hsub, vsub;
70 
71 static void apply_unsharp( uint8_t *dst, int dst_stride,
72  const uint8_t *src, int src_stride,
73  int width, int height, FilterParam *fp)
74 {
75  uint32_t **sc = fp->sc;
76  uint32_t sr[(MAX_SIZE * MAX_SIZE) - 1], tmp1, tmp2;
77 
78  int32_t res;
79  int x, y, z;
80  const uint8_t *src2;
81 
82  if (!fp->amount) {
83  if (dst_stride == src_stride)
84  memcpy(dst, src, src_stride * height);
85  else
86  for (y = 0; y < height; y++, dst += dst_stride, src += src_stride)
87  memcpy(dst, src, width);
88  return;
89  }
90 
91  for (y = 0; y < 2 * fp->steps_y; y++)
92  memset(sc[y], 0, sizeof(sc[y][0]) * (width + 2 * fp->steps_x));
93 
94  for (y = -fp->steps_y; y < height + fp->steps_y; y++) {
95  if (y < height)
96  src2 = src;
97 
98  memset(sr, 0, sizeof(sr[0]) * (2 * fp->steps_x - 1));
99  for (x = -fp->steps_x; x < width + fp->steps_x; x++) {
100  tmp1 = x <= 0 ? src2[0] : x >= width ? src2[width-1] : src2[x];
101  for (z = 0; z < fp->steps_x * 2; z += 2) {
102  tmp2 = sr[z + 0] + tmp1; sr[z + 0] = tmp1;
103  tmp1 = sr[z + 1] + tmp2; sr[z + 1] = tmp2;
104  }
105  for (z = 0; z < fp->steps_y * 2; z += 2) {
106  tmp2 = sc[z + 0][x + fp->steps_x] + tmp1; sc[z + 0][x + fp->steps_x] = tmp1;
107  tmp1 = sc[z + 1][x + fp->steps_x] + tmp2; sc[z + 1][x + fp->steps_x] = tmp2;
108  }
109  if (x >= fp->steps_x && y >= fp->steps_y) {
110  const uint8_t *srx = src - fp->steps_y * src_stride + x - fp->steps_x;
111  uint8_t *dsx = dst - fp->steps_y * dst_stride + x - fp->steps_x;
112 
113  res = (int32_t)*srx + ((((int32_t) * srx - (int32_t)((tmp1 + fp->halfscale) >> fp->scalebits)) * fp->amount) >> 16);
114  *dsx = av_clip_uint8(res);
115  }
116  }
117  if (y >= 0) {
118  dst += dst_stride;
119  src += src_stride;
120  }
121  }
122 }
123 
124 static void set_filter_param(FilterParam *fp, int msize_x, int msize_y, float amount)
125 {
126  fp->msize_x = msize_x;
127  fp->msize_y = msize_y;
128  fp->amount = amount * 65536.0;
129 
130  fp->steps_x = msize_x / 2;
131  fp->steps_y = msize_y / 2;
132  fp->scalebits = (fp->steps_x + fp->steps_y) * 2;
133  fp->halfscale = 1 << (fp->scalebits - 1);
134 }
135 
137 {
138  UnsharpContext *unsharp = ctx->priv;
139 
140  set_filter_param(&unsharp->luma, unsharp->lmsize_x, unsharp->lmsize_y, unsharp->lamount);
141  set_filter_param(&unsharp->chroma, unsharp->cmsize_x, unsharp->cmsize_y, unsharp->camount);
142 
143  return 0;
144 }
145 
147 {
148  enum AVPixelFormat pix_fmts[] = {
152  };
153 
155 
156  return 0;
157 }
158 
159 static void init_filter_param(AVFilterContext *ctx, FilterParam *fp, const char *effect_type, int width)
160 {
161  int z;
162  const char *effect;
163 
164  effect = fp->amount == 0 ? "none" : fp->amount < 0 ? "blur" : "sharpen";
165 
166  av_log(ctx, AV_LOG_VERBOSE, "effect:%s type:%s msize_x:%d msize_y:%d amount:%0.2f\n",
167  effect, effect_type, fp->msize_x, fp->msize_y, fp->amount / 65535.0);
168 
169  for (z = 0; z < 2 * fp->steps_y; z++)
170  fp->sc[z] = av_malloc(sizeof(*(fp->sc[z])) * (width + 2 * fp->steps_x));
171 }
172 
173 static int config_props(AVFilterLink *link)
174 {
175  UnsharpContext *unsharp = link->dst->priv;
177 
178  unsharp->hsub = desc->log2_chroma_w;
179  unsharp->vsub = desc->log2_chroma_h;
180 
181  init_filter_param(link->dst, &unsharp->luma, "luma", link->w);
182  init_filter_param(link->dst, &unsharp->chroma, "chroma", AV_CEIL_RSHIFT(link->w, unsharp->hsub));
183 
184  return 0;
185 }
186 
188 {
189  int z;
190 
191  for (z = 0; z < 2 * fp->steps_y; z++)
192  av_free(fp->sc[z]);
193 }
194 
196 {
197  UnsharpContext *unsharp = ctx->priv;
198 
199  free_filter_param(&unsharp->luma);
200  free_filter_param(&unsharp->chroma);
201 }
202 
203 static int filter_frame(AVFilterLink *link, AVFrame *in)
204 {
205  UnsharpContext *unsharp = link->dst->priv;
206  AVFilterLink *outlink = link->dst->outputs[0];
207  AVFrame *out;
208  int cw = AV_CEIL_RSHIFT(link->w, unsharp->hsub);
209  int ch = AV_CEIL_RSHIFT(link->h, unsharp->vsub);
210 
211  out = ff_get_video_buffer(outlink, outlink->w, outlink->h);
212  if (!out) {
213  av_frame_free(&in);
214  return AVERROR(ENOMEM);
215  }
216  av_frame_copy_props(out, in);
217 
218  apply_unsharp(out->data[0], out->linesize[0], in->data[0], in->linesize[0], link->w, link->h, &unsharp->luma);
219  apply_unsharp(out->data[1], out->linesize[1], in->data[1], in->linesize[1], cw, ch, &unsharp->chroma);
220  apply_unsharp(out->data[2], out->linesize[2], in->data[2], in->linesize[2], cw, ch, &unsharp->chroma);
221 
222  av_frame_free(&in);
223  return ff_filter_frame(outlink, out);
224 }
225 
226 #define OFFSET(x) offsetof(UnsharpContext, x)
227 #define FLAGS AV_OPT_FLAG_VIDEO_PARAM
228 static const AVOption options[] = {
229  { "luma_msize_x", "luma matrix horizontal size", OFFSET(lmsize_x), AV_OPT_TYPE_INT, { .i64 = 5 }, MIN_SIZE, MAX_SIZE, FLAGS },
230  { "luma_msize_y", "luma matrix vertical size", OFFSET(lmsize_y), AV_OPT_TYPE_INT, { .i64 = 5 }, MIN_SIZE, MAX_SIZE, FLAGS },
231  { "luma_amount", "luma effect strength", OFFSET(lamount), AV_OPT_TYPE_FLOAT, { .dbl = 1 }, -2, 5, FLAGS },
232  { "chroma_msize_x", "chroma matrix horizontal size", OFFSET(cmsize_x), AV_OPT_TYPE_INT, { .i64 = 5 }, MIN_SIZE, MAX_SIZE, FLAGS },
233  { "chroma_msize_y", "chroma matrix vertical size", OFFSET(cmsize_y), AV_OPT_TYPE_INT, { .i64 = 5 }, MIN_SIZE, MAX_SIZE, FLAGS },
234  { "chroma_amount", "chroma effect strength", OFFSET(camount), AV_OPT_TYPE_FLOAT, { .dbl = 0 }, -2, 5, FLAGS },
235  { NULL },
236 };
237 
238 static const AVClass unsharp_class = {
239  .class_name = "unsharp",
240  .item_name = av_default_item_name,
241  .option = options,
242  .version = LIBAVUTIL_VERSION_INT,
243 };
244 
246  {
247  .name = "default",
248  .type = AVMEDIA_TYPE_VIDEO,
249  .filter_frame = filter_frame,
250  .config_props = config_props,
251  },
252  { NULL }
253 };
254 
256  {
257  .name = "default",
258  .type = AVMEDIA_TYPE_VIDEO,
259  },
260  { NULL }
261 };
262 
264  .name = "unsharp",
265  .description = NULL_IF_CONFIG_SMALL("Sharpen or blur the input video."),
266 
267  .priv_size = sizeof(UnsharpContext),
268  .priv_class = &unsharp_class,
269 
270  .init = init,
271  .uninit = uninit,
273 
274  .inputs = avfilter_vf_unsharp_inputs,
275 
276  .outputs = avfilter_vf_unsharp_outputs,
277 };
void * av_malloc(size_t size)
Allocate a block of size bytes with alignment suitable for all memory accesses (including vectors if ...
Definition: mem.c:62
const AVPixFmtDescriptor * av_pix_fmt_desc_get(enum AVPixelFormat pix_fmt)
Definition: pixdesc.c:1768
This structure describes decoded (raw) audio or video data.
Definition: frame.h:140
static const AVFilterPad avfilter_vf_unsharp_inputs[]
Definition: vf_unsharp.c:245
AVOption.
Definition: opt.h:234
#define FLAGS
Definition: vf_unsharp.c:227
planar YUV 4:4:4, 24bpp, (1 Cr & Cb sample per 1x1 Y samples)
Definition: pixfmt.h:64
static const AVFilterPad outputs[]
Definition: af_ashowinfo.c:259
Main libavfilter public API header.
memory handling functions
const char * desc
Definition: nvenc.c:101
static av_cold int init(AVFilterContext *ctx)
Definition: vf_unsharp.c:136
av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (%s)\, len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt), use_generic ? ac->func_descr_generic :ac->func_descr)
AVFrame * ff_get_video_buffer(AVFilterLink *link, int w, int h)
Request a picture buffer with a specific set of permissions.
Definition: video.c:57
#define MAX_SIZE
Definition: vf_unsharp.c:49
uint8_t log2_chroma_w
Amount to shift the luma width right to find the chroma width.
Definition: pixdesc.h:91
AVFilterFormats * ff_make_format_list(const int *fmts)
Create a list of supported formats.
Definition: formats.c:165
static void apply_unsharp(uint8_t *dst, int dst_stride, const uint8_t *src, int src_stride, int width, int height, FilterParam *fp)
Definition: vf_unsharp.c:71
const char * name
Pad name.
Definition: internal.h:41
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:39
static void set_filter_param(FilterParam *fp, int msize_x, int msize_y, float amount)
Definition: vf_unsharp.c:124
int ff_filter_frame(AVFilterLink *link, AVFrame *frame)
Send a frame of data to the next filter.
Definition: avfilter.c:747
uint8_t
#define av_cold
Definition: attributes.h:66
AVOptions.
#define OFFSET(x)
Definition: vf_unsharp.c:226
AVFilter ff_vf_unsharp
Definition: vf_unsharp.c:263
static const AVFilterPad avfilter_vf_unsharp_outputs[]
Definition: vf_unsharp.c:255
planar YUV 4:4:0 full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV440P and setting color_range...
Definition: pixfmt.h:97
planar YUV 4:2:2, 16bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV422P and setting col...
Definition: pixfmt.h:72
#define AV_LOG_VERBOSE
Detailed information.
Definition: log.h:140
void ff_set_common_formats(AVFilterContext *ctx, AVFilterFormats *formats)
A helper for query_formats() which sets all links to the same list of formats.
Definition: formats.c:389
FilterParam luma
luma parameters (width, height, amount)
Definition: vf_unsharp.c:66
A filter pad used for either input or output.
Definition: internal.h:35
#define src
Definition: vp8dsp.c:254
void av_free(void *ptr)
Free a memory block which has been allocated with av_malloc(z)() or av_realloc(). ...
Definition: mem.c:190
uint8_t log2_chroma_h
Amount to shift the luma height right to find the chroma height.
Definition: pixdesc.h:100
#define AVERROR(e)
Definition: error.h:43
void av_frame_free(AVFrame **frame)
Free the frame and any dynamically allocated objects in it, e.g.
Definition: frame.c:80
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:148
#define MIN_SIZE
Definition: vf_unsharp.c:48
void * priv
private data for use by the filter
Definition: avfilter.h:277
static const AVClass unsharp_class
Definition: vf_unsharp.c:238
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:63
int msize_x
matrix width
Definition: vf_unsharp.c:52
static const AVOption options[]
Definition: vf_unsharp.c:228
int scalebits
bits to shift pixel
Definition: vf_unsharp.c:57
planar YUV 4:2:0, 12bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV420P and setting col...
Definition: pixfmt.h:71
static av_cold void uninit(AVFilterContext *ctx)
Definition: vf_unsharp.c:195
int32_t
AVFormatContext * ctx
Definition: movenc.c:48
static void free_filter_param(FilterParam *fp)
Definition: vf_unsharp.c:187
static int filter_frame(AVFilterLink *link, AVFrame *in)
Definition: vf_unsharp.c:203
LIBAVUTIL_VERSION_INT
Definition: eval.c:55
static int config_props(AVFilterLink *link)
Definition: vf_unsharp.c:173
int steps_x
horizontal step count
Definition: vf_unsharp.c:55
int amount
effect amount
Definition: vf_unsharp.c:54
NULL
Definition: eval.c:55
static int width
Definition: utils.c:156
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:158
av_default_item_name
Definition: dnxhdenc.c:55
Descriptor that unambiguously describes how the bits of a pixel are stored in the up to 4 data planes...
Definition: pixdesc.h:80
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(const int16_t *) pi >> 8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(const int32_t *) pi >> 24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) #define SET_CONV_FUNC_GROUP(ofmt, ifmt) static void set_generic_function(AudioConvert *ac) { } void ff_audio_convert_free(AudioConvert **ac) { if(! *ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);} AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, int sample_rate, int apply_map) { AudioConvert *ac;int in_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) return NULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method !=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt) > 2) { ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc) { av_free(ac);return NULL;} return ac;} in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar) { ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar ? ac->channels :1;} else if(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;else ac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);return ac;} int ff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in) { int use_generic=1;int len=in->nb_samples;int p;if(ac->dc) { av_log(ac->avr, AV_LOG_TRACE, "%d samples - audio_convert: %s to %s (dithered)\", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));return ff_convert_dither(ac-> in
planar YUV 4:1:0, 9bpp, (1 Cr & Cb sample per 4x4 Y samples)
Definition: pixfmt.h:65
Describe the class of an AVClass context structure.
Definition: log.h:34
Filter definition.
Definition: avfilter.h:120
int32_t halfscale
amount to add to pixel
Definition: vf_unsharp.c:58
static const AVFilterPad inputs[]
Definition: af_ashowinfo.c:248
static int query_formats(AVFilterContext *ctx)
Definition: vf_unsharp.c:146
const char * name
Filter name.
Definition: avfilter.h:124
AVFilterLink ** outputs
array of pointers to output links
Definition: avfilter.h:274
static enum AVPixelFormat pix_fmts[]
Definition: libkvazaar.c:257
int steps_y
vertical step count
Definition: vf_unsharp.c:56
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:146
int msize_y
matrix height
Definition: vf_unsharp.c:53
int height
Definition: gxfenc.c:72
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:59
common internal and external API header
planar YUV 4:4:4, 24bpp, full scale (JPEG), deprecated in favor of AV_PIX_FMT_YUV444P and setting col...
Definition: pixfmt.h:73
planar YUV 4:1:1, 12bpp, (1 Cr & Cb sample per 4x1 Y samples)
Definition: pixfmt.h:66
An instance of a filter.
Definition: avfilter.h:262
FILE * out
Definition: movenc.c:54
planar YUV 4:4:0 (1 Cr & Cb sample per 1x2 Y samples)
Definition: pixfmt.h:96
static void init_filter_param(AVFilterContext *ctx, FilterParam *fp, const char *effect_type, int width)
Definition: vf_unsharp.c:159
internal API functions
AVPixelFormat
Pixel format.
Definition: pixfmt.h:57
FilterParam chroma
chroma parameters (width, height, amount)
Definition: vf_unsharp.c:67
int av_frame_copy_props(AVFrame *dst, const AVFrame *src)
Copy only "metadata" fields from src to dst.
Definition: frame.c:386
uint32_t * sc[(MAX_SIZE *MAX_SIZE) - 1]
finite state machine storage
Definition: vf_unsharp.c:59
#define AV_CEIL_RSHIFT(a, b)
Fast a / (1 << b) rounded toward +inf, assuming a >= 0 and b >= 0.
Definition: common.h:57