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 * @return < 0 in case of error, 0 otherwise 40 */ 41 int avpriv_do_elbg(int *points, int dim, int numpoints, int *codebook, 42 int numCB, int num_steps, int *closest_cb, 43 AVLFG *rand_state); 44 45 /** 46 * Initialize the **codebook vector for the elbg algorithm. If you have already 47 * a codebook and you want to refine it, you shouldn't call this function. 48 * If numpoints < 8*numCB this function fills **codebook with random numbers. 49 * If not, it calls avpriv_do_elbg for a (smaller) random sample of the points in 50 * **points. Get the same parameters as avpriv_do_elbg. 51 * @return < 0 in case of error, 0 otherwise 52 */ 53 int avpriv_init_elbg(int *points, int dim, int numpoints, int *codebook, 54 int numCB, int num_steps, int *closest_cb, 55 AVLFG *rand_state); 56 57 #endif /* AVCODEC_ELBG_H */ 58