motion.h

Go to the documentation of this file.

 
#ifndef MOTION_H
#define MOTION_H
 
#ifdef __cplusplus
extern "C"{
#endif 
 
#include <stdint.h>
#include <stdbool.h>
#include <stdlib.h>
 
#define mmax(a,b) \
   ({ __typeof__ (a) _a = (a); \
       __typeof__ (b) _b = (b); \
     _a > _b ? _a : _b; })
 
#define mmin(a,b) \
   ({ __typeof__ (a) _a = (a); \
       __typeof__ (b) _b = (b); \
     _a < _b ? _a : _b; })
 
#define WINDOW 5 
 
#define LK_OPTICAL_FLOW_8BIT    0
#define LK_OPTICAL_FLOW     1
#define BLOCK_MATCHING_ARPS 2
#define BLOCK_MATCHING_EPZS     3
 
typedef struct { int16_t x, y; } Vector16_t;
 
typedef struct {
    int16_t  vx;     
    int16_t  vy;     
    uint16_t mag2;   
} MotionVector16_t;
 
typedef struct {
    int8_t  vx;      
    int8_t  vy;      
} MotionVector8_t;
 
typedef struct MotionEstPredictor {
    int mvs[10][2];  
    int nb;          
} MotionEstPredictor;
 
 
typedef struct MotionEstContext{    
    char name[20];
    uint8_t *data_cur,                  
            *data_ref;                  
    int method;                         
    int max;                            
    int width,                          
        height,                         
    //  linewidth,                      ///< images width * depth (= width * 3 if rgb)  
        b_width,                        
        b_height,                       
        b_count;                        
    
    int mbSize,                         
        log2_mbSize;                    
    int search_param;                   
 
    int pred_x,                         
        pred_y;                         
    MotionEstPredictor preds[2];        
 
    MotionVector16_t *mv_table[3];      
 
    uint64_t (*get_cost) (struct MotionEstContext *self, int x_mb, int y_mb, int x_mv, int y_mv);
    bool (*motion_func) (struct MotionEstContext *self);    
} MotionEstContext;
 
void uninit(MotionEstContext *ctx);
 
bool init_context(MotionEstContext *ctx);
 
bool motion_estimation(MotionEstContext *ctx, uint8_t *img_prev, uint8_t *img_cur);
 
uint64_t me_comp_sad(MotionEstContext *me_ctx, int x_mb, int y_mb, int x_mv, int y_mv);
 
bool LK_optical_flow(MotionEstContext *);
 
bool LK_optical_flow8(const uint8_t *src1, const uint8_t *src2, uint8_t *v, int w, int h);
 
bool motionEstARPS(MotionEstContext *) ;
 
bool motionEstEPZS(MotionEstContext *);
 
uint8_t *motionComp(const uint8_t *imgI, const MotionVector16_t *motionVect, size_t w, size_t h, 
        size_t mbSize);
 
#ifdef __cplusplus
}
#endif
#endif