Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit f5c91df0 authored by Manisha Jajoo's avatar Manisha Jajoo Committed by Pawin Vongmasa
Browse files

VTS: add CSDFlush test for A/V Decoders 

Test: atest VtsHalMediaC2V1_0TargetAudioDecTest
Test: atest VtsHalMediaC2V1_0TargetVideoDecTest

Bug: 133639884

Change-Id: Ia151ede1b1f01693ea7a7132567285edbf132e06
parent e6641671

media/codec2/hidl/1.0/vts/functional/audio/VtsHalMediaC2V1_0TargetAudioDecTest.cpp

+114 −0
Original line number Diff line number Diff line
@@ -37,6 +37,8 @@ using android::C2AllocatorIon; 
static std::vector<std::tuple<std::string, std::string, std::string, std::string>>

        kDecodeTestParameters;



static std::vector<std::tuple<std::string, std::string, std::string>> kCsdFlushTestParameters;



// Resource directory

static std::string sResourceDir = "";
@@ -831,6 +833,109 @@ TEST_P(Codec2AudioDecHidlTest, DecodeTestEmptyBuffersInserted) { 
    ASSERT_EQ(mComponent->stop(), C2_OK);

}



class Codec2AudioDecCsdInputTests

    : public Codec2AudioDecHidlTestBase,

      public ::testing::WithParamInterface<std::tuple<std::string, std::string, std::string>> {

    void getParams() {

        mInstanceName = std::get<0>(GetParam());

        mComponentName = std::get<1>(GetParam());

    }

};



// Test the codecs for the following

// start - csd - data… - (with/without)flush - data… - flush - data…

TEST_P(Codec2AudioDecCsdInputTests, CSDFlushTest) {

    description("Tests codecs for flush at different states");

    if (mDisableTest) GTEST_SKIP() << "Test is disabled";



    char mURL[512], info[512];

    android::Vector<FrameInfo> Info;



    strcpy(mURL, sResourceDir.c_str());

    strcpy(info, sResourceDir.c_str());

    GetURLForComponent(mCompName, mURL, info);

    if (!strcmp(mURL, sResourceDir.c_str())) {

        ALOGV("EMPTY INPUT sResourceDir.c_str() %s mURL  %s ", sResourceDir.c_str(), mURL);

        return;

    }

    ALOGV("mURL : %s", mURL);



    int32_t numCsds = populateInfoVector(info, &Info, mTimestampDevTest, &mTimestampUslist);

    ASSERT_GE(numCsds, 0) << "Error in parsing input info file";



    int32_t bitStreamInfo[2] = {0};

    if (mCompName == raw) {

        bitStreamInfo[0] = 8000;

        bitStreamInfo[1] = 1;

    } else {

        ASSERT_NO_FATAL_FAILURE(getInputChannelInfo(mComponent, mCompName, bitStreamInfo));

    }

    if (!setupConfigParam(mComponent, bitStreamInfo)) {

        std::cout << "[   WARN   ] Test Skipped \n";

        return;

    }



    ASSERT_EQ(mComponent->start(), C2_OK);

    std::ifstream eleStream;

    eleStream.open(mURL, std::ifstream::binary);

    ASSERT_EQ(eleStream.is_open(), true);



    bool signalEOS = false;

    bool flushCsd = !std::get<2>(GetParam()).compare("true");

    ALOGV("sending %d csd data ", numCsds);

    int framesToDecode = numCsds;

    ASSERT_NO_FATAL_FAILURE(decodeNFrames(mComponent, mQueueLock, mQueueCondition, mWorkQueue,

                                          mFlushedIndices, mLinearPool, eleStream, &Info, 0,

                                          framesToDecode, false));



    c2_status_t err = C2_OK;

    std::list<std::unique_ptr<C2Work>> flushedWork;

    if (numCsds && flushCsd) {

        // We wait for all the CSD buffers to get consumed.

        // Once we have received all CSD work back, we call flush

        waitOnInputConsumption(mQueueLock, mQueueCondition, mWorkQueue);



        err = mComponent->flush(C2Component::FLUSH_COMPONENT, &flushedWork);

        ASSERT_EQ(err, C2_OK);

        waitOnInputConsumption(mQueueLock, mQueueCondition, mWorkQueue,

                               MAX_INPUT_BUFFERS - flushedWork.size());

        ASSERT_NO_FATAL_FAILURE(

                verifyFlushOutput(flushedWork, mWorkQueue, mFlushedIndices, mQueueLock));

        ASSERT_EQ(mWorkQueue.size(), MAX_INPUT_BUFFERS);

        oBufferMetaData.clear();

    }



    int offset = framesToDecode;

    while (1) {

        framesToDecode = c2_min(FLUSH_INTERVAL, (int)Info.size() - offset);

        if (framesToDecode < FLUSH_INTERVAL) signalEOS = true;

        ASSERT_NO_FATAL_FAILURE(decodeNFrames(mComponent, mQueueLock, mQueueCondition, mWorkQueue,

                                              mFlushedIndices, mLinearPool, eleStream, &Info,

                                              offset, framesToDecode, signalEOS));

        offset += framesToDecode;

        err = mComponent->flush(C2Component::FLUSH_COMPONENT, &flushedWork);

        ASSERT_EQ(err, C2_OK);

        // blocking call to ensures application to Wait till remaining

        // 'non-flushed' inputs are consumed

        waitOnInputConsumption(mQueueLock, mQueueCondition, mWorkQueue,

                               MAX_INPUT_BUFFERS - flushedWork.size());

        ASSERT_NO_FATAL_FAILURE(

                verifyFlushOutput(flushedWork, mWorkQueue, mFlushedIndices, mQueueLock));

        ASSERT_EQ(mWorkQueue.size(), MAX_INPUT_BUFFERS);

        if (signalEOS || offset >= (int)Info.size()) {

            break;

        }

    }

    if (!signalEOS) {

        ASSERT_NO_FATAL_FAILURE(testInputBuffer(mComponent, mQueueLock, mWorkQueue,

                                                C2FrameData::FLAG_END_OF_STREAM, false));

        waitOnInputConsumption(mQueueLock, mQueueCondition, mWorkQueue);

    }

    eleStream.close();

    ASSERT_EQ(mWorkQueue.size(), MAX_INPUT_BUFFERS);

    ASSERT_EQ(mComponent->stop(), C2_OK);

}



