/* -*- 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_