media/omx/1.0/vts/functional/video/VtsHalMediaOmxV1_0TargetVideoDecTest.cpp

   1 /*
   2  * Copyright (C) 2017 The Android Open Source Project
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #define LOG_TAG "media_omx_hidl_video_dec_test"
  18 #include <android-base/logging.h>
  19
  20 #include <android/hardware/graphics/allocator/2.0/IAllocator.h>
  21 #include <android/hardware/graphics/mapper/2.0/IMapper.h>
  22 #include <android/hardware/graphics/mapper/2.0/types.h>
  23 #include <android/hardware/media/omx/1.0/IOmx.h>
  24 #include <android/hardware/media/omx/1.0/IOmxNode.h>
  25 #include <android/hardware/media/omx/1.0/IOmxObserver.h>
  26 #include <android/hardware/media/omx/1.0/types.h>
  27 #include <android/hidl/allocator/1.0/IAllocator.h>
  28 #include <android/hidl/memory/1.0/IMapper.h>
  29 #include <android/hidl/memory/1.0/IMemory.h>
  30
  31 using ::android::hardware::graphics::common::V1_0::BufferUsage;
  32 using ::android::hardware::graphics::common::V1_0::PixelFormat;
  33 using ::android::hardware::media::omx::V1_0::IOmx;
  34 using ::android::hardware::media::omx::V1_0::IOmxObserver;
  35 using ::android::hardware::media::omx::V1_0::IOmxNode;
  36 using ::android::hardware::media::omx::V1_0::Message;
  37 using ::android::hardware::media::omx::V1_0::CodecBuffer;
  38 using ::android::hardware::media::omx::V1_0::PortMode;
  39 using ::android::hidl::allocator::V1_0::IAllocator;
  40 using ::android::hidl::memory::V1_0::IMemory;
  41 using ::android::hidl::memory::V1_0::IMapper;
  42 using ::android::hardware::Return;
  43 using ::android::hardware::Void;
  44 using ::android::hardware::hidl_vec;
  45 using ::android::hardware::hidl_string;
  46 using ::android::sp;
  47
  48 #include <VtsHalHidlTargetTestBase.h>
  49 #include <getopt.h>
  50 #include <media_hidl_test_common.h>
  51 #include <media_video_hidl_test_common.h>
  52 #include <fstream>
  53
  54 // A class for test environment setup
  55 class ComponentTestEnvironment : public ::testing::Environment {
  56    public:
  57     virtual void SetUp() {}
  58     virtual void TearDown() {}
  59
  60     ComponentTestEnvironment() : instance("default"), res("/sdcard/media/") {}
  61
  62     void setInstance(const char* _instance) { instance = _instance; }
  63
  64     void setComponent(const char* _component) { component = _component; }
  65
  66     void setRole(const char* _role) { role = _role; }
  67
  68     void setRes(const char* _res) { res = _res; }
  69
  70     const hidl_string getInstance() const { return instance; }
  71
  72     const hidl_string getComponent() const { return component; }
  73
  74     const hidl_string getRole() const { return role; }
  75
  76     const hidl_string getRes() const { return res; }
  77
  78     int initFromOptions(int argc, char** argv) {
  79         static struct option options[] = {
  80             {"instance", required_argument, 0, 'I'},
  81             {"component", required_argument, 0, 'C'},
  82             {"role", required_argument, 0, 'R'},
  83             {"res", required_argument, 0, 'P'},
  84             {0, 0, 0, 0}};
  85
  86         while (true) {
  87             int index = 0;
  88             int c = getopt_long(argc, argv, "I:C:R:P:", options, &index);
  89             if (c == -1) {
  90                 break;
  91             }
  92
  93             switch (c) {
  94                 case 'I':
  95                     setInstance(optarg);
  96                     break;
  97                 case 'C':
  98                     setComponent(optarg);
  99                     break;
 100                 case 'R':
 101                     setRole(optarg);
 102                     break;
 103                 case 'P':
 104                     setRes(optarg);
 105                     break;
 106                 case '?':
 107                     break;
 108             }
 109         }
 110
 111         if (optind < argc) {
 112             fprintf(stderr,
 113                     "unrecognized option: %s\n\n"
 114                     "usage: %s <gtest options> <test options>\n\n"
 115                     "test options are:\n\n"
 116                     "-I, --instance: HAL instance to test\n"
 117                     "-C, --component: OMX component to test\n"
 118                     "-R, --role: OMX component Role\n"
 119                     "-P, --res: Resource files directory location\n",
 120                     argv[optind ?: 1], argv[0]);
 121             return 2;
 122         }
 123         return 0;
 124     }
 125
 126    private:
 127     hidl_string instance;
 128     hidl_string component;
 129     hidl_string role;
 130     hidl_string res;
 131 };
 132
 133 static ComponentTestEnvironment* gEnv = nullptr;
 134
 135 // video decoder test fixture class
 136 class VideoDecHidlTest : public ::testing::VtsHalHidlTargetTestBase {
 137    private:
 138     typedef ::testing::VtsHalHidlTargetTestBase Super;
 139    public:
 140     ::std::string getTestCaseInfo() const override {
 141         return ::std::string() +
 142                 "Component: " + gEnv->getComponent().c_str() + " | " +
 143                 "Role: " + gEnv->getRole().c_str() + " | " +
 144                 "Instance: " + gEnv->getInstance().c_str() + " | " +
 145                 "Res: " + gEnv->getRes().c_str();
 146     }
 147
 148     virtual void SetUp() override {
 149         Super::SetUp();
 150         disableTest = false;
 151         android::hardware::media::omx::V1_0::Status status;
 152         omx = Super::getService<IOmx>(gEnv->getInstance());
 153         ASSERT_NE(omx, nullptr);
 154         observer =
 155             new CodecObserver([this](Message msg, const BufferInfo* buffer) {
 156                 handleMessage(msg, buffer);
 157             });
 158         ASSERT_NE(observer, nullptr);
 159         if (strncmp(gEnv->getComponent().c_str(), "OMX.", 4) != 0)
 160             disableTest = true;
 161         EXPECT_TRUE(omx->allocateNode(
 162                            gEnv->getComponent(), observer,
 163                            [&](android::hardware::media::omx::V1_0::Status _s,
 164                                sp<IOmxNode> const& _nl) {
 165                                status = _s;
 166                                this->omxNode = _nl;
 167                            })
 168                         .isOk());
 169         ASSERT_NE(omxNode, nullptr);
 170         ASSERT_NE(gEnv->getRole().empty(), true) << "Invalid Component Role";
 171         struct StringToName {
 172             const char* Name;
 173             standardComp CompName;
 174         };
 175         const StringToName kStringToName[] = {
 176             {"h263", h263}, {"avc", avc}, {"mpeg2", mpeg2}, {"mpeg4", mpeg4},
 177             {"hevc", hevc}, {"vp8", vp8}, {"vp9", vp9},
 178         };
 179         const size_t kNumStringToName =
 180             sizeof(kStringToName) / sizeof(kStringToName[0]);
 181         const char* pch;
 182         char substring[OMX_MAX_STRINGNAME_SIZE];
 183         strcpy(substring, gEnv->getRole().c_str());
 184         pch = strchr(substring, '.');
 185         ASSERT_NE(pch, nullptr);
 186         compName = unknown_comp;
 187         for (size_t i = 0; i < kNumStringToName; ++i) {
 188             if (!strcasecmp(pch + 1, kStringToName[i].Name)) {
 189                 compName = kStringToName[i].CompName;
 190                 break;
 191             }
 192         }
 193         if (compName == unknown_comp) disableTest = true;
 194         struct CompToCompression {
 195             standardComp CompName;
 196             OMX_VIDEO_CODINGTYPE eCompressionFormat;
 197         };
 198         static const CompToCompression kCompToCompression[] = {
 199             {h263, OMX_VIDEO_CodingH263},   {avc, OMX_VIDEO_CodingAVC},
 200             {mpeg2, OMX_VIDEO_CodingMPEG2}, {mpeg4, OMX_VIDEO_CodingMPEG4},
 201             {hevc, OMX_VIDEO_CodingHEVC},   {vp8, OMX_VIDEO_CodingVP8},
 202             {vp9, OMX_VIDEO_CodingVP9},
 203         };
 204         static const size_t kNumCompToCompression =
 205             sizeof(kCompToCompression) / sizeof(kCompToCompression[0]);
 206         size_t i;
 207         for (i = 0; i < kNumCompToCompression; ++i) {
 208             if (kCompToCompression[i].CompName == compName) {
 209                 eCompressionFormat = kCompToCompression[i].eCompressionFormat;
 210                 break;
 211             }
 212         }
 213         if (i == kNumCompToCompression) disableTest = true;
 214         portMode[0] = portMode[1] = PortMode::PRESET_BYTE_BUFFER;
 215         eosFlag = false;
 216         framesReceived = 0;
 217         timestampUs = 0;
 218         timestampDevTest = false;
 219         isSecure = false;
 220         size_t suffixLen = strlen(".secure");
 221         if (strlen(gEnv->getComponent().c_str()) >= suffixLen) {
 222             isSecure =
 223                 !strcmp(gEnv->getComponent().c_str() +
 224                             strlen(gEnv->getComponent().c_str()) - suffixLen,
 225                         ".secure");
 226         }
 227         if (isSecure) disableTest = true;
 228         if (disableTest) std::cout << "[          ] Warning !  Test Disabled\n";
 229     }
 230
 231     virtual void TearDown() override {
 232         if (omxNode != nullptr) {
 233             EXPECT_TRUE((omxNode->freeNode()).isOk());
 234             omxNode = nullptr;
 235         }
 236         Super::TearDown();
 237     }
 238
 239     // callback function to process messages received by onMessages() from IL
 240     // client.
 241     void handleMessage(Message msg, const BufferInfo* buffer) {
 242         (void)buffer;
 243         if (msg.type == Message::Type::FILL_BUFFER_DONE) {
 244             if (msg.data.extendedBufferData.flags & OMX_BUFFERFLAG_EOS) {
 245                 eosFlag = true;
 246             }
 247             if (msg.data.extendedBufferData.rangeLength != 0) {
 248                 framesReceived += 1;
 249                 // For decoder components current timestamp always exceeds
 250                 // previous timestamp
 251                 EXPECT_GE(msg.data.extendedBufferData.timestampUs, timestampUs);
 252                 timestampUs = msg.data.extendedBufferData.timestampUs;
 253                 // Test if current timestamp is among the list of queued
 254                 // timestamps
 255                 if (timestampDevTest) {
 256                     bool tsHit = false;
 257                     android::List<uint64_t>::iterator it =
 258                         timestampUslist.begin();
 259                     while (it != timestampUslist.end()) {
 260                         if (*it == timestampUs) {
 261                             timestampUslist.erase(it);
 262                             tsHit = true;
 263                             break;
 264                         }
 265                         it++;
 266                     }
 267                     if (tsHit == false) {
 268                         if (timestampUslist.empty() == false) {
 269                             EXPECT_EQ(tsHit, true)
 270                                 << "TimeStamp not recognized";
 271                         } else {
 272                             std::cout
 273                                 << "[          ] Warning ! Received non-zero "
 274                                    "output / TimeStamp not recognized \n";
 275                         }
 276                     }
 277                 }
 278 #define WRITE_OUTPUT 0
 279 #if WRITE_OUTPUT
 280                 static int count = 0;
 281                 FILE* ofp = nullptr;
 282                 if (count)
 283                     ofp = fopen("out.bin", "ab");
 284                 else
 285                     ofp = fopen("out.bin", "wb");
 286                 if (ofp != nullptr &&
 287                     portMode[1] == PortMode::PRESET_BYTE_BUFFER) {
 288                     fwrite(static_cast<void*>(buffer->mMemory->getPointer()),
 289                            sizeof(char),
 290                            msg.data.extendedBufferData.rangeLength, ofp);
 291                     fclose(ofp);
 292                     count++;
 293                 }
 294 #endif
 295             }
 296         }
 297     }
 298
 299     enum standardComp {
 300         h263,
 301         avc,
 302         mpeg2,
 303         mpeg4,
 304         hevc,
 305         vp8,
 306         vp9,
 307         unknown_comp,
 308     };
 309
 310     sp<IOmx> omx;
 311     sp<CodecObserver> observer;
 312     sp<IOmxNode> omxNode;
 313     standardComp compName;
 314     OMX_VIDEO_CODINGTYPE eCompressionFormat;
 315     bool disableTest;
 316     PortMode portMode[2];
 317     bool eosFlag;
 318     uint32_t framesReceived;
 319     uint64_t timestampUs;
 320     ::android::List<uint64_t> timestampUslist;
 321     bool timestampDevTest;
 322     bool isSecure;
 323
 324    protected:
 325     static void description(const std::string& description) {
 326         RecordProperty("description", description);
 327     }
 328 };
 329
 330 // Set Default port param.
 331 void setDefaultPortParam(sp<IOmxNode> omxNode, OMX_U32 portIndex,
 332                          OMX_VIDEO_CODINGTYPE eCompressionFormat,
 333                          OMX_COLOR_FORMATTYPE eColorFormat,
 334                          OMX_U32 nFrameWidth = 352, OMX_U32 nFrameHeight = 288,
 335                          OMX_U32 nBitrate = 0,
 336                          OMX_U32 xFramerate = (24U << 16)) {
 337     switch ((int)eCompressionFormat) {
 338         case OMX_VIDEO_CodingUnused:
 339             setupRAWPort(omxNode, portIndex, nFrameWidth, nFrameHeight,
 340                          nBitrate, xFramerate, eColorFormat);
 341             break;
 342         default:
 343             break;
 344     }
 345 }
 346
 347 // In decoder components, often the input port parameters get updated upon
 348 // parsing the header of elementary stream. Client needs to collect this
 349 // information to reconfigure other ports that share data with this input
 350 // port.
 351 void getInputChannelInfo(sp<IOmxNode> omxNode, OMX_U32 kPortIndexInput,
 352                          uint32_t* nFrameWidth, uint32_t* nFrameHeight,
 353                          uint32_t* xFramerate) {
 354     android::hardware::media::omx::V1_0::Status status;
 355     *nFrameWidth = 352;
 356     *nFrameHeight = 288;
 357     *xFramerate = (24U << 16);
 358
 359     OMX_PARAM_PORTDEFINITIONTYPE portDef;
 360     status = getPortParam(omxNode, OMX_IndexParamPortDefinition,
 361                           kPortIndexInput, &portDef);
 362     EXPECT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 363     if (status == ::android::hardware::media::omx::V1_0::Status::OK) {
 364         *nFrameWidth = portDef.format.video.nFrameWidth;
 365         *nFrameHeight = portDef.format.video.nFrameHeight;
 366         *xFramerate = portDef.format.video.xFramerate;
 367     }
 368 }
 369
 370 // LookUpTable of clips and metadata for component testing
 371 void GetURLForComponent(VideoDecHidlTest::standardComp comp, char* mURL,
 372                         char* info) {
 373     struct CompToURL {
 374         VideoDecHidlTest::standardComp comp;
 375         const char* mURL;
 376         const char* info;
 377     };
 378     static const CompToURL kCompToURL[] = {
 379         {VideoDecHidlTest::standardComp::avc,
 380          "bbb_avc_1920x1080_5000kbps_30fps.h264",
 381          "bbb_avc_1920x1080_5000kbps_30fps.info"},
 382         {VideoDecHidlTest::standardComp::hevc,
 383          "bbb_hevc_640x360_1600kbps_30fps.hevc",
 384          "bbb_hevc_640x360_1600kbps_30fps.info"},
 385         {VideoDecHidlTest::standardComp::mpeg2,
 386          "bbb_mpeg2_176x144_105kbps_25fps.m2v",
 387          "bbb_mpeg2_176x144_105kbps_25fps.info"},
 388         {VideoDecHidlTest::standardComp::h263,
 389          "bbb_h263_352x288_300kbps_12fps.h263",
 390          "bbb_h263_352x288_300kbps_12fps.info"},
 391         {VideoDecHidlTest::standardComp::mpeg4,
 392          "bbb_mpeg4_1280x720_1000kbps_25fps.m4v",
 393          "bbb_mpeg4_1280x720_1000kbps_25fps.info"},
 394         {VideoDecHidlTest::standardComp::vp8, "bbb_vp8_640x360_2mbps_30fps.vp8",
 395          "bbb_vp8_640x360_2mbps_30fps.info"},
 396         {VideoDecHidlTest::standardComp::vp9,
 397          "bbb_vp9_640x360_1600kbps_30fps.vp9",
 398          "bbb_vp9_640x360_1600kbps_30fps.info"},
 399     };
 400
 401     for (size_t i = 0; i < sizeof(kCompToURL) / sizeof(kCompToURL[0]); ++i) {
 402         if (kCompToURL[i].comp == comp) {
 403             strcat(mURL, kCompToURL[i].mURL);
 404             strcat(info, kCompToURL[i].info);
 405             return;
 406         }
 407     }
 408 }
 409
 410 void allocateGraphicBuffers(sp<IOmxNode> omxNode, OMX_U32 portIndex,
 411                             android::Vector<BufferInfo>* buffArray,
 412                             uint32_t nFrameWidth, uint32_t nFrameHeight,
 413                             int32_t* nStride, uint32_t count) {
 414     android::hardware::media::omx::V1_0::Status status;
 415     sp<android::hardware::graphics::allocator::V2_0::IAllocator> allocator =
 416         android::hardware::graphics::allocator::V2_0::IAllocator::getService();
 417     ASSERT_NE(nullptr, allocator.get());
 418
 419     sp<android::hardware::graphics::mapper::V2_0::IMapper> mapper =
 420         android::hardware::graphics::mapper::V2_0::IMapper::getService();
 421     ASSERT_NE(mapper.get(), nullptr);
 422
 423     android::hardware::graphics::mapper::V2_0::IMapper::BufferDescriptorInfo
 424         descriptorInfo;
 425     uint32_t usage;
 426
 427     descriptorInfo.width = nFrameWidth;
 428     descriptorInfo.height = nFrameHeight;
 429     descriptorInfo.layerCount = 1;
 430     descriptorInfo.format = PixelFormat::RGBA_8888;
 431     descriptorInfo.usage = static_cast<uint64_t>(BufferUsage::CPU_READ_OFTEN);
 432     omxNode->getGraphicBufferUsage(
 433         portIndex,
 434         [&status, &usage](android::hardware::media::omx::V1_0::Status _s,
 435                           uint32_t _n1) {
 436             status = _s;
 437             usage = _n1;
 438         });
 439     if (status == android::hardware::media::omx::V1_0::Status::OK) {
 440         descriptorInfo.usage |= usage;
 441     }
 442
 443     ::android::hardware::hidl_vec<uint32_t> descriptor;
 444     android::hardware::graphics::mapper::V2_0::Error error;
 445     mapper->createDescriptor(
 446         descriptorInfo, [&error, &descriptor](
 447                             android::hardware::graphics::mapper::V2_0::Error _s,
 448                             ::android::hardware::hidl_vec<uint32_t> _n1) {
 449             error = _s;
 450             descriptor = _n1;
 451         });
 452     EXPECT_EQ(error, android::hardware::graphics::mapper::V2_0::Error::NONE);
 453
 454     EXPECT_EQ(buffArray->size(), count);
 455     allocator->allocate(
 456         descriptor, count,
 457         [&](android::hardware::graphics::mapper::V2_0::Error _s, uint32_t _n1,
 458             const ::android::hardware::hidl_vec<
 459                 ::android::hardware::hidl_handle>& _n2) {
 460             ASSERT_EQ(android::hardware::graphics::mapper::V2_0::Error::NONE,
 461                       _s);
 462             *nStride = _n1;
 463             ASSERT_EQ(count, _n2.size());
 464             for (uint32_t i = 0; i < count; i++) {
 465                 buffArray->editItemAt(i).omxBuffer.nativeHandle = _n2[i];
 466                 buffArray->editItemAt(i).omxBuffer.attr.anwBuffer.width =
 467                     nFrameWidth;
 468                 buffArray->editItemAt(i).omxBuffer.attr.anwBuffer.height =
 469                     nFrameHeight;
 470                 buffArray->editItemAt(i).omxBuffer.attr.anwBuffer.stride = _n1;
 471                 buffArray->editItemAt(i).omxBuffer.attr.anwBuffer.format =
 472                     descriptorInfo.format;
 473                 buffArray->editItemAt(i).omxBuffer.attr.anwBuffer.usage =
 474                     descriptorInfo.usage;
 475                 buffArray->editItemAt(i).omxBuffer.attr.anwBuffer.layerCount =
 476                     descriptorInfo.layerCount;
 477                 buffArray->editItemAt(i).omxBuffer.attr.anwBuffer.id =
 478                     (*buffArray)[i].id;
 479             }
 480         });
 481 }
 482
 483 // port settings reconfiguration during runtime. reconfigures frame dimensions
 484 void portReconfiguration(sp<IOmxNode> omxNode, sp<CodecObserver> observer,
 485                          android::Vector<BufferInfo>* iBuffer,
 486                          android::Vector<BufferInfo>* oBuffer,
 487                          OMX_U32 kPortIndexInput, OMX_U32 kPortIndexOutput,
 488                          Message msg, PortMode oPortMode, void* args) {
 489     android::hardware::media::omx::V1_0::Status status;
 490     (void)args;
 491
 492     if (msg.data.eventData.event == OMX_EventPortSettingsChanged) {
 493         ASSERT_EQ(msg.data.eventData.data1, kPortIndexOutput);
 494         if (msg.data.eventData.data2 == OMX_IndexParamPortDefinition ||
 495             msg.data.eventData.data2 == 0) {
 496             status = omxNode->sendCommand(
 497                 toRawCommandType(OMX_CommandPortDisable), kPortIndexOutput);
 498             ASSERT_EQ(status, android::hardware::media::omx::V1_0::Status::OK);
 499
 500             status = observer->dequeueMessage(&msg, DEFAULT_TIMEOUT, iBuffer,
 501                                               oBuffer);
 502             if (status ==
 503                 android::hardware::media::omx::V1_0::Status::TIMED_OUT) {
 504                 for (size_t i = 0; i < oBuffer->size(); ++i) {
 505                     // test if client got all its buffers back
 506                     EXPECT_EQ((*oBuffer)[i].owner, client);
 507                     // free the buffers
 508                     status =
 509                         omxNode->freeBuffer(kPortIndexOutput, (*oBuffer)[i].id);
 510                     ASSERT_EQ(status,
 511                               android::hardware::media::omx::V1_0::Status::OK);
 512                 }
 513                 status = observer->dequeueMessage(&msg, DEFAULT_TIMEOUT,
 514                                                   iBuffer, oBuffer);
 515                 ASSERT_EQ(status,
 516                           android::hardware::media::omx::V1_0::Status::OK);
 517                 ASSERT_EQ(msg.type, Message::Type::EVENT);
 518                 ASSERT_EQ(msg.data.eventData.event, OMX_EventCmdComplete);
 519                 ASSERT_EQ(msg.data.eventData.data1, OMX_CommandPortDisable);
 520                 ASSERT_EQ(msg.data.eventData.data2, kPortIndexOutput);
 521
 522                 // set Port Params
 523                 uint32_t nFrameWidth, nFrameHeight, xFramerate;
 524                 OMX_COLOR_FORMATTYPE eColorFormat =
 525                     OMX_COLOR_FormatYUV420Planar;
 526                 getInputChannelInfo(omxNode, kPortIndexInput, &nFrameWidth,
 527                                     &nFrameHeight, &xFramerate);
 528                 setDefaultPortParam(omxNode, kPortIndexOutput,
 529                                     OMX_VIDEO_CodingUnused, eColorFormat,
 530                                     nFrameWidth, nFrameHeight, 0, xFramerate);
 531
 532                 // If you can disable a port, then you should be able to
 533                 // enable
 534                 // it as well
 535                 status = omxNode->sendCommand(
 536                     toRawCommandType(OMX_CommandPortEnable), kPortIndexOutput);
 537                 ASSERT_EQ(status,
 538                           android::hardware::media::omx::V1_0::Status::OK);
 539
 540                 // do not enable the port until all the buffers are supplied
 541                 status = observer->dequeueMessage(&msg, DEFAULT_TIMEOUT,
 542                                                   iBuffer, oBuffer);
 543                 ASSERT_EQ(
 544                     status,
 545                     android::hardware::media::omx::V1_0::Status::TIMED_OUT);
 546
 547                 allocatePortBuffers(omxNode, oBuffer, kPortIndexOutput,
 548                                     oPortMode);
 549                 if (oPortMode != PortMode::PRESET_BYTE_BUFFER) {
 550                     OMX_PARAM_PORTDEFINITIONTYPE portDef;
 551
 552                     status = getPortParam(omxNode, OMX_IndexParamPortDefinition,
 553                                           kPortIndexOutput, &portDef);
 554                     ASSERT_EQ(
 555                         status,
 556                         ::android::hardware::media::omx::V1_0::Status::OK);
 557                     allocateGraphicBuffers(omxNode, kPortIndexOutput, oBuffer,
 558                                            portDef.format.video.nFrameWidth,
 559                                            portDef.format.video.nFrameHeight,
 560                                            &portDef.format.video.nStride,
 561                                            portDef.nBufferCountActual);
 562                 }
 563                 status = observer->dequeueMessage(&msg, DEFAULT_TIMEOUT,
 564                                                   iBuffer, oBuffer);
 565                 ASSERT_EQ(status,
 566                           android::hardware::media::omx::V1_0::Status::OK);
 567                 ASSERT_EQ(msg.type, Message::Type::EVENT);
 568                 ASSERT_EQ(msg.data.eventData.data1, OMX_CommandPortEnable);
 569                 ASSERT_EQ(msg.data.eventData.data2, kPortIndexOutput);
 570
 571                 // dispatch output buffers
 572                 for (size_t i = 0; i < oBuffer->size(); i++) {
 573                     dispatchOutputBuffer(omxNode, oBuffer, i, oPortMode);
 574                 }
 575             } else {
 576                 ASSERT_TRUE(false);
 577             }
 578         } else if (msg.data.eventData.data2 ==
 579                    OMX_IndexConfigCommonOutputCrop) {
 580             std::cout << "[          ] Warning ! OMX_EventPortSettingsChanged/ "
 581                          "OMX_IndexConfigCommonOutputCrop not handled \n";
 582         } else if (msg.data.eventData.data2 == OMX_IndexVendorStartUnused + 3) {
 583             std::cout << "[          ] Warning ! OMX_EventPortSettingsChanged/ "
 584                          "kDescribeColorAspectsIndex not handled \n";
 585         }
 586     } else if (msg.data.eventData.event == OMX_EventError) {
 587         std::cout << "[          ] Warning ! OMX_EventError/ "
 588                      "Decode Frame Call might be failed \n";
 589         return;
 590     } else {
 591         // something unexpected happened
 592         ASSERT_TRUE(false);
 593     }
 594 }
 595
 596 // blocking call to ensures application to Wait till all the inputs are consumed
 597 void waitOnInputConsumption(sp<IOmxNode> omxNode, sp<CodecObserver> observer,
 598                             android::Vector<BufferInfo>* iBuffer,
 599                             android::Vector<BufferInfo>* oBuffer,
 600                             OMX_U32 kPortIndexInput, OMX_U32 kPortIndexOutput,
 601                             PortMode oPortMode) {
 602     android::hardware::media::omx::V1_0::Status status;
 603     Message msg;
 604     int timeOut = TIMEOUT_COUNTER_Q;
 605
 606     while (timeOut--) {
 607         size_t i = 0;
 608         status =
 609             observer->dequeueMessage(&msg, DEFAULT_TIMEOUT_Q, iBuffer, oBuffer);
 610         if (status == android::hardware::media::omx::V1_0::Status::OK) {
 611             EXPECT_EQ(msg.type, Message::Type::EVENT);
 612             portReconfiguration(omxNode, observer, iBuffer, oBuffer,
 613                                 kPortIndexInput, kPortIndexOutput, msg,
 614                                 oPortMode, nullptr);
 615         }
 616         // status == TIMED_OUT, it could be due to process time being large
 617         // than DEFAULT_TIMEOUT or component needs output buffers to start
 618         // processing.
 619         for (; i < iBuffer->size(); i++) {
 620             if ((*iBuffer)[i].owner != client) break;
 621         }
 622         if (i == iBuffer->size()) break;
 623
 624         // Dispatch an output buffer assuming outQueue.empty() is true
 625         size_t index;
 626         if ((index = getEmptyBufferID(oBuffer)) < oBuffer->size()) {
 627             dispatchOutputBuffer(omxNode, oBuffer, index, oPortMode);
 628             timeOut = TIMEOUT_COUNTER_Q;
 629         }
 630     }
 631 }
 632
 633 // Decode N Frames
 634 void decodeNFrames(sp<IOmxNode> omxNode, sp<CodecObserver> observer,
 635                    android::Vector<BufferInfo>* iBuffer,
 636                    android::Vector<BufferInfo>* oBuffer,
 637                    OMX_U32 kPortIndexInput, OMX_U32 kPortIndexOutput,
 638                    std::ifstream& eleStream, android::Vector<FrameData>* Info,
 639                    int offset, int range, PortMode oPortMode,
 640                    bool signalEOS = true) {
 641     android::hardware::media::omx::V1_0::Status status;
 642     Message msg;
 643
 644     // dispatch output buffers
 645     for (size_t i = 0; i < oBuffer->size(); i++) {
 646         dispatchOutputBuffer(omxNode, oBuffer, i, oPortMode);
 647     }
 648     // dispatch input buffers
 649     uint32_t flags = 0;
 650     int frameID = offset;
 651     for (size_t i = 0; (i < iBuffer->size()) && (frameID < (int)Info->size()) &&
 652                        (frameID < (offset + range));
 653          i++) {
 654         char* ipBuffer = static_cast<char*>(
 655             static_cast<void*>((*iBuffer)[i].mMemory->getPointer()));
 656         ASSERT_LE((*Info)[frameID].bytesCount,
 657                   static_cast<int>((*iBuffer)[i].mMemory->getSize()));
 658         eleStream.read(ipBuffer, (*Info)[frameID].bytesCount);
 659         ASSERT_EQ(eleStream.gcount(), (*Info)[frameID].bytesCount);
 660         flags = (*Info)[frameID].flags;
 661         if (signalEOS && ((frameID == (int)Info->size() - 1) ||
 662                           (frameID == (offset + range - 1))))
 663             flags |= OMX_BUFFERFLAG_EOS;
 664         dispatchInputBuffer(omxNode, iBuffer, i, (*Info)[frameID].bytesCount,
 665                             flags, (*Info)[frameID].timestamp);
 666         frameID++;
 667     }
 668
 669     int timeOut = TIMEOUT_COUNTER_Q;
 670     bool iQueued, oQueued;
 671     while (1) {
 672         iQueued = oQueued = false;
 673         status =
 674             observer->dequeueMessage(&msg, DEFAULT_TIMEOUT_Q, iBuffer, oBuffer);
 675
 676         // Port Reconfiguration
 677         if (status == android::hardware::media::omx::V1_0::Status::OK &&
 678             msg.type == Message::Type::EVENT) {
 679             portReconfiguration(omxNode, observer, iBuffer, oBuffer,
 680                                 kPortIndexInput, kPortIndexOutput, msg,
 681                                 oPortMode, nullptr);
 682         }
 683
 684         if (frameID == (int)Info->size() || frameID == (offset + range)) break;
 685
 686         // Dispatch input buffer
 687         size_t index = 0;
 688         if ((index = getEmptyBufferID(iBuffer)) < iBuffer->size()) {
 689             char* ipBuffer = static_cast<char*>(
 690                 static_cast<void*>((*iBuffer)[index].mMemory->getPointer()));
 691             ASSERT_LE((*Info)[frameID].bytesCount,
 692                       static_cast<int>((*iBuffer)[index].mMemory->getSize()));
 693             eleStream.read(ipBuffer, (*Info)[frameID].bytesCount);
 694             ASSERT_EQ(eleStream.gcount(), (*Info)[frameID].bytesCount);
 695             flags = (*Info)[frameID].flags;
 696             if (signalEOS && ((frameID == (int)Info->size() - 1) ||
 697                               (frameID == (offset + range - 1))))
 698                 flags |= OMX_BUFFERFLAG_EOS;
 699             dispatchInputBuffer(omxNode, iBuffer, index,
 700                                 (*Info)[frameID].bytesCount, flags,
 701                                 (*Info)[frameID].timestamp);
 702             frameID++;
 703             iQueued = true;
 704         }
 705         if ((index = getEmptyBufferID(oBuffer)) < oBuffer->size()) {
 706             dispatchOutputBuffer(omxNode, oBuffer, index, oPortMode);
 707             oQueued = true;
 708         }
 709         if (iQueued || oQueued)
 710             timeOut = TIMEOUT_COUNTER_Q;
 711         else
 712             timeOut--;
 713         if (timeOut == 0) {
 714             EXPECT_TRUE(false) << "Wait on Input/Output is found indefinite";
 715             break;
 716         }
 717     }
 718 }
 719
 720 // set component role
 721 TEST_F(VideoDecHidlTest, SetRole) {
 722     description("Test Set Component Role");
 723     if (disableTest) return;
 724     android::hardware::media::omx::V1_0::Status status;
 725     status = setRole(omxNode, gEnv->getRole().c_str());
 726     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 727 }
 728
 729 // port format enumeration
 730 TEST_F(VideoDecHidlTest, EnumeratePortFormat) {
 731     description("Test Component on Mandatory Port Parameters (Port Format)");
 732     if (disableTest) return;
 733     android::hardware::media::omx::V1_0::Status status;
 734     uint32_t kPortIndexInput = 0, kPortIndexOutput = 1;
 735     OMX_COLOR_FORMATTYPE eColorFormat = OMX_COLOR_FormatYUV420Planar;
 736     OMX_U32 xFramerate = (24U << 16);
 737     status = setRole(omxNode, gEnv->getRole().c_str());
 738     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 739     OMX_PORT_PARAM_TYPE params;
 740     status = getParam(omxNode, OMX_IndexParamVideoInit, &params);
 741     if (status == ::android::hardware::media::omx::V1_0::Status::OK) {
 742         ASSERT_EQ(params.nPorts, 2U);
 743         kPortIndexInput = params.nStartPortNumber;
 744         kPortIndexOutput = kPortIndexInput + 1;
 745     }
 746     status = setVideoPortFormat(omxNode, kPortIndexInput, eCompressionFormat,
 747                                 OMX_COLOR_FormatUnused, 0U);
 748     EXPECT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 749     status =
 750         setVideoPortFormat(omxNode, kPortIndexOutput, OMX_VIDEO_CodingUnused,
 751                            eColorFormat, xFramerate);
 752     EXPECT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 753 }
 754
 755 // test port settings reconfiguration, elementary stream decode and timestamp
 756 // deviation
 757 TEST_F(VideoDecHidlTest, DecodeTest) {
 758     description("Tests Port Reconfiguration, Decode and timestamp deviation");
 759     if (disableTest) return;
 760     android::hardware::media::omx::V1_0::Status status;
 761     uint32_t kPortIndexInput = 0, kPortIndexOutput = 1;
 762     status = setRole(omxNode, gEnv->getRole().c_str());
 763     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 764     OMX_PORT_PARAM_TYPE params;
 765     status = getParam(omxNode, OMX_IndexParamVideoInit, &params);
 766     if (status == ::android::hardware::media::omx::V1_0::Status::OK) {
 767         ASSERT_EQ(params.nPorts, 2U);
 768         kPortIndexInput = params.nStartPortNumber;
 769         kPortIndexOutput = kPortIndexInput + 1;
 770     }
 771     char mURL[512], info[512];
 772     strcpy(mURL, gEnv->getRes().c_str());
 773     strcpy(info, gEnv->getRes().c_str());
 774     GetURLForComponent(compName, mURL, info);
 775
 776     std::ifstream eleStream, eleInfo;
 777
 778     eleInfo.open(info);
 779     ASSERT_EQ(eleInfo.is_open(), true);
 780     android::Vector<FrameData> Info;
 781     int bytesCount = 0;
 782     uint32_t flags = 0;
 783     uint32_t timestamp = 0;
 784     timestampDevTest = true;
 785     while (1) {
 786         if (!(eleInfo >> bytesCount)) break;
 787         eleInfo >> flags;
 788         eleInfo >> timestamp;
 789         Info.push_back({bytesCount, flags, timestamp});
 790         if (flags != OMX_BUFFERFLAG_CODECCONFIG)
 791             timestampUslist.push_back(timestamp);
 792     }
 793     eleInfo.close();
 794
 795     // set port mode
 796     portMode[0] = PortMode::PRESET_BYTE_BUFFER;
 797     portMode[1] = PortMode::DYNAMIC_ANW_BUFFER;
 798     status = omxNode->setPortMode(kPortIndexInput, portMode[0]);
 799     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 800     status = omxNode->setPortMode(kPortIndexOutput, portMode[1]);
 801     if (status != ::android::hardware::media::omx::V1_0::Status::OK) {
 802         portMode[1] = PortMode::PRESET_BYTE_BUFFER;
 803         status = omxNode->setPortMode(kPortIndexOutput, portMode[1]);
 804         ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 805     }
 806
 807     // set Port Params
 808     uint32_t nFrameWidth, nFrameHeight, xFramerate;
 809     OMX_COLOR_FORMATTYPE eColorFormat = OMX_COLOR_FormatYUV420Planar;
 810     getInputChannelInfo(omxNode, kPortIndexInput, &nFrameWidth, &nFrameHeight,
 811                         &xFramerate);
 812     setDefaultPortParam(omxNode, kPortIndexOutput, OMX_VIDEO_CodingUnused,
 813                         eColorFormat, nFrameWidth, nFrameHeight, 0, xFramerate);
 814     omxNode->prepareForAdaptivePlayback(kPortIndexOutput, false, 1920, 1080);
 815
 816     android::Vector<BufferInfo> iBuffer, oBuffer;
 817
 818     // set state to idle
 819     changeStateLoadedtoIdle(omxNode, observer, &iBuffer, &oBuffer,
 820                             kPortIndexInput, kPortIndexOutput, portMode);
 821     // set state to executing
 822     changeStateIdletoExecute(omxNode, observer);
 823
 824     if (portMode[1] != PortMode::PRESET_BYTE_BUFFER) {
 825         OMX_PARAM_PORTDEFINITIONTYPE portDef;
 826
 827         status = getPortParam(omxNode, OMX_IndexParamPortDefinition,
 828                               kPortIndexOutput, &portDef);
 829         ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 830         allocateGraphicBuffers(
 831             omxNode, kPortIndexOutput, &oBuffer,
 832             portDef.format.video.nFrameWidth, portDef.format.video.nFrameHeight,
 833             &portDef.format.video.nStride, portDef.nBufferCountActual);
 834     }
 835
 836     // Port Reconfiguration
 837     eleStream.open(mURL, std::ifstream::binary);
 838     ASSERT_EQ(eleStream.is_open(), true);
 839     decodeNFrames(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
 840                   kPortIndexOutput, eleStream, &Info, 0, (int)Info.size(),
 841                   portMode[1]);
 842     eleStream.close();
 843     waitOnInputConsumption(omxNode, observer, &iBuffer, &oBuffer,
 844                            kPortIndexInput, kPortIndexOutput, portMode[1]);
 845     testEOS(omxNode, observer, &iBuffer, &oBuffer, false, eosFlag, portMode,
 846             portReconfiguration, kPortIndexInput, kPortIndexOutput, nullptr);
 847     EXPECT_EQ(timestampUslist.empty(), true);
 848     // set state to idle
 849     changeStateExecutetoIdle(omxNode, observer, &iBuffer, &oBuffer);
 850     // set state to executing
 851     changeStateIdletoLoaded(omxNode, observer, &iBuffer, &oBuffer,
 852                             kPortIndexInput, kPortIndexOutput);
 853 }
 854
 855 // end of sequence test
 856 TEST_F(VideoDecHidlTest, EOSTest_M) {
 857     description("Test End of stream monkeying");
 858     if (disableTest) return;
 859     android::hardware::media::omx::V1_0::Status status;
 860     uint32_t kPortIndexInput = 0, kPortIndexOutput = 1;
 861     status = setRole(omxNode, gEnv->getRole().c_str());
 862     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 863     OMX_PORT_PARAM_TYPE params;
 864     status = getParam(omxNode, OMX_IndexParamVideoInit, &params);
 865     if (status == ::android::hardware::media::omx::V1_0::Status::OK) {
 866         ASSERT_EQ(params.nPorts, 2U);
 867         kPortIndexInput = params.nStartPortNumber;
 868         kPortIndexOutput = kPortIndexInput + 1;
 869     }
 870
 871     // set Port Params
 872     uint32_t nFrameWidth, nFrameHeight, xFramerate;
 873     OMX_COLOR_FORMATTYPE eColorFormat = OMX_COLOR_FormatYUV420Planar;
 874     getInputChannelInfo(omxNode, kPortIndexInput, &nFrameWidth, &nFrameHeight,
 875                         &xFramerate);
 876     setDefaultPortParam(omxNode, kPortIndexOutput, OMX_VIDEO_CodingUnused,
 877                         eColorFormat, nFrameWidth, nFrameHeight, 0, xFramerate);
 878
 879     // set port mode
 880     PortMode portMode[2];
 881     portMode[0] = portMode[1] = PortMode::PRESET_BYTE_BUFFER;
 882     status = omxNode->setPortMode(kPortIndexInput, portMode[0]);
 883     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 884     status = omxNode->setPortMode(kPortIndexOutput, portMode[1]);
 885     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 886
 887     android::Vector<BufferInfo> iBuffer, oBuffer;
 888
 889     // set state to idle
 890     changeStateLoadedtoIdle(omxNode, observer, &iBuffer, &oBuffer,
 891                             kPortIndexInput, kPortIndexOutput, portMode);
 892     // set state to executing
 893     changeStateIdletoExecute(omxNode, observer);
 894
 895     // request EOS at the start
 896     testEOS(omxNode, observer, &iBuffer, &oBuffer, true, eosFlag, portMode,
 897             portReconfiguration, kPortIndexInput, kPortIndexOutput, nullptr);
 898     flushPorts(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
 899                kPortIndexOutput);
 900     EXPECT_GE(framesReceived, 0U);
 901     framesReceived = 0;
 902     timestampUs = 0;
 903
 904     // set state to idle
 905     changeStateExecutetoIdle(omxNode, observer, &iBuffer, &oBuffer);
 906     // set state to executing
 907     changeStateIdletoLoaded(omxNode, observer, &iBuffer, &oBuffer,
 908                             kPortIndexInput, kPortIndexOutput);
 909 }
 910
 911 // end of sequence test
 912 TEST_F(VideoDecHidlTest, ThumbnailTest) {
 913     description("Test Request for thumbnail");
 914     if (disableTest) return;
 915     android::hardware::media::omx::V1_0::Status status;
 916     uint32_t kPortIndexInput = 0, kPortIndexOutput = 1;
 917     status = setRole(omxNode, gEnv->getRole().c_str());
 918     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 919     OMX_PORT_PARAM_TYPE params;
 920     status = getParam(omxNode, OMX_IndexParamVideoInit, &params);
 921     if (status == ::android::hardware::media::omx::V1_0::Status::OK) {
 922         ASSERT_EQ(params.nPorts, 2U);
 923         kPortIndexInput = params.nStartPortNumber;
 924         kPortIndexOutput = kPortIndexInput + 1;
 925     }
 926     char mURL[512], info[512];
 927     strcpy(mURL, gEnv->getRes().c_str());
 928     strcpy(info, gEnv->getRes().c_str());
 929     GetURLForComponent(compName, mURL, info);
 930
 931     std::ifstream eleStream, eleInfo;
 932
 933     eleInfo.open(info);
 934     ASSERT_EQ(eleInfo.is_open(), true);
 935     android::Vector<FrameData> Info;
 936     int bytesCount = 0;
 937     uint32_t flags = 0;
 938     uint32_t timestamp = 0;
 939     while (1) {
 940         if (!(eleInfo >> bytesCount)) break;
 941         eleInfo >> flags;
 942         eleInfo >> timestamp;
 943         Info.push_back({bytesCount, flags, timestamp});
 944     }
 945     eleInfo.close();
 946
 947     // set Port Params
 948     uint32_t nFrameWidth, nFrameHeight, xFramerate;
 949     OMX_COLOR_FORMATTYPE eColorFormat = OMX_COLOR_FormatYUV420Planar;
 950     getInputChannelInfo(omxNode, kPortIndexInput, &nFrameWidth, &nFrameHeight,
 951                         &xFramerate);
 952     setDefaultPortParam(omxNode, kPortIndexOutput, OMX_VIDEO_CodingUnused,
 953                         eColorFormat, nFrameWidth, nFrameHeight, 0, xFramerate);
 954
 955     // set port mode
 956     PortMode portMode[2];
 957     portMode[0] = portMode[1] = PortMode::PRESET_BYTE_BUFFER;
 958     status = omxNode->setPortMode(kPortIndexInput, portMode[0]);
 959     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 960     status = omxNode->setPortMode(kPortIndexOutput, portMode[1]);
 961     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
 962
 963     android::Vector<BufferInfo> iBuffer, oBuffer;
 964
 965     // set state to idle
 966     changeStateLoadedtoIdle(omxNode, observer, &iBuffer, &oBuffer,
 967                             kPortIndexInput, kPortIndexOutput, portMode);
 968     // set state to executing
 969     changeStateIdletoExecute(omxNode, observer);
 970
 971     // request EOS for thumbnail
 972     size_t i = 0;
 973     while (!(Info[i].flags & OMX_BUFFERFLAG_SYNCFRAME)) i++;
 974     eleStream.open(mURL, std::ifstream::binary);
 975     ASSERT_EQ(eleStream.is_open(), true);
 976     decodeNFrames(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
 977                   kPortIndexOutput, eleStream, &Info, 0, i + 1, portMode[1]);
 978     eleStream.close();
 979     waitOnInputConsumption(omxNode, observer, &iBuffer, &oBuffer,
 980                            kPortIndexInput, kPortIndexOutput, portMode[1]);
 981     testEOS(omxNode, observer, &iBuffer, &oBuffer, false, eosFlag, portMode,
 982             portReconfiguration, kPortIndexInput, kPortIndexOutput, nullptr);
 983     flushPorts(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
 984                kPortIndexOutput);
 985     EXPECT_GE(framesReceived, 1U);
 986     framesReceived = 0;
 987     timestampUs = 0;
 988
 989     eleStream.open(mURL, std::ifstream::binary);
 990     ASSERT_EQ(eleStream.is_open(), true);
 991     decodeNFrames(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
 992                   kPortIndexOutput, eleStream, &Info, 0, i + 1, portMode[1],
 993                   false);
 994     eleStream.close();
 995     waitOnInputConsumption(omxNode, observer, &iBuffer, &oBuffer,
 996                            kPortIndexInput, kPortIndexOutput, portMode[1]);
 997     testEOS(omxNode, observer, &iBuffer, &oBuffer, true, eosFlag, portMode,
 998             portReconfiguration, kPortIndexInput, kPortIndexOutput, nullptr);
 999     flushPorts(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
1000                kPortIndexOutput);
1001     EXPECT_GE(framesReceived, 1U);
1002     framesReceived = 0;
1003     timestampUs = 0;
1004
1005     // set state to idle
1006     changeStateExecutetoIdle(omxNode, observer, &iBuffer, &oBuffer);
1007     // set state to executing
1008     changeStateIdletoLoaded(omxNode, observer, &iBuffer, &oBuffer,
1009                             kPortIndexInput, kPortIndexOutput);
1010 }
1011
1012 // end of sequence test
1013 TEST_F(VideoDecHidlTest, SimpleEOSTest) {
1014     description("Test End of stream");
1015     if (disableTest) return;
1016     android::hardware::media::omx::V1_0::Status status;
1017     uint32_t kPortIndexInput = 0, kPortIndexOutput = 1;
1018     status = setRole(omxNode, gEnv->getRole().c_str());
1019     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
1020     OMX_PORT_PARAM_TYPE params;
1021     status = getParam(omxNode, OMX_IndexParamVideoInit, &params);
1022     if (status == ::android::hardware::media::omx::V1_0::Status::OK) {
1023         ASSERT_EQ(params.nPorts, 2U);
1024         kPortIndexInput = params.nStartPortNumber;
1025         kPortIndexOutput = kPortIndexInput + 1;
1026     }
1027     char mURL[512], info[512];
1028     strcpy(mURL, gEnv->getRes().c_str());
1029     strcpy(info, gEnv->getRes().c_str());
1030     GetURLForComponent(compName, mURL, info);
1031
1032     std::ifstream eleStream, eleInfo;
1033
1034     eleInfo.open(info);
1035     ASSERT_EQ(eleInfo.is_open(), true);
1036     android::Vector<FrameData> Info;
1037     int bytesCount = 0;
1038     uint32_t flags = 0;
1039     uint32_t timestamp = 0;
1040     while (1) {
1041         if (!(eleInfo >> bytesCount)) break;
1042         eleInfo >> flags;
1043         eleInfo >> timestamp;
1044         Info.push_back({bytesCount, flags, timestamp});
1045     }
1046     eleInfo.close();
1047
1048     // set Port Params
1049     uint32_t nFrameWidth, nFrameHeight, xFramerate;
1050     OMX_COLOR_FORMATTYPE eColorFormat = OMX_COLOR_FormatYUV420Planar;
1051     getInputChannelInfo(omxNode, kPortIndexInput, &nFrameWidth, &nFrameHeight,
1052                         &xFramerate);
1053     setDefaultPortParam(omxNode, kPortIndexOutput, OMX_VIDEO_CodingUnused,
1054                         eColorFormat, nFrameWidth, nFrameHeight, 0, xFramerate);
1055
1056     // set port mode
1057     PortMode portMode[2];
1058     portMode[0] = portMode[1] = PortMode::PRESET_BYTE_BUFFER;
1059     status = omxNode->setPortMode(kPortIndexInput, portMode[0]);
1060     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
1061     status = omxNode->setPortMode(kPortIndexOutput, portMode[1]);
1062     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
1063
1064     android::Vector<BufferInfo> iBuffer, oBuffer;
1065
1066     // set state to idle
1067     changeStateLoadedtoIdle(omxNode, observer, &iBuffer, &oBuffer,
1068                             kPortIndexInput, kPortIndexOutput, portMode);
1069     // set state to executing
1070     changeStateIdletoExecute(omxNode, observer);
1071
1072     // request EOS at the end
1073     eleStream.open(mURL, std::ifstream::binary);
1074     ASSERT_EQ(eleStream.is_open(), true);
1075     decodeNFrames(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
1076                   kPortIndexOutput, eleStream, &Info, 0, (int)Info.size(),
1077                   portMode[1], false);
1078     eleStream.close();
1079     waitOnInputConsumption(omxNode, observer, &iBuffer, &oBuffer,
1080                            kPortIndexInput, kPortIndexOutput, portMode[1]);
1081     testEOS(omxNode, observer, &iBuffer, &oBuffer, true, eosFlag, portMode,
1082             portReconfiguration, kPortIndexInput, kPortIndexOutput, nullptr);
1083     flushPorts(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
1084                kPortIndexOutput);
1085     framesReceived = 0;
1086     timestampUs = 0;
1087
1088     // set state to idle
1089     changeStateExecutetoIdle(omxNode, observer, &iBuffer, &oBuffer);
1090     // set state to executing
1091     changeStateIdletoLoaded(omxNode, observer, &iBuffer, &oBuffer,
1092                             kPortIndexInput, kPortIndexOutput);
1093 }
1094
1095 // test input/output port flush
1096 TEST_F(VideoDecHidlTest, FlushTest) {
1097     description("Test Flush");
1098     if (disableTest) return;
1099     android::hardware::media::omx::V1_0::Status status;
1100     uint32_t kPortIndexInput = 0, kPortIndexOutput = 1;
1101     status = setRole(omxNode, gEnv->getRole().c_str());
1102     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
1103     OMX_PORT_PARAM_TYPE params;
1104     status = getParam(omxNode, OMX_IndexParamVideoInit, &params);
1105     if (status == ::android::hardware::media::omx::V1_0::Status::OK) {
1106         ASSERT_EQ(params.nPorts, 2U);
1107         kPortIndexInput = params.nStartPortNumber;
1108         kPortIndexOutput = kPortIndexInput + 1;
1109     }
1110     char mURL[512], info[512];
1111     strcpy(mURL, gEnv->getRes().c_str());
1112     strcpy(info, gEnv->getRes().c_str());
1113     GetURLForComponent(compName, mURL, info);
1114
1115     std::ifstream eleStream, eleInfo;
1116
1117     eleInfo.open(info);
1118     ASSERT_EQ(eleInfo.is_open(), true);
1119     android::Vector<FrameData> Info;
1120     int bytesCount = 0;
1121     uint32_t flags = 0;
1122     uint32_t timestamp = 0;
1123     while (1) {
1124         if (!(eleInfo >> bytesCount)) break;
1125         eleInfo >> flags;
1126         eleInfo >> timestamp;
1127         Info.push_back({bytesCount, flags, timestamp});
1128     }
1129     eleInfo.close();
1130
1131     // set Port Params
1132     uint32_t nFrameWidth, nFrameHeight, xFramerate;
1133     OMX_COLOR_FORMATTYPE eColorFormat = OMX_COLOR_FormatYUV420Planar;
1134     getInputChannelInfo(omxNode, kPortIndexInput, &nFrameWidth, &nFrameHeight,
1135                         &xFramerate);
1136     setDefaultPortParam(omxNode, kPortIndexOutput, OMX_VIDEO_CodingUnused,
1137                         eColorFormat, nFrameWidth, nFrameHeight, 0, xFramerate);
1138
1139     // set port mode
1140     PortMode portMode[2];
1141     portMode[0] = portMode[1] = PortMode::PRESET_BYTE_BUFFER;
1142     status = omxNode->setPortMode(kPortIndexInput, portMode[0]);
1143     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
1144     status = omxNode->setPortMode(kPortIndexOutput, portMode[1]);
1145     ASSERT_EQ(status, ::android::hardware::media::omx::V1_0::Status::OK);
1146
1147     android::Vector<BufferInfo> iBuffer, oBuffer;
1148
1149     // set state to idle
1150     changeStateLoadedtoIdle(omxNode, observer, &iBuffer, &oBuffer,
1151                             kPortIndexInput, kPortIndexOutput, portMode);
1152     // set state to executing
1153     changeStateIdletoExecute(omxNode, observer);
1154
1155     // Decode 128 frames and flush. here 128 is chosen to ensure there is a key
1156     // frame after this so that the below section can be convered for all
1157     // components
1158     int nFrames = 128;
1159     eleStream.open(mURL, std::ifstream::binary);
1160     ASSERT_EQ(eleStream.is_open(), true);
1161     decodeNFrames(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
1162                   kPortIndexOutput, eleStream, &Info, 0, nFrames, portMode[1],
1163                   false);
1164     flushPorts(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
1165                kPortIndexOutput);
1166     framesReceived = 0;
1167
1168     // Seek to next key frame and start decoding till the end
1169     int index = nFrames;
1170     bool keyFrame = false;
1171     while (index < (int)Info.size()) {
1172         if ((Info[index].flags & OMX_BUFFERFLAG_SYNCFRAME) ==
1173             OMX_BUFFERFLAG_SYNCFRAME) {
1174             timestampUs = Info[index - 1].timestamp;
1175             keyFrame = true;
1176             break;
1177         }
1178         eleStream.ignore(Info[index].bytesCount);
1179         index++;
1180     }
1181     if (keyFrame) {
1182         decodeNFrames(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
1183                       kPortIndexOutput, eleStream, &Info, index,
1184                       Info.size() - index, portMode[1], false);
1185     }
1186     flushPorts(omxNode, observer, &iBuffer, &oBuffer, kPortIndexInput,
1187                kPortIndexOutput);
1188     framesReceived = 0;
1189
1190     // set state to idle
1191     changeStateExecutetoIdle(omxNode, observer, &iBuffer, &oBuffer);
1192     // set state to executing
1193     changeStateIdletoLoaded(omxNode, observer, &iBuffer, &oBuffer,
1194                             kPortIndexInput, kPortIndexOutput);
1195 }
1196
1197 int main(int argc, char** argv) {
1198     gEnv = new ComponentTestEnvironment();
1199     ::testing::AddGlobalTestEnvironment(gEnv);
1200     ::testing::InitGoogleTest(&argc, argv);
1201     int status = gEnv->initFromOptions(argc, argv);
1202     if (status == 0) {
1203         status = RUN_ALL_TESTS();
1204         ALOGI("Test result = %d", status);
1205     }
1206     return status;
1207 }