#ifndef GPU_COMMAND_BUFFER_SERVICE_PROGRAM_MANAGER_H_
#define GPU_COMMAND_BUFFER_SERVICE_PROGRAM_MANAGER_H_
#include <stddef.h>
#include <stdint.h>
#include <map>
#include <memory>
#include <string>
#include <vector>
#include "base/check_op.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/raw_ref.h"
#include "base/memory/ref_counted.h"
#include "gpu/command_buffer/service/common_decoder.h"
#include "gpu/command_buffer/service/gl_utils.h"
#include "gpu/command_buffer/service/shader_manager.h"
#include "gpu/gpu_gles2_export.h"
namespace gpu {
class DecoderClient;
struct GpuPreferences;
namespace gles2 {
class FeatureInfo;
class ProgramCache;
class ProgramManager;
class ProgressReporter;
class Shader;
class ShaderManager;
enum class UniformApiType : uint32_t {
kUniformNone = 0,
kUniform1i = 1 << 0,
kUniform2i = 1 << 1,
kUniform3i = 1 << 2,
kUniform4i = 1 << 3,
kUniform1f = 1 << 4,
kUniform2f = 1 << 5,
kUniform3f = 1 << 6,
kUniform4f = 1 << 7,
kUniformMatrix2f = 1 << 8,
kUniformMatrix3f = 1 << 9,
kUniformMatrix4f = 1 << 10,
kUniform1ui = 1 << 11,
kUniform2ui = 1 << 12,
kUniform3ui = 1 << 13,
kUniform4ui = 1 << 14,
kUniformMatrix2x3f = 1 << 15,
kUniformMatrix2x4f = 1 << 16,
kUniformMatrix3x2f = 1 << 17,
kUniformMatrix3x4f = 1 << 18,
kUniformMatrix4x2f = 1 << 19,
kUniformMatrix4x3f = 1 << 20,
};
inline constexpr UniformApiType operator|(UniformApiType a, UniformApiType b) {
return static_cast<UniformApiType>(uint32_t(a) | uint32_t(b));
}
inline constexpr UniformApiType operator&(UniformApiType a, UniformApiType b) {
return static_cast<UniformApiType>(uint32_t(a) & uint32_t(b));
}
class GPU_GLES2_EXPORT Program : public base::RefCounted<Program> {
public:
static const int kMaxAttachedShaders = 2;
enum VaryingsPackingOption { kCountOnlyStaticallyUsed, kCountAll };
struct ProgramOutputInfo {
ProgramOutputInfo(GLuint _color_name,
GLuint _index,
const std::string& _name)
: color_name(_color_name), index(_index), name(_name) {}
ProgramOutputInfo() : color_name(0), index(0) {}
GLuint color_name;
GLuint index;
std::string name;
};
struct UniformInfo {
UniformInfo();
UniformInfo(const UniformInfo& other);
UniformInfo(const std::string& client_name,
GLint client_location_base,
GLenum _type,
bool _is_array,
const std::vector<GLint>& service_locations);
~UniformInfo();
bool IsSampler() const {
switch (type) {
case GL_SAMPLER_2D:
case GL_SAMPLER_2D_RECT_ARB:
case GL_SAMPLER_CUBE:
case GL_SAMPLER_EXTERNAL_OES:
case GL_SAMPLER_3D:
case GL_SAMPLER_2D_SHADOW:
case GL_SAMPLER_2D_ARRAY:
case GL_SAMPLER_2D_ARRAY_SHADOW:
case GL_SAMPLER_CUBE_SHADOW:
case GL_INT_SAMPLER_2D:
case GL_INT_SAMPLER_3D:
case GL_INT_SAMPLER_CUBE:
case GL_INT_SAMPLER_2D_ARRAY:
case GL_UNSIGNED_INT_SAMPLER_2D:
case GL_UNSIGNED_INT_SAMPLER_3D:
case GL_UNSIGNED_INT_SAMPLER_CUBE:
case GL_UNSIGNED_INT_SAMPLER_2D_ARRAY:
return true;
default:
return false;
}
}
GLsizei size;
GLenum type;
UniformApiType accepts_api_type;
GLint fake_location_base;
bool is_array;
std::string name;
std::vector<GLint> element_locations;
std::vector<GLuint> texture_units;
};
struct VertexAttrib {
VertexAttrib(GLsizei size,
GLenum type,
const std::string& name,
GLint location,
size_t location_count)
: size(size),
type(type),
location(location),
location_count(location_count),
name(name) {}
GLsizei size;
GLenum type;
GLint location;
size_t location_count;
std::string name;
};
struct UniformBlockSizeInfo {
uint32_t binding;
uint32_t data_size;
};
template <typename T>
class ShaderVariableLocationEntry {
public:
ShaderVariableLocationEntry()
: shader_variable_(nullptr), inactive_(false) {}
bool IsUnused() const { return !shader_variable_ && !inactive_; }
bool IsInactive() const { return inactive_; }
bool IsActive() const { return shader_variable_ != nullptr; }
void SetInactive() {
shader_variable_ = nullptr;
inactive_ = true;
}
void SetActive(T* shader_variable) {
DCHECK(shader_variable);
shader_variable_ = shader_variable;
inactive_ = false;
}
const T* shader_variable() const {
DCHECK(IsActive());
return shader_variable_;
}
T* shader_variable() {
DCHECK(IsActive());
return shader_variable_;
}
private:
raw_ptr<T> shader_variable_;
bool inactive_;
};
typedef std::vector<UniformInfo> UniformInfoVector;
typedef std::vector<ShaderVariableLocationEntry<UniformInfo>>
UniformLocationVector;
typedef std::vector<VertexAttrib> AttribInfoVector;
typedef std::vector<ProgramOutputInfo> ProgramOutputInfoVector;
typedef std::vector<int> SamplerIndices;
typedef std::map<std::string, GLint> LocationMap;
typedef std::map<std::string, std::pair<GLuint, GLuint>> LocationIndexMap;
typedef std::vector<std::string> StringVector;
Program(ProgramManager* manager, GLuint service_id);
GLuint service_id() const {
return service_id_;
}
const SamplerIndices& sampler_indices() {
return sampler_indices_;
}
const AttribInfoVector& GetAttribInfos() const {
return attrib_infos_;
}
const VertexAttrib* GetAttribInfo(GLint index) const {
return (static_cast<size_t>(index) < attrib_infos_.size())
? &attrib_infos_[index]
: nullptr;
}
GLint GetAttribLocation(const std::string& original_name) const;
const VertexAttrib* GetAttribInfoByLocation(GLuint location) const {
if (location < attrib_location_to_index_map_.size()) {
GLint index = attrib_location_to_index_map_[location];
if (index >= 0) {
return &attrib_infos_[index];
}
}
return nullptr;
}
const UniformInfo* GetUniformInfo(GLint index) const;
const std::string* GetAttribMappedName(
const std::string& original_name) const;
const std::string* GetUniformMappedName(
const std::string& original_name) const;
const std::string* GetOriginalNameFromHashedName(
const std::string& hashed_name) const;
const sh::Varying* GetVaryingInfo(const std::string& hashed_name) const;
const sh::InterfaceBlock* GetInterfaceBlockInfo(
const std::string& hashed_name) const;
GLint GetUniformFakeLocation(const std::string& name) const;
const UniformInfo* GetUniformInfoByFakeLocation(
GLint fake_location, GLint* real_location, GLint* array_index) const;
bool IsInactiveUniformLocationByFakeLocation(GLint fake_location) const;
const ProgramOutputInfo* GetProgramOutputInfo(
const std::string& original_name) const;
void GetProgramInfo(
ProgramManager* manager, CommonDecoder::Bucket* bucket) const;
bool GetUniformBlocks(CommonDecoder::Bucket* bucket) const;
bool GetTransformFeedbackVaryings(CommonDecoder::Bucket* bucket) const;
bool GetUniformsES3(CommonDecoder::Bucket* bucket) const;
GLint GetFragDataLocation(const std::string& original_name) const;
GLint GetFragDataIndex(const std::string& original_name) const;
bool SetSamplers(
GLint num_texture_units, GLint fake_location,
GLsizei count, const GLint* value);
bool IsDeleted() const {
return deleted_;
}
void GetProgramiv(GLenum pname, GLint* params);
bool IsValid() const {
return valid_;
}
bool IsShaderAttached(Shader* shader);
bool AttachShader(ShaderManager* manager, Shader* shader);
void DetachShader(ShaderManager* manager, Shader* shader);
void CompileAttachedShaders();
bool AttachedShadersExist() const;
bool CanLink() const;
bool Link(ShaderManager* manager,
VaryingsPackingOption varyings_packing_option,
DecoderClient* client);
void Validate();
const std::string* log_info() const {
return log_info_.get();
}
bool InUse() const {
DCHECK_GE(use_count_, 0);
return use_count_ != 0;
}
void SetAttribLocationBinding(const std::string& attrib, GLint location) {
bind_attrib_location_map_[attrib] = location;
}
bool SetUniformLocationBinding(const std::string& name, GLint location);
bool DetectShaderVersionMismatch() const;
void SetProgramOutputLocationBinding(const std::string& name,
GLuint colorName);
void SetProgramOutputLocationIndexedBinding(const std::string& name,
GLuint colorName,
GLuint index);
bool DetectAttribLocationBindingConflicts() const;
bool DetectUniformLocationBindingConflicts() const;
bool DetectUniformsMismatch(std::string* conflicting_name) const;
bool DetectInterfaceBlocksMismatch(std::string* conflicting_name) const;
bool DetectVaryingsMismatch(std::string* conflicting_name) const;
bool DetectProgramOutputLocationBindingConflicts() const;
bool DetectBuiltInInvariantConflicts() const;
bool DetectGlobalNameConflicts(std::string* conflicting_name) const;
bool CheckVaryingsPacking(VaryingsPackingOption option) const;
void TransformFeedbackVaryings(GLsizei count, const char* const* varyings,
GLenum buffer_mode);
const LocationMap& bind_attrib_location_map() const {
return bind_attrib_location_map_;
}
const std::vector<std::string>& effective_transform_feedback_varyings()
const {
return effective_transform_feedback_varyings_;
}
GLenum effective_transform_feedback_buffer_mode() const {
return effective_transform_feedback_buffer_mode_;
}
GLint draw_id_uniform_location() const { return draw_id_uniform_location_; }
GLint base_vertex_uniform_location() const {
return base_vertex_uniform_location_;
}
GLint base_instance_uniform_location() const {
return base_instance_uniform_location_;
}
uint32_t fragment_output_type_mask() const {
return fragment_output_type_mask_;
}
uint32_t fragment_output_written_mask() const {
return fragment_output_written_mask_;
}
const std::vector<uint32_t>& vertex_input_base_type_mask() const {
return vertex_input_base_type_mask_;
}
const std::vector<uint32_t>& vertex_input_active_mask() const {
return vertex_input_active_mask_;
}
void SetUniformBlockBinding(GLuint index, GLuint binding);
const std::vector<UniformBlockSizeInfo>& uniform_block_size_info() const {
return uniform_block_size_info_;
}
const std::vector<GLsizeiptr>& GetTransformFeedbackVaryingSizes() const {
return transform_feedback_data_size_per_vertex_;
}
private:
friend class base::RefCounted<Program>;
friend class ProgramManager;
~Program();
void set_log_info(const char* str) {
log_info_.reset(str ? new std::string(str) : nullptr);
}
void ClearLinkStatus() {
link_status_ = false;
}
void IncUseCount() {
++use_count_;
}
void DecUseCount() {
--use_count_;
DCHECK_GE(use_count_, 0);
}
void MarkAsDeleted() {
DCHECK(!deleted_);
deleted_ = true;
}
void Reset();
void Update();
bool UpdateUniforms();
void UpdateProgramOutputs();
void UpdateFragmentOutputBaseTypes();
void UpdateVertexInputBaseTypes();
void UpdateUniformBlockSizeInfo();
void UpdateTransformFeedbackInfo();
std::string ProcessLogInfo(const std::string& log);
void UpdateLogInfo();
void ClearUniforms(std::vector<uint8_t>* zero_buffer);
void UpdateDrawIDUniformLocation();
void UpdateBaseVertexUniformLocation();
void UpdateBaseInstanceUniformLocation();
void ExecuteBindAttribLocationCalls();
bool ExecuteTransformFeedbackVaryingsCall();
void ExecuteProgramOutputBindCalls();
void GetVertexAttribData(
const std::string& name, std::string* original_name, GLenum* type) const;
void DetachShaders(ShaderManager* manager);
static inline size_t GetUniformLocationIndexFromFakeLocation(
GLint fake_location) {
return static_cast<size_t>(fake_location & 0xFFFF);
}
static inline size_t GetArrayElementIndexFromFakeLocation(
GLint fake_location) {
return static_cast<size_t>((fake_location >> 16) & 0xFFFF);
}
const FeatureInfo& feature_info() const;
void ClearVertexInputMasks();
raw_ptr<ProgramManager> manager_;
int use_count_;
GLsizei max_attrib_name_length_;
AttribInfoVector attrib_infos_;
std::vector<GLint> attrib_location_to_index_map_;
GLsizei max_uniform_name_length_;
UniformInfoVector uniform_infos_;
UniformLocationVector uniform_locations_;
SamplerIndices sampler_indices_;
ProgramOutputInfoVector program_output_infos_;
GLuint service_id_;
scoped_refptr<Shader> attached_shaders_[kMaxAttachedShaders];
scoped_refptr<Shader> shaders_from_last_successful_link_[kMaxAttachedShaders];
bool deleted_;
bool valid_;
bool link_status_;
bool uniforms_cleared_;
GLint draw_id_uniform_location_;
GLint base_vertex_uniform_location_;
GLint base_instance_uniform_location_;
std::unique_ptr<std::string> log_info_;
LocationMap bind_attrib_location_map_;
LocationMap bind_uniform_location_map_;
std::vector<std::string> transform_feedback_varyings_;
GLenum transform_feedback_buffer_mode_;
std::vector<std::string> effective_transform_feedback_varyings_;
GLenum effective_transform_feedback_buffer_mode_;
std::vector<GLsizeiptr> transform_feedback_data_size_per_vertex_;
LocationIndexMap bind_program_output_location_index_map_;
std::vector<UniformBlockSizeInfo> uniform_block_size_info_;
uint32_t fragment_output_type_mask_;
uint32_t fragment_output_written_mask_;
std::vector<uint32_t> vertex_input_base_type_mask_;
std::vector<uint32_t> vertex_input_active_mask_;
};
class GPU_GLES2_EXPORT ProgramManager {
public:
ProgramManager(ProgramCache* program_cache,
uint32_t max_varying_vectors,
uint32_t max_draw_buffers,
uint32_t max_dual_source_draw_buffers,
uint32_t max_vertex_attribs,
const GpuPreferences& gpu_preferences,
FeatureInfo* feature_info,
gl::ProgressReporter* progress_reporter);
ProgramManager(const ProgramManager&) = delete;
ProgramManager& operator=(const ProgramManager&) = delete;
~ProgramManager();
void Destroy(bool have_context);
Program* CreateProgram(GLuint client_id, GLuint service_id);
Program* GetProgram(GLuint client_id);
bool GetClientId(GLuint service_id, GLuint* client_id) const;
ProgramCache* program_cache() const;
void MarkAsDeleted(ShaderManager* shader_manager, Program* program);
void UseProgram(Program* program);
void UnuseProgram(ShaderManager* shader_manager, Program* program);
void ClearUniforms(Program* program);
void UpdateDrawIDUniformLocation(Program* program);
void UpdateBaseVertexUniformLocation(Program* program);
void UpdateBaseInstanceUniformLocation(Program* program);
static bool HasBuiltInPrefix(const std::string& name);
bool IsOwned(Program* program) const;
bool HasCachedCompileStatus(Shader* shader) const;
static int32_t MakeFakeLocation(int32_t index, int32_t element);
uint32_t max_varying_vectors() const { return max_varying_vectors_; }
uint32_t max_draw_buffers() const { return max_draw_buffers_; }
uint32_t max_dual_source_draw_buffers() const {
return max_dual_source_draw_buffers_;
}
uint32_t max_vertex_attribs() const { return max_vertex_attribs_; }
private:
friend class Program;
void StartTracking(Program* program);
void StopTracking(Program* program);
void RemoveProgramInfoIfUnused(
ShaderManager* shader_manager, Program* program);
typedef std::map<GLuint, scoped_refptr<Program> > ProgramMap;
ProgramMap programs_;
unsigned int program_count_;
bool have_context_;
std::vector<uint8_t> zero_;
raw_ptr<ProgramCache> program_cache_;
uint32_t max_varying_vectors_;
uint32_t max_draw_buffers_;
uint32_t max_dual_source_draw_buffers_;
uint32_t max_vertex_attribs_;
const raw_ref<const GpuPreferences> gpu_preferences_;
scoped_refptr<FeatureInfo> feature_info_;
raw_ptr<gl::ProgressReporter> progress_reporter_;
};
inline const FeatureInfo& Program::feature_info() const {
return *manager_->feature_info_.get();
}
}
}
#endif