transformation_add_constant_boolean.h 2.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. // Copyright (c) 2019 Google LLC
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #ifndef SOURCE_FUZZ_TRANSFORMATION_ADD_BOOLEAN_CONSTANT_H_
  15. #define SOURCE_FUZZ_TRANSFORMATION_ADD_BOOLEAN_CONSTANT_H_
  16. #include "source/fuzz/protobufs/spirvfuzz_protobufs.h"
  17. #include "source/fuzz/transformation.h"
  18. #include "source/fuzz/transformation_context.h"
  19. #include "source/opt/ir_context.h"
  20. namespace spvtools {
  21. namespace fuzz {
  22. class TransformationAddConstantBoolean : public Transformation {
  23. public:
  24. explicit TransformationAddConstantBoolean(
  25. protobufs::TransformationAddConstantBoolean message);
  26. TransformationAddConstantBoolean(uint32_t fresh_id, bool is_true,
  27. bool is_irrelevant);
  28. // - |message_.fresh_id| must not be used by the module.
  29. // - The module must already contain OpTypeBool.
  30. bool IsApplicable(
  31. opt::IRContext* ir_context,
  32. const TransformationContext& transformation_context) const override;
  33. // - Adds OpConstantTrue (OpConstantFalse) to the module with id
  34. // |message_.fresh_id| if |message_.is_true| holds (does not hold).
  35. // - Also, creates an IdIsIrrelevant fact about |fresh_id| if |is_irrelevant|
  36. // is true.
  37. void Apply(opt::IRContext* ir_context,
  38. TransformationContext* transformation_context) const override;
  39. std::unordered_set<uint32_t> GetFreshIds() const override;
  40. protobufs::Transformation ToMessage() const override;
  41. private:
  42. protobufs::TransformationAddConstantBoolean message_;
  43. };
  44. } // namespace fuzz
  45. } // namespace spvtools
  46. #endif // SOURCE_FUZZ_TRANSFORMATION_ADD_BOOLEAN_CONSTANT_H_