1 files changed, 115 insertions, 0 deletions
diff --git a/dom/canvas/WebGLBuffer.h b/dom/canvas/WebGLBuffer.h
new file mode 100644
index 000000000..883712aad
--- /dev/null
+++ b/dom/canvas/WebGLBuffer.h
@@ -0,0 +1,115 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef WEBGL_BUFFER_H_
+#define WEBGL_BUFFER_H_
+
+#include "GLDefs.h"
+#include "mozilla/LinkedList.h"
+#include "mozilla/UniquePtr.h"
+#include "nsWrapperCache.h"
+
+#include "WebGLObjectModel.h"
+#include "WebGLTypes.h"
+
+namespace mozilla {
+
+class WebGLElementArrayCache;
+
+class WebGLBuffer final
+    : public nsWrapperCache
+    , public WebGLRefCountedObject<WebGLBuffer>
+    , public LinkedListElement<WebGLBuffer>
+{
+    friend class WebGLContext;
+    friend class WebGL2Context;
+    friend class WebGLTexture;
+
+public:
+    enum class Kind {
+        Undefined,
+        ElementArray,
+        OtherData
+    };
+
+    WebGLBuffer(WebGLContext* webgl, GLuint buf);
+
+    void SetContentAfterBind(GLenum target);
+    Kind Content() const { return mContent; }
+
+    void Delete();
+
+    size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const;
+
+    GLenum Usage() const { return mUsage; }
+    size_t ByteLength() const { return mByteLength; }
+
+    bool ElementArrayCacheBufferData(const void* ptr, size_t bufferSizeInBytes);
+
+    void ElementArrayCacheBufferSubData(size_t pos, const void* ptr,
+                                        size_t updateSizeInBytes);
+
+    bool Validate(GLenum type, uint32_t max_allowed, size_t first, size_t count) const;
+    bool ValidateRange(const char* funcName, size_t byteOffset, size_t byteLen) const;
+
+    bool IsElementArrayUsedWithMultipleTypes() const;
+
+    WebGLContext* GetParentObject() const {
+        return mContext;
+    }
+
+    virtual JSObject* WrapObject(JSContext* cx, JS::Handle<JSObject*> givenProto) override;
+
+    bool ValidateCanBindToTarget(const char* funcName, GLenum target);
+    void BufferData(GLenum target, size_t size, const void* data, GLenum usage);
+
+    ////
+
+    static void AddBindCount(GLenum target, WebGLBuffer* buffer, int8_t addVal) {
+        if (!buffer)
+            return;
+
+        if (target == LOCAL_GL_TRANSFORM_FEEDBACK_BUFFER) {
+            MOZ_ASSERT_IF(addVal < 0, buffer->mTFBindCount >= size_t(-addVal));
+            buffer->mTFBindCount += addVal;
+        } else {
+            MOZ_ASSERT_IF(addVal < 0, buffer->mNonTFBindCount >= size_t(-addVal));
+            buffer->mNonTFBindCount += addVal;
+        }
+    }
+
+    static void SetSlot(GLenum target, WebGLBuffer* newBuffer,
+                        WebGLRefPtr<WebGLBuffer>* const out_slot)
+    {
+        WebGLBuffer* const oldBuffer = *out_slot;
+        AddBindCount(target, oldBuffer, -1);
+        AddBindCount(target, newBuffer, +1);
+        *out_slot = newBuffer;
+    }
+
+    bool IsBoundForTF() const { return bool(mTFBindCount); }
+    bool IsBoundForNonTF() const { return bool(mNonTFBindCount); }
+
+    ////
+
+    const GLenum mGLName;
+
+    NS_INLINE_DECL_CYCLE_COLLECTING_NATIVE_REFCOUNTING(WebGLBuffer)
+    NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_NATIVE_CLASS(WebGLBuffer)
+
+protected:
+    ~WebGLBuffer();
+
+    Kind mContent;
+    GLenum mUsage;
+    size_t mByteLength;
+    UniquePtr<WebGLElementArrayCache> mCache;
+    size_t mTFBindCount;
+    size_t mNonTFBindCount;
+};
+
+} // namespace mozilla
+
+#endif // WEBGL_BUFFER_H_