Rev 6118 | Rev 6135 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 6117 | serge | 1 | #include |
| 2 | #include |
||
| 3 | #include |
||
| 4 | #include |
||
| 5 | #include |
||
| 6 | #include |
||
| 7 | #include |
||
| 8 | #include |
||
| 9 | #include |
||
| 10 | #include |
||
| 11 | #include |
||
| 12 | #include |
||
| 13 | #include "winlib/winlib.h" |
||
| 14 | #include "fplay.h" |
||
| 15 | |||
| 16 | struct hw_profile |
||
| 17 | { |
||
| 18 | enum AVCodecID av_codec; |
||
| 19 | int ff_profile; |
||
| 20 | uint64_t va_profile; |
||
| 21 | }; |
||
| 22 | |||
| 23 | |||
| 24 | #define ENTER() printf("enter %s\n",__FUNCTION__) |
||
| 25 | #define LEAVE() printf("leave %s\n",__FUNCTION__) |
||
| 26 | #define FAIL() printf("fail %s\n",__FUNCTION__) |
||
| 27 | |||
| 28 | |||
| 29 | #if DEBUG |
||
| 30 | # define D(x) x |
||
| 31 | # define bug printf |
||
| 32 | #else |
||
| 33 | # define D(x) |
||
| 34 | #endif |
||
| 35 | |||
| 36 | #undef ARRAY_ELEMS |
||
| 37 | #define ARRAY_ELEMS(a) (sizeof(a) / sizeof((a)[0])) |
||
| 38 | |||
| 39 | static int drm_fd = 0; |
||
| 40 | static struct vaapi_context *v_context; |
||
| 41 | |||
| 42 | static VASurfaceID v_surface_id[4]; |
||
| 43 | |||
| 44 | #define HAS_HEVC VA_CHECK_VERSION(0, 38, 0) |
||
| 45 | #define HAS_VP9 (VA_CHECK_VERSION(0, 38, 1) && defined(FF_PROFILE_VP9_0)) |
||
| 46 | |||
| 47 | #define PE(av_codec_id, ff_profile, vdp_profile) \ |
||
| 48 | {AV_CODEC_ID_ ## av_codec_id, FF_PROFILE_ ## ff_profile, \ |
||
| 49 | VAProfile ## vdp_profile} |
||
| 50 | |||
| 51 | static const struct hw_profile profiles[] = { |
||
| 52 | PE(MPEG2VIDEO, MPEG2_MAIN, MPEG2Main), |
||
| 53 | PE(MPEG2VIDEO, MPEG2_SIMPLE, MPEG2Simple), |
||
| 54 | PE(MPEG4, MPEG4_ADVANCED_SIMPLE, MPEG4AdvancedSimple), |
||
| 55 | PE(MPEG4, MPEG4_MAIN, MPEG4Main), |
||
| 56 | PE(MPEG4, MPEG4_SIMPLE, MPEG4Simple), |
||
| 57 | PE(H264, H264_HIGH, H264High), |
||
| 58 | PE(H264, H264_MAIN, H264Main), |
||
| 59 | PE(H264, H264_BASELINE, H264Baseline), |
||
| 60 | PE(VC1, VC1_ADVANCED, VC1Advanced), |
||
| 61 | PE(VC1, VC1_MAIN, VC1Main), |
||
| 62 | PE(VC1, VC1_SIMPLE, VC1Simple), |
||
| 63 | PE(WMV3, VC1_ADVANCED, VC1Advanced), |
||
| 64 | PE(WMV3, VC1_MAIN, VC1Main), |
||
| 65 | PE(WMV3, VC1_SIMPLE, VC1Simple), |
||
| 66 | #if HAS_HEVC |
||
| 67 | PE(HEVC, HEVC_MAIN, HEVCMain), |
||
| 68 | PE(HEVC, HEVC_MAIN_10, HEVCMain10), |
||
| 69 | #endif |
||
| 70 | #if HAS_VP9 |
||
| 71 | PE(VP9, VP9_0, VP9Profile0), |
||
| 72 | #endif |
||
| 73 | {0} |
||
| 74 | }; |
||
| 75 | |||
| 76 | int va_check_codec_support(enum AVCodecID id) |
||
| 77 | { |
||
| 78 | for (int n = 0; profiles[n].av_codec; n++) { |
||
| 79 | if (profiles[n].av_codec == id) |
||
| 80 | return 1; |
||
| 81 | } |
||
| 82 | return 0; |
||
| 83 | } |
||
| 84 | |||
| 85 | static int vaapi_check_status(VAStatus status, const char *msg) |
||
| 86 | { |
||
| 87 | if (status != VA_STATUS_SUCCESS) { |
||
| 88 | fprintf(stderr, "[%s] %s: %s\n", PACKAGE_NAME, msg, vaErrorStr(status)); |
||
| 89 | return 0; |
||
| 90 | } |
||
| 91 | return 1; |
||
| 92 | }; |
||
| 93 | |||
| 94 | static const char *string_of_VADisplayAttribType(VADisplayAttribType type) |
||
| 95 | { |
||
| 96 | switch (type) { |
||
| 97 | #define TYPE(type) \ |
||
| 98 | case VADisplayAttrib##type: return "VADisplayAttrib" #type |
||
| 99 | TYPE(Brightness); |
||
| 100 | TYPE(Contrast); |
||
| 101 | TYPE(Hue); |
||
| 102 | TYPE(Saturation); |
||
| 103 | TYPE(BackgroundColor); |
||
| 104 | #if !VA_CHECK_VERSION(0,34,0) |
||
| 105 | TYPE(DirectSurface); |
||
| 106 | #endif |
||
| 107 | #if VA_CHECK_VERSION(0,32,0) |
||
| 108 | TYPE(Rotation); |
||
| 109 | #endif |
||
| 110 | #undef TYPE |
||
| 111 | default: break; |
||
| 112 | } |
||
| 113 | return " |
||
| 114 | } |
||
| 115 | |||
| 116 | static const char *string_of_VAProfile(VAProfile profile) |
||
| 117 | { |
||
| 118 | switch (profile) { |
||
| 119 | #define PROFILE(profile) \ |
||
| 120 | case VAProfile##profile: return "VAProfile" #profile |
||
| 121 | PROFILE(MPEG2Simple); |
||
| 122 | PROFILE(MPEG2Main); |
||
| 123 | PROFILE(MPEG4Simple); |
||
| 124 | PROFILE(MPEG4AdvancedSimple); |
||
| 125 | PROFILE(MPEG4Main); |
||
| 126 | #if VA_CHECK_VERSION(0,32,0) |
||
| 127 | PROFILE(JPEGBaseline); |
||
| 128 | PROFILE(H263Baseline); |
||
| 129 | PROFILE(H264ConstrainedBaseline); |
||
| 130 | #endif |
||
| 131 | PROFILE(H264Baseline); |
||
| 132 | PROFILE(H264Main); |
||
| 133 | PROFILE(H264High); |
||
| 134 | PROFILE(VC1Simple); |
||
| 135 | PROFILE(VC1Main); |
||
| 136 | PROFILE(VC1Advanced); |
||
| 137 | #undef PROFILE |
||
| 138 | default: break; |
||
| 139 | } |
||
| 140 | return " |
||
| 141 | } |
||
| 142 | |||
| 143 | static const char *string_of_VAEntrypoint(VAEntrypoint entrypoint) |
||
| 144 | { |
||
| 145 | switch (entrypoint) { |
||
| 146 | #define ENTRYPOINT(entrypoint) \ |
||
| 147 | case VAEntrypoint##entrypoint: return "VAEntrypoint" #entrypoint |
||
| 148 | ENTRYPOINT(VLD); |
||
| 149 | ENTRYPOINT(IZZ); |
||
| 150 | ENTRYPOINT(IDCT); |
||
| 151 | ENTRYPOINT(MoComp); |
||
| 152 | ENTRYPOINT(Deblocking); |
||
| 153 | #if VA_CHECK_VERSION(0,32,0) |
||
| 154 | ENTRYPOINT(EncSlice); |
||
| 155 | ENTRYPOINT(EncPicture); |
||
| 156 | #endif |
||
| 157 | #undef ENTRYPOINT |
||
| 158 | default: break; |
||
| 159 | } |
||
| 160 | return " |
||
| 161 | } |
||
| 162 | |||
| 163 | VADisplay va_open_display(void) |
||
| 164 | { |
||
| 165 | VADisplay va_dpy; |
||
| 166 | |||
| 167 | drm_fd = get_service("DISPLAY"); |
||
| 168 | if (drm_fd == 0) |
||
| 169 | return NULL; |
||
| 170 | |||
| 171 | va_dpy = vaGetDisplayDRM(drm_fd); |
||
| 172 | if (va_dpy) |
||
| 173 | return va_dpy; |
||
| 174 | |||
| 175 | drm_fd = 0; |
||
| 176 | return NULL; |
||
| 177 | }; |
||
| 178 | |||
| 6121 | serge | 179 | void *vaapi_init(VADisplay display) |
| 6117 | serge | 180 | { |
| 181 | struct vaapi_context *vaapi; |
||
| 182 | int major_version, minor_version; |
||
| 183 | int i, num_display_attrs, max_display_attrs; |
||
| 184 | VADisplayAttribute *display_attrs = NULL; |
||
| 185 | VAStatus status; |
||
| 186 | |||
| 187 | if (v_context) |
||
| 188 | return 0; |
||
| 189 | |||
| 190 | if (!display) |
||
| 191 | goto error; |
||
| 192 | D(bug("VA display %p\n", display)); |
||
| 193 | |||
| 194 | status = vaInitialize(display, &major_version, &minor_version); |
||
| 195 | if (!vaapi_check_status(status, "vaInitialize()")) |
||
| 196 | goto error; |
||
| 197 | D(bug("VA API version %d.%d\n", major_version, minor_version)); |
||
| 198 | |||
| 199 | max_display_attrs = vaMaxNumDisplayAttributes(display); |
||
| 200 | display_attrs = malloc(max_display_attrs * sizeof(display_attrs[0])); |
||
| 201 | if (!display_attrs) |
||
| 202 | goto error; |
||
| 203 | |||
| 204 | num_display_attrs = 0; /* XXX: workaround old GMA500 bug */ |
||
| 205 | status = vaQueryDisplayAttributes(display, display_attrs, &num_display_attrs); |
||
| 206 | if (!vaapi_check_status(status, "vaQueryDisplayAttributes()")) |
||
| 207 | goto error; |
||
| 208 | D(bug("%d display attributes available\n", num_display_attrs)); |
||
| 209 | for (i = 0; i < num_display_attrs; i++) { |
||
| 210 | VADisplayAttribute * const display_attr = &display_attrs[i]; |
||
| 211 | D(bug(" %-32s (%s/%s) min %d max %d value 0x%x\n", |
||
| 212 | string_of_VADisplayAttribType(display_attr->type), |
||
| 213 | (display_attr->flags & VA_DISPLAY_ATTRIB_GETTABLE) ? "get" : "---", |
||
| 214 | (display_attr->flags & VA_DISPLAY_ATTRIB_SETTABLE) ? "set" : "---", |
||
| 215 | display_attr->min_value, |
||
| 216 | display_attr->max_value, |
||
| 217 | display_attr->value)); |
||
| 218 | } |
||
| 219 | |||
| 220 | if ((vaapi = calloc(1, sizeof(*vaapi))) == NULL) |
||
| 221 | goto error; |
||
| 222 | vaapi->display = display; |
||
| 223 | vaapi->config_id = VA_INVALID_ID; |
||
| 224 | vaapi->context_id = VA_INVALID_ID; |
||
| 225 | |||
| 226 | v_context = vaapi; |
||
| 227 | |||
| 6121 | serge | 228 | return vaapi; |
| 6117 | serge | 229 | |
| 230 | error: |
||
| 231 | free(display_attrs); |
||
| 6121 | serge | 232 | return NULL; |
| 6117 | serge | 233 | } |
| 234 | |||
| 235 | static int has_profile(struct vaapi_context *vaapi, VAProfile profile) |
||
| 236 | { |
||
| 237 | VAProfile *profiles; |
||
| 238 | int n_profiles; |
||
| 239 | VAStatus status; |
||
| 240 | int i; |
||
| 241 | |||
| 242 | profiles = calloc(vaMaxNumProfiles(vaapi->display), sizeof(profiles[0])); |
||
| 243 | |||
| 244 | status = vaQueryConfigProfiles(vaapi->display,profiles,&n_profiles); |
||
| 245 | |||
| 246 | if (!vaapi_check_status(status, "vaQueryConfigProfiles()")) |
||
| 247 | return 0; |
||
| 248 | |||
| 249 | D(bug("%d profiles available\n", n_profiles)); |
||
| 250 | |||
| 251 | for (i = 0; i < n_profiles; i++) |
||
| 252 | { |
||
| 253 | if (profiles[i] == profile) |
||
| 254 | return 1; |
||
| 255 | } |
||
| 256 | return 0; |
||
| 257 | } |
||
| 258 | |||
| 259 | static int has_entrypoint(struct vaapi_context *vaapi, VAProfile profile, VAEntrypoint entrypoint) |
||
| 260 | { |
||
| 261 | VAEntrypoint *entrypoints; |
||
| 262 | int n_entrypoints; |
||
| 263 | VAStatus status; |
||
| 264 | int i; |
||
| 265 | |||
| 266 | entrypoints = calloc(vaMaxNumEntrypoints(vaapi->display), sizeof(entrypoints[0])); |
||
| 267 | |||
| 268 | status = vaQueryConfigEntrypoints(vaapi->display, profile, |
||
| 269 | entrypoints, &n_entrypoints); |
||
| 270 | if (!vaapi_check_status(status, "vaQueryConfigEntrypoints()")) |
||
| 271 | return 0; |
||
| 272 | |||
| 273 | D(bug("%d entrypoints available for %s\n", n_entrypoints, |
||
| 274 | string_of_VAProfile(profile))); |
||
| 275 | |||
| 276 | for (i = 0; i < n_entrypoints; i++) |
||
| 277 | { |
||
| 278 | if (entrypoints[i] == entrypoint) |
||
| 279 | return 1; |
||
| 280 | } |
||
| 281 | return 0; |
||
| 282 | } |
||
| 283 | |||
| 284 | static int vaapi_init_decoder(VAProfile profile, |
||
| 285 | VAEntrypoint entrypoint, |
||
| 286 | unsigned int picture_width, |
||
| 287 | unsigned int picture_height) |
||
| 288 | { |
||
| 289 | struct vaapi_context* const vaapi = v_context; |
||
| 290 | VAConfigAttrib attrib; |
||
| 291 | VAConfigID config_id = VA_INVALID_ID; |
||
| 292 | VAContextID context_id = VA_INVALID_ID; |
||
| 293 | VAStatus status; |
||
| 294 | |||
| 295 | ENTER(); |
||
| 296 | if (!vaapi) |
||
| 297 | { |
||
| 298 | FAIL(); |
||
| 299 | return -1; |
||
| 300 | }; |
||
| 301 | |||
| 302 | if (!has_profile(vaapi, profile)) |
||
| 303 | { |
||
| 304 | FAIL(); |
||
| 305 | return -1; |
||
| 306 | }; |
||
| 307 | |||
| 308 | if (!has_entrypoint(vaapi, profile, entrypoint)) |
||
| 309 | { |
||
| 310 | FAIL(); |
||
| 311 | return -1; |
||
| 312 | }; |
||
| 313 | |||
| 314 | if (vaapi->config_id != VA_INVALID_ID) |
||
| 315 | vaDestroyConfig(vaapi->display, vaapi->config_id); |
||
| 316 | |||
| 317 | attrib.type = VAConfigAttribRTFormat; |
||
| 318 | |||
| 319 | printf("vaGetConfigAttributes\n"); |
||
| 320 | status = vaGetConfigAttributes(vaapi->display, profile, entrypoint, |
||
| 321 | &attrib, 1); |
||
| 322 | if (!vaapi_check_status(status, "vaGetConfigAttributes()")) |
||
| 323 | { |
||
| 324 | FAIL(); |
||
| 325 | return -1; |
||
| 326 | } |
||
| 327 | |||
| 328 | if ((attrib.value & VA_RT_FORMAT_YUV420) == 0) |
||
| 329 | { |
||
| 330 | printf("Chroma format not supported.\n"); |
||
| 331 | FAIL(); |
||
| 332 | return -1; |
||
| 333 | }; |
||
| 334 | |||
| 335 | printf("vaCreateConfig\n"); |
||
| 336 | status = vaCreateConfig(vaapi->display, profile, entrypoint, |
||
| 337 | &attrib, 1, &config_id); |
||
| 338 | if (!vaapi_check_status(status, "vaCreateConfig()")) |
||
| 339 | { |
||
| 340 | FAIL(); |
||
| 341 | return -1; |
||
| 342 | } |
||
| 343 | |||
| 344 | printf("vaCreateSurfaces %dx%d\n",picture_width,picture_height); |
||
| 345 | status = vaCreateSurfaces(vaapi->display, VA_RT_FORMAT_YUV420, picture_width, picture_height, |
||
| 346 | v_surface_id,4,NULL,0); |
||
| 347 | printf("v_surface_id_3 %x\n", v_surface_id[3]); |
||
| 348 | if (!vaapi_check_status(status, "vaCreateSurfaces()")) |
||
| 349 | { |
||
| 350 | FAIL(); |
||
| 351 | return -1; |
||
| 352 | }; |
||
| 353 | { |
||
| 354 | VAImage vaimage; |
||
| 355 | VABufferInfo info = {0}; |
||
| 356 | |||
| 357 | vaDeriveImage(vaapi->display,v_surface_id[0],&vaimage); |
||
| 358 | printf("vaDeriveImage: %x fourcc: %x\n" |
||
| 359 | "offset0: %d pitch0: %d\n" |
||
| 360 | "offset1: %d pitch1: %d\n" |
||
| 361 | "offset2: %d pitch2: %d\n", |
||
| 362 | vaimage.buf, vaimage.format.fourcc, |
||
| 363 | vaimage.offsets[0],vaimage.pitches[0], |
||
| 364 | vaimage.offsets[1],vaimage.pitches[1], |
||
| 365 | vaimage.offsets[2],vaimage.pitches[2]); |
||
| 366 | |||
| 367 | info.mem_type = VA_SURFACE_ATTRIB_MEM_TYPE_KERNEL_DRM; |
||
| 368 | vaAcquireBufferHandle(vaapi->display, vaimage.buf, &info); |
||
| 369 | printf("vaAcquireBufferHandle: %x type: %x\n" |
||
| 370 | "mem type: %x mem size: %x\n", |
||
| 371 | info.handle, info.type, info.mem_type, info.mem_size); |
||
| 372 | |||
| 373 | vaReleaseBufferHandle(vaapi->display, vaimage.buf); |
||
| 374 | |||
| 375 | vaDestroyImage(vaapi->display,vaimage.image_id); |
||
| 376 | }; |
||
| 377 | |||
| 378 | printf("vaCreateContext %dx%d\n",picture_width,picture_height); |
||
| 379 | status = vaCreateContext(vaapi->display, config_id, |
||
| 380 | picture_width, picture_height, |
||
| 381 | VA_PROGRESSIVE, |
||
| 382 | v_surface_id, 4, |
||
| 383 | &context_id); |
||
| 384 | if (!vaapi_check_status(status, "vaCreateContext()")) |
||
| 385 | { |
||
| 386 | FAIL(); |
||
| 387 | return -1; |
||
| 388 | }; |
||
| 389 | |||
| 390 | vaapi->config_id = config_id; |
||
| 391 | vaapi->context_id = context_id; |
||
| 392 | LEAVE(); |
||
| 393 | return 0; |
||
| 394 | } |
||
| 395 | |||
| 396 | |||
| 397 | static enum PixelFormat get_format(struct AVCodecContext *avctx, |
||
| 398 | const enum AVPixelFormat *fmt) |
||
| 399 | { |
||
| 400 | int i, profile; |
||
| 401 | |||
| 402 | ENTER(); |
||
| 403 | |||
| 404 | // for (int i = 0; fmt[i] != AV_PIX_FMT_NONE; i++) |
||
| 405 | // printf(" %s", av_get_pix_fmt_name(fmt[i])); |
||
| 406 | |||
| 407 | for (i = 0; fmt[i] != PIX_FMT_NONE; i++) { |
||
| 408 | printf("pixformat %x\n", fmt[i]); |
||
| 409 | if (fmt[i] != AV_PIX_FMT_VAAPI_VLD) |
||
| 410 | continue; |
||
| 411 | |||
| 412 | switch (avctx->codec_id) |
||
| 413 | { |
||
| 414 | case CODEC_ID_MPEG2VIDEO: |
||
| 415 | profile = VAProfileMPEG2Main; |
||
| 416 | break; |
||
| 417 | case CODEC_ID_MPEG4: |
||
| 418 | case CODEC_ID_H263: |
||
| 419 | profile = VAProfileMPEG4AdvancedSimple; |
||
| 420 | break; |
||
| 421 | case CODEC_ID_H264: |
||
| 422 | profile = VAProfileH264High; |
||
| 423 | break; |
||
| 424 | case CODEC_ID_WMV3: |
||
| 425 | profile = VAProfileVC1Main; |
||
| 426 | break; |
||
| 427 | case CODEC_ID_VC1: |
||
| 428 | profile = VAProfileVC1Advanced; |
||
| 429 | break; |
||
| 430 | default: |
||
| 431 | profile = -1; |
||
| 432 | break; |
||
| 433 | } |
||
| 434 | if (profile >= 0) { |
||
| 435 | if (vaapi_init_decoder(profile, VAEntrypointVLD, avctx->width, avctx->height) == 0) |
||
| 436 | { |
||
| 437 | avctx->hwaccel_context = v_context; |
||
| 438 | LEAVE(); |
||
| 439 | return fmt[i]; ; |
||
| 440 | } |
||
| 441 | } |
||
| 442 | } |
||
| 443 | FAIL(); |
||
| 444 | return PIX_FMT_NONE; |
||
| 445 | } |
||
| 446 | |||
| 447 | struct av_surface |
||
| 448 | { |
||
| 449 | int w; |
||
| 450 | int h; |
||
| 451 | VASurfaceID id; |
||
| 452 | }; |
||
| 453 | |||
| 454 | static void av_release_buffer(void *opaque, uint8_t *data) |
||
| 455 | { |
||
| 456 | struct av_surface surface = *(struct av_surface*)data; |
||
| 457 | // VDPAUContext *ctx = opaque; |
||
| 458 | |||
| 459 | // ctx->video_surface_destroy(surface); |
||
| 460 | av_freep(&data); |
||
| 461 | } |
||
| 462 | |||
| 463 | static int get_buffer2(AVCodecContext *avctx, AVFrame *pic, int flags) |
||
| 464 | { |
||
| 6118 | serge | 465 | vst_t *vst = (vst_t*)avctx->opaque; |
| 466 | void *surface = (void *)(uintptr_t)v_surface_id[vst->dfx]; |
||
| 6117 | serge | 467 | |
| 468 | pic->data[3] = surface; |
||
| 469 | |||
| 470 | struct av_surface *avsurface; |
||
| 471 | surface = av_malloc(sizeof(*avsurface)); |
||
| 472 | if (!surface) |
||
| 473 | return AVERROR(ENOMEM); |
||
| 474 | |||
| 475 | pic->buf[0] = av_buffer_create((uint8_t*)avsurface, sizeof(*avsurface), |
||
| 476 | av_release_buffer, avctx, |
||
| 477 | AV_BUFFER_FLAG_READONLY); |
||
| 478 | return 0; |
||
| 479 | } |
||
| 480 | |||
| 481 | struct vaapi_context va_context_storage; |
||
| 482 | |||
| 6118 | serge | 483 | int fplay_init_context(vst_t *vst) |
| 6117 | serge | 484 | { |
| 6118 | serge | 485 | AVCodecContext *vCtx = vst->vCtx; |
| 6117 | serge | 486 | |
| 6121 | serge | 487 | if(va_check_codec_support(vCtx->codec_id)) |
| 488 | { |
||
| 489 | VADisplay dpy; |
||
| 6117 | serge | 490 | |
| 6121 | serge | 491 | dpy = va_open_display(); |
| 492 | vst->hwCtx = vaapi_init(dpy); |
||
| 6117 | serge | 493 | |
| 6121 | serge | 494 | if(vst->hwCtx != NULL) |
| 495 | { |
||
| 496 | for(int i = 0; i < 4; i++) |
||
| 497 | { |
||
| 498 | int ret; |
||
| 6117 | serge | 499 | |
| 6121 | serge | 500 | ret = avpicture_alloc(&vst->vframe[i].picture, AV_PIX_FMT_BGRA, |
| 501 | vst->vCtx->width, vst->vCtx->height); |
||
| 502 | if ( ret != 0 ) |
||
| 503 | { |
||
| 504 | printf("Cannot alloc video buffer\n\r"); |
||
| 505 | return ret; |
||
| 506 | }; |
||
| 507 | vst->vframe[i].format = AV_PIX_FMT_BGRA; |
||
| 508 | vst->vframe[i].pts = 0; |
||
| 509 | vst->vframe[i].ready = 0; |
||
| 510 | }; |
||
| 6117 | serge | 511 | |
| 6121 | serge | 512 | vst->hwdec = 1; |
| 513 | vCtx->opaque = vst; |
||
| 514 | vCtx->thread_count = 1; |
||
| 515 | vCtx->get_format = get_format; |
||
| 516 | vCtx->get_buffer2 = get_buffer2; |
||
| 517 | return 0; |
||
| 518 | }; |
||
| 519 | }; |
||
| 6117 | serge | 520 | |
| 6121 | serge | 521 | vst->hwdec = 0; |
| 522 | |||
| 523 | for(int i = 0; i < 4; i++) |
||
| 524 | { |
||
| 525 | int ret; |
||
| 526 | |||
| 527 | ret = avpicture_alloc(&vst->vframe[i].picture, vst->vCtx->pix_fmt, |
||
| 528 | vst->vCtx->width, vst->vCtx->height); |
||
| 529 | if ( ret != 0 ) |
||
| 530 | { |
||
| 531 | printf("Cannot alloc video buffer\n\r"); |
||
| 532 | return ret; |
||
| 533 | }; |
||
| 534 | vst->vframe[i].format = vst->vCtx->pix_fmt; |
||
| 535 | vst->vframe[i].pts = 0; |
||
| 536 | vst->vframe[i].ready = 0; |
||
| 537 | }; |
||
| 538 | |||
| 6117 | serge | 539 | return 0; |
| 540 | } |
||
| 541 | |||
| 6121 | serge | 542 | |
| 6117 | serge | 543 | struct SwsContext *vacvt_ctx; |
| 544 | |||
| 6118 | serge | 545 | void va_convert_picture(vst_t *vst, int width, int height, AVPicture *pic) |
| 6117 | serge | 546 | { |
| 547 | uint8_t *src_data[4]; |
||
| 548 | int src_linesize[4]; |
||
| 549 | VAImage vaimage; |
||
| 550 | VAStatus status; |
||
| 551 | uint8_t *vdata; |
||
| 552 | struct vaapi_context* const vaapi = v_context; |
||
| 553 | |||
| 6118 | serge | 554 | vaSyncSurface(vaapi->display,v_surface_id[vst->dfx]); |
| 6117 | serge | 555 | |
| 6118 | serge | 556 | status = vaDeriveImage(vaapi->display,v_surface_id[vst->dfx],&vaimage); |
| 6117 | serge | 557 | if (!vaapi_check_status(status, "vaDeriveImage()")) |
| 558 | { |
||
| 559 | FAIL(); |
||
| 560 | return; |
||
| 561 | }; |
||
| 562 | |||
| 563 | static int once = 2; |
||
| 564 | |||
| 6118 | serge | 565 | if(once && vst->dfx == 0) |
| 6117 | serge | 566 | { |
| 567 | VABufferInfo info = {0}; |
||
| 568 | |||
| 569 | printf("vaDeriveImage: %x fourcc: %x\n" |
||
| 570 | "offset0: %d pitch0: %d\n" |
||
| 571 | "offset1: %d pitch1: %d\n" |
||
| 572 | "offset2: %d pitch2: %d\n", |
||
| 573 | vaimage.buf, vaimage.format.fourcc, |
||
| 574 | vaimage.offsets[0],vaimage.pitches[0], |
||
| 575 | vaimage.offsets[1],vaimage.pitches[1], |
||
| 576 | vaimage.offsets[2],vaimage.pitches[2]); |
||
| 577 | |||
| 578 | info.mem_type = VA_SURFACE_ATTRIB_MEM_TYPE_KERNEL_DRM; |
||
| 579 | status = vaAcquireBufferHandle(vaapi->display, vaimage.buf, &info); |
||
| 580 | if (vaapi_check_status(status, "vaAcquireBufferHandle()")) |
||
| 581 | { |
||
| 582 | printf("vaAcquireBufferHandle: %x type: %x\n" |
||
| 583 | "mem type: %x mem size: %d\n", |
||
| 584 | info.handle, info.type, info.mem_type, info.mem_size); |
||
| 585 | |||
| 586 | vaReleaseBufferHandle(vaapi->display, vaimage.buf); |
||
| 587 | } |
||
| 588 | once--; |
||
| 589 | }; |
||
| 590 | |||
| 591 | src_linesize[0] = vaimage.pitches[0]; |
||
| 592 | src_linesize[1] = vaimage.pitches[1]; |
||
| 593 | src_linesize[2] = vaimage.pitches[2]; |
||
| 594 | src_linesize[3] = 0; |
||
| 595 | |||
| 596 | status = vaMapBuffer(vaapi->display,vaimage.buf,(void **)&vdata); |
||
| 597 | if (!vaapi_check_status(status, "vaMapBuffer()")) |
||
| 598 | { |
||
| 599 | FAIL(); |
||
| 600 | return; |
||
| 601 | }; |
||
| 602 | |||
| 603 | // printf("vdata: %x offset0: %d offset1: %d offset2: %d\n", vdata, |
||
| 604 | // vaimage.offsets[0], |
||
| 605 | // vaimage.offsets[1], |
||
| 606 | // vaimage.offsets[2]); |
||
| 607 | |||
| 608 | src_data[0] = vdata + vaimage.offsets[0]; |
||
| 609 | src_data[1] = vdata + vaimage.offsets[1]; |
||
| 610 | src_data[2] = vdata + vaimage.offsets[2]; |
||
| 611 | src_data[3] = 0; |
||
| 612 | |||
| 613 | vacvt_ctx = sws_getCachedContext(vacvt_ctx, width, height, AV_PIX_FMT_NV12, |
||
| 614 | width, height, AV_PIX_FMT_BGRA, |
||
| 615 | SWS_FAST_BILINEAR, NULL, NULL, NULL); |
||
| 616 | if(vacvt_ctx == NULL) |
||
| 617 | { |
||
| 618 | printf("Cannot initialize the conversion context!\n"); |
||
| 619 | return ; |
||
| 620 | }; |
||
| 621 | |||
| 622 | sws_scale(vacvt_ctx, (const uint8_t* const *)src_data, src_linesize, 0, height, pic->data, pic->linesize); |
||
| 623 | |||
| 624 | vaUnmapBuffer (vaapi->display, vaimage.buf); |
||
| 625 | vaDestroyImage(vaapi->display, vaimage.image_id); |
||
| 626 | } |