INSTANTIATE_TEST_SUITE_P(PerInstance, Codec2AudioDecHidlTest, testing::ValuesIn(kTestParameters),

                         android::hardware::PrintInstanceTupleNameToString<>);
@@ -839,6 +944,10 @@ INSTANTIATE_TEST_SUITE_P(StreamIndexAndEOS, Codec2AudioDecDecodeTest, 
                         testing::ValuesIn(kDecodeTestParameters),

                         android::hardware::PrintInstanceTupleNameToString<>);



INSTANTIATE_TEST_SUITE_P(CsdInputs, Codec2AudioDecCsdInputTests,

                         testing::ValuesIn(kCsdFlushTestParameters),

                         android::hardware::PrintInstanceTupleNameToString<>);



}  // anonymous namespace



int main(int argc, char** argv) {
@@ -852,6 +961,11 @@ int main(int argc, char** argv) { 
                std::make_tuple(std::get<0>(params), std::get<1>(params), "1", "false"));

        kDecodeTestParameters.push_back(

                std::make_tuple(std::get<0>(params), std::get<1>(params), "1", "true"));



        kCsdFlushTestParameters.push_back(

                std::make_tuple(std::get<0>(params), std::get<1>(params), "true"));

        kCsdFlushTestParameters.push_back(

                std::make_tuple(std::get<0>(params), std::get<1>(params), "false"));

    }



    // Set the resource directory based on command line args.

media/codec2/hidl/1.0/vts/functional/video/VtsHalMediaC2V1_0TargetVideoDecTest.cpp

+115 −0
Original line number Diff line number Diff line
@@ -43,6 +43,8 @@ using android::C2AllocatorIon; 
static std::vector<std::tuple<std::string, std::string, std::string, std::string>>

        kDecodeTestParameters;



static std::vector<std::tuple<std::string, std::string, std::string>> kCsdFlushTestParameters;



// Resource directory

static std::string sResourceDir = "";
@@ -1001,6 +1003,110 @@ TEST_P(Codec2VideoDecHidlTest, DecodeTestEmptyBuffersInserted) { 
    }

}



class Codec2VideoDecCsdInputTests

    : public Codec2VideoDecHidlTestBase,

      public ::testing::WithParamInterface<std::tuple<std::string, std::string, std::string>> {

    void getParams() {

        mInstanceName = std::get<0>(GetParam());

        mComponentName = std::get<1>(GetParam());

    }

};



// Test the codecs for the following

// start - csd - data… - (with/without)flush - data… - flush - data…

