123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139 |
- // SPDX-License-Identifier: LGPL-2.1 OR BSD-3-Clause
- /*
- * Copyright (c) 2019, Chips&Media
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- *
- * 1. Redistributions of source code must retain the above copyright notice, this
- * list of conditions and the following disclaimer.
- * 2. Redistributions in binary form must reproduce the above copyright notice,
- * this list of conditions and the following disclaimer in the documentation
- * and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
- * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
- * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
- * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
- * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
- * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
- #include <assert.h>
- #include <string.h>
- #include "main_helper.h"
- FILE * CreatePicInfoXmlFile(
- const char *yuvPath
- )
- {
- char strXmlFilePath[256];
- FILE *fp;
- if (yuvPath != NULL) {
- sprintf(strXmlFilePath, "%s.xml", yuvPath);
- }
- else {
- strcpy(strXmlFilePath, "out.xml");
- }
- if ((fp=osal_fopen(strXmlFilePath, "wb")) == NULL) {
- printf("%s:%d failed to open %s\n", __FUNCTION__, __LINE__, strXmlFilePath);
- return NULL;
- }
- fputs("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n",fp);
- fputs("<ffprobe>\r\n", fp);
- fputs("<frames>\r\n", fp);
- return fp;
- }
- void ClosePicInfoXMLFile(
- FILE *fpXML
- )
- {
- if (fpXML != NULL) {
- fputs("</frames>\r\n", fpXML);
- fputs("</ffprobe>\r\n", fpXML);
- osal_fclose(fpXML);
- }
- }
- void SavePicInfoToXml(
- FILE* fpXml,
- DecOutputInfo* fbInfo
- )
- {
- //BOOL bCrop;
- BOOL bKeyFrame;
- Uint32 pkt_pts;
- Uint32 pkt_dts;
- Uint32 pkt_pos;
- double pkt_pts_sec;
- double pkt_dts_sec;
- Uint32 pkt_size;
- Uint32 width;
- Uint32 height;
- Uint32 bitdepthY;
- Uint32 bitdepthC;
- char strFormat[256] = {"yuv420p"};
- char strPicType[8];
- pkt_size = fbInfo->consumedByte;
- pkt_pos = 0;
- pkt_size = 0;
- pkt_pts = 0;
- pkt_dts = 0;
- pkt_pts_sec = pkt_pts/1000.0;
- pkt_dts_sec = pkt_dts/1000.0;
- bKeyFrame = FALSE;
- switch(fbInfo->picType) {
- case PIC_TYPE_I:
- strcpy(strPicType, "I");
- bKeyFrame = TRUE;
- break;
- case PIC_TYPE_P:
- strcpy(strPicType, "P");
- break;
- case PIC_TYPE_B:
- strcpy(strPicType, "B");
- break;
- default:
- strcpy(strPicType, "B");
- break;
- }
- if (fbInfo->dispFrame.format == FORMAT_420) {
- if (fbInfo->dispFrame.cbcrInterleave == TRUE) {
- strcpy(strFormat, "nv12");
- }
- else {
- strcpy(strFormat, "yuv420p");
- }
- if (fbInfo->dispFrame.nv21 == TRUE) {
- strcpy(strFormat, "nv21");
- }
- }
- width = fbInfo->dispPicWidth;
- height = fbInfo->dispPicHeight;
- bitdepthY = fbInfo->dispFrame.lumaBitDepth;
- bitdepthC = fbInfo->dispFrame.chromaBitDepth;
- fprintf(fpXml, "<frame media_type=\"video\" key_frame=\"%d\" pkt_pts=\"%d\" pkt_pts_time=\"%.1f\" "
- "pkt_dts=\"%d\" pkt_dts_time=\"%.1f\" pkt_pos=\"%d\" pkt_size=\"%d\" width=\"%d\" "
- "height=\"%d\" pix_fmt=\"%s\" pict_type=\"%s\" coded_picture_number=\"0\" display_picture_number=\"0\""
- "interlaced_frame=\"0\" top_field_first=\"0\" repeat_pict=\"0\" bitdepthY=\"%d\" bitdepthC=\"%d\"/>\r\n",
- bKeyFrame, pkt_pts, pkt_pts_sec, pkt_dts, pkt_dts_sec, pkt_pos, pkt_size, width, height, strFormat, strPicType, bitdepthY, bitdepthC);
- osal_fflush(fpXml);
- return;
- }
-
|