159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta/* 259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Copyright (c) 2009-2010 jMonkeyEngine 359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * All rights reserved. 459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * Redistribution and use in source and binary forms, with or without 659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * modification, are permitted provided that the following conditions are 759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * met: 859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions of source code must retain the above copyright 1059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer. 1159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Redistributions in binary form must reproduce the above copyright 1359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * notice, this list of conditions and the following disclaimer in the 1459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * documentation and/or other materials provided with the distribution. 1559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 1659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * * Neither the name of 'jMonkeyEngine' nor the names of its contributors 1759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * may be used to endorse or promote products derived from this software 1859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * without specific prior written permission. 1959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 2059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 2259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 2359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR 2459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 2559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 2659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 2759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF 2859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING 2959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 3059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 3159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 3259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapackage com.jme3.shader.plugins; 3459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.asset.AssetInfo; 3659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.asset.AssetKey; 3759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.asset.AssetLoader; 3859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.asset.AssetManager; 3959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.BufferedReader; 4059b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.IOException; 4159b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.InputStream; 4259b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.InputStreamReader; 4359b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.*; 4459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta/** 4659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * GLSL File parser that supports #import pre-processor statement 4759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 4859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapublic class GLSLLoader implements AssetLoader { 4959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private AssetManager owner; 5159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private Map<String, DependencyNode> dependCache = new HashMap<String, DependencyNode>(); 5259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private class DependencyNode { 5459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private String shaderSource; 5659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private String shaderName; 5759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private final Set<DependencyNode> dependsOn = new HashSet<DependencyNode>(); 5959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private final Set<DependencyNode> dependOnMe = new HashSet<DependencyNode>(); 6059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public DependencyNode(String shaderName){ 6259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta this.shaderName = shaderName; 6359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 6459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void setSource(String source){ 6659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta this.shaderSource = source; 6759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 6859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void addDependency(DependencyNode node){ 7059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (this.dependsOn.contains(node)) 7159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return; // already contains dependency 7259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// System.out.println(shaderName + " depend on "+node.shaderName); 7459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta this.dependsOn.add(node); 7559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta node.dependOnMe.add(this); 7659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 7759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 7959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private class GlslDependKey extends AssetKey<InputStream> { 8159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public GlslDependKey(String name){ 8259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta super(name); 8359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta @Override 8559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public boolean shouldCache(){ 8659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return false; 8759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private DependencyNode loadNode(InputStream in, String nodeName) throws IOException{ 9159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta DependencyNode node = new DependencyNode(nodeName); 9259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (in == null) 9359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new IOException("Dependency "+nodeName+" cannot be found."); 9459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta StringBuilder sb = new StringBuilder(); 9659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta BufferedReader r = new BufferedReader(new InputStreamReader(in)); 9759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta while (r.ready()){ 9859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String ln = r.readLine(); 9959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (ln.startsWith("#import ")){ 10059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ln = ln.substring(8).trim(); 10159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (ln.startsWith("\"") && ln.endsWith("\"") && ln.length() > 3){ 10259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // import user code 10359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // remove quotes to get filename 10459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta ln = ln.substring(1, ln.length()-1); 10559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (ln.equals(nodeName)) 10659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new IOException("Node depends on itself."); 10759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 10859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // check cache first 10959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta DependencyNode dependNode = dependCache.get(ln); 11059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (dependNode == null){ 11159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta GlslDependKey key = new GlslDependKey(ln); 11259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // make sure not to register an input stream with 11359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // the cache.. 11459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta InputStream stream = (InputStream) owner.loadAsset(key); 11559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta dependNode = loadNode(stream, ln); 11659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta node.addDependency(dependNode); 11859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// }else if (ln.startsWith("uniform") || ln.startsWith("varying") || ln.startsWith("attribute")){ 12059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// // these variables are included as dependencies as well 12159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// DependencyNode dependNode = dependCache.get(ln); 12259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// if (dependNode == null){ 12359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// // the source and name are the same for variable dependencies 12459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// dependNode = new DependencyNode(ln); 12559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// dependNode.setSource(ln); 12659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// dependCache.put(ln, dependNode); 12759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// } 12859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// node.addDependency(dependNode); 12959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 13059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sb.append(ln).append('\n'); 13159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta r.close(); 13459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta node.setSource(sb.toString()); 13659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta dependCache.put(nodeName, node); 13759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return node; 13859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private DependencyNode nextIndependentNode(List<DependencyNode> checkedNodes){ 14159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta Collection<DependencyNode> allNodes = dependCache.values(); 14259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (allNodes == null || allNodes.isEmpty()) 14359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return null; 14459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (DependencyNode node : allNodes){ 14659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (node.dependsOn.isEmpty()){ 14759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return node; 14859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 14959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // circular dependency found.. 15259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (DependencyNode node : allNodes){ 15359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta System.out.println(node.shaderName); 15459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta throw new RuntimeException("Circular dependency."); 15659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private String resolveDependencies(DependencyNode root){ 15959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta StringBuilder sb = new StringBuilder(); 16059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta List<DependencyNode> checkedNodes = new ArrayList<DependencyNode>(); 16259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta checkedNodes.add(root); 16359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta while (true){ 16459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta DependencyNode indepnNode = nextIndependentNode(checkedNodes); 16559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (indepnNode == null) 16659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta break; 16759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sb.append(indepnNode.shaderSource).append('\n'); 16959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta dependCache.remove(indepnNode.shaderName); 17059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 17159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // take out this dependency 17259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (Iterator<DependencyNode> iter = indepnNode.dependOnMe.iterator(); 17359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta iter.hasNext();){ 17459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta DependencyNode dependNode = iter.next(); 17559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta iter.remove(); 17659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta dependNode.dependsOn.remove(indepnNode); 17759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 18059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// System.out.println(sb.toString()); 18159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// System.out.println("--------------------------------------------------"); 18259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 18359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return sb.toString(); 18459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 18559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 18659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta /** 18759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * 18859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @param owner 18959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @param in 19059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @param extension 19159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @param key 19259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @return 19359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta * @throws java.io.IOException 19459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta */ 19559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public Object load(AssetInfo info) throws IOException { 19659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // The input stream provided is for the vertex shader, 19759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // to retrieve the fragment shader, use the content manager 19859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta this.owner = info.getManager(); 19959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (info.getKey().getExtension().equals("glsllib")){ 20059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // NOTE: Loopback, GLSLLIB is loaded by this loader 20159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // and needs data as InputStream 20259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return info.openStream(); 20359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta }else{ 20459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // GLSLLoader wants result as String for 20559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // fragment shader 20659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta DependencyNode rootNode = loadNode(info.openStream(), "[main]"); 20759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String code = resolveDependencies(rootNode); 20859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta dependCache.clear(); 20959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return code; 21059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 21159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 21259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 21359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta} 214