Make shader and program object lifetimes match OpenGL ES spec....

gpu/command_buffer/service/program_manager_unittest.cc

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "gpu/command_buffer/service/program_manager.h"
 
 #include <algorithm>
 
 #include "base/scoped_ptr.h"
 #include "base/string_number_conversions.h"
@@ skipping 43 matching lines @@
 
 TEST_F(ProgramManagerTest, Basic) {
   const GLuint kClient1Id = 1;
   const GLuint kService1Id = 11;
   const GLuint kClient2Id = 2;
   // Check we can create program.
   manager_.CreateProgramInfo(kClient1Id, kService1Id);
   // Check program got created.
   ProgramManager::ProgramInfo* info1 = manager_.GetProgramInfo(kClient1Id);
   ASSERT_TRUE(info1 != NULL);
-  EXPECT_EQ(kService1Id, info1->service_id());
-  EXPECT_FALSE(info1->CanLink());
-  EXPECT_STREQ("", info1->log_info().c_str());
   GLuint client_id = 0;
   EXPECT_TRUE(manager_.GetClientId(info1->service_id(), &client_id));
   EXPECT_EQ(kClient1Id, client_id);
   // Check we get nothing for a non-existent program.
   EXPECT_TRUE(manager_.GetProgramInfo(kClient2Id) == NULL);
-  // Check trying to a remove non-existent programs does not crash.
-  manager_.RemoveProgramInfo(kClient2Id);
-  // Check we can't get the program after we remove it.
-  manager_.RemoveProgramInfo(kClient1Id);
-  EXPECT_TRUE(manager_.GetProgramInfo(kClient1Id) == NULL);
 }
 
 TEST_F(ProgramManagerTest, Destroy) {
   const GLuint kClient1Id = 1;
   const GLuint kService1Id = 11;
   // Check we can create program.
   manager_.CreateProgramInfo(kClient1Id, kService1Id);
   // Check program got created.
   ProgramManager::ProgramInfo* info1 =
       manager_.GetProgramInfo(kClient1Id);
   ASSERT_TRUE(info1 != NULL);
   EXPECT_CALL(*gl_, DeleteProgram(kService1Id))
       .Times(1)
       .RetiresOnSaturation();
   manager_.Destroy(true);
   // Check the resources were released.
   info1 = manager_.GetProgramInfo(kClient1Id);
   ASSERT_TRUE(info1 == NULL);
 }
 
+TEST_F(ProgramManagerTest, ProgramInfo) {
+  const GLuint kClient1Id = 1;
+  const GLuint kService1Id = 11;
+  // Check we can create program.
+  manager_.CreateProgramInfo(kClient1Id, kService1Id);
+  // Check program got created.
+  ProgramManager::ProgramInfo* info1 = manager_.GetProgramInfo(kClient1Id);
+  ASSERT_TRUE(info1 != NULL);
+  EXPECT_EQ(kService1Id, info1->service_id());
+  EXPECT_FALSE(info1->InUse());
+  EXPECT_FALSE(info1->IsValid());
+  EXPECT_FALSE(info1->IsDeleted());
+  EXPECT_FALSE(info1->CanLink());
+  EXPECT_STREQ("", info1->log_info().c_str());
+}
+
 class ProgramManagerWithShaderTest : public testing::Test {
  public:
   ProgramManagerWithShaderTest()
       : program_info_(NULL) {
   }
 
   ~ProgramManagerWithShaderTest() {
     manager_.Destroy(false);
   }
 
@@ skipping 317 matching lines @@
   shader_manager.CreateShaderInfo(
       kVShaderClientId, kVShaderServiceId, GL_VERTEX_SHADER);
   ShaderManager::ShaderInfo* vshader = shader_manager.GetShaderInfo(
       kVShaderClientId);
   vshader->SetStatus(true, "", NULL);
   shader_manager.CreateShaderInfo(
       kFShaderClientId, kFShaderServiceId, GL_FRAGMENT_SHADER);
   ShaderManager::ShaderInfo* fshader = shader_manager.GetShaderInfo(
       kFShaderClientId);
   fshader->SetStatus(true, "", NULL);
-  EXPECT_TRUE(program_info->AttachShader(vshader));
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, vshader));
   EXPECT_FALSE(program_info->CanLink());
-  EXPECT_TRUE(program_info->AttachShader(fshader));
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, fshader));
   EXPECT_TRUE(program_info->CanLink());
-  program_info->DetachShader(vshader);
+  program_info->DetachShader(&shader_manager, vshader);
   EXPECT_FALSE(program_info->CanLink());
