2 * Copyright (C) 2011 The Android Open Source Project
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include "GLSharedGroup.h"
19 /**** KeyedVector utilities ****/
22 static void clearObjectMap(android::DefaultKeyedVector<GLuint, T>& v) {
23 for (size_t i = 0; i < v.size(); i++)
28 /**** BufferData ****/
30 BufferData::BufferData() : m_size(0), m_usage(0), m_mapped(false) {};
31 BufferData::BufferData(GLsizeiptr size, void * data) : m_size(size), m_usage(0), m_mapped(false)
34 if (size>0) buffer = m_fixedBuffer.alloc(size);
35 if (data) memcpy(buffer, data, size);
38 /**** ProgramData ****/
39 ProgramData::ProgramData() : m_numIndexes(0),
46 void ProgramData::initProgramData(GLuint numIndexes)
49 m_numIndexes = numIndexes;
51 m_Indexes = new IndexInfo[numIndexes];
52 m_locShiftWAR = false;
55 bool ProgramData::isInitialized()
60 ProgramData::~ProgramData()
66 void ProgramData::setIndexInfo(GLuint index, GLint base, GLint size, GLenum type)
68 if (index>=m_numIndexes)
70 m_Indexes[index].base = base;
71 m_Indexes[index].size = size;
72 m_Indexes[index].type = type;
74 m_Indexes[index].appBase = m_Indexes[index-1].appBase +
75 m_Indexes[index-1].size;
78 m_Indexes[index].appBase = 0;
80 m_Indexes[index].hostLocsPerElement = 1;
81 m_Indexes[index].flags = 0;
82 m_Indexes[index].samplerValue = 0;
85 void ProgramData::setIndexFlags(GLuint index, GLuint flags)
87 if (index >= m_numIndexes)
89 m_Indexes[index].flags |= flags;
92 GLuint ProgramData::getIndexForLocation(GLint location)
94 GLuint index = m_numIndexes;
96 for (GLuint i=0;i<m_numIndexes;++i)
98 GLint dist = location - m_Indexes[i].base;
100 (minDist < 0 || dist < minDist)) {
108 GLenum ProgramData::getTypeForLocation(GLint location)
110 GLuint index = getIndexForLocation(location);
111 if (index<m_numIndexes) {
112 return m_Indexes[index].type;
117 void ProgramData::setupLocationShiftWAR()
119 m_locShiftWAR = false;
120 for (GLuint i=0; i<m_numIndexes; i++) {
121 if (0 != (m_Indexes[i].base & 0xffff)) {
125 // if we have one uniform at location 0, we do not need the WAR.
126 if (m_numIndexes > 1) {
127 m_locShiftWAR = true;
131 GLint ProgramData::locationWARHostToApp(GLint hostLoc, GLint arrIndex)
133 if (!m_locShiftWAR) return hostLoc;
135 GLuint index = getIndexForLocation(hostLoc);
136 if (index<m_numIndexes) {
138 m_Indexes[index].hostLocsPerElement =
139 (hostLoc - m_Indexes[index].base) / arrIndex;
141 return m_Indexes[index].appBase + arrIndex;
146 GLint ProgramData::locationWARAppToHost(GLint appLoc)
148 if (!m_locShiftWAR) return appLoc;
150 for(GLuint i=0; i<m_numIndexes; i++) {
151 GLint elemIndex = appLoc - m_Indexes[i].appBase;
152 if (elemIndex >= 0 && elemIndex < m_Indexes[i].size) {
153 return m_Indexes[i].base +
154 elemIndex * m_Indexes[i].hostLocsPerElement;
160 GLint ProgramData::getNextSamplerUniform(GLint index, GLint* val, GLenum* target)
162 for (GLint i = index + 1; i >= 0 && i < (GLint)m_numIndexes; i++) {
163 if (m_Indexes[i].type == GL_SAMPLER_2D) {
164 if (val) *val = m_Indexes[i].samplerValue;
166 if (m_Indexes[i].flags & INDEX_FLAG_SAMPLER_EXTERNAL) {
167 *target = GL_TEXTURE_EXTERNAL_OES;
169 *target = GL_TEXTURE_2D;
178 bool ProgramData::setSamplerUniform(GLint appLoc, GLint val, GLenum* target)
180 for (GLuint i = 0; i < m_numIndexes; i++) {
181 GLint elemIndex = appLoc - m_Indexes[i].appBase;
182 if (elemIndex >= 0 && elemIndex < m_Indexes[i].size) {
183 if (m_Indexes[i].type == GL_TEXTURE_2D) {
184 m_Indexes[i].samplerValue = val;
186 if (m_Indexes[i].flags & INDEX_FLAG_SAMPLER_EXTERNAL) {
187 *target = GL_TEXTURE_EXTERNAL_OES;
189 *target = GL_TEXTURE_2D;
199 bool ProgramData::attachShader(GLuint shader)
201 size_t n = m_shaders.size();
202 for (size_t i = 0; i < n; i++) {
203 if (m_shaders[i] == shader) {
207 // AKA m_shaders.push_back(), but that has an ambiguous call to insertAt()
208 // due to the default parameters. This is the desired insertAt() overload.
209 m_shaders.insertAt(shader, m_shaders.size(), 1);
213 bool ProgramData::detachShader(GLuint shader)
215 size_t n = m_shaders.size();
216 for (size_t i = 0; i < n; i++) {
217 if (m_shaders[i] == shader) {
218 m_shaders.removeAt(i);
225 /***** GLSharedGroup ****/
227 GLSharedGroup::GLSharedGroup() :
228 m_buffers(android::DefaultKeyedVector<GLuint, BufferData*>(NULL)),
229 m_programs(android::DefaultKeyedVector<GLuint, ProgramData*>(NULL)),
230 m_shaders(android::DefaultKeyedVector<GLuint, ShaderData*>(NULL)),
231 m_shaderPrograms(android::DefaultKeyedVector<GLuint, ShaderProgramData*>(NULL))
235 GLSharedGroup::~GLSharedGroup()
239 clearObjectMap(m_buffers);
240 clearObjectMap(m_programs);
241 clearObjectMap(m_shaders);
242 clearObjectMap(m_shaderPrograms);
245 bool GLSharedGroup::isShaderOrProgramObject(GLuint obj)
247 android::AutoMutex _lock(m_lock);
248 return ((m_shaders.valueFor(obj)!=NULL) ||
249 (m_programs.valueFor(obj)!=NULL) ||
250 (m_shaderPrograms.valueFor(m_shaderProgramIdMap[obj]) !=NULL));
253 BufferData * GLSharedGroup::getBufferData(GLuint bufferId)
255 android::AutoMutex _lock(m_lock);
256 return m_buffers.valueFor(bufferId);
259 SharedTextureDataMap* GLSharedGroup::getTextureData() {
260 return &m_textureRecs;
263 void GLSharedGroup::addBufferData(GLuint bufferId, GLsizeiptr size, void * data)
265 android::AutoMutex _lock(m_lock);
266 m_buffers.add(bufferId, new BufferData(size, data));
269 void GLSharedGroup::updateBufferData(GLuint bufferId, GLsizeiptr size, void * data)
271 android::AutoMutex _lock(m_lock);
272 ssize_t idx = m_buffers.indexOfKey(bufferId);
274 delete m_buffers.valueAt(idx);
275 m_buffers.editValueAt(idx) = new BufferData(size, data);
277 m_buffers.add(bufferId, new BufferData(size, data));
281 void GLSharedGroup::setBufferUsage(GLuint bufferId, GLenum usage) {
282 android::AutoMutex _lock(m_lock);
283 ssize_t idx = m_buffers.indexOfKey(bufferId);
285 m_buffers.editValueAt(idx)->m_usage = usage;
289 void GLSharedGroup::setBufferMapped(GLuint bufferId, bool mapped) {
290 BufferData * buf = m_buffers.valueFor(bufferId);
292 buf->m_mapped = mapped;
295 GLenum GLSharedGroup::getBufferUsage(GLuint bufferId) {
296 BufferData * buf = m_buffers.valueFor(bufferId);
301 bool GLSharedGroup::isBufferMapped(GLuint bufferId) {
302 BufferData * buf = m_buffers.valueFor(bufferId);
303 if (!buf) return false;
304 return buf->m_mapped;
307 GLenum GLSharedGroup::subUpdateBufferData(GLuint bufferId, GLintptr offset, GLsizeiptr size, void * data)
309 android::AutoMutex _lock(m_lock);
310 BufferData * buf = m_buffers.valueFor(bufferId);
311 if ((!buf) || (buf->m_size < offset+size) || (offset < 0) || (size<0)) return GL_INVALID_VALUE;
313 //it's safe to update now
314 memcpy((char*)buf->m_fixedBuffer.ptr() + offset, data, size);
316 buf->m_indexRangeCache.invalidateRange((size_t)offset, (size_t)size);
320 void GLSharedGroup::deleteBufferData(GLuint bufferId)
322 android::AutoMutex _lock(m_lock);
323 ssize_t idx = m_buffers.indexOfKey(bufferId);
325 delete m_buffers.valueAt(idx);
326 m_buffers.removeItemsAt(idx);
330 void GLSharedGroup::addProgramData(GLuint program)
332 android::AutoMutex _lock(m_lock);
333 ProgramData *pData = m_programs.valueFor(program);
336 m_programs.removeItem(program);
340 m_programs.add(program,new ProgramData());
343 void GLSharedGroup::initProgramData(GLuint program, GLuint numIndexes)
345 android::AutoMutex _lock(m_lock);
346 ProgramData *pData = m_programs.valueFor(program);
349 pData->initProgramData(numIndexes);
353 bool GLSharedGroup::isProgramInitialized(GLuint program)
355 android::AutoMutex _lock(m_lock);
356 ProgramData* pData = m_programs.valueFor(program);
359 return pData->isInitialized();
361 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return false;
362 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap[program]);
364 return spData->programData->isInitialized();
369 void GLSharedGroup::deleteProgramData(GLuint program)
371 android::AutoMutex _lock(m_lock);
372 ProgramData *pData = m_programs.valueFor(program);
376 m_programs.removeItem(program);
378 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return;
379 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap[program]);
383 m_shaderPrograms.removeItem(m_shaderProgramIdMap[program]);
384 m_shaderProgramIdMap.erase(program);
387 // No such thing for separable shader programs.
388 void GLSharedGroup::attachShader(GLuint program, GLuint shader)
390 android::AutoMutex _lock(m_lock);
391 ProgramData* programData = m_programs.valueFor(program);
392 ssize_t idx = m_shaders.indexOfKey(shader);
393 if (programData && idx >= 0) {
394 if (programData->attachShader(shader)) {
395 refShaderDataLocked(idx);
400 void GLSharedGroup::detachShader(GLuint program, GLuint shader)
402 android::AutoMutex _lock(m_lock);
403 ProgramData* programData = m_programs.valueFor(program);
404 ssize_t idx = m_shaders.indexOfKey(shader);
405 if (programData && idx >= 0) {
406 if (programData->detachShader(shader)) {
407 unrefShaderDataLocked(idx);
412 // Not needed/used for separate shader programs.
413 void GLSharedGroup::setProgramIndexInfo(GLuint program, GLuint index, GLint base, GLint size, GLenum type, const char* name)
415 android::AutoMutex _lock(m_lock);
416 ProgramData* pData = m_programs.valueFor(program);
419 pData->setIndexInfo(index,base,size,type);
421 if (type == GL_SAMPLER_2D) {
422 size_t n = pData->getNumShaders();
423 for (size_t i = 0; i < n; i++) {
424 GLuint shaderId = pData->getShader(i);
425 ShaderData* shader = m_shaders.valueFor(shaderId);
426 if (!shader) continue;
427 ShaderData::StringList::iterator nameIter = shader->samplerExternalNames.begin();
428 ShaderData::StringList::iterator nameEnd = shader->samplerExternalNames.end();
429 while (nameIter != nameEnd) {
430 if (*nameIter == name) {
431 pData->setIndexFlags(index, ProgramData::INDEX_FLAG_SAMPLER_EXTERNAL);
441 GLenum GLSharedGroup::getProgramUniformType(GLuint program, GLint location)
443 android::AutoMutex _lock(m_lock);
444 ProgramData* pData = m_programs.valueFor(program);
447 type = pData->getTypeForLocation(location);
449 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return type;
450 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap[program]);
452 type = spData->programData->getTypeForLocation(location);
457 bool GLSharedGroup::isProgram(GLuint program)
459 android::AutoMutex _lock(m_lock);
460 ProgramData* pData = m_programs.valueFor(program);
461 if (pData) return true;
462 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return false;
463 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap[program]);
464 if (spData) return true;
468 void GLSharedGroup::setupLocationShiftWAR(GLuint program)
470 android::AutoMutex _lock(m_lock);
471 ProgramData* pData = m_programs.valueFor(program);
472 if (pData) pData->setupLocationShiftWAR();
475 GLint GLSharedGroup::locationWARHostToApp(GLuint program, GLint hostLoc, GLint arrIndex)
477 android::AutoMutex _lock(m_lock);
478 ProgramData* pData = m_programs.valueFor(program);
479 if (pData) return pData->locationWARHostToApp(hostLoc, arrIndex);
480 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return hostLoc;
481 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap[program]);
482 if (spData) return spData->programData->locationWARHostToApp(hostLoc, arrIndex);
486 GLint GLSharedGroup::locationWARAppToHost(GLuint program, GLint appLoc)
488 android::AutoMutex _lock(m_lock);
489 ProgramData* pData = m_programs.valueFor(program);
490 if (pData) return pData->locationWARAppToHost(appLoc);
491 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return appLoc;
492 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap[program]);
493 if (spData) return spData->programData->locationWARAppToHost(appLoc);
497 bool GLSharedGroup::needUniformLocationWAR(GLuint program)
499 android::AutoMutex _lock(m_lock);
500 ProgramData* pData = m_programs.valueFor(program);
501 if (pData) return pData->needUniformLocationWAR();
502 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return false;
503 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap[program]);
504 if (spData) return spData->programData->needUniformLocationWAR();
508 GLint GLSharedGroup::getNextSamplerUniform(GLuint program, GLint index, GLint* val, GLenum* target) const
510 android::AutoMutex _lock(m_lock);
511 ProgramData* pData = m_programs.valueFor(program);
512 if (pData) return pData->getNextSamplerUniform(index, val, target);
513 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return -1;
514 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap.find(program)->second);
515 if (spData) return spData->programData->getNextSamplerUniform(index, val, target);
519 bool GLSharedGroup::setSamplerUniform(GLuint program, GLint appLoc, GLint val, GLenum* target)
521 android::AutoMutex _lock(m_lock);
522 ProgramData* pData = m_programs.valueFor(program);
523 if (pData) return pData->setSamplerUniform(appLoc, val, target);
524 if (m_shaderProgramIdMap.find(program) == m_shaderProgramIdMap.end()) return false;
525 ShaderProgramData* spData = m_shaderPrograms.valueFor(m_shaderProgramIdMap[program]);
526 if (spData) return spData->programData->setSamplerUniform(appLoc, val, target);
530 bool GLSharedGroup::isShader(GLuint shader)
532 android::AutoMutex _lock(m_lock);
533 ShaderData* pData = m_shaders.valueFor(shader);
534 return (pData!=NULL);
537 bool GLSharedGroup::addShaderData(GLuint shader)
539 android::AutoMutex _lock(m_lock);
540 ShaderData* data = new ShaderData;
542 if (m_shaders.add(shader, data) < 0) {
551 ShaderData* GLSharedGroup::getShaderData(GLuint shader)
553 android::AutoMutex _lock(m_lock);
554 return m_shaders.valueFor(shader);
557 void GLSharedGroup::unrefShaderData(GLuint shader)
559 android::AutoMutex _lock(m_lock);
560 ssize_t idx = m_shaders.indexOfKey(shader);
562 unrefShaderDataLocked(idx);
566 void GLSharedGroup::refShaderDataLocked(ssize_t shaderIdx)
568 assert(shaderIdx >= 0 && shaderIdx <= m_shaders.size());
569 ShaderData* data = m_shaders.valueAt(shaderIdx);
573 void GLSharedGroup::unrefShaderDataLocked(ssize_t shaderIdx)
575 assert(shaderIdx >= 0 && shaderIdx <= m_shaders.size());
576 ShaderData* data = m_shaders.valueAt(shaderIdx);
577 if (--data->refcount == 0) {
579 m_shaders.removeItemsAt(shaderIdx);
583 uint32_t GLSharedGroup::addNewShaderProgramData() {
584 android::AutoMutex _lock(m_lock);
585 ShaderProgramData* data = new ShaderProgramData;
586 uint32_t currId = m_shaderProgramId;
587 ALOGD("%s: new data %p id %u", __FUNCTION__, data, currId);
588 m_shaderPrograms.add(currId, data);
593 void GLSharedGroup::associateGLShaderProgram(GLuint shaderProgramName, uint32_t shaderProgramId) {
594 android::AutoMutex _lock(m_lock);
595 m_shaderProgramIdMap[shaderProgramName] = shaderProgramId;
598 ShaderProgramData* GLSharedGroup::getShaderProgramDataById(uint32_t id) {
599 android::AutoMutex _lock(m_lock);
600 ShaderProgramData* res = m_shaderPrograms.valueFor(id);
601 ALOGD("%s: id=%u res=%p", __FUNCTION__, id, res);
605 ShaderProgramData* GLSharedGroup::getShaderProgramData(GLuint shaderProgramName) {
606 android::AutoMutex _lock(m_lock);
607 return m_shaderPrograms.valueFor(m_shaderProgramIdMap[shaderProgramName]);
610 void GLSharedGroup::deleteShaderProgramDataById(uint32_t id) {
611 android::AutoMutex _lock(m_lock);
612 ShaderProgramData* data = m_shaderPrograms.valueFor(id);
614 m_shaderPrograms.removeItemsAt(id);
618 void GLSharedGroup::deleteShaderProgramData(GLuint shaderProgramName) {
619 android::AutoMutex _lock(m_lock);
620 uint32_t id = m_shaderProgramIdMap[shaderProgramName];
621 ShaderProgramData* data = m_shaderPrograms.valueFor(id);
623 m_shaderPrograms.removeItemsAt(id);
624 m_shaderProgramIdMap.erase(shaderProgramName);
627 void GLSharedGroup::initShaderProgramData(GLuint shaderProgram, GLuint numIndices) {
628 ShaderProgramData* spData = getShaderProgramData(shaderProgram);
629 spData->programData->initProgramData(numIndices);
632 void GLSharedGroup::setShaderProgramIndexInfo(GLuint shaderProgram, GLuint index, GLint base, GLint size, GLenum type, const char* name) {
633 ShaderProgramData* spData = getShaderProgramData(shaderProgram);
634 ProgramData* pData = spData->programData;
635 ShaderData* sData = spData->shaderData;
639 pData->setIndexInfo(index, base, size, type);
641 if (type == GL_SAMPLER_2D) {
642 ShaderData::StringList::iterator nameIter = sData->samplerExternalNames.begin();
643 ShaderData::StringList::iterator nameEnd = sData->samplerExternalNames.end();
644 while (nameIter != nameEnd) {
645 if (*nameIter == name) {
646 pData->setIndexFlags(index, ProgramData::INDEX_FLAG_SAMPLER_EXTERNAL);
655 void GLSharedGroup::setupShaderProgramLocationShiftWAR(GLuint shaderProgram) {
656 ShaderProgramData* spData = getShaderProgramData(shaderProgram);
657 spData->programData->setupLocationShiftWAR();