source/fuzz/transformation_add_image_sample_unused_components.h

   1 // Copyright (c) 2020 André Perez Maselco
   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
  15 #ifndef SOURCE_FUZZ_TRANSFORMATION_ADD_IMAGE_SAMPLE_UNUSED_COMPONENTS_H_
  16 #define SOURCE_FUZZ_TRANSFORMATION_ADD_IMAGE_SAMPLE_UNUSED_COMPONENTS_H_
  17
  18 #include "source/fuzz/protobufs/spirvfuzz_protobufs.h"
  19 #include "source/fuzz/transformation.h"
  20 #include "source/fuzz/transformation_context.h"
  21 #include "source/opt/ir_context.h"
  22
  23 namespace spvtools {
  24 namespace fuzz {
  25
  26 class TransformationAddImageSampleUnusedComponents : public Transformation {
  27  public:
  28   explicit TransformationAddImageSampleUnusedComponents(
  29       protobufs::TransformationAddImageSampleUnusedComponents message);
  30
  31   TransformationAddImageSampleUnusedComponents(
  32       uint32_t coordinate_with_unused_components_id,
  33       const protobufs::InstructionDescriptor& instruction_descriptor);
  34
  35   // - |coordinate_with_unused_components_id| must identify a vector such that
  36   //   the first components match the components of the image sample coordinate.
  37   // - |message_.instruction_descriptor| must identify an image sample
  38   //   instruction
  39   bool IsApplicable(
  40       opt::IRContext* ir_context,
  41       const TransformationContext& transformation_context) const override;
  42
  43   // Add unused components to an image sample coordinate by replacing the
  44   // coordinate with |coordinate_with_unused_components_id|.
  45   void Apply(opt::IRContext* ir_context,
  46              TransformationContext* transformation_context) const override;
  47
  48   std::unordered_set<uint32_t> GetFreshIds() const override;
  49
  50   protobufs::Transformation ToMessage() const override;
  51
  52  private:
  53   protobufs::TransformationAddImageSampleUnusedComponents message_;
  54 };
  55
  56 }  // namespace fuzz
  57 }  // namespace spvtools
  58
  59 #endif  // SOURCE_FUZZ_TRANSFORMATION_ADD_IMAGE_SAMPLE_UNUSED_COMPONENTS_H_