dom/canvas/WebGLShader.h
author Sylvestre Ledru <sledru@mozilla.com>
Fri, 24 Feb 2017 17:04:50 +0100
changeset 489671 05d9746016f47666c00390aacc9f9d62c8ffffb4
parent 489668 cbb8fdf1daf98a15f7d57f6b08d273bdf96aa1a0
permissions -rw-r--r--
Move to 99 chars instead of 80 MozReview-Commit-ID: 6NxbMuFVI7e

/* -*- 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_SHADER_H_
#define WEBGL_SHADER_H_

#include <map>
#include <string>
#include <vector>

#include "GLDefs.h"
#include "mozilla/LinkedList.h"
#include "mozilla/MemoryReporting.h"
#include "nsString.h"
#include "nsWrapperCache.h"

#include "WebGLObjectModel.h"

namespace mozilla {

namespace webgl {
class ShaderValidator;
} // namespace webgl

class WebGLShader final : public nsWrapperCache,
                          public WebGLRefCountedObject<WebGLShader>,
                          public LinkedListElement<WebGLShader>
{
  friend class WebGLContext;
  friend class WebGLProgram;

public:
  WebGLShader(WebGLContext* webgl, GLenum type);

protected:
  ~WebGLShader();

public:
  // GL funcs
  void CompileShader();
  JS::Value GetShaderParameter(GLenum pname) const;
  void GetShaderInfoLog(nsAString* out) const;
  void GetShaderSource(nsAString* out) const;
  void GetShaderTranslatedSource(nsAString* out) const;
  void ShaderSource(const nsAString& source);

  // Util funcs
  bool CanLinkTo(const WebGLShader* prev, nsCString* const out_log) const;
  size_t CalcNumSamplerUniforms() const;
  size_t NumAttributes() const;
  bool FindAttribUserNameByMappedName(const nsACString& mappedName,
                                      nsCString* const out_userName) const;
  bool FindVaryingByMappedName(const nsACString& mappedName,
                               nsCString* const out_userName,
                               bool* const out_isArray) const;
  bool FindUniformByMappedName(const nsACString& mappedName,
                               nsCString* const out_userName,
                               bool* const out_isArray) const;
  bool UnmapUniformBlockName(const nsACString& baseMappedName,
                             nsCString* const out_baseUserName) const;

  void EnumerateFragOutputs(std::map<nsCString, const nsCString>& out_FragOutputs) const;

  bool IsCompiled() const { return mTranslationSuccessful && mCompilationSuccessful; }

private:
  void BindAttribLocation(GLuint prog, const nsCString& userName, GLuint index) const;
  void MapTransformFeedbackVaryings(const std::vector<nsString>& varyings,
                                    std::vector<std::string>* out_mappedVaryings) const;

public:
  // Other funcs
  size_t SizeOfIncludingThis(mozilla::MallocSizeOf mallocSizeOf) const;
  void Delete();

  WebGLContext* GetParentObject() const { return mContext; }

  virtual JSObject* WrapObject(JSContext* js, JS::Handle<JSObject*> givenProto) override;

  NS_INLINE_DECL_CYCLE_COLLECTING_NATIVE_REFCOUNTING(WebGLShader)
  NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_NATIVE_CLASS(WebGLShader)

public:
  const GLuint mGLName;
  const GLenum mType;

protected:
  nsString mSource;
  nsCString mCleanSource;

  UniquePtr<webgl::ShaderValidator> mValidator;
  nsCString mValidationLog;
  bool mTranslationSuccessful;
  nsCString mTranslatedSource;

  bool mCompilationSuccessful;
  nsCString mCompilationLog;
};

} // namespace mozilla

#endif // WEBGL_SHADER_H_