Shader.cc 2.83 KB
Newer Older
Robert Menzel's avatar
Robert Menzel committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
////////////////////////////////////////////////////////////////////////////////
// Copyright (c) 2011, Computer Graphics Group RWTH Aachen University         //
// All rights reserved.                                                       //
////////////////////////////////////////////////////////////////////////////////

#include <ACGL/Resource/Shader.hh>
#include <ACGL/GLUtils/Tools.hh>
#include <ACGL/Utils/Log.hh>
#include <ACGL/Utils/StringOperations.hh>

#include <iostream>
#include <fstream>

using namespace ACGL::Utils;
using namespace ACGL::GLUtils;
using namespace ACGL::GLUtils::Tools;
using namespace ACGL::Resource;

bool Shader::setFromFile(const std::string& _filename)
{
    std::string line = "";
    std::string fileContent = "";
    
    std::ifstream fileStream(_filename.c_str(), std::ifstream::in);
    
    if(fileStream.is_open())
    {
        while (fileStream.good())
        {
            std::getline(fileStream,line);
            fileContent += line + "\n";
        }
        fileStream.close();
    }
    else
    {
        error() << "Failed to open file: " << _filename << std::endl;
        return false;
    }
    
    return setSource(fileContent);
}

bool Shader::setSource(const std::string& _source)
{
    std::istringstream stream(_source);
    std::string line = "";

    mAttributes.clear();

    while (stream.good())
    {
        std::getline(stream,line);
        if(StringOperations::startsWith(line, "attribute"))
        {
            std::vector<std::string> tokens = StringOperations::split(line, ' ');
            if(tokens.size() > 2)
            {
                std::vector<std::string> tokensOfToken = StringOperations::split(tokens[2], ';');
                if(tokensOfToken.size() > 0)
                {
                    mAttributes.push_back(tokensOfToken[0]);
                }
            }
        }
    }
    return compile(_source.c_str());
}

bool Shader::compile(const char* _pProgramText) const
{
    glShaderSource(mContext, 1, &_pProgramText, NULL);
    openGLRareError();
    glCompileShader(mContext);
    
    if ( openGLCommonErrorOccured() )
    {
        return false;
    }

    // check for shader compile errors:
    GLint shaderError;
    glGetShaderiv(mContext, GL_COMPILE_STATUS, &shaderError);
    if(shaderError != GL_TRUE)
    {
        // yes, errors
        error() << "Shader compile error: " << std::endl;
    }

    GLsizei length = 0;
    glGetShaderiv(mContext, GL_INFO_LOG_LENGTH, &length);
    if(length > 1)
    {
        // a compile log can get produced even if there were no errors, but warnings!
        GLchar* pInfo = new char[length + 1];
        glGetShaderInfoLog(mContext,  length, &length, pInfo);
        error() << "Compile log: " << std::string(pInfo) << std::endl;
        delete[] pInfo;
    }
    return (shaderError == GL_TRUE); // return true if we encountered no errors
}