Commit | Line | Data |
---|---|---|

ddcf4b5d | 1 | /* |

3a66812f | 2 | * Copyright (C) 2007 Vitor Sessak <vitor1001@gmail.com> |

ddcf4b5d | 3 | * |

2912e87a | 4 | * This file is part of Libav. |

ddcf4b5d | 5 | * |

2912e87a | 6 | * Libav is free software; you can redistribute it and/or |

ddcf4b5d VS |
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 | * | |

2912e87a | 11 | * Libav is distributed in the hope that it will be useful, |

ddcf4b5d VS |
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 | |

2912e87a | 17 | * License along with Libav; if not, write to the Free Software |

ddcf4b5d VS |
18 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |

19 | */ | |

20 | ||

98790382 SS |
21 | #ifndef AVCODEC_ELBG_H |

22 | #define AVCODEC_ELBG_H | |

699b3f99 | 23 | |

e814a59f | 24 | #include "libavutil/lfg.h" |

ddcf4b5d VS |
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. | |

573e0118 | 38 | * @param rand_state A random number generator state. Should be already initialized by av_lfg_init(). |

ddcf4b5d VS |
39 | */ |

40 | void ff_do_elbg(int *points, int dim, int numpoints, int *codebook, | |

41 | int numCB, int num_steps, int *closest_cb, | |

9aa2ab42 | 42 | AVLFG *rand_state); |

ddcf4b5d VS |
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 ff_do_elbg for a (smaller) random sample of the points in | |

49 | * **points. Get the same parameters as ff_do_elbg. | |

50 | */ | |

51 | void ff_init_elbg(int *points, int dim, int numpoints, int *codebook, | |

52 | int numCB, int num_steps, int *closest_cb, | |

9aa2ab42 | 53 | AVLFG *rand_state); |

699b3f99 | 54 | |

98790382 | 55 | #endif /* AVCODEC_ELBG_H */ |