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; 3459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 3559b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport com.jme3.export.*; 3659b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.io.IOException; 3759b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.Map; 3859b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.SortedMap; 3959b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartaimport java.util.TreeMap; 4059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4159b2e6871c65f58fdad78cd7229c292f6a177578Scott Bartapublic class DefineList implements Savable { 4259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private final SortedMap<String, String> defines = new TreeMap<String, String>(); 4459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta private String compiled = null; 4559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void write(JmeExporter ex) throws IOException{ 4759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta OutputCapsule oc = ex.getCapsule(this); 4859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 4959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String[] keys = new String[defines.size()]; 5059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String[] vals = new String[defines.size()]; 5159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int i = 0; 5359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (Map.Entry<String, String> define : defines.entrySet()){ 5459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta keys[i] = define.getKey(); 5559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta vals[i] = define.getValue(); 5659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta i++; 5759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 5859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 5959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta oc.write(keys, "keys", null); 6059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta oc.write(vals, "vals", null); 6159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // for compatability only with older versions 6359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta oc.write(compiled, "compiled", null); 6459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 6559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void read(JmeImporter im) throws IOException{ 6759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta InputCapsule ic = im.getCapsule(this); 6859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 6959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String[] keys = ic.readStringArray("keys", null); 7059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String[] vals = ic.readStringArray("vals", null); 7159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (int i = 0; i < keys.length; i++){ 7259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta defines.put(keys[i], vals[i]); 7359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 7459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = ic.readString("compiled", null); 7659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 7759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 7859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void clear() { 7959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta defines.clear(); 8059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = ""; 8159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 8359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public String get(String key){ 8459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // I do not see the point of forcing a rebuild on get() 8559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // so I'm commenting it out. -pspeed 8659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta //compiled = null; 8759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return defines.get(key); 8859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 8959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// public void set(String key, String val){ 9159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// compiled = null; 9259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// defines.put(key, val); 9359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta// } 9459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 9559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public boolean set(String key, VarType type, Object val){ 9659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (val == null){ 9759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta defines.remove(key); 9859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = null; 9959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return true; 10059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 10159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 10259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta switch (type){ 10359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta case Boolean: 10459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if ( ((Boolean) val).booleanValue() ) { 10559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // same literal, != should work 10659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if( defines.put(key, "1") != "1" ) { 10759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = null; 10859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return true; 10959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } else if (defines.containsKey(key)) { 11159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta defines.remove(key); 11259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = null; 11359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return true; 11459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 11559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 11659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta break; 11759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta case Float: 11859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta case Int: 11959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta String original = defines.put(key, val.toString()); 12059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (!val.equals(original)) { 12159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = null; 12259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return true; 12359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 12459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta break; 12559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta default: 12659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta // same literal, != should work 12759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (defines.put(key, "1") != "1") { 12859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = null; 12959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return true; 13059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta break; 13259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return false; 13559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 13659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 13759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public boolean remove(String key){ 13859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (defines.remove(key) != null) { 13959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = null; 14059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return true; 14159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 14259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return false; 14459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 14559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 14659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public void addFrom(DefineList other){ 14759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (other == null) 14859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return; 14959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = null; 15159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta defines.putAll(other.defines); 15259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 15359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 15459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public String getCompiled(){ 15559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (compiled == null){ 15659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta StringBuilder sb = new StringBuilder(); 15759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (Map.Entry<String, String> entry : defines.entrySet()){ 15859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sb.append("#define ").append(entry.getKey()).append(" "); 15959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sb.append(entry.getValue()).append('\n'); 16059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 16159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta compiled = sb.toString(); 16259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 16359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return compiled; 16459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 16559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 16659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta @Override 16759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta public String toString(){ 16859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta StringBuilder sb = new StringBuilder(); 16959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta int i = 0; 17059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta for (Map.Entry<String, String> entry : defines.entrySet()) { 17159b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sb.append(entry.getKey()); 17259b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta if (i != defines.size() - 1) 17359b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta sb.append(", "); 17459b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 17559b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta i++; 17659b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17759b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta return sb.toString(); 17859b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta } 17959b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta 18059b2e6871c65f58fdad78cd7229c292f6a177578Scott Barta} 181