reader shaders as assets

CMakeLists.txt

@@ -18,8 +18,8 @@ set(MATH quaternion.cpp vector.cpp matrix.cpp rayCaster.cpp)
 set(RENDER lineRenderer.cpp particleEmitter.cpp camera.cpp light.cpp material.cpp mesh.cpp meshData.cpp model.cpp node.cpp quad.cpp box.cpp root.cpp shader.cpp texture.cpp)
 set(ANIM animationController.cpp animationChannel.cpp animation.cpp animatable.cpp keyframeChannel.cpp driver.cpp)
 set(ARMATURE skeleton.cpp bone.cpp ikSolver.cpp)
-set(ASSET_MANAGER assetManager.cpp textAsset.h imageAsset.h modelAsset.h fontAsset.h)
+set(ASSET_MANAGER assetManager.cpp shaderAsset.h textAsset.h imageAsset.h modelAsset.h fontAsset.h)
-set(ASSET_READERS imageReader.cpp fontReader.cpp modelReader.cpp xmlModelReader.cpp)
+set(ASSET_READERS shaderReader.cpp imageReader.cpp fontReader.cpp modelReader.cpp xmlModelReader.cpp)
 
 if(BUILD_WITH_ASSIMP)
 	set(ASSET_READERS ${ASSET_READERS} assimpModelReader.cpp)

assetManager.cpp

@@ -1,8 +1,13 @@
 #include "assetManager.h"
 #include "abstractAssetReader.h"
 #include "util.h"
+#include "root.h"
+#include "node.h"
+#include "mesh.h"
+#include "material.h"
 #include "imageReader.h"
 #include "fontReader.h"
+#include "shaderReader.h"
 #include "xmlModelReader.h"
 
 #include <tinydir.h>
@@ -71,6 +76,11 @@ namespace vb01{
 			if(find(modelFormats.begin(), modelFormats.end(), format) != modelFormats.end())
 				assetReader = XmlModelReader::getSingleton();
 
+			vector<string> shaderFormats = vector<string>{"vert", "frag", "geo"};
+
+			if(find(shaderFormats.begin(), shaderFormats.end(), format) != shaderFormats.end())
+				assetReader = ShaderReader::getSingleton();
+
 			if(assetReader)
 				assets.push_back(assetReader->readAsset(file));
 			else
@@ -85,4 +95,28 @@ namespace vb01{
 		
 		return nullptr;
 	}
+
+	void AssetManager::editAsset(string path, Asset &newAsset){
+		ShaderAsset *oldAsset = (ShaderAsset*)getAsset(path);
+
+		if(oldAsset){
+			oldAsset->shaderString = ((ShaderAsset&)newAsset).shaderString;
+
+			Root *root = Root::getSingleton();
+			vector<Node*> descendants;
+			root->getRootNode()->getDescendants(descendants);
+			root->getGuiNode()->getDescendants(descendants);
+
+			for(Node *desc : descendants){
+				vector<Mesh*> meshes = desc->getMeshes();
+
+				for(Mesh *mesh : meshes){
+					Material *mat = mesh->getMaterial();
+
+					if(mat)
+						mat->getShader()->loadShaders();
+				}
+			}
+		}
+	}
 }

assetManager.h

@@ -12,6 +12,7 @@ namespace vb01{
 		public:
 			static AssetManager* getSingleton();
 			void load(std::string, bool = false);
+			void editAsset(std::string, Asset&);
 			Asset* getAsset(std::string);
 			inline Asset* getAsset(int i){return assets[i];}
 			inline std::vector<Asset*> getAssets(){return assets;}

shader.cpp

@@ -1,20 +1,19 @@
 #include "glad.h"
 #include <glfw3.h>
 #include <iostream>
-#include <sstream>
-#include <fstream>
 
-#include "shader.h"
 #include <glm/gtc/matrix_transform.hpp>
 #include <glm/gtc/type_ptr.hpp>
 
+#include "shader.h"
+#include "assetManager.h"
+
 using namespace std;
 using namespace glm;
 
 namespace vb01{
 	Shader::Shader(string shaderPath, bool geometry){
 		this->geometry = geometry;
-		this->path = shaderPath;
 
 		initShaders(shaderPath + ".vert", shaderPath + ".frag", shaderPath + ".geo");
 		loadShaders();
@@ -22,7 +21,6 @@ namespace vb01{
 
 	Shader::Shader(string vertShader, string fragShader){
 		this->geometry = false;
-		this->path = vertShader;
 
 		initShaders(vertShader, fragShader, "");
 		loadShaders();
@@ -30,19 +28,16 @@ namespace vb01{
 
 	Shader::Shader(string vertShader, string fragShader, string geoShader){
 		this->geometry = true;
-		this->path = vertShader;
 
 		initShaders(vertShader, fragShader, geoShader);
 		loadShaders();
 	}
 
-	Shader::~Shader(){}
-
 	string Shader::getName(){
-			int dirId = path.find_last_of('/');
+		int dirId = vString->path.find_last_of('/');
-			string name = (dirId != -1 ? path.substr(dirId + 1) : path);
+		string name = (dirId != -1 ? vString->path.substr(dirId + 1) : vString->path);
 
-			int dotId = path.find_last_of('.');
+		int dotId = name.find_last_of('.');
 
 		if(dotId != -1)
 			name = name.substr(0, dotId);
@@ -51,67 +46,12 @@ namespace vb01{
 	}
 
 	void Shader::initShaders(string vertShaderPath, string fragShaderPath, string geoShaderPath){
-		ifstream vertShaderFile, fragShaderFile;
+		AssetManager *am = AssetManager::getSingleton();
-		vertShaderFile.open(vertShaderPath);
+		vString = (ShaderAsset*)am->getAsset(vertShaderPath);
-		fragShaderFile.open(fragShaderPath);
+		fString = (ShaderAsset*)am->getAsset(fragShaderPath);
 
-		stringstream vertShaderStream, fragShaderStream;
+		if(geometry)
-		vertShaderStream << vertShaderFile.rdbuf();
+			gString = (ShaderAsset*)am->getAsset(geoShaderPath);
-		fragShaderStream << fragShaderFile.rdbuf();
-		
-		vertShaderFile.close();
-		fragShaderFile.close();
-		
-		vString = vertShaderStream.str();
-		fString = fragShaderStream.str();
-
-		if(geometry){
-			ifstream geoShaderFile;
-			geoShaderFile.open(geoShaderPath);
-
-			stringstream geoShaderStream;
-			geoShaderStream << geoShaderFile.rdbuf();
-
-			geoShaderFile.close();
-			gString = geoShaderStream.str();
-		}
-	}
-
-	void Shader::editShader(ShaderType type, int line, string insertion){
-		replaceLine(type, line, insertion);
-		loadShaders();
-	}
-
-	void Shader::replaceLine(ShaderType type, int line, string insertion){
-		string *shaderString;
-
-		switch(type){
-			case VERTEX_SHADER:
-				shaderString = &vString;
-				break;
-			case FRAGMENT_SHADER:
-				shaderString = &fString;
-				break;
-			case GEOMETRY_SHADER:
-				shaderString = &gString;
-				break;
-		}
-
-		int numPassedLines = 0, lineStart = -1, lineEnd = -1;
-		for(int i = 0; i < shaderString->length(); i++)
-			if(shaderString[0][i] == '\n'){
-				if(numPassedLines == line - 1)
-					lineStart = i + 1;
-
-				if(numPassedLines == line){
-					lineEnd = i;
-					break;
-				}
-
-				numPassedLines++;
-			}
-
-		*shaderString = shaderString->substr(0, lineStart) + insertion + shaderString->substr(lineEnd);
 	}
 
 	void Shader::pushShader(u32 &type, string &sString, int glType, ErrorType errorType){
@@ -124,11 +64,11 @@ namespace vb01{
 
 	void Shader::loadShaders(){
 		u32 vert, geo, frag;
-		pushShader(vert, vString, GL_VERTEX_SHADER, VERTEX_ERROR);
+		pushShader(vert, vString->shaderString, GL_VERTEX_SHADER, VERTEX_ERROR);
-		pushShader(frag, fString, GL_FRAGMENT_SHADER, FRAGMENT_ERROR);
+		pushShader(frag, fString->shaderString, GL_FRAGMENT_SHADER, FRAGMENT_ERROR);
 
 		if(geometry)
-			pushShader(geo, gString, GL_GEOMETRY_SHADER, GEOMETRY_ERROR);
+			pushShader(geo, gString->shaderString, GL_GEOMETRY_SHADER, GEOMETRY_ERROR);
 
 		id = glCreateProgram();
 		glAttachShader(id, vert);

shader.h

@@ -4,6 +4,7 @@
 #include <string>
 #include <glm/mat4x4.hpp>
 
+#include "shaderAsset.h"
 #include "vector.h"
 #include "util.h"
 
@@ -17,10 +18,10 @@ namespace vb01{
 			Shader(std::string, bool = false);
 			Shader(std::string, std::string);
 			Shader(std::string, std::string, std::string);
-			~Shader();
+			~Shader(){}
 			std::string getName();
-			void setNumLights(int);
 			void use();
+			void loadShaders();
 			void setMat4(glm::mat4, std::string);
 			void setVec4(Vector4, std::string);
 			void setVec3(Vector3, std::string);
@@ -29,17 +30,16 @@ namespace vb01{
 			void setBool(bool, std::string);
 			void setInt(int, std::string);
 			void setUnsignedInt(u32, std::string);
-			void editShader(ShaderType, int, std::string);
+			void editShader(ShaderType, int, std::string){}
 			inline bool isGeometry(){return geometry;}
 		private:
 			void initShaders(std::string, std::string, std::string);
 			void replaceLine(ShaderType, int, std::string);
-			void loadShaders();
 			void pushShader(u32&, std::string&, int, ErrorType);
 			void checkCompileErrors(u32, ErrorType);
 			u32 id;
 			bool geometry = false;
-			std::string path, vString, fString, gString;
+			ShaderAsset *vString = nullptr, *fString = nullptr, *gString = nullptr;
 
 			friend class ShaderTest;
 	};

shaderAsset.h

@@ -0,0 +1,17 @@
+#ifndef SHADER_ASSET_H
+#define SHADER_ASSET_H
+
+#include "asset.h" 
+
+namespace vb01{
+	struct ShaderAsset : public Asset{
+		std::string shaderString = ""; 
+
+		ShaderAsset(std::string p, std::string str){
+			path = p;
+			shaderString = str;
+		}
+	};
+}
+
+#endif

shaderReader.cpp

@@ -0,0 +1,30 @@
+#include "shaderReader.h"
+#include "shaderAsset.h"
+
+#include <sstream>
+#include <fstream>
+
+namespace vb01{
+	using namespace std;
+
+	static ShaderReader *shaderReader = nullptr;
+
+	ShaderReader* ShaderReader::getSingleton(){
+		if(!shaderReader)
+			shaderReader = new ShaderReader();
+		
+		return shaderReader;
+	}
+
+	Asset* ShaderReader::readAsset(string path){
+		ifstream shaderFile;
+		shaderFile.open(path);
+		
+		stringstream shaderStream;
+		shaderStream << shaderFile.rdbuf();
+		
+		shaderFile.close();
+
+		return new ShaderAsset(path, shaderStream.str());
+	}
+}

shaderReader.h

@@ -0,0 +1,16 @@
+#ifndef SHADER_READER_H
+#define SHADER_READER_H
+
+#include "abstractAssetReader.h"
+
+namespace vb01{
+	class ShaderReader : public AbstractAssetReader{
+		public:
+			static ShaderReader* getSingleton();
+			Asset* readAsset(std::string);
+		private:
+			ShaderReader(){}
+	};
+}
+
+#endif