93 lines
3.3 KiB
C++
93 lines
3.3 KiB
C++
// 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
|