mirror of
https://github.com/bkaradzic/bgfx.git
synced 2026-02-19 05:23:00 +01:00
125 lines
4.2 KiB
C++
125 lines
4.2 KiB
C++
// Copyright (c) 2022-2025 Arm Ltd.
|
|
//
|
|
// 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 SOURCE_OPT_GRAPH_H_
|
|
#define SOURCE_OPT_GRAPH_H_
|
|
|
|
#include "source/opt/instruction.h"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
|
|
struct Graph {
|
|
// Creates a graph instance declared by the given OpGraph instruction
|
|
// |def_inst|.
|
|
inline explicit Graph(std::unique_ptr<Instruction> def_inst);
|
|
explicit Graph(const Graph& f) = delete;
|
|
|
|
// Creates a clone of the graph in the given |context|
|
|
//
|
|
// The parent module will default to null and needs to be explicitly set by
|
|
// the user.
|
|
Graph* Clone(IRContext*) const;
|
|
|
|
// The OpGraph instruction that begins the definition of this graph.
|
|
Instruction& DefInst() { return *def_inst_; }
|
|
const Instruction& DefInst() const { return *def_inst_; }
|
|
|
|
// Appends an input to this graph.
|
|
inline void AddInput(std::unique_ptr<Instruction> inst);
|
|
|
|
// Appends an instruction to this graph.
|
|
inline void AddInstruction(std::unique_ptr<Instruction> inst);
|
|
|
|
// Appends an output to this graph.
|
|
inline void AddOutput(std::unique_ptr<Instruction> inst);
|
|
|
|
// Saves the given graph end instruction.
|
|
void SetGraphEnd(std::unique_ptr<Instruction> end_inst);
|
|
|
|
// Returns the given graph end instruction.
|
|
inline Instruction* EndInst() { return end_inst_.get(); }
|
|
inline const Instruction* EndInst() const { return end_inst_.get(); }
|
|
|
|
// Returns graph's id
|
|
inline uint32_t result_id() const { return def_inst_->result_id(); }
|
|
|
|
// Returns graph's return type id
|
|
inline uint32_t type_id() const { return def_inst_->type_id(); }
|
|
|
|
// Return a read-only reference to the instructions that define the body of
|
|
// the graph.
|
|
const std::vector<std::unique_ptr<Instruction>>& instructions() const {
|
|
return insts_;
|
|
}
|
|
|
|
// Return a read-only reference to the instructions that define the inputs
|
|
// of the graph.
|
|
const std::vector<std::unique_ptr<Instruction>>& inputs() const {
|
|
return inputs_;
|
|
}
|
|
|
|
// Return a read-only reference to the instructions that define the outputs
|
|
// of the graph.
|
|
const std::vector<std::unique_ptr<Instruction>>& outputs() const {
|
|
return outputs_;
|
|
}
|
|
|
|
// Runs the given function |f| on instructions in this graph, in order,
|
|
// and optionally on debug line instructions that might precede them and
|
|
// non-semantic instructions that succceed the function.
|
|
void ForEachInst(const std::function<void(Instruction*)>& f,
|
|
bool run_on_debug_line_insts = false,
|
|
bool run_on_non_semantic_insts = false);
|
|
void ForEachInst(const std::function<void(const Instruction*)>& f,
|
|
bool run_on_debug_line_insts = false,
|
|
bool run_on_non_semantic_insts = false) const;
|
|
|
|
private:
|
|
// The OpGraph instruction that begins the definition of this graph.
|
|
std::unique_ptr<Instruction> def_inst_;
|
|
// All inputs to this graph.
|
|
std::vector<std::unique_ptr<Instruction>> inputs_;
|
|
// All instructions describing this graph
|
|
std::vector<std::unique_ptr<Instruction>> insts_;
|
|
// All outputs of this graph.
|
|
std::vector<std::unique_ptr<Instruction>> outputs_;
|
|
// The OpGraphEnd instruction.
|
|
std::unique_ptr<Instruction> end_inst_;
|
|
};
|
|
|
|
inline Graph::Graph(std::unique_ptr<Instruction> def_inst)
|
|
: def_inst_(std::move(def_inst)) {}
|
|
|
|
inline void Graph::AddInput(std::unique_ptr<Instruction> inst) {
|
|
inputs_.emplace_back(std::move(inst));
|
|
}
|
|
|
|
inline void Graph::AddInstruction(std::unique_ptr<Instruction> inst) {
|
|
insts_.emplace_back(std::move(inst));
|
|
}
|
|
|
|
inline void Graph::AddOutput(std::unique_ptr<Instruction> inst) {
|
|
outputs_.emplace_back(std::move(inst));
|
|
}
|
|
|
|
inline void Graph::SetGraphEnd(std::unique_ptr<Instruction> end_inst) {
|
|
end_inst_ = std::move(end_inst);
|
|
}
|
|
|
|
} // namespace opt
|
|
} // namespace spvtools
|
|
|
|
#endif // SOURCE_OPT_GRAPH_H_
|