Home | History | Annotate | Download | only in tests
      1 /*
      2  * Copyright (C) 2009 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 #ifndef OMX_HARNESS_H_
     18 
     19 #define OMX_HARNESS_H_
     20 
     21 #include <media/IOMX.h>
     22 #include <utils/List.h>
     23 #include <utils/Vector.h>
     24 #include <utils/threads.h>
     25 
     26 #include <binder/MemoryDealer.h>
     27 #include <android/hidl/allocator/1.0/IAllocator.h>
     28 #include <android/hidl/memory/1.0/IMemory.h>
     29 #include <OMX_Component.h>
     30 
     31 namespace android {
     32 
     33 class MemoryDealer;
     34 
     35 struct Harness : public RefBase {
     36     typedef hidl::memory::V1_0::IMemory TMemory;
     37     typedef hardware::hidl_memory hidl_memory;
     38     enum BufferFlags {
     39         kBufferBusy = 1
     40     };
     41     struct Buffer {
     42         IOMX::buffer_id mID;
     43         sp<IMemory> mMemory;
     44         hidl_memory mHidlMemory;
     45         uint32_t mFlags;
     46     };
     47 
     48     Harness();
     49 
     50     status_t initCheck() const;
     51 
     52     status_t dequeueMessageForNode(omx_message *msg, int64_t timeoutUs = -1);
     53 
     54     status_t dequeueMessageForNodeIgnoringBuffers(
     55             Vector<Buffer> *inputBuffers,
     56             Vector<Buffer> *outputBuffers,
     57             omx_message *msg, int64_t timeoutUs = -1);
     58 
     59     status_t getPortDefinition(
     60             OMX_U32 portIndex, OMX_PARAM_PORTDEFINITIONTYPE *def);
     61 
     62     status_t allocatePortBuffers(
     63             OMX_U32 portIndex, Vector<Buffer> *buffers);
     64 
     65     status_t setRole(const char *role);
     66 
     67     status_t testStateTransitions(
     68             const char *componentName, const char *componentRole);
     69 
     70     status_t testSeek(
     71             const char *componentName, const char *componentRole);
     72 
     73     status_t test(
     74             const char *componentName, const char *componentRole);
     75 
     76     status_t testAll();
     77 
     78 protected:
     79     virtual ~Harness();
     80 
     81 private:
     82     typedef hidl::allocator::V1_0::IAllocator IAllocator;
     83 
     84     friend struct NodeReaper;
     85     struct CodecObserver;
     86 
     87     Mutex mLock;
     88 
     89     status_t mInitCheck;
     90     sp<IOMX> mOMX;
     91     sp<IOMXNode> mOMXNode;
     92     List<omx_message> mMessageQueue;
     93     Condition mMessageAddedCondition;
     94     int32_t mLastMsgGeneration;
     95     int32_t mCurGeneration;
     96     sp<IAllocator> mAllocator;
     97 
     98     status_t initOMX();
     99 
    100     bool handleBufferMessage(
    101             const omx_message &msg,
    102             Vector<Buffer> *inputBuffers,
    103             Vector<Buffer> *outputBuffers);
    104 
    105     void handleMessages(int32_t gen, const std::list<omx_message> &messages);
    106 
    107     Harness(const Harness &);
    108     Harness &operator=(const Harness &);
    109 };
    110 
    111 }  // namespace android
    112 
    113 #endif  // OMX_HARNESS_H_
    114