Mercurial > libavcodec.hg
annotate libopenjpeg.c @ 9477:94edfc2a9b8b libavcodec
Remove a pointless right-shift in xan decoder.
author | reimar |
---|---|
date | Fri, 17 Apr 2009 17:54:55 +0000 |
parents | 54bc8a2727b0 |
children | 5df11d49abb7 |
rev | line source |
---|---|
8747 | 1 /* |
2 * JPEG 2000 decoding support via OpenJPEG | |
3 * Copyright (c) 2009 Jaikrishnan Menon <realityman@gmx.net> | |
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 | |
22 /** | |
23 * @file libavcodec/libopenjpeg.c | |
24 * JPEG 2000 decoder using libopenjpeg | |
25 */ | |
26 | |
27 #include "avcodec.h" | |
28 #include "libavutil/intreadwrite.h" | |
29 #define OPJ_STATIC | |
8787
9575568668c4
Use default system include path for an installed OpenJPEG library.
diego
parents:
8747
diff
changeset
|
30 #include <openjpeg.h> |
8747 | 31 |
32 #define JP2_SIG_TYPE 0x6A502020 | |
33 #define JP2_SIG_VALUE 0x0D0A870A | |
34 | |
35 typedef struct { | |
36 opj_dparameters_t dec_params; | |
37 AVFrame image; | |
38 } LibOpenJPEGContext; | |
39 | |
40 static int check_image_attributes(opj_image_t *image) | |
41 { | |
42 return(image->comps[0].dx == image->comps[1].dx && | |
43 image->comps[1].dx == image->comps[2].dx && | |
44 image->comps[0].dy == image->comps[1].dy && | |
45 image->comps[1].dy == image->comps[2].dy && | |
46 image->comps[0].prec == image->comps[1].prec && | |
47 image->comps[1].prec == image->comps[2].prec); | |
48 } | |
49 | |
50 static av_cold int libopenjpeg_decode_init(AVCodecContext *avctx) | |
51 { | |
52 LibOpenJPEGContext *ctx = avctx->priv_data; | |
53 | |
54 opj_set_default_decoder_parameters(&ctx->dec_params); | |
55 avctx->coded_frame = &ctx->image; | |
56 return 0; | |
57 } | |
58 | |
59 static int libopenjpeg_decode_frame(AVCodecContext *avctx, | |
60 void *data, int *data_size, | |
9355
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
8787
diff
changeset
|
61 AVPacket *avpkt) |
8747 | 62 { |
9355
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
8787
diff
changeset
|
63 const uint8_t *buf = avpkt->data; |
54bc8a2727b0
Implement avcodec_decode_video2(), _audio3() and _subtitle2() which takes an
rbultje
parents:
8787
diff
changeset
|
64 int buf_size = avpkt->size; |
8747 | 65 LibOpenJPEGContext *ctx = avctx->priv_data; |
66 AVFrame *picture = &ctx->image, *output = data; | |
67 opj_dinfo_t *dec; | |
68 opj_cio_t *stream; | |
69 opj_image_t *image; | |
70 int width, height, has_alpha = 0, ret = -1; | |
71 int x, y, index; | |
72 uint8_t *img_ptr; | |
73 int adjust[4]; | |
74 | |
75 *data_size = 0; | |
76 | |
77 // Check if input is a raw jpeg2k codestream or in jp2 wrapping | |
78 if((AV_RB32(buf) == 12) && | |
79 (AV_RB32(buf + 4) == JP2_SIG_TYPE) && | |
80 (AV_RB32(buf + 8) == JP2_SIG_VALUE)) { | |
81 dec = opj_create_decompress(CODEC_JP2); | |
82 } else { | |
83 dec = opj_create_decompress(CODEC_J2K); | |
84 } | |
85 | |
86 if(!dec) { | |
87 av_log(avctx, AV_LOG_ERROR, "Error initializing decoder.\n"); | |
88 return -1; | |
89 } | |
90 opj_set_event_mgr((opj_common_ptr)dec, NULL, NULL); | |
91 | |
92 // Tie decoder with decoding parameters | |
93 opj_setup_decoder(dec, &ctx->dec_params); | |
94 stream = opj_cio_open((opj_common_ptr)dec, buf, buf_size); | |
95 if(!stream) { | |
96 av_log(avctx, AV_LOG_ERROR, "Codestream could not be opened for reading.\n"); | |
97 opj_destroy_decompress(dec); | |
98 return -1; | |
99 } | |
100 | |
101 // Decode the codestream | |
102 image = opj_decode_with_info(dec, stream, NULL); | |
103 opj_cio_close(stream); | |
104 if(!image) { | |
105 av_log(avctx, AV_LOG_ERROR, "Error decoding codestream.\n"); | |
106 opj_destroy_decompress(dec); | |
107 return -1; | |
108 } | |
109 width = image->comps[0].w; | |
110 height = image->comps[0].h; | |
111 if(avcodec_check_dimensions(avctx, width, height) < 0) { | |
112 av_log(avctx, AV_LOG_ERROR, "%dx%d dimension invalid.\n", width, height); | |
113 goto done; | |
114 } | |
115 avcodec_set_dimensions(avctx, width, height); | |
116 | |
117 switch(image->numcomps) | |
118 { | |
119 case 1: avctx->pix_fmt = PIX_FMT_GRAY8; | |
120 break; | |
121 case 3: if(check_image_attributes(image)) { | |
122 avctx->pix_fmt = PIX_FMT_RGB24; | |
123 } else { | |
124 avctx->pix_fmt = PIX_FMT_GRAY8; | |
125 av_log(avctx, AV_LOG_ERROR, "Only first component will be used.\n"); | |
126 } | |
127 break; | |
128 case 4: has_alpha = 1; | |
129 avctx->pix_fmt = PIX_FMT_RGB32; | |
130 break; | |
131 default: av_log(avctx, AV_LOG_ERROR, "%d components unsupported.\n", image->numcomps); | |
132 goto done; | |
133 } | |
134 | |
135 if(picture->data[0]) | |
136 avctx->release_buffer(avctx, picture); | |
137 | |
138 if(avctx->get_buffer(avctx, picture) < 0) { | |
139 av_log(avctx, AV_LOG_ERROR, "Couldn't allocate image buffer.\n"); | |
140 return -1; | |
141 } | |
142 | |
143 for(x = 0; x < image->numcomps; x++) { | |
144 adjust[x] = FFMAX(image->comps[x].prec - 8, 0); | |
145 } | |
146 | |
147 for(y = 0; y < height; y++) { | |
148 index = y*width; | |
149 img_ptr = picture->data[0] + y*picture->linesize[0]; | |
150 for(x = 0; x < width; x++, index++) { | |
151 *img_ptr++ = image->comps[0].data[index] >> adjust[0]; | |
152 if(image->numcomps > 2 && check_image_attributes(image)) { | |
153 *img_ptr++ = image->comps[1].data[index] >> adjust[1]; | |
154 *img_ptr++ = image->comps[2].data[index] >> adjust[2]; | |
155 if(has_alpha) | |
156 *img_ptr++ = image->comps[3].data[index] >> adjust[3]; | |
157 } | |
158 } | |
159 } | |
160 | |
161 *output = ctx->image; | |
162 *data_size = sizeof(AVPicture); | |
163 ret = buf_size; | |
164 | |
165 done: | |
166 opj_image_destroy(image); | |
167 opj_destroy_decompress(dec); | |
168 return ret; | |
169 } | |
170 | |
171 static av_cold int libopenjpeg_decode_close(AVCodecContext *avctx) | |
172 { | |
173 LibOpenJPEGContext *ctx = avctx->priv_data; | |
174 | |
175 if(ctx->image.data[0]) | |
176 avctx->release_buffer(avctx, &ctx->image); | |
177 return 0 ; | |
178 } | |
179 | |
180 | |
181 AVCodec libopenjpeg_decoder = { | |
182 "libopenjpeg", | |
183 CODEC_TYPE_VIDEO, | |
184 CODEC_ID_JPEG2000, | |
185 sizeof(LibOpenJPEGContext), | |
186 libopenjpeg_decode_init, | |
187 NULL, | |
188 libopenjpeg_decode_close, | |
189 libopenjpeg_decode_frame, | |
190 NULL, | |
191 .long_name = NULL_IF_CONFIG_SMALL("OpenJPEG based JPEG 2000 decoder"), | |
192 } ; |