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

ddcf4b5d VS |
1 | /* |

2 | * Copyright (C) 2007 Vitor <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 | ||

699b3f99 MR |
21 | #ifndef AVCODEC_ELBG_H |

22 | #define AVCODEC_ELBG_H | |

23 | ||

ddcf4b5d VS |
24 | #include "random.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 initialised by av_init_random. | |

39 | */ | |

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

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

42 | AVRandomState *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 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, | |

53 | AVRandomState *rand_state); | |

699b3f99 MR |
54 | |

55 | #endif |