Check the flat qualifier from the correct member

[libs/gl.git] / source / glsl / builtin.cpp

diff --git a/source/glsl/builtin.cpp b/source/glsl/builtin.cpp
index 83e480e5cf417b5943abb7841ebcb32256f8136e..217509a20b74c61e83091feda6dc856dd9a3cfcc 100644 (file)
--- a/source/glsl/builtin.cpp
+++ b/source/glsl/builtin.cpp
@@ -1,3 +1,4 @@
+#include <msp/core/algorithm.h>
 #include <msp/gl/resources.h>
 #include <msp/io/seekable.h>
 #include "builtin.h"
@@ -9,7 +10,7 @@ namespace Msp {
 namespace GL {
 namespace SL {
 
-void add_builtin_type(Stage &stage, const std::string &name, BasicTypeDeclaration::Kind kind, unsigned size, unsigned sign)
+void add_builtin_type(Stage &stage, const string &name, BasicTypeDeclaration::Kind kind, unsigned size, unsigned sign)
 {
        RefPtr<BasicTypeDeclaration> type = new BasicTypeDeclaration;
        type->source = BUILTIN_SOURCE;
@@ -62,10 +63,8 @@ const Stage *get_builtins(Stage::Type type)
 
        if(type==Stage::SHARED)
                return &module->shared;
-       for(list<Stage>::iterator i=module->stages.begin(); i!=module->stages.end(); ++i)
-               if(i->type==type)
-                       return &*i;
-       return 0;
+       auto i = find_member(module->stages, type, &Stage::type);
+       return (i!=module->stages.end() ? &*i : 0);
 }
 
 } // namespace SL