| // Copyright (C) 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_GRAPH_INCLUDE_PREBUILTGRAPH_H_ |
| #define COMPUTEPIPE_RUNNER_GRAPH_INCLUDE_PREBUILTGRAPH_H_ |
| |
| #include <string> |
| |
| #include "InputFrame.h" |
| #include "Options.pb.h" |
| #include "PrebuiltEngineInterface.h" |
| #include "RunnerComponent.h" |
| #include "types/Status.h" |
| |
| namespace android { |
| namespace automotive { |
| namespace computepipe { |
| namespace graph { |
| |
| enum PrebuiltGraphState { |
| RUNNING = 0, |
| UNINITIALIZED, |
| FLUSHING, |
| STOPPED, |
| }; |
| |
| enum PrebuiltGraphType { |
| LOCAL = 0, |
| REMOTE = 1, |
| }; |
| |
| class PrebuiltGraph : public runner::RunnerComponentInterface { |
| public: |
| // Gets the graph type. |
| virtual PrebuiltGraphType GetGraphType() const = 0; |
| |
| // Gets the PrebuiltGraphState |
| virtual PrebuiltGraphState GetGraphState() const = 0; |
| |
| // Gets the graph status and reports any error code or if the status is OK. |
| virtual Status GetStatus() const = 0; |
| |
| // Gets the error message from the graph. |
| virtual std::string GetErrorMessage() const = 0; |
| |
| // Gets the supported graph config options. |
| virtual const proto::Options& GetSupportedGraphConfigs() const = 0; |
| |
| // Sets input stream data. The string is expected to be a serialized proto |
| // the definition of which is known to the graph. |
| virtual Status SetInputStreamData(int streamIndex, int64_t timestamp, |
| const std::string& streamData) = 0; |
| |
| // Sets pixel data to the specified input stream index. |
| virtual Status SetInputStreamPixelData(int streamIndex, int64_t timestamp, |
| const runner::InputFrame& inputFrame) = 0; |
| |
| // Start graph profiling. |
| virtual Status StartGraphProfiling() = 0; |
| |
| // Stop graph profiling. |
| virtual Status StopGraphProfiling() = 0; |
| |
| // Collects debugging and profiling information for the graph. The graph |
| // needs to be started with debugging enabled in order to get valid info. |
| virtual std::string GetDebugInfo() = 0; |
| }; |
| |
| PrebuiltGraph* GetLocalGraphFromLibrary( |
| const std::string& prebuiltLib, std::weak_ptr<PrebuiltEngineInterface> engineInterface); |
| |
| std::unique_ptr<PrebuiltGraph> GetRemoteGraphFromAddress( |
| const std::string& address, std::weak_ptr<PrebuiltEngineInterface> engineInterface); |
| |
| } // namespace graph |
| } // namespace computepipe |
| } // namespace automotive |
| } // namespace android |
| |
| #endif // COMPUTEPIPE_RUNNER_GRAPH_INCLUDE_PREBUILTGRAPH_H_ |