Commit | Line | Data |
---|---|---|
2ba45a60 DM |
1 | /* |
2 | * Copyright (C) 2007 Vitor Sessak <vitor1001@gmail.com> | |
3 | * | |
4 | * This file is part of FFmpeg. | |
5 | * | |
6 | * FFmpeg is free software; you can redistribute it and/or | |
7 | * modify it under the terms of the GNU Lesser General Public | |
8 | * License as published by the Free Software Foundation; either | |
9 | * version 2.1 of the License, or (at your option) any later version. | |
10 | * | |
11 | * FFmpeg is distributed in the hope that it will be useful, | |
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
14 | * Lesser General Public License for more details. | |
15 | * | |
16 | * You should have received a copy of the GNU Lesser General Public | |
17 | * License along with FFmpeg; if not, write to the Free Software | |
18 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
19 | */ | |
20 | ||
21 | #ifndef AVCODEC_ELBG_H | |
22 | #define AVCODEC_ELBG_H | |
23 | ||
24 | #include "libavutil/lfg.h" | |
25 | ||
26 | /** | |
27 | * Implementation of the Enhanced LBG Algorithm | |
28 | * Based on the paper "Neural Networks 14:1219-1237" that can be found in | |
29 | * http://citeseer.ist.psu.edu/patan01enhanced.html . | |
30 | * | |
31 | * @param points Input points. | |
32 | * @param dim Dimension of the points. | |
33 | * @param numpoints Num of points in **points. | |
34 | * @param codebook Pointer to the output codebook. Must be allocated. | |
35 | * @param numCB Number of points in the codebook. | |
36 | * @param num_steps The maximum number of steps. One step is already a good compromise between time and quality. | |
37 | * @param closest_cb Return the closest codebook to each point. Must be allocated. | |
38 | * @param rand_state A random number generator state. Should be already initialized by av_lfg_init(). | |
39 | */ | |
40 | void avpriv_do_elbg(int *points, int dim, int numpoints, int *codebook, | |
41 | int numCB, int num_steps, int *closest_cb, | |
42 | AVLFG *rand_state); | |
43 | ||
44 | /** | |
45 | * Initialize the **codebook vector for the elbg algorithm. If you have already | |
46 | * a codebook and you want to refine it, you shouldn't call this function. | |
47 | * If numpoints < 8*numCB this function fills **codebook with random numbers. | |
48 | * If not, it calls avpriv_do_elbg for a (smaller) random sample of the points in | |
49 | * **points. Get the same parameters as avpriv_do_elbg. | |
50 | */ | |
51 | void avpriv_init_elbg(int *points, int dim, int numpoints, int *codebook, | |
52 | int numCB, int num_steps, int *closest_cb, | |
53 | AVLFG *rand_state); | |
54 | ||
55 | #endif /* AVCODEC_ELBG_H */ |