#include "config.h"
#include "WebGLProgram.h"
#if ENABLE(WEBGL)
#include "InspectorInstrumentation.h"
#include "ScriptExecutionContext.h"
#include "WebGLContextGroup.h"
#include "WebGLRenderingContextBase.h"
#include "WebGLShader.h"
#include <JavaScriptCore/SlotVisitor.h>
#include <JavaScriptCore/SlotVisitorInlines.h>
#include <wtf/Lock.h>
#include <wtf/Locker.h>
#include <wtf/NeverDestroyed.h>
#if !USE(ANGLE)
#include "GraphicsContextGLOpenGL.h"
#endif
namespace WebCore {
HashMap<WebGLProgram*, WebGLRenderingContextBase*>& WebGLProgram::instances(const LockHolder&)
{
static NeverDestroyed<HashMap<WebGLProgram*, WebGLRenderingContextBase*>> instances;
return instances;
}
Lock& WebGLProgram::instancesMutex()
{
static Lock mutex;
return mutex;
}
Ref<WebGLProgram> WebGLProgram::create(WebGLRenderingContextBase& ctx)
{
return adoptRef(*new WebGLProgram(ctx));
}
WebGLProgram::WebGLProgram(WebGLRenderingContextBase& ctx)
: WebGLSharedObject(ctx)
, ContextDestructionObserver(ctx.scriptExecutionContext())
{
ASSERT(scriptExecutionContext());
{
LockHolder lock(instancesMutex());
instances(lock).add(this, &ctx);
}
setObject(ctx.graphicsContextGL()->createProgram());
}
WebGLProgram::~WebGLProgram()
{
InspectorInstrumentation::willDestroyWebGLProgram(*this);
{
LockHolder lock(instancesMutex());
ASSERT(instances(lock).contains(this));
instances(lock).remove(this);
}
if (!hasGroupOrContext())
return;
runDestructor();
}
void WebGLProgram::contextDestroyed()
{
InspectorInstrumentation::willDestroyWebGLProgram(*this);
ContextDestructionObserver::contextDestroyed();
}
void WebGLProgram::deleteObjectImpl(const AbstractLocker& locker, GraphicsContextGL* context3d, PlatformGLObject obj)
{
context3d->deleteProgram(obj);
if (m_vertexShader) {
m_vertexShader->onDetached(locker, context3d);
m_vertexShader = nullptr;
}
if (m_fragmentShader) {
m_fragmentShader->onDetached(locker, context3d);
m_fragmentShader = nullptr;
}
}
unsigned WebGLProgram::numActiveAttribLocations()
{
cacheInfoIfNeeded();
return m_activeAttribLocations.size();
}
GCGLint WebGLProgram::getActiveAttribLocation(GCGLuint index)
{
cacheInfoIfNeeded();
if (index >= numActiveAttribLocations())
return -1;
return m_activeAttribLocations[index];
}
bool WebGLProgram::isUsingVertexAttrib0()
{
cacheInfoIfNeeded();
for (unsigned ii = 0; ii < numActiveAttribLocations(); ++ii) {
if (!getActiveAttribLocation(ii))
return true;
}
return false;
}
bool WebGLProgram::getLinkStatus()
{
cacheInfoIfNeeded();
return m_linkStatus;
}
void WebGLProgram::setLinkStatus(bool status)
{
cacheInfoIfNeeded();
m_linkStatus = status;
}
void WebGLProgram::increaseLinkCount()
{
++m_linkCount;
m_infoValid = false;
}
WebGLShader* WebGLProgram::getAttachedShader(GCGLenum type)
{
switch (type) {
case GraphicsContextGL::VERTEX_SHADER:
return m_vertexShader.get();
case GraphicsContextGL::FRAGMENT_SHADER:
return m_fragmentShader.get();
default:
return 0;
}
}
bool WebGLProgram::attachShader(const AbstractLocker&, WebGLShader* shader)
{
if (!shader || !shader->object())
return false;
switch (shader->getType()) {
case GraphicsContextGL::VERTEX_SHADER:
if (m_vertexShader)
return false;
m_vertexShader = shader;
return true;
case GraphicsContextGL::FRAGMENT_SHADER:
if (m_fragmentShader)
return false;
m_fragmentShader = shader;
return true;
default:
return false;
}
}
bool WebGLProgram::detachShader(const AbstractLocker&, WebGLShader* shader)
{
if (!shader || !shader->object())
return false;
switch (shader->getType()) {
case GraphicsContextGL::VERTEX_SHADER:
if (m_vertexShader != shader)
return false;
m_vertexShader = nullptr;
return true;
case GraphicsContextGL::FRAGMENT_SHADER:
if (m_fragmentShader != shader)
return false;
m_fragmentShader = nullptr;
return true;
default:
return false;
}
}
void WebGLProgram::addMembersToOpaqueRoots(const AbstractLocker&, JSC::SlotVisitor& visitor)
{
visitor.addOpaqueRoot(m_vertexShader.get());
visitor.addOpaqueRoot(m_fragmentShader.get());
}
void WebGLProgram::cacheActiveAttribLocations(GraphicsContextGL* context3d)
{
m_activeAttribLocations.clear();
GCGLint numAttribs = context3d->getProgrami(object(), GraphicsContextGL::ACTIVE_ATTRIBUTES);
m_activeAttribLocations.resize(static_cast<size_t>(numAttribs));
for (int i = 0; i < numAttribs; ++i) {
GraphicsContextGL::ActiveInfo info;
#if USE(ANGLE)
context3d->getActiveAttrib(object(), i, info);
#else
static_cast<GraphicsContextGLOpenGL*>(context3d)->getActiveAttribImpl(object(), i, info);
#endif
m_activeAttribLocations[i] = context3d->getAttribLocation(object(), info.name);
}
}
void WebGLProgram::cacheInfoIfNeeded()
{
if (m_infoValid)
return;
if (!object())
return;
GraphicsContextGL* context = getAGraphicsContextGL();
if (!context)
return;
GCGLint linkStatus = context->getProgrami(object(), GraphicsContextGL::LINK_STATUS);
m_linkStatus = linkStatus;
if (m_linkStatus) {
cacheActiveAttribLocations(context);
m_requiredTransformFeedbackBufferCount = m_requiredTransformFeedbackBufferCountAfterNextLink;
}
m_infoValid = true;
}
}
#endif // ENABLE(WEBGL)