#!/bin/bash
#*******************************************************************************
# Encoder binary comparison test model
#             -- Compared with benchmark version using SHA-1 string
#             -- Test bit stream under folder  openh264/res
#             -- SHA-1 string of benchmark version for  all cases  of all bit streams
#                 under folder  openh264/test/encoder_binary_comparion/SHA1Table
#             -- For more detail,please refer to file AboutTest.
#
#brief:
#             -- Transform test bit stream into YUV as  test sequence,
#                1) Called by run_OneBitStream.sh  before testing  all cases
#                2) eg:
#                    -- input:    ./run_BitStreamToYUV.sh   TestABC.264
#                    -- output:   TestABC.264_382X288.yuv
#             -- Usage:  run_BitStreamToYUV.sh   ${BitStreamFile}
#
#
# date:    10/06/2014 Created
#*******************************************************************************

#  usage:    run_ParseDecoderLog          $Decoder_LogFile
#  eg:       input: run_ParseDecoderLog   test.264.log
#            output 1024  720
run_ParseDecoderLog()
{
    if [ ! $# -eq 1 ]
    then
        echo "usage:  run_ParseDecoderLog  \$Decoder_LogFile"
        return 1
    fi
    local LogFile=$1
    local Width=""
    local Height=""
    while read line
    do
        if [[  $line =~  "iWidth"   ]]
        then
            Width=`echo $line | awk 'BEGIN  {FS="[:\n]"} {print $2}'`
        fi
        if [[  $line =~  "height"   ]]
        then
             Height=`echo $line | awk 'BEGIN  {FS="[:\n]"} {print $2}'`
        fi
    done < ${LogFile}
    echo "${Width}  ${Height}"
}

#usage: run_BitStream2YUV    $BitstreamName  $OutputYUVName $LogFile
run_BitStream2YUV()
{
    if [ ! $# -eq 3 ]
    then
        echo "usage: run_BitStream2YUV  \$BitstreamName \$OutputYUVName \$LogFile   "
        return 1
    fi
    local BitStreamName=$1
    local OutputYUVName=$2
    local LogFile=$3
    if [ ! -f ${BitStreamName}  ]
    then
        echo "bit stream file does not exist!"
        echo "detected by run_BitStreamToYUV.sh"
        return 1
    fi
    #decode bitstream
    ./h264dec  ${BitStreamName}  ${OutputYUVName} 2> ${LogFile}
    return 0
}

#usage: run_RegularizeYUVName $BitstreamName $OutputYUVName $LogFile
run_RegularizeYUVName()
{
    if [ ! $# -eq 3 ]
    then
        echo "usage: run_RegularizeYUVName  \$BitstreamName  \$OutputYUVName \$LogFile "
        return 1
    fi
    local BitStreamName=$1
    local OrignName=$2
    local LogFile=$3
    local RegularizedYUVName=""
    declare -a aDecodedYUVInfo
    aDecodedYUVInfo=(`run_ParseDecoderLog  ${LogFile}`)
    BitStreamName=`echo ${BitStreamName} | awk 'BEGIN {FS="/"} {print $NF}'`
    RegularizedYUVName="${BitStreamName}_${aDecodedYUVInfo[0]}x${aDecodedYUVInfo[1]}.yuv"
    mv -f  ${OrignName}   ${RegularizedYUVName}
    echo ""
    echo "file: ${OrignName}   has been renamed as: ${RegularizedYUVName}"
    echo ""
    return 0
}

#usage: runMain    ${BitStreamName}
runMain()
{
    if [ ! $# -eq 1 ]
    then
        echo "usage: runMain  \${BitStreamName} "
        return 1
    fi
    local BitStreameFile=$1
    local BitSteamName=`echo ${BitStreameFile} | awk 'BEGIN {FS="/"} {print $NF}'`
    local DecoderLogFile="${BitSteamName}_h264dec.log"
    local DecodedYUVName="${BitSteamName}_dec.yuv"
    local RegularizedName=""

    #**********************
    #decoded test bit stream
    run_BitStream2YUV  ${BitStreameFile}  ${DecodedYUVName}  ${DecoderLogFile}
    if [  ! $?  -eq 0  ]
    then
            echo "bit stream decoded  failed!"
        return 1
    fi
    #*********************
    #regularized  YUV name
    run_RegularizeYUVName  ${BitStreameFile}  ${DecodedYUVName}  ${DecoderLogFile}
    return 0
}
BitStreamFile=$1
runMain    ${BitStreamFile}