| // Copyright 2020 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_PIXEL_STREAM_MANAGER_H |
| #define COMPUTEPIPE_RUNNER_STREAM_MANAGER_PIXEL_STREAM_MANAGER_H |
| |
| #include <vndk/hardware_buffer.h> |
| |
| #include <map> |
| #include <memory> |
| #include <mutex> |
| #include <vector> |
| |
| #include "InputFrame.h" |
| #include "MemHandle.h" |
| #include "RunnerComponent.h" |
| #include "StreamManager.h" |
| #include "StreamManagerInit.h" |
| |
| namespace android { |
| namespace automotive { |
| namespace computepipe { |
| namespace runner { |
| namespace stream_manager { |
| |
| class PixelMemHandle : public MemHandle { |
| public: |
| explicit PixelMemHandle(int bufferId, int streamId, int additionalUsageFlags = 0); |
| |
| virtual ~PixelMemHandle(); |
| |
| // Delete copy and move constructors as well as copy and move assignment operators. |
| PixelMemHandle(const PixelMemHandle&) = delete; |
| PixelMemHandle(PixelMemHandle&&) = delete; |
| PixelMemHandle operator=(const PixelMemHandle&) = delete; |
| PixelMemHandle operator=(PixelMemHandle&&) = delete; |
| |
| /** |
| * Overrides 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; |
| |
| /* Sets frame info */ |
| Status setFrameData(uint64_t timestamp, const InputFrame& inputFrame); |
| |
| private: |
| const int mBufferId; |
| const int mStreamId; |
| AHardwareBuffer_Desc mDesc; |
| AHardwareBuffer* mBuffer; |
| uint64_t mTimestamp; |
| int mUsage; |
| }; |
| |
| class PixelStreamManager : 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 pixel packet produced by graph stream |
| Status queuePacket(const InputFrame& frame, uint64_t timestamp) override; |
| /* Make a copy of the packet. */ |
| std::shared_ptr<MemHandle> clonePacket(std::shared_ptr<MemHandle> handle) override; |
| |
| Status handleExecutionPhase(const RunnerEvent& e) override; |
| Status handleStopWithFlushPhase(const RunnerEvent& e) override; |
| Status handleStopImmediatePhase(const RunnerEvent& e) override; |
| |
| explicit PixelStreamManager(std::string name, int streamId); |
| ~PixelStreamManager() = default; |
| |
| private: |
| void freeAllPackets(); |
| std::mutex mLock; |
| std::mutex mStateLock; |
| int mStreamId; |
| uint32_t mMaxInFlightPackets; |
| std::shared_ptr<StreamEngineInterface> mEngine; |
| |
| struct BufferMetadata { |
| int outstandingRefCount; |
| std::shared_ptr<PixelMemHandle> handle; |
| }; |
| |
| std::map<int, BufferMetadata> mBuffersInUse; |
| std::vector<std::shared_ptr<PixelMemHandle>> mBuffersReady; |
| }; |
| |
| } // namespace stream_manager |
| } // namespace runner |
| } // namespace computepipe |
| } // namespace automotive |
| } // namespace android |
| |
| #endif // COMPUTEPIPE_RUNNER_STREAM_MANAGER_PIXEL_STREAM_MANAGER_H |