mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-10-19 03:20:14 +00:00
eb4653a67f
To make the decoration manger available everywhere, and to reduce the number of times it needs to be build, I add one the IRContext. As the same time, I move code that modifies decoration instruction into the IRContext from mempass and the decoration manager. This will make it easier to keep everything up to date. This should take care of issue #928.
110 lines
4.5 KiB
C++
110 lines
4.5 KiB
C++
// Copyright (c) 2017 Pierre Moreau
|
|
//
|
|
// 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 LIBSPIRV_OPT_DECORATION_MANAGER_H_
|
|
#define LIBSPIRV_OPT_DECORATION_MANAGER_H_
|
|
|
|
#include <functional>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "instruction.h"
|
|
#include "module.h"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
namespace analysis {
|
|
|
|
// A class for analyzing and managing decorations in an ir::Module.
|
|
class DecorationManager {
|
|
public:
|
|
// Constructs a decoration manager from the given |module|
|
|
explicit DecorationManager(ir::Module* module) : module_(module) {
|
|
AnalyzeDecorations();
|
|
}
|
|
DecorationManager() = delete;
|
|
|
|
// Removes all decorations from |id|, which should not be a group ID.
|
|
void RemoveDecorationsFrom(uint32_t id);
|
|
|
|
// Removes all decorations from the result id of |inst|.
|
|
void RemoveDecoration(ir::Instruction* inst);
|
|
|
|
// Returns a vector of all decorations affecting |id|. If a group is applied
|
|
// to |id|, the decorations of that group are returned rather than the group
|
|
// decoration instruction. If |include_linkage| is not set, linkage
|
|
// decorations won't be returned.
|
|
std::vector<ir::Instruction*> GetDecorationsFor(uint32_t id,
|
|
bool include_linkage);
|
|
std::vector<const ir::Instruction*> GetDecorationsFor(
|
|
uint32_t id, bool include_linkage) const;
|
|
// Returns whether two IDs have the same decorations. Two SpvOpGroupDecorate
|
|
// instructions that apply the same decorations but to different IDs, still
|
|
// count as being the same.
|
|
bool HaveTheSameDecorations(uint32_t id1, uint32_t id2) const;
|
|
// Returns whether two decorations are the same. SpvOpDecorateId is currently
|
|
// not handled and will return false no matter what.
|
|
bool AreDecorationsTheSame(const ir::Instruction* inst1,
|
|
const ir::Instruction* inst2) const;
|
|
|
|
// |f| is run on each decoration instruction for |id| with decoration
|
|
// |decoration|. Processed are all decorations which target |id| either
|
|
// directly or indirectly by Decoration Groups.
|
|
void ForEachDecoration(uint32_t id, uint32_t decoration,
|
|
std::function<void(const ir::Instruction&)> f);
|
|
|
|
// Clone all decorations from one id |from|.
|
|
// The cloned decorations are assigned to the given id |to| and are
|
|
// added to the module. The purpose is to decorate cloned instructions.
|
|
// This function does not check if the id |to| is already decorated.
|
|
// Function |f| can be used to update context information and is called
|
|
// with |false|, before an instruction is going to be changed and
|
|
// with |true| afterwards.
|
|
void CloneDecorations(uint32_t from, uint32_t to, std::function<void(ir::Instruction&, bool)> f);
|
|
|
|
// Informs the decoration manager of a new decoration that it needs to track.
|
|
void AddDecoration(ir::Instruction* inst);
|
|
|
|
private:
|
|
// Removes the instruction from the set of decorations targeting |target_id|.
|
|
void RemoveInstructionFromTarget(ir::Instruction* inst,
|
|
const uint32_t target_id);
|
|
|
|
using IdToDecorationInstsMap =
|
|
std::unordered_map<uint32_t, std::vector<ir::Instruction*>>;
|
|
// Analyzes the defs and uses in the given |module| and populates data
|
|
// structures in this class. Does nothing if |module| is nullptr.
|
|
void AnalyzeDecorations();
|
|
|
|
template <typename T>
|
|
std::vector<T> InternalGetDecorationsFor(uint32_t id, bool include_linkage);
|
|
|
|
// Mapping from ids to the instructions applying a decoration to them. In
|
|
// other words, for each id you get all decoration instructions referencing
|
|
// that id, be it directly (SpvOpDecorate, SpvOpMemberDecorate and
|
|
// SpvOpDecorateId), or indirectly (SpvOpGroupDecorate,
|
|
// SpvOpMemberGroupDecorate).
|
|
IdToDecorationInstsMap id_to_decoration_insts_;
|
|
// Mapping from group ids to all the decoration instructions they apply.
|
|
IdToDecorationInstsMap group_to_decoration_insts_;
|
|
// The enclosing module.
|
|
ir::Module* module_;
|
|
};
|
|
|
|
} // namespace analysis
|
|
} // namespace opt
|
|
} // namespace spvtools
|
|
|
|
#endif // LIBSPIRV_OPT_DECORATION_MANAGER_H_
|