mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-10-18 19:20:05 +00:00
c94501352d
A std::set is used instead of std::vector, where the elements are ordered by member index first. Decorations for fields are now looked up by going over the range of decorations for the member only, instead of the whole set. In an ANGLE test that generates a struct with 4096 members, validation goes down from ~140ms to ~90ms. On debug builds, the difference is more pronounced, going down from ~2.5s to ~600ms.
99 lines
3.8 KiB
C++
99 lines
3.8 KiB
C++
// Copyright (c) 2017 Google Inc.
|
|
//
|
|
// 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_VAL_DECORATION_H_
|
|
#define SOURCE_VAL_DECORATION_H_
|
|
|
|
#include <cstdint>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include "source/latest_version_spirv_header.h"
|
|
|
|
namespace spvtools {
|
|
namespace val {
|
|
|
|
// An object of this class represents a specific decoration including its
|
|
// parameters (if any). Decorations are used by OpDecorate and OpMemberDecorate,
|
|
// and they describe certain properties that can be assigned to one or several
|
|
// <id>s.
|
|
//
|
|
// A Decoration object contains the decoration type (an enum), associated
|
|
// literal parameters, and struct member index. If the decoration does not apply
|
|
// to a struct member, then the index is kInvalidIndex. A Decoration object does
|
|
// not store the target Id, i.e. the Id to which it applies. It is
|
|
// possible for the same decoration to be applied to several <id>s (and they
|
|
// might be assigned using separate SPIR-V instructions, possibly using an
|
|
// assignment through GroupDecorate).
|
|
//
|
|
// Example 1: Decoration for an object<id> with no parameters:
|
|
// OpDecorate %obj Flat
|
|
// dec_type_ = SpvDecorationFlat
|
|
// params_ = empty vector
|
|
// struct_member_index_ = kInvalidMember
|
|
//
|
|
// Example 2: Decoration for an object<id> with two parameters:
|
|
// OpDecorate %obj LinkageAttributes "link" Import
|
|
// dec_type_ = SpvDecorationLinkageAttributes
|
|
// params_ = vector { link, Import }
|
|
// struct_member_index_ = kInvalidMember
|
|
//
|
|
// Example 3: Decoration for a member of a structure with one parameter:
|
|
// OpMemberDecorate %struct 2 Offset 2
|
|
// dec_type_ = SpvDecorationOffset
|
|
// params_ = vector { 2 }
|
|
// struct_member_index_ = 2
|
|
//
|
|
class Decoration {
|
|
public:
|
|
enum { kInvalidMember = -1 };
|
|
Decoration(SpvDecoration t,
|
|
const std::vector<uint32_t>& parameters = std::vector<uint32_t>(),
|
|
uint32_t member_index = kInvalidMember)
|
|
: dec_type_(t), params_(parameters), struct_member_index_(member_index) {}
|
|
|
|
void set_struct_member_index(uint32_t index) { struct_member_index_ = index; }
|
|
int struct_member_index() const { return struct_member_index_; }
|
|
SpvDecoration dec_type() const { return dec_type_; }
|
|
std::vector<uint32_t>& params() { return params_; }
|
|
const std::vector<uint32_t>& params() const { return params_; }
|
|
|
|
inline bool operator<(const Decoration& rhs) const {
|
|
// Note: Sort by struct_member_index_ first, then type, so look up can be
|
|
// efficient using lower_bound() and upper_bound().
|
|
if (struct_member_index_ < rhs.struct_member_index_) return true;
|
|
if (rhs.struct_member_index_ < struct_member_index_) return false;
|
|
if (dec_type_ < rhs.dec_type_) return true;
|
|
if (rhs.dec_type_ < dec_type_) return false;
|
|
return params_ < rhs.params_;
|
|
}
|
|
inline bool operator==(const Decoration& rhs) const {
|
|
return (dec_type_ == rhs.dec_type_ && params_ == rhs.params_ &&
|
|
struct_member_index_ == rhs.struct_member_index_);
|
|
}
|
|
|
|
private:
|
|
SpvDecoration dec_type_;
|
|
std::vector<uint32_t> params_;
|
|
|
|
// If the decoration applies to a member of a structure type, then the index
|
|
// of the member is stored here. Otherwise, this is kInvalidIndex.
|
|
int struct_member_index_;
|
|
};
|
|
|
|
} // namespace val
|
|
} // namespace spvtools
|
|
|
|
#endif // SOURCE_VAL_DECORATION_H_
|