Go to most recent revision | Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
6144 | serge | 1 | #include |
2 | |||
3 | #include |
||
4 | #include |
||
5 | #include |
||
6 | |||
7 | #include |
||
8 | #include |
||
9 | #include |
||
10 | #include |
||
11 | #include |
||
12 | #include "winlib/winlib.h" |
||
13 | |||
14 | #include "sound.h" |
||
15 | #include "fplay.h" |
||
16 | |||
17 | static struct decoder* init_ffmpeg_decoder(vst_t *vst) |
||
18 | { |
||
19 | AVCodecContext *vCtx = vst->vCtx; |
||
20 | struct decoder *decoder; |
||
21 | vframe_t *vframe; |
||
22 | int i, ret; |
||
23 | |||
24 | decoder = calloc(1, sizeof(struct decoder)); |
||
25 | if(decoder == NULL) |
||
26 | return NULL; |
||
27 | |||
28 | decoder->Frame = av_frame_alloc(); |
||
29 | if(decoder->Frame == NULL) |
||
30 | goto err_0; |
||
31 | |||
32 | decoder->nframes = 4; |
||
33 | |||
34 | for(i = 0; i < decoder->nframes; i++) |
||
35 | { |
||
36 | vframe = &decoder->vframes[i]; |
||
37 | |||
38 | ret = avpicture_alloc(&vframe->picture, vCtx->pix_fmt, |
||
39 | vCtx->width, vCtx->height); |
||
40 | if ( ret != 0 ) |
||
41 | goto err_1; |
||
42 | |||
43 | vframe->format = vCtx->pix_fmt; |
||
44 | vframe->index = i; |
||
45 | vframe->pts = 0; |
||
46 | vframe->ready = 0; |
||
47 | list_add_tail(&vframe->list, &vst->input_list); |
||
48 | }; |
||
49 | |||
50 | if(avcodec_open2(vCtx, vst->vCodec, NULL) < 0) |
||
51 | { |
||
52 | printf("Error while opening codec for input stream %d\n", |
||
53 | vst->vStream); |
||
54 | goto err_1; |
||
55 | }; |
||
56 | |||
57 | decoder->name = vst->vCodec->name; |
||
58 | decoder->pix_fmt = vCtx->pix_fmt; |
||
59 | decoder->width = vCtx->width; |
||
60 | decoder->height = vCtx->height; |
||
61 | decoder->codec_id = vCtx->codec_id; |
||
62 | |||
63 | return decoder; |
||
64 | |||
65 | err_1: |
||
66 | for(i = i-1; i >= 0; i--) |
||
67 | { |
||
68 | vframe = &decoder->vframes[i]; |
||
69 | avpicture_free(&vframe->picture); |
||
70 | }; |
||
71 | av_frame_free(&decoder->Frame); |
||
72 | err_0: |
||
73 | free(decoder); |
||
74 | return NULL; |
||
75 | } |
||
76 | |||
77 | int init_video_decoder(vst_t *vst) |
||
78 | { |
||
79 | AVCodecContext *vCtx = vst->vCtx; |
||
80 | |||
81 | vst->vCodec = avcodec_find_decoder(vCtx->codec_id); |
||
82 | |||
83 | if(vst->vCodec == NULL) |
||
84 | { |
||
85 | printf("Unsupported codec with id %d for input stream %d\n", |
||
86 | vst->vCtx->codec_id, vst->vStream); |
||
87 | return -1; |
||
88 | } |
||
89 | |||
90 | vst->decoder = va_init_decoder(vst); |
||
91 | if(vst->decoder == NULL) |
||
92 | vst->decoder = init_ffmpeg_decoder(vst); |
||
93 | |||
94 | if(vst->decoder != NULL) |
||
95 | { |
||
96 | printf("%dx%d %s %s%s decoder\n", |
||
97 | vst->decoder->width, vst->decoder->height, |
||
98 | av_get_pix_fmt_name(vst->decoder->pix_fmt), |
||
99 | vst->decoder->is_hw == 0 ? "ffmpeg ":"vaapi ", |
||
100 | vst->decoder->name); |
||
101 | |||
102 | return 0; |
||
103 | }; |
||
104 | |||
105 | return -1; |
||
106 | } |
||
107 | |||
108 | void fini_video_decoder(vst_t *vst) |
||
109 | { |
||
110 | avcodec_close(vst->vCtx); |
||
111 | |||
112 | if(vst->decoder->is_hw != 0) |
||
113 | return; |
||
114 | |||
115 | for(int i = 0; i < vst->decoder->nframes; i++) |
||
116 | { |
||
117 | vframe_t *vframe; |
||
118 | vframe = &vst->decoder->vframes[i]; |
||
119 | avpicture_free(&vframe->picture); |
||
120 | }; |
||
121 | av_frame_free(&vst->decoder->Frame); |
||
122 | free(vst->decoder); |
||
123 | }; |
||
124 | |||
125 | static vframe_t *get_input_frame(vst_t *vst) |
||
126 | { |
||
127 | vframe_t *vframe = NULL; |
||
128 | |||
129 | mutex_lock(&vst->input_lock); |
||
130 | if(!list_empty(&vst->input_list)) |
||
131 | { |
||
132 | vframe = list_first_entry(&vst->input_list, vframe_t, list); |
||
133 | list_del(&vframe->list); |
||
134 | } |
||
135 | mutex_unlock(&vst->input_lock); |
||
136 | |||
137 | return vframe; |
||
138 | } |
||
139 | |||
140 | static void put_output_frame(vst_t *vst, vframe_t *vframe) |
||
141 | { |
||
142 | mutex_lock(&vst->output_lock); |
||
143 | if(list_empty(&vst->output_list)) |
||
144 | list_add_tail(&vframe->list, &vst->output_list); |
||
145 | else |
||
146 | { |
||
147 | vframe_t *cur; |
||
148 | |||
149 | cur = list_first_entry(&vst->output_list,vframe_t,list); |
||
150 | if(vframe->pts < cur->pts) |
||
151 | list_add_tail(&vframe->list, &vst->output_list); |
||
152 | else |
||
153 | { |
||
154 | list_for_each_entry_reverse(cur,&vst->output_list,list) |
||
155 | { |
||
156 | if(vframe->pts > cur->pts) |
||
157 | { |
||
158 | list_add(&vframe->list, &cur->list); |
||
159 | break; |
||
160 | }; |
||
161 | }; |
||
162 | }; |
||
163 | }; |
||
164 | vst->frames_count++; |
||
165 | mutex_unlock(&vst->output_lock); |
||
166 | }; |
||
167 | |||
168 | int decode_video(vst_t* vst) |
||
169 | { |
||
170 | struct decoder* decoder = vst->decoder; |
||
171 | double pts; |
||
172 | AVPacket pkt; |
||
173 | |||
174 | int frameFinished; |
||
175 | |||
176 | if(decoder->active_frame == NULL) |
||
177 | decoder->active_frame = get_input_frame(vst); |
||
178 | |||
179 | if(decoder->active_frame == NULL) |
||
180 | return -1; |
||
181 | |||
182 | if( get_packet(&vst->q_video, &pkt) == 0 ) |
||
183 | return 0; |
||
184 | |||
185 | frameFinished = 0; |
||
186 | |||
187 | mutex_lock(&vst->gpu_lock); |
||
188 | |||
189 | if(avcodec_decode_video2(vst->vCtx, decoder->Frame, &frameFinished, &pkt) <= 0) |
||
190 | printf("video decoder error\n"); |
||
191 | |||
192 | if(frameFinished) |
||
193 | { |
||
194 | vframe_t *vframe = decoder->active_frame; |
||
195 | AVPicture *dst_pic; |
||
196 | |||
197 | if(decoder->is_hw) |
||
198 | pts = pkt.pts; |
||
199 | else |
||
200 | pts = av_frame_get_best_effort_timestamp(decoder->Frame); |
||
201 | |||
202 | pts*= av_q2d(vst->video_time_base); |
||
203 | |||
204 | dst_pic = &vframe->picture; |
||
205 | |||
206 | if(vframe->is_hw_pic == 0) |
||
207 | av_image_copy(dst_pic->data, dst_pic->linesize, |
||
208 | (const uint8_t**)decoder->Frame->data, |
||
209 | decoder->Frame->linesize, vst->vCtx->pix_fmt, vst->vCtx->width, vst->vCtx->height); |
||
210 | else |
||
211 | va_create_planar(vst, vframe); |
||
212 | |||
213 | vframe->pts = pts*1000.0; |
||
214 | vframe->pkt_pts = pkt.pts*av_q2d(vst->video_time_base)*1000.0; |
||
215 | vframe->ready = 1; |
||
216 | |||
217 | put_output_frame(vst, vframe); |
||
218 | |||
219 | // printf("decoded index: %d pts: %f pkt_pts %f pkt_dts %f\n", |
||
220 | // vst->dfx, vst->vframe[vst->dfx].pts, |
||
221 | // vst->vframe[vst->dfx].pkt_pts, vst->vframe[vst->dfx].pkt_dts); |
||
222 | |||
223 | decoder->active_frame = NULL; |
||
224 | }; |
||
225 | av_frame_unref(decoder->Frame); |
||
226 | mutex_unlock(&vst->gpu_lock); |
||
227 | |||
228 | av_free_packet(&pkt); |
||
229 | |||
230 | return 1; |
||
231 | }=>>>>> |
||
232 |