1 files changed, 149 insertions, 0 deletions
diff --git a/chromium/third_party/dawn/src/dawn_native/AttachmentState.cpp b/chromium/third_party/dawn/src/dawn_native/AttachmentState.cpp
new file mode 100644
index 00000000000..d21b01885f9
--- /dev/null
+++ b/chromium/third_party/dawn/src/dawn_native/AttachmentState.cpp
@@ -0,0 +1,149 @@
+// Copyright 2019 The Dawn Authors
+//
+// 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.
+
+#include "dawn_native/AttachmentState.h"
+
+#include "common/BitSetIterator.h"
+#include "common/HashUtils.h"
+#include "dawn_native/Device.h"
+#include "dawn_native/Texture.h"
+
+namespace dawn_native {
+
+    AttachmentStateBlueprint::AttachmentStateBlueprint(
+        const RenderBundleEncoderDescriptor* descriptor)
+        : mSampleCount(descriptor->sampleCount) {
+        for (uint32_t i = 0; i < descriptor->colorFormatsCount; ++i) {
+            mColorAttachmentsSet.set(i);
+            mColorFormats[i] = descriptor->colorFormats[i];
+        }
+        mDepthStencilFormat = descriptor->depthStencilFormat;
+    }
+
+    AttachmentStateBlueprint::AttachmentStateBlueprint(const RenderPipelineDescriptor* descriptor)
+        : mSampleCount(descriptor->sampleCount) {
+        for (uint32_t i = 0; i < descriptor->colorStateCount; ++i) {
+            ASSERT(descriptor->colorStates[i] != nullptr);
+            mColorAttachmentsSet.set(i);
+            mColorFormats[i] = descriptor->colorStates[i]->format;
+        }
+        if (descriptor->depthStencilState != nullptr) {
+            mDepthStencilFormat = descriptor->depthStencilState->format;
+        }
+    }
+
+    AttachmentStateBlueprint::AttachmentStateBlueprint(const RenderPassDescriptor* descriptor) {
+        for (uint32_t i = 0; i < descriptor->colorAttachmentCount; ++i) {
+            TextureViewBase* attachment = descriptor->colorAttachments[i]->attachment;
+            mColorAttachmentsSet.set(i);
+            mColorFormats[i] = attachment->GetFormat().format;
+            if (mSampleCount == 0) {
+                mSampleCount = attachment->GetTexture()->GetSampleCount();
+            } else {
+                ASSERT(mSampleCount == attachment->GetTexture()->GetSampleCount());
+            }
+        }
+        if (descriptor->depthStencilAttachment != nullptr) {
+            TextureViewBase* attachment = descriptor->depthStencilAttachment->attachment;
+            mDepthStencilFormat = attachment->GetFormat().format;
+            if (mSampleCount == 0) {
+                mSampleCount = attachment->GetTexture()->GetSampleCount();
+            } else {
+                ASSERT(mSampleCount == attachment->GetTexture()->GetSampleCount());
+            }
+        }
+        ASSERT(mSampleCount > 0);
+    }
+
+    AttachmentStateBlueprint::AttachmentStateBlueprint(const AttachmentStateBlueprint& rhs) =
+        default;
+
+    size_t AttachmentStateBlueprint::HashFunc::operator()(
+        const AttachmentStateBlueprint* attachmentState) const {
+        size_t hash = 0;
+
+        // Hash color formats
+        HashCombine(&hash, attachmentState->mColorAttachmentsSet);
+        for (uint32_t i : IterateBitSet(attachmentState->mColorAttachmentsSet)) {
+            HashCombine(&hash, attachmentState->mColorFormats[i]);
+        }
+
+        // Hash depth stencil attachment
+        HashCombine(&hash, attachmentState->mDepthStencilFormat);
+
+        // Hash sample count
+        HashCombine(&hash, attachmentState->mSampleCount);
+
+        return hash;
+    }
+
+    bool AttachmentStateBlueprint::EqualityFunc::operator()(
+        const AttachmentStateBlueprint* a,
+        const AttachmentStateBlueprint* b) const {
+        // Check set attachments
+        if (a->mColorAttachmentsSet != b->mColorAttachmentsSet) {
+            return false;
+        }
+
+        // Check color formats
+        for (uint32_t i : IterateBitSet(a->mColorAttachmentsSet)) {
+            if (a->mColorFormats[i] != b->mColorFormats[i]) {
+                return false;
+            }
+        }
+
+        // Check depth stencil format
+        if (a->mDepthStencilFormat != b->mDepthStencilFormat) {
+            return false;
+        }
+
+        // Check sample count
+        if (a->mSampleCount != b->mSampleCount) {
+            return false;
+        }
+
+        return true;
+    }
+
+    AttachmentState::AttachmentState(DeviceBase* device, const AttachmentStateBlueprint& blueprint)
+        : AttachmentStateBlueprint(blueprint), RefCounted(), mDevice(device) {
+    }
+
+    AttachmentState::~AttachmentState() {
+        mDevice->UncacheAttachmentState(this);
+    }
+
+    std::bitset<kMaxColorAttachments> AttachmentState::GetColorAttachmentsMask() const {
+        return mColorAttachmentsSet;
+    }
+
+    dawn::TextureFormat AttachmentState::GetColorAttachmentFormat(uint32_t index) const {
+        ASSERT(mColorAttachmentsSet[index]);
+        return mColorFormats[index];
+    }
+
+    bool AttachmentState::HasDepthStencilAttachment() const {
+        return mDepthStencilFormat != dawn::TextureFormat::Undefined;
+    }
+
+    dawn::TextureFormat AttachmentState::GetDepthStencilFormat() const {
+        ASSERT(HasDepthStencilAttachment());
+        return mDepthStencilFormat;
+    }
+
+    uint32_t AttachmentState::GetSampleCount() const {
+        return mSampleCount;
+    }
+
+}  // namespace dawn_native