mirror of
https://github.com/KhronosGroup/SPIRV-Cross.git
synced 2024-11-15 00:11:06 +00:00
d92de00cc1
This is a fairly fundamental change on how IDs are handled.
It serves many purposes:
- Improve performance. We only need to iterate over IDs which are
relevant at any one time.
- Makes sure we iterate through IDs in SPIR-V module declaration order
rather than ID space. IDs don't have to be monotonically increasing,
which was an assumption SPIRV-Cross used to have. It has apparently
never been a problem until now.
- Support LUTs of structs. We do this by interleaving declaration of
constants and struct types in SPIR-V module order.
To support this, the ParsedIR interface needed to change slightly.
Before setting any ID with variant_set<T> we let ParsedIR know
that an ID with a specific type has been added. The surface for change
should be minimal.
ParsedIR will maintain a per-type list of IDs which the cross-compiler
will need to consider for later.
Instead of looping over ir.ids[] (which can be extremely large), we loop
over types now, using:
ir.for_each_typed_id<SPIRVariable>([&](uint32_t id, SPIRVariable &var) {
handle_variable(var);
});
Now we make sure that we're never looking at irrelevant types.
96 lines
2.2 KiB
C++
96 lines
2.2 KiB
C++
/*
|
|
* Copyright 2018-2019 Arm Limited
|
|
*
|
|
* 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 SPIRV_CROSS_PARSER_HPP
|
|
#define SPIRV_CROSS_PARSER_HPP
|
|
|
|
#include "spirv_cross_parsed_ir.hpp"
|
|
#include <stdint.h>
|
|
#include <vector>
|
|
|
|
namespace spirv_cross
|
|
{
|
|
class Parser
|
|
{
|
|
public:
|
|
Parser(const uint32_t *spirv_data, size_t word_count);
|
|
Parser(std::vector<uint32_t> spirv);
|
|
|
|
void parse();
|
|
|
|
ParsedIR &get_parsed_ir()
|
|
{
|
|
return ir;
|
|
}
|
|
|
|
private:
|
|
ParsedIR ir;
|
|
SPIRFunction *current_function = nullptr;
|
|
SPIRBlock *current_block = nullptr;
|
|
|
|
void parse(const Instruction &instr);
|
|
const uint32_t *stream(const Instruction &instr) const;
|
|
|
|
template <typename T, typename... P>
|
|
T &set(uint32_t id, P &&... args)
|
|
{
|
|
ir.add_typed_id(static_cast<Types>(T::type), id);
|
|
auto &var = variant_set<T>(ir.ids.at(id), std::forward<P>(args)...);
|
|
var.self = id;
|
|
return var;
|
|
}
|
|
|
|
template <typename T>
|
|
T &get(uint32_t id)
|
|
{
|
|
return variant_get<T>(ir.ids.at(id));
|
|
}
|
|
|
|
template <typename T>
|
|
T *maybe_get(uint32_t id)
|
|
{
|
|
if (ir.ids.at(id).get_type() == static_cast<Types>(T::type))
|
|
return &get<T>(id);
|
|
else
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename T>
|
|
const T &get(uint32_t id) const
|
|
{
|
|
return variant_get<T>(ir.ids.at(id));
|
|
}
|
|
|
|
template <typename T>
|
|
const T *maybe_get(uint32_t id) const
|
|
{
|
|
if (ir.ids.at(id).get_type() == T::type)
|
|
return &get<T>(id);
|
|
else
|
|
return nullptr;
|
|
}
|
|
|
|
// This must be an ordered data structure so we always pick the same type aliases.
|
|
std::vector<uint32_t> global_struct_cache;
|
|
|
|
bool types_are_logically_equivalent(const SPIRType &a, const SPIRType &b) const;
|
|
bool variable_storage_is_aliased(const SPIRVariable &v) const;
|
|
void make_constant_null(uint32_t id, uint32_t type);
|
|
};
|
|
} // namespace spirv_cross
|
|
|
|
#endif
|