-  EXPECT_TRUE(program_info->AttachShader(vshader));
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, vshader));
   EXPECT_TRUE(program_info->CanLink());
-  program_info->DetachShader(fshader);
+  program_info->DetachShader(&shader_manager, fshader);
   EXPECT_FALSE(program_info->CanLink());
-  EXPECT_FALSE(program_info->AttachShader(vshader));
+  EXPECT_FALSE(program_info->AttachShader(&shader_manager, vshader));
   EXPECT_FALSE(program_info->CanLink());
-  EXPECT_TRUE(program_info->AttachShader(fshader));
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, fshader));
   EXPECT_TRUE(program_info->CanLink());
   vshader->SetStatus(false, "", NULL);
   EXPECT_FALSE(program_info->CanLink());
   vshader->SetStatus(true, "", NULL);
   EXPECT_TRUE(program_info->CanLink());
   fshader->SetStatus(false, "", NULL);
   EXPECT_FALSE(program_info->CanLink());
   fshader->SetStatus(true, "", NULL);
   EXPECT_TRUE(program_info->CanLink());
   shader_manager.Destroy(false);
@@ skipping 121 matching lines @@
   const size_t kNumAttribs= arraysize(kAttribs);
   const size_t kNumUniforms = arraysize(kUniforms);
   static const GLuint kClientProgramId = 1234;
   static const GLuint kServiceProgramId = 5679;
   SetupShader(kAttribs, kNumAttribs, kUniforms, kNumUniforms,
               kServiceProgramId);
   manager_.CreateProgramInfo(kClientProgramId, kServiceProgramId);
   ProgramManager::ProgramInfo* program_info =
       manager_.GetProgramInfo(kClientProgramId);
   ASSERT_TRUE(program_info != NULL);
-  EXPECT_TRUE(program_info->AttachShader(shader_info));
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, shader_info));
   program_info->Update();
   // Check that we got the good type, not the bad.
   // Check Attribs
   for (unsigned index = 0; index < kNumAttribs; ++index) {
     const ProgramManager::ProgramInfo::VertexAttribInfo* attrib_info =
         program_info->GetAttribInfo(index);
     ASSERT_TRUE(attrib_info != NULL);
     ShaderTranslator::VariableMap::const_iterator it = attrib_map.find(
         attrib_info->name);
     ASSERT_TRUE(it != attrib_map.end());
     EXPECT_EQ(it->first, attrib_info->name);
     EXPECT_EQ(static_cast<GLenum>(it->second.type), attrib_info->type);
     EXPECT_EQ(it->second.size, attrib_info->size);
   }
   // Check Uniforms
   for (unsigned index = 0; index < kNumUniforms; ++index) {
     const ProgramManager::ProgramInfo::UniformInfo* uniform_info =
         program_info->GetUniformInfo(index);
     ASSERT_TRUE(uniform_info != NULL);
     ShaderTranslator::VariableMap::const_iterator it = uniform_map.find(
         uniform_info->name);
     ASSERT_TRUE(it != uniform_map.end());
     EXPECT_EQ(it->first, uniform_info->name);
     EXPECT_EQ(static_cast<GLenum>(it->second.type), uniform_info->type);
     EXPECT_EQ(it->second.size, uniform_info->size);
   }
   shader_manager.Destroy(false);
 }
 