TEST_P(Codec2VideoDecCsdInputTests, CSDFlushTest) {

    description("Tests codecs for flush at different states");

    if (mDisableTest) GTEST_SKIP() << "Test is disabled";



    char mURL[512], info[512];



    android::Vector<FrameInfo> Info;



    strcpy(mURL, sResourceDir.c_str());

    strcpy(info, sResourceDir.c_str());

    GetURLForComponent(mCompName, mURL, info);



    int32_t numCsds = populateInfoVector(info, &Info, mTimestampDevTest, &mTimestampUslist);

    ASSERT_GE(numCsds, 0) << "Error in parsing input info file";



    ASSERT_EQ(mComponent->start(), C2_OK);



    ALOGV("mURL : %s", mURL);

    std::ifstream eleStream;

    eleStream.open(mURL, std::ifstream::binary);

    ASSERT_EQ(eleStream.is_open(), true);

    bool flushedDecoder = false;

    bool signalEOS = false;

    bool keyFrame = false;

    bool flushCsd = !std::get<2>(GetParam()).compare("true");



    ALOGV("sending %d csd data ", numCsds);

    int framesToDecode = numCsds;

    ASSERT_NO_FATAL_FAILURE(decodeNFrames(mComponent, mQueueLock, mQueueCondition, mWorkQueue,

                                          mFlushedIndices, mLinearPool, eleStream, &Info, 0,

                                          framesToDecode, false));

    c2_status_t err = C2_OK;

    std::list<std::unique_ptr<C2Work>> flushedWork;

    if (numCsds && flushCsd) {

        // We wait for all the CSD buffers to get consumed.

        // Once we have received all CSD work back, we call flush

        waitOnInputConsumption(mQueueLock, mQueueCondition, mWorkQueue);



        err = mComponent->flush(C2Component::FLUSH_COMPONENT, &flushedWork);

        ASSERT_EQ(err, C2_OK);

        flushedDecoder = true;

        waitOnInputConsumption(mQueueLock, mQueueCondition, mWorkQueue,

                               MAX_INPUT_BUFFERS - flushedWork.size());

        ASSERT_NO_FATAL_FAILURE(

                verifyFlushOutput(flushedWork, mWorkQueue, mFlushedIndices, mQueueLock));

        ASSERT_EQ(mWorkQueue.size(), MAX_INPUT_BUFFERS);

    }



    int offset = framesToDecode;

    uint32_t flags = 0;

    while (1) {

        while (offset < (int)Info.size()) {

            flags = 0;

            if (Info[offset].flags) flags = 1u << (Info[offset].flags - 1);

            if (flags & SYNC_FRAME) {

                keyFrame = true;

                break;

            }

            eleStream.ignore(Info[offset].bytesCount);

            offset++;

        }

        if (keyFrame) {

            framesToDecode = c2_min(FLUSH_INTERVAL, (int)Info.size() - offset);

            if (framesToDecode < FLUSH_INTERVAL) signalEOS = true;

            ASSERT_NO_FATAL_FAILURE(decodeNFrames(

                    mComponent, mQueueLock, mQueueCondition, mWorkQueue, mFlushedIndices,

                    mLinearPool, eleStream, &Info, offset, framesToDecode, signalEOS));

            offset += framesToDecode;

        }

        err = mComponent->flush(C2Component::FLUSH_COMPONENT, &flushedWork);

        ASSERT_EQ(err, C2_OK);

        keyFrame = false;

        // blocking call to ensures application to Wait till remaining

        // 'non-flushed' inputs are consumed

        waitOnInputConsumption(mQueueLock, mQueueCondition, mWorkQueue,

                               MAX_INPUT_BUFFERS - flushedWork.size());

        ASSERT_NO_FATAL_FAILURE(

                verifyFlushOutput(flushedWork, mWorkQueue, mFlushedIndices, mQueueLock));

        ASSERT_EQ(mWorkQueue.size(), MAX_INPUT_BUFFERS);

        if (signalEOS || offset >= (int)Info.size()) {

            break;

        }

    }

    if (!signalEOS) {

        ASSERT_NO_FATAL_FAILURE(testInputBuffer(mComponent, mQueueLock, mWorkQueue,

                                                C2FrameData::FLAG_END_OF_STREAM, false));

        waitOnInputConsumption(mQueueLock, mQueueCondition, mWorkQueue);

    }

    eleStream.close();

    ASSERT_EQ(mWorkQueue.size(), MAX_INPUT_BUFFERS);

    ASSERT_EQ(mComponent->stop(), C2_OK);

}



INSTANTIATE_TEST_SUITE_P(PerInstance, Codec2VideoDecHidlTest, testing::ValuesIn(kTestParameters),

                         android::hardware::PrintInstanceTupleNameToString<>);
@@ -1009,6 +1115,10 @@ INSTANTIATE_TEST_SUITE_P(StreamIndexAndEOS, Codec2VideoDecDecodeTest, 
                         testing::ValuesIn(kDecodeTestParameters),

                         android::hardware::PrintInstanceTupleNameToString<>);



INSTANTIATE_TEST_SUITE_P(CsdInputs, Codec2VideoDecCsdInputTests,

                         testing::ValuesIn(kCsdFlushTestParameters),

                         android::hardware::PrintInstanceTupleNameToString<>);



}  // anonymous namespace



// TODO : Video specific configuration Test
@@ -1027,6 +1137,11 @@ int main(int argc, char** argv) { 
                std::make_tuple(std::get<0>(params), std::get<1>(params), "2", "false"));

        kDecodeTestParameters.push_back(

                std::make_tuple(std::get<0>(params), std::get<1>(params), "2", "true"));



        kCsdFlushTestParameters.push_back(

                std::make_tuple(std::get<0>(params), std::get<1>(params), "true"));

        kCsdFlushTestParameters.push_back(

                std::make_tuple(std::get<0>(params), std::get<1>(params), "false"));

    }



    // Set the resource directory based on command line args.