Statistics
| Branch: | Revision:

ffmpeg / libavfilter / libmpcodecs / vf_rotate.c @ e4852fb3

History | View | Annotate | Download (4.75 KB)

1
/*
2
 * This file is part of MPlayer.
3
 *
4
 * MPlayer is free software; you can redistribute it and/or modify
5
 * it under the terms of the GNU General Public License as published by
6
 * the Free Software Foundation; either version 2 of the License, or
7
 * (at your option) any later version.
8
 *
9
 * MPlayer is distributed in the hope that it will be useful,
10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 * GNU General Public License for more details.
13
 *
14
 * You should have received a copy of the GNU General Public License along
15
 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
16
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
17
 */
18

    
19
#include <stdio.h>
20
#include <stdlib.h>
21
#include <string.h>
22
#include <inttypes.h>
23

    
24
#include "config.h"
25
#include "mp_msg.h"
26

    
27
#include "img_format.h"
28
#include "mp_image.h"
29
#include "vf.h"
30

    
31
struct vf_priv_s {
32
    int direction;
33
};
34

    
35
static void rotate(unsigned char* dst,unsigned char* src,int dststride,int srcstride,int w,int h,int bpp,int dir){
36
    int y;
37
    if(dir&1){
38
        src+=srcstride*(w-1);
39
        srcstride*=-1;
40
    }
41
    if(dir&2){
42
        dst+=dststride*(h-1);
43
        dststride*=-1;
44
    }
45

    
46
    for(y=0;y<h;y++){
47
        int x;
48
        switch(bpp){
49
        case 1:
50
            for(x=0;x<w;x++) dst[x]=src[y+x*srcstride];
51
            break;
52
        case 2:
53
            for(x=0;x<w;x++) *((short*)(dst+x*2))=*((short*)(src+y*2+x*srcstride));
54
            break;
55
        case 3:
56
            for(x=0;x<w;x++){
57
                dst[x*3+0]=src[0+y*3+x*srcstride];
58
                dst[x*3+1]=src[1+y*3+x*srcstride];
59
                dst[x*3+2]=src[2+y*3+x*srcstride];
60
            }
61
            break;
62
        case 4:
63
            for(x=0;x<w;x++) *((int*)(dst+x*4))=*((int*)(src+y*4+x*srcstride));
64
        }
65
        dst+=dststride;
66
    }
67
}
68

    
69
//===========================================================================//
70

    
71
static int config(struct vf_instance *vf,
72
        int width, int height, int d_width, int d_height,
73
        unsigned int flags, unsigned int outfmt){
74
    if (vf->priv->direction & 4) {
75
        if (width<height) vf->priv->direction&=3;
76
    }
77
    if (vf->priv->direction & 4){
78
        vf->put_image=vf_next_put_image; // passthru mode!
79
        if (vf->next->draw_slice) vf->draw_slice=vf_next_draw_slice;
80
/* FIXME: this should be in an other procedure in vf.c; that should always check
81
     whether the filter after the passthrough one still (not)supports slices */
82
        return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
83
    }
84
    return vf_next_config(vf,height,width,d_height,d_width,flags,outfmt);
85
}
86

    
87
static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts){
88
    mp_image_t *dmpi;
89

    
90
    // hope we'll get DR buffer:
91
    dmpi=vf_get_image(vf->next,mpi->imgfmt,
92
        MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
93
        mpi->h, mpi->w);
94

    
95
    if(mpi->flags&MP_IMGFLAG_PLANAR){
96
        rotate(dmpi->planes[0],mpi->planes[0],
97
               dmpi->stride[0],mpi->stride[0],
98
               dmpi->w,dmpi->h,1,vf->priv->direction);
99
        rotate(dmpi->planes[1],mpi->planes[1],
100
               dmpi->stride[1],mpi->stride[1],
101
               dmpi->w>>mpi->chroma_x_shift,dmpi->h>>mpi->chroma_y_shift,1,vf->priv->direction);
102
        rotate(dmpi->planes[2],mpi->planes[2],
103
               dmpi->stride[2],mpi->stride[2],
104
               dmpi->w>>mpi->chroma_x_shift,dmpi->h>>mpi->chroma_y_shift,1,vf->priv->direction);
105
    } else {
106
        rotate(dmpi->planes[0],mpi->planes[0],
107
               dmpi->stride[0],mpi->stride[0],
108
               dmpi->w,dmpi->h,dmpi->bpp>>3,vf->priv->direction);
109
        dmpi->planes[1] = mpi->planes[1]; // passthrough rgb8 palette
110
    }
111

    
112
    return vf_next_put_image(vf,dmpi, pts);
113
}
114

    
115
//===========================================================================//
116

    
117
static int query_format(struct vf_instance *vf, unsigned int fmt){
118
    if(IMGFMT_IS_RGB(fmt) || IMGFMT_IS_BGR(fmt)) return vf_next_query_format(vf, fmt);
119
    // we can support only symmetric (chroma_x_shift==chroma_y_shift) YUV formats:
120
    switch(fmt) {
121
        case IMGFMT_YV12:
122
        case IMGFMT_I420:
123
        case IMGFMT_IYUV:
124
        case IMGFMT_YVU9:
125
//        case IMGFMT_IF09:
126
        case IMGFMT_Y8:
127
        case IMGFMT_Y800:
128
        case IMGFMT_444P:
129
            return vf_next_query_format(vf, fmt);
130
    }
131
    return 0;
132
}
133

    
134
static int vf_open(vf_instance_t *vf, char *args){
135
    vf->config=config;
136
    vf->put_image=put_image;
137
    vf->query_format=query_format;
138
    vf->priv=malloc(sizeof(struct vf_priv_s));
139
    vf->priv->direction=args?atoi(args):0;
140
    return 1;
141
}
142

    
143
const vf_info_t vf_info_rotate = {
144
    "rotate",
145
    "rotate",
146
    "A'rpi",
147
    "",
148
    vf_open,
149
    NULL
150
};
151

    
152
//===========================================================================//