+TEST_F(ProgramManagerWithShaderTest, ProgramInfoUseCount) {
+  ShaderManager shader_manager;
+  ProgramManager::ProgramInfo* program_info =
+      manager_.GetProgramInfo(kClientProgramId);
+  ASSERT_TRUE(program_info != NULL);
+  EXPECT_FALSE(program_info->CanLink());
+  const GLuint kVShaderClientId = 2001;
+  const GLuint kFShaderClientId = 2002;
+  const GLuint kVShaderServiceId = 3001;
+  const GLuint kFShaderServiceId = 3002;
+  shader_manager.CreateShaderInfo(
+      kVShaderClientId, kVShaderServiceId, GL_VERTEX_SHADER);
+  ShaderManager::ShaderInfo* vshader = shader_manager.GetShaderInfo(
+      kVShaderClientId);
+  ASSERT_TRUE(vshader != NULL);
+  vshader->SetStatus(true, "", NULL);
+  shader_manager.CreateShaderInfo(
+      kFShaderClientId, kFShaderServiceId, GL_FRAGMENT_SHADER);
+  ShaderManager::ShaderInfo* fshader = shader_manager.GetShaderInfo(
+      kFShaderClientId);
+  ASSERT_TRUE(fshader != NULL);
+  fshader->SetStatus(true, "", NULL);
+  EXPECT_FALSE(vshader->InUse());
+  EXPECT_FALSE(fshader->InUse());
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, vshader));
+  EXPECT_TRUE(vshader->InUse());
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, fshader));
+  EXPECT_TRUE(fshader->InUse());
+  EXPECT_TRUE(program_info->CanLink());
+  EXPECT_FALSE(program_info->InUse());
+  EXPECT_FALSE(program_info->IsDeleted());
+  manager_.UseProgram(program_info);
+  EXPECT_TRUE(program_info->InUse());
+  manager_.UseProgram(program_info);
+  EXPECT_TRUE(program_info->InUse());
+  manager_.MarkAsDeleted(&shader_manager, program_info);
+  EXPECT_TRUE(program_info->IsDeleted());
+  ProgramManager::ProgramInfo* info2 =
+      manager_.GetProgramInfo(kClientProgramId);
+  EXPECT_EQ(program_info, info2);
+  manager_.UnuseProgram(&shader_manager, program_info);
+  EXPECT_TRUE(program_info->InUse());
+  // this should delete the info.
+  manager_.UnuseProgram(&shader_manager, program_info);
+  info2 = manager_.GetProgramInfo(kClientProgramId);
+  EXPECT_TRUE(info2 == NULL);
+  EXPECT_FALSE(vshader->InUse());
+  EXPECT_FALSE(fshader->InUse());
+  shader_manager.Destroy(false);
+}
+
+TEST_F(ProgramManagerWithShaderTest, ProgramInfoUseCount2) {
+  ShaderManager shader_manager;
+  ProgramManager::ProgramInfo* program_info =
+      manager_.GetProgramInfo(kClientProgramId);
+  ASSERT_TRUE(program_info != NULL);
+  EXPECT_FALSE(program_info->CanLink());
+  const GLuint kVShaderClientId = 2001;
+  const GLuint kFShaderClientId = 2002;
+  const GLuint kVShaderServiceId = 3001;
+  const GLuint kFShaderServiceId = 3002;
+  shader_manager.CreateShaderInfo(
+      kVShaderClientId, kVShaderServiceId, GL_VERTEX_SHADER);
+  ShaderManager::ShaderInfo* vshader = shader_manager.GetShaderInfo(
+      kVShaderClientId);
+  ASSERT_TRUE(vshader != NULL);
+  vshader->SetStatus(true, "", NULL);
+  shader_manager.CreateShaderInfo(
+      kFShaderClientId, kFShaderServiceId, GL_FRAGMENT_SHADER);
+  ShaderManager::ShaderInfo* fshader = shader_manager.GetShaderInfo(
+      kFShaderClientId);
+  ASSERT_TRUE(fshader != NULL);
+  fshader->SetStatus(true, "", NULL);
+  EXPECT_FALSE(vshader->InUse());
+  EXPECT_FALSE(fshader->InUse());
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, vshader));
+  EXPECT_TRUE(vshader->InUse());
+  EXPECT_TRUE(program_info->AttachShader(&shader_manager, fshader));
+  EXPECT_TRUE(fshader->InUse());
+  EXPECT_TRUE(program_info->CanLink());
+  EXPECT_FALSE(program_info->InUse());
+  EXPECT_FALSE(program_info->IsDeleted());
+  manager_.UseProgram(program_info);
+  EXPECT_TRUE(program_info->InUse());
+  manager_.UseProgram(program_info);
+  EXPECT_TRUE(program_info->InUse());
+  manager_.UnuseProgram(&shader_manager, program_info);
+  EXPECT_TRUE(program_info->InUse());
+  manager_.UnuseProgram(&shader_manager, program_info);
+  EXPECT_FALSE(program_info->InUse());
+  ProgramManager::ProgramInfo* info2 =
+      manager_.GetProgramInfo(kClientProgramId);
+  EXPECT_EQ(program_info, info2);
+  // this should delete the program.
+  manager_.MarkAsDeleted(&shader_manager, program_info);
+  info2 = manager_.GetProgramInfo(kClientProgramId);
+  EXPECT_TRUE(info2 == NULL);
+  EXPECT_FALSE(vshader->InUse());
+  EXPECT_FALSE(fshader->InUse());
+  shader_manager.Destroy(false);
+}
+
 }  // namespace gles2
 }  // namespace gpu