You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
2.9 KiB
C++
104 lines
2.9 KiB
C++
/*=========================================================================
|
|
|
|
Program: Visualization Toolkit
|
|
Module: vtkLightingHelper.cxx
|
|
|
|
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
|
|
All rights reserved.
|
|
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
|
|
|
|
This software is distributed WITHOUT ANY WARRANTY; without even
|
|
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
PURPOSE. See the above copyright notice for more information.
|
|
|
|
=========================================================================*/
|
|
#include "vtkLightingHelper.h"
|
|
|
|
#include "vtkObjectFactory.h"
|
|
#include "vtkShaderProgram2.h"
|
|
#include "vtkShader2Collection.h"
|
|
#include "vtkgl.h"
|
|
#include "vtkOpenGLError.h"
|
|
|
|
extern const char * vtkLightingHelper_s;
|
|
|
|
//----------------------------------------------------------------------------
|
|
vtkStandardNewMacro(vtkLightingHelper);
|
|
|
|
//----------------------------------------------------------------------------
|
|
vtkCxxSetObjectMacro(vtkLightingHelper, Shader, vtkShaderProgram2);
|
|
|
|
//----------------------------------------------------------------------------
|
|
vtkLightingHelper::vtkLightingHelper()
|
|
{
|
|
this->Shader = 0;
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
vtkLightingHelper::~vtkLightingHelper()
|
|
{
|
|
this->SetShader(0);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
void vtkLightingHelper::Initialize(vtkShaderProgram2* pgm,
|
|
vtkShader2Type mode)
|
|
{
|
|
if (this->Shader != pgm)
|
|
{
|
|
this->SetShader(pgm);
|
|
if (pgm)
|
|
{
|
|
vtkShader2 *s=vtkShader2::New();
|
|
s->SetSourceCode(vtkLightingHelper_s);
|
|
s->SetType(mode);
|
|
s->SetContext(this->Shader->GetContext());
|
|
this->Shader->GetShaders()->AddItem(s);
|
|
s->Delete();
|
|
}
|
|
}
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
void vtkLightingHelper::PrepareForRendering()
|
|
{
|
|
#ifndef NDEBUG
|
|
GLint ivalue;
|
|
glGetIntegerv(vtkgl::CURRENT_PROGRAM, &ivalue);
|
|
if (ivalue != 0)
|
|
{
|
|
vtkErrorMacro("PrepareForRendering() cannot be called after a shader program has been bound.");
|
|
return;
|
|
}
|
|
#endif
|
|
|
|
vtkOpenGLClearErrorMacro();
|
|
|
|
for (int cc=0; cc < VTK_MAX_LIGHTS; cc++)
|
|
{
|
|
// use the light's 4th diffuse component to store an enabled bit
|
|
GLfloat lightDiffuse[4];
|
|
glGetLightfv(GL_LIGHT0 + cc, GL_DIFFUSE, lightDiffuse);
|
|
|
|
// enable/disable the light for fixed function
|
|
if (glIsEnabled(GL_LIGHT0 + cc))
|
|
{
|
|
lightDiffuse[3] = 1;
|
|
}
|
|
else
|
|
{
|
|
lightDiffuse[3] = 0;
|
|
}
|
|
glLightfv(GL_LIGHT0 + cc, GL_DIFFUSE, lightDiffuse);
|
|
}
|
|
|
|
vtkOpenGLCheckErrorMacro("failed after enabling lights");
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
void vtkLightingHelper::PrintSelf(ostream& os, vtkIndent indent)
|
|
{
|
|
this->Superclass::PrintSelf(os, indent);
|
|
os << indent << "Shader: " << this->Shader << endl;
|
|
}
|