public/lib/escher/vk/shader_variant_args.h

// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef LIB_ESCHER_VK_SHADER_VARIANT_ARGS_H_
#define LIB_ESCHER_VK_SHADER_VARIANT_ARGS_H_

#include <string>
#include <utility>
#include <vector>

#include "lib/escher/util/hashable.h"

namespace escher {

// Contains preprocessor definitions to be passed to a shader compiler in order
// to obtain a variant of a ShaderProgram or ShaderModule.  The consumers of
// these args are not required to sort the definitions into a canonical order;
// e.g. if you pass {FOO=1,BAR=2} and {BAR=2,FOO=1} to a ShaderProgramFactory,
// it is likely to generate two different program variants that compile to
// identical VkPipelines.
class ShaderVariantArgs : public Hashable {
 public:
  explicit ShaderVariantArgs(
      std::vector<std::pair<std::string, std::string>> defs)
      : definitions_(std::move(defs)) {}

  ShaderVariantArgs() {}
  ShaderVariantArgs(ShaderVariantArgs&& other) = default;
  ShaderVariantArgs(const ShaderVariantArgs& other) = default;

  ShaderVariantArgs& operator=(ShaderVariantArgs&& other) = default;

  // |Hashable|.
  bool operator==(const ShaderVariantArgs& other) const;
  bool operator!=(const ShaderVariantArgs& other) const;

  // Get/set the name/value pairs of preprocessor definitions that are used in
  // this variant of a ShaderProgram or ShaderModule.
  const std::vector<std::pair<std::string, std::string>>& definitions() const;
  void set_definitions(std::vector<std::pair<std::string, std::string>> defs);

 private:
  // |Hashable|.
  Hash GenerateHash() const override;

  std::vector<std::pair<std::string, std::string>> definitions_;
};

// Inline function definitions.

inline bool ShaderVariantArgs::operator==(
    const ShaderVariantArgs& other) const {
  return hash() == other.hash() && definitions_ == other.definitions_;
}

inline bool ShaderVariantArgs::operator!=(
    const ShaderVariantArgs& other) const {
  return !(*this == other);
}

inline const std::vector<std::pair<std::string, std::string>>&
ShaderVariantArgs::definitions() const {
  return definitions_;
}

inline void ShaderVariantArgs::set_definitions(
    std::vector<std::pair<std::string, std::string>> defs) {
  definitions_ = std::move(defs);
  InvalidateHash();
}

}  // namespace escher

#endif  // LIB_ESCHER_VK_SHADER_VARIANT_ARGS_H_