| // Copyright (C) 2019 The Android Open Source Project |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #ifndef COMPUTEPIPE_RUNNER_STREAM_MANAGER_SEMANTIC_MANAGER_H |
| #define COMPUTEPIPE_RUNNER_STREAM_MANAGER_SEMANTIC_MANAGER_H |
| |
| #include <mutex> |
| |
| #include "InputFrame.h" |
| #include "OutputConfig.pb.h" |
| #include "RunnerComponent.h" |
| #include "StreamManager.h" |
| #include "StreamManagerInit.h" |
| |
| namespace android { |
| namespace automotive { |
| namespace computepipe { |
| namespace runner { |
| namespace stream_manager { |
| |
| class SemanticHandle : public MemHandle { |
| public: |
| static constexpr uint32_t kMaxSemanticDataSize = 1024; |
| /** |
| * Override mem handle methods |
| */ |
| int getStreamId() const override; |
| int getBufferId() const override; |
| proto::PacketType getType() const override; |
| uint64_t getTimeStamp() const override; |
| uint32_t getSize() const override; |
| const char* getData() const override; |
| AHardwareBuffer* getHardwareBuffer() const override; |
| /* set info for the memory. Make a copy */ |
| Status setMemInfo(int streamId, const char* data, uint32_t size, uint64_t timestamp, |
| const proto::PacketType& type); |
| /* Destroy local copy */ |
| ~SemanticHandle(); |
| |
| private: |
| char* mData = nullptr; |
| uint32_t mSize; |
| uint64_t mTimestamp; |
| proto::PacketType mType; |
| int mStreamId; |
| }; |
| |
| class SemanticManager : public StreamManager, StreamManagerInit { |
| public: |
| void setEngineInterface(std::shared_ptr<StreamEngineInterface> engine) override; |
| /* Set Max in flight packets based on client specification */ |
| Status setMaxInFlightPackets(uint32_t maxPackets) override; |
| /* Free previously dispatched packet. Once client has confirmed usage */ |
| Status freePacket(int bufferId) override; |
| /* Queue packet produced by graph stream */ |
| Status queuePacket(const char* data, const uint32_t size, uint64_t timestamp) override; |
| /* Queues an image packet produced by graph stream */ |
| Status queuePacket(const InputFrame& inputData, uint64_t timestamp) override; |
| /* Make a copy of the packet. */ |
| std::shared_ptr<MemHandle> clonePacket(std::shared_ptr<MemHandle> handle) override; |
| /* Override handling of Runner Engine Events */ |
| void notifyEndOfStream(); |
| |
| Status handleExecutionPhase(const RunnerEvent& e) override; |
| Status handleStopWithFlushPhase(const RunnerEvent& e) override; |
| Status handleStopImmediatePhase(const RunnerEvent& e) override; |
| |
| explicit SemanticManager(std::string name, int streamId, const proto::PacketType& type); |
| ~SemanticManager() = default; |
| |
| private: |
| std::mutex mStateLock; |
| int mStreamId; |
| std::shared_ptr<StreamEngineInterface> mEngine; |
| }; |
| } // namespace stream_manager |
| } // namespace runner |
| } // namespace computepipe |
| } // namespace automotive |
| } // namespace android |
| #endif |