Change license headers to say 'FFmpeg' instead of 'this program/this library'
[ffmpeg.git] / libavformat / yuv.c
1 /*
2  * .Y.U.V image format
3  * Copyright (c) 2003 Fabrice Bellard.
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 #include "avformat.h"
22
23 static int sizes[][2] = {
24     { 640, 480 },
25     { 720, 480 },
26     { 720, 576 },
27     { 352, 288 },
28     { 352, 240 },
29     { 160, 128 },
30     { 512, 384 },
31     { 640, 352 },
32     { 640, 240 },
33 };
34
35 static int infer_size(int *width_ptr, int *height_ptr, int size)
36 {
37     int i;
38
39     for(i=0;i<sizeof(sizes)/sizeof(sizes[0]);i++) {
40         if ((sizes[i][0] * sizes[i][1]) == size) {
41             *width_ptr = sizes[i][0];
42             *height_ptr = sizes[i][1];
43             return 0;
44         }
45     }
46     return -1;
47 }
48
49 static int yuv_read(ByteIOContext *f,
50                     int (*alloc_cb)(void *opaque, AVImageInfo *info), void *opaque)
51 {
52     ByteIOContext pb1, *pb = &pb1;
53     int img_size, ret;
54     char fname[1024], *p;
55     int size;
56     URLContext *h;
57     AVImageInfo info1, *info = &info1;
58
59     img_size = url_fsize(f);
60
61     /* XXX: hack hack */
62     h = url_fileno(f);
63     url_get_filename(h, fname, sizeof(fname));
64
65     if (infer_size(&info->width, &info->height, img_size) < 0) {
66         return AVERROR_IO;
67     }
68     info->pix_fmt = PIX_FMT_YUV420P;
69
70     ret = alloc_cb(opaque, info);
71     if (ret)
72         return ret;
73
74     size = info->width * info->height;
75
76     p = strrchr(fname, '.');
77     if (!p || p[1] != 'Y')
78         return AVERROR_IO;
79
80     get_buffer(f, info->pict.data[0], size);
81
82     p[1] = 'U';
83     if (url_fopen(pb, fname, URL_RDONLY) < 0)
84         return AVERROR_IO;
85
86     get_buffer(pb, info->pict.data[1], size / 4);
87     url_fclose(pb);
88
89     p[1] = 'V';
90     if (url_fopen(pb, fname, URL_RDONLY) < 0)
91         return AVERROR_IO;
92
93     get_buffer(pb, info->pict.data[2], size / 4);
94     url_fclose(pb);
95     return 0;
96 }
97
98 static int yuv_write(ByteIOContext *pb2, AVImageInfo *info)
99 {
100     ByteIOContext pb1, *pb;
101     char fname[1024], *p;
102     int i, j, width, height;
103     uint8_t *ptr;
104     URLContext *h;
105     static const char *ext = "YUV";
106
107     /* XXX: hack hack */
108     h = url_fileno(pb2);
109     url_get_filename(h, fname, sizeof(fname));
110
111     p = strrchr(fname, '.');
112     if (!p || p[1] != 'Y')
113         return AVERROR_IO;
114
115     width = info->width;
116     height = info->height;
117
118     for(i=0;i<3;i++) {
119         if (i == 1) {
120             width >>= 1;
121             height >>= 1;
122         }
123
124         if (i >= 1) {
125             pb = &pb1;
126             p[1] = ext[i];
127             if (url_fopen(pb, fname, URL_WRONLY) < 0)
128                 return AVERROR_IO;
129         } else {
130             pb = pb2;
131         }
132
133         ptr = info->pict.data[i];
134         for(j=0;j<height;j++) {
135             put_buffer(pb, ptr, width);
136             ptr += info->pict.linesize[i];
137         }
138         put_flush_packet(pb);
139         if (i >= 1) {
140             url_fclose(pb);
141         }
142     }
143     return 0;
144 }
145
146 static int yuv_probe(AVProbeData *pd)
147 {
148     if (match_ext(pd->filename, "Y"))
149         return AVPROBE_SCORE_MAX;
150     else
151         return 0;
152 }
153
154 AVImageFormat yuv_image_format = {
155     "yuv",
156     "Y",
157     yuv_probe,
158     yuv_read,
159     (1 << PIX_FMT_YUV420P),
160     yuv_write,
161 };