screenshot: improve quality by using additional swscale flags
[mplayer.git] / libmpcodecs / mp_image.c
blob45426eb6736870e405d072d40fefdf26af4a49fe
1 /*
2 * This file is part of MPlayer.
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.
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.
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.
19 #include "config.h"
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <string.h>
25 #include "talloc.h"
27 #include "libmpcodecs/img_format.h"
28 #include "libmpcodecs/mp_image.h"
30 #include "libvo/fastmemcpy.h"
31 #include "libavutil/mem.h"
33 void mp_image_alloc_planes(mp_image_t *mpi) {
34 // IF09 - allocate space for 4. plane delta info - unused
35 if (mpi->imgfmt == IMGFMT_IF09) {
36 mpi->planes[0]=av_malloc(mpi->bpp*mpi->width*(mpi->height+2)/8+
37 mpi->chroma_width*mpi->chroma_height);
38 } else
39 mpi->planes[0]=av_malloc(mpi->bpp*mpi->width*(mpi->height+2)/8);
40 if (!mpi->planes[0])
41 abort(); //out of memory
42 if (mpi->flags&MP_IMGFLAG_PLANAR) {
43 int bpp = IMGFMT_IS_YUVP16(mpi->imgfmt)? 2 : 1;
44 // YV12/I420/YVU9/IF09. feel free to add other planar formats here...
45 mpi->stride[0]=mpi->stride[3]=bpp*mpi->width;
46 if(mpi->num_planes > 2){
47 mpi->stride[1]=mpi->stride[2]=bpp*mpi->chroma_width;
48 if(mpi->flags&MP_IMGFLAG_SWAPPED){
49 // I420/IYUV (Y,U,V)
50 mpi->planes[1]=mpi->planes[0]+mpi->stride[0]*mpi->height;
51 mpi->planes[2]=mpi->planes[1]+mpi->stride[1]*mpi->chroma_height;
52 if (mpi->num_planes > 3)
53 mpi->planes[3]=mpi->planes[2]+mpi->stride[2]*mpi->chroma_height;
54 } else {
55 // YV12,YVU9,IF09 (Y,V,U)
56 mpi->planes[2]=mpi->planes[0]+mpi->stride[0]*mpi->height;
57 mpi->planes[1]=mpi->planes[2]+mpi->stride[1]*mpi->chroma_height;
58 if (mpi->num_planes > 3)
59 mpi->planes[3]=mpi->planes[1]+mpi->stride[1]*mpi->chroma_height;
61 } else {
62 // NV12/NV21
63 mpi->stride[1]=mpi->chroma_width;
64 mpi->planes[1]=mpi->planes[0]+mpi->stride[0]*mpi->height;
66 } else {
67 mpi->stride[0]=mpi->width*mpi->bpp/8;
68 if (mpi->flags & MP_IMGFLAG_RGB_PALETTE)
69 mpi->planes[1] = av_malloc(1024);
71 mpi->flags|=MP_IMGFLAG_ALLOCATED;
74 mp_image_t* alloc_mpi(int w, int h, unsigned long int fmt) {
75 mp_image_t* mpi = new_mp_image(w,h);
77 mp_image_setfmt(mpi,fmt);
78 mp_image_alloc_planes(mpi);
80 return mpi;
83 void copy_mpi(mp_image_t *dmpi, mp_image_t *mpi) {
84 if(mpi->flags&MP_IMGFLAG_PLANAR){
85 memcpy_pic(dmpi->planes[0],mpi->planes[0], mpi->w, mpi->h,
86 dmpi->stride[0],mpi->stride[0]);
87 memcpy_pic(dmpi->planes[1],mpi->planes[1], mpi->chroma_width, mpi->chroma_height,
88 dmpi->stride[1],mpi->stride[1]);
89 memcpy_pic(dmpi->planes[2], mpi->planes[2], mpi->chroma_width, mpi->chroma_height,
90 dmpi->stride[2],mpi->stride[2]);
91 } else {
92 memcpy_pic(dmpi->planes[0],mpi->planes[0],
93 mpi->w*(dmpi->bpp/8), mpi->h,
94 dmpi->stride[0],mpi->stride[0]);
98 void mp_image_setfmt(mp_image_t* mpi,unsigned int out_fmt){
99 mpi->flags&=~(MP_IMGFLAG_PLANAR|MP_IMGFLAG_YUV|MP_IMGFLAG_SWAPPED);
100 mpi->imgfmt=out_fmt;
101 // compressed formats
102 if(out_fmt == IMGFMT_MPEGPES || IMGFMT_IS_HWACCEL(out_fmt)){
103 mpi->bpp=0;
104 return;
106 mpi->num_planes=1;
107 if (IMGFMT_IS_RGB(out_fmt)) {
108 if (IMGFMT_RGB_DEPTH(out_fmt) < 8 && !(out_fmt&128))
109 mpi->bpp = IMGFMT_RGB_DEPTH(out_fmt);
110 else
111 mpi->bpp=(IMGFMT_RGB_DEPTH(out_fmt)+7)&(~7);
112 return;
114 if (IMGFMT_IS_BGR(out_fmt)) {
115 if (IMGFMT_BGR_DEPTH(out_fmt) < 8 && !(out_fmt&128))
116 mpi->bpp = IMGFMT_BGR_DEPTH(out_fmt);
117 else
118 mpi->bpp=(IMGFMT_BGR_DEPTH(out_fmt)+7)&(~7);
119 mpi->flags|=MP_IMGFLAG_SWAPPED;
120 return;
122 mpi->flags|=MP_IMGFLAG_YUV;
123 mpi->num_planes=3;
124 if (mp_get_chroma_shift(out_fmt, NULL, NULL, NULL)) {
125 mpi->flags|=MP_IMGFLAG_PLANAR;
126 mpi->bpp = mp_get_chroma_shift(out_fmt, &mpi->chroma_x_shift, &mpi->chroma_y_shift, NULL);
127 mpi->chroma_width = mpi->width >> mpi->chroma_x_shift;
128 mpi->chroma_height = mpi->height >> mpi->chroma_y_shift;
130 switch(out_fmt){
131 case IMGFMT_I420:
132 case IMGFMT_IYUV:
133 mpi->flags|=MP_IMGFLAG_SWAPPED;
134 case IMGFMT_YV12:
135 return;
136 case IMGFMT_420A:
137 case IMGFMT_IF09:
138 mpi->num_planes=4;
139 case IMGFMT_YVU9:
140 case IMGFMT_444P:
141 case IMGFMT_422P:
142 case IMGFMT_411P:
143 case IMGFMT_440P:
144 case IMGFMT_444P16_LE:
145 case IMGFMT_444P16_BE:
146 case IMGFMT_444P10_LE:
147 case IMGFMT_444P10_BE:
148 case IMGFMT_444P9_LE:
149 case IMGFMT_444P9_BE:
150 case IMGFMT_422P16_LE:
151 case IMGFMT_422P16_BE:
152 case IMGFMT_422P10_LE:
153 case IMGFMT_422P10_BE:
154 case IMGFMT_420P16_LE:
155 case IMGFMT_420P16_BE:
156 case IMGFMT_420P10_LE:
157 case IMGFMT_420P10_BE:
158 case IMGFMT_420P9_LE:
159 case IMGFMT_420P9_BE:
160 return;
161 case IMGFMT_Y800:
162 case IMGFMT_Y8:
163 /* they're planar ones, but for easier handling use them as packed */
164 mpi->flags&=~MP_IMGFLAG_PLANAR;
165 mpi->num_planes=1;
166 return;
167 case IMGFMT_UYVY:
168 mpi->flags|=MP_IMGFLAG_SWAPPED;
169 case IMGFMT_YUY2:
170 mpi->bpp=16;
171 mpi->num_planes=1;
172 return;
173 case IMGFMT_NV12:
174 mpi->flags|=MP_IMGFLAG_SWAPPED;
175 case IMGFMT_NV21:
176 mpi->flags|=MP_IMGFLAG_PLANAR;
177 mpi->bpp=12;
178 mpi->num_planes=2;
179 mpi->chroma_width=(mpi->width>>0);
180 mpi->chroma_height=(mpi->height>>1);
181 mpi->chroma_x_shift=0;
182 mpi->chroma_y_shift=1;
183 return;
185 mp_msg(MSGT_DECVIDEO,MSGL_WARN,"mp_image: unknown out_fmt: 0x%X\n",out_fmt);
186 mpi->bpp=0;
189 static int mp_image_destructor(void *ptr)
191 mp_image_t *mpi = ptr;
193 if(mpi->flags&MP_IMGFLAG_ALLOCATED){
194 /* because we allocate the whole image at once */
195 av_free(mpi->planes[0]);
196 if (mpi->flags & MP_IMGFLAG_RGB_PALETTE)
197 av_free(mpi->planes[1]);
200 return 0;
203 mp_image_t* new_mp_image(int w,int h){
204 mp_image_t* mpi = talloc_zero(NULL, mp_image_t);
205 talloc_set_destructor(mpi, mp_image_destructor);
206 mpi->width=mpi->w=w;
207 mpi->height=mpi->h=h;
208 return mpi;
211 void free_mp_image(mp_image_t* mpi){
212 talloc_free(mpi);