// 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.

#include "lib/escher/vk/impl/pipeline_layout_cache.h"

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

namespace escher {
namespace impl {

PipelineLayoutCache::PipelineLayoutCache(ResourceRecycler* recycler)
    : recycler_(recycler) {}

PipelineLayoutCache::~PipelineLayoutCache() = default;

const PipelineLayoutPtr& PipelineLayoutCache::ObtainPipelineLayout(
    const PipelineLayoutSpec& spec) {
  Hasher h;
  h.struc(spec.descriptor_set_layouts);
  h.struc(spec.push_constant_ranges);
  h.u32(spec.attribute_mask);

  Hash hash = h.value();
  auto it = layouts_.find(hash);
  if (it != end(layouts_)) {
    FXL_DCHECK(it->second->spec() == spec);
    return it->second;
  }

  auto pair = layouts_.insert(std::make_pair(
      hash, fxl::MakeRefCounted<PipelineLayout>(recycler_, spec)));
  return pair.first->second;
}

void PipelineLayoutCache::Clear() { layouts_.clear(); }

}  // namespace impl
}  // namespace escher