Load compute shaders the easy way

Author: scott

jme3-core/src/main/java/com/jme3/shadow/SdsmFitter.java

@@ -31,8 +31,6 @@
  */
 package com.jme3.shadow;
 
-import com.jme3.asset.AssetInfo;
-import com.jme3.asset.AssetKey;
 import com.jme3.asset.AssetManager;
 import com.jme3.math.Matrix4f;
 import com.jme3.math.Vector2f;
@@ -45,9 +43,10 @@
 import com.jme3.renderer.opengl.ShaderStorageBufferObject;
 import com.jme3.texture.Texture;
 
-import java.io.IOException;
-import java.io.InputStream;
-import java.util.*;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.LinkedList;
+import java.util.List;
 
 /**
  * Compute shader used in SDSM.
@@ -310,8 +309,8 @@ public SdsmFitter(GL4 gl, Renderer renderer, AssetManager assetManager) {
         this.renderer = renderer;
 
         // Load compute shaders
-        String reduceSource = loadShaderSource(assetManager, REDUCE_DEPTH_SHADER);
-        String fitSource = loadShaderSource(assetManager, FIT_FRUSTUMS_SHADER);
+        String reduceSource = (String)assetManager.loadAsset(REDUCE_DEPTH_SHADER);
+        String fitSource = (String)assetManager.loadAsset(FIT_FRUSTUMS_SHADER);
 
         depthReduceShader = new ComputeShader(gl, reduceSource);
         fitFrustumsShader = new ComputeShader(gl, fitSource);
@@ -449,16 +448,6 @@ private SdsmResultHolder getResultHolderForUse() {
         }
     }
 
-    private static String loadShaderSource(AssetManager assetManager, String resourcePath) {
-        //TODO: Should these shaders get special loaders or something?
-        AssetInfo info = assetManager.locateAsset(new AssetKey<>(resourcePath));
-        try (InputStream is = info.openStream()) {
-            return new Scanner(is).useDelimiter("\\A").next();
-        } catch (IOException e) {
-            throw new RendererException("Failed to load shader: " + resourcePath);
-        }
-    }
-
     private static float getProjectionToViewZ(float near, float far, float projZPos) {
         float a = far / (far - near);
         float b = far * near / (near - far);

jme3-core/src/main/resources/com/jme3/asset/General.cfg

@@ -20,7 +20,7 @@ LOADER com.jme3.scene.plugins.ogre.MeshLoader : meshxml, mesh.xml
 LOADER com.jme3.scene.plugins.ogre.SkeletonLoader : skeletonxml, skeleton.xml
 LOADER com.jme3.scene.plugins.ogre.MaterialLoader : material
 LOADER com.jme3.scene.plugins.ogre.SceneLoader : scene
-LOADER com.jme3.shader.plugins.GLSLLoader : vert, frag, geom, tsctrl, tseval, glsl, glsllib
+LOADER com.jme3.shader.plugins.GLSLLoader : vert, frag, geom, tsctrl, tseval, glsl, glsllib, comp
 LOADER com.jme3.scene.plugins.fbx.FbxLoader : fbx
 LOADER com.jme3.scene.plugins.gltf.GltfLoader : gltf
 LOADER com.jme3.scene.plugins.gltf.BinLoader : bin