Merge pull request #167 from GreenCubes/master

Implemented FBX model loader
experimental
normen 11 years ago
commit fb40a14de2
  1. 1
      jme3-core/src/main/resources/com/jme3/asset/Desktop.cfg
  2. 1
      jme3-plugins/build.gradle
  3. 77
      jme3-plugins/src/fbx/java/com/jme3/scene/plugins/fbx/AnimationList.java
  4. 77
      jme3-plugins/src/fbx/java/com/jme3/scene/plugins/fbx/ContentTextureKey.java
  5. 89
      jme3-plugins/src/fbx/java/com/jme3/scene/plugins/fbx/ContentTextureLocator.java
  6. 54
      jme3-plugins/src/fbx/java/com/jme3/scene/plugins/fbx/SceneKey.java
  7. 1586
      jme3-plugins/src/fbx/java/com/jme3/scene/plugins/fbx/SceneLoader.java
  8. 64
      jme3-plugins/src/fbx/java/com/jme3/scene/plugins/fbx/file/FBXElement.java
  9. 42
      jme3-plugins/src/fbx/java/com/jme3/scene/plugins/fbx/file/FBXFile.java
  10. 225
      jme3-plugins/src/fbx/java/com/jme3/scene/plugins/fbx/file/FBXReader.java

@ -22,3 +22,4 @@ LOADER com.jme3.scene.plugins.ogre.MaterialLoader : material
LOADER com.jme3.scene.plugins.ogre.SceneLoader : scene LOADER com.jme3.scene.plugins.ogre.SceneLoader : scene
LOADER com.jme3.scene.plugins.blender.BlenderModelLoader : blend LOADER com.jme3.scene.plugins.blender.BlenderModelLoader : blend
LOADER com.jme3.shader.plugins.GLSLLoader : vert, frag, glsl, glsllib LOADER com.jme3.shader.plugins.GLSLLoader : vert, frag, glsl, glsllib
LOADER com.jme3.scene.plugins.fbx.SceneLoader : fbx

@ -6,6 +6,7 @@ sourceSets {
main { main {
java { java {
srcDir 'src/ogre/java' srcDir 'src/ogre/java'
srcDir 'src/fbx/java'
srcDir 'src/xml/java' srcDir 'src/xml/java'
} }
} }

@ -0,0 +1,77 @@
/*
* Copyright (c) 2009-2014 jMonkeyEngine
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.jme3.scene.plugins.fbx;
import java.util.ArrayList;
import java.util.List;
/**
* Defines animations set that will be created while loading FBX scene
* <p>Animation <code>name</code> is using to access animation via {@link AnimControl}.<br>
* <code>firstFrame</code> and <code>lastFrame</code> defines animation time interval.<br>
* Use <code>layerName</code> also to define source animation layer in the case of multiple layers in the scene.<br>
* Skeletal animations will be created if only scene contain skeletal bones</p>
*/
public class AnimationList {
List<AnimInverval> list = new ArrayList<AnimInverval>();
/**
* Use in the case of multiple animation layers in FBX asset
* @param name - animation name to assess via {@link AnimControl}
* @param layerName - source layer
*/
public void add(String name, int firstFrame, int lastFrame) {
add(name, null, firstFrame, lastFrame);
}
/**
* Use in the case of multiple animation layers in FBX asset
* @param name - animation name to assess via {@link AnimControl}
* @param layerName - source layer
*/
public void add(String name, String layerName, int firstFrame, int lastFrame) {
AnimInverval cue = new AnimInverval();
cue.name = name;
cue.layerName = layerName;
cue.firstFrame = firstFrame;
cue.lastFrame = lastFrame;
list.add(cue);
}
static class AnimInverval {
String name;
String layerName;
int firstFrame;
int lastFrame;
}
}

@ -0,0 +1,77 @@
/*
* Copyright (c) 2009-2014 jMonkeyEngine
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.jme3.scene.plugins.fbx;
import com.jme3.asset.TextureKey;
import com.jme3.export.InputCapsule;
import com.jme3.export.JmeExporter;
import com.jme3.export.JmeImporter;
import com.jme3.export.OutputCapsule;
import java.io.IOException;
/**
* Used to load textures from image binary content.
* <p>Filename is required to acquire proper type asset loader according to extension.</p>
*/
public class ContentTextureKey extends TextureKey {
private byte[] content;
public ContentTextureKey(String name, byte[] content) {
super(name);
this.content = content;
}
public byte[] getContent() {
return content;
}
@Override
public String toString() {
return super.toString() + " " + content.length + " bytes";
}
@Override
public void write(JmeExporter ex) throws IOException {
super.write(ex);
OutputCapsule oc = ex.getCapsule(this);
oc.write(content, "content", new byte[0]);
}
@Override
public void read(JmeImporter im) throws IOException {
super.read(im);
InputCapsule ic = im.getCapsule(this);
content = ic.readByteArray("content", new byte[0]);
}
}

@ -0,0 +1,89 @@
/*
* Copyright (c) 2009-2014 jMonkeyEngine
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.jme3.scene.plugins.fbx;
import com.jme3.asset.AssetInfo;
import com.jme3.asset.AssetKey;
import com.jme3.asset.AssetLocator;
import com.jme3.asset.AssetManager;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
* Used to locate a resource based on a {@link ContentTextureKey}.
*/
public class ContentTextureLocator implements AssetLocator {
private static final Logger logger = Logger.getLogger(ContentTextureLocator.class.getName());
@Override
public void setRootPath(String rootPath) {
}
@SuppressWarnings("rawtypes")
@Override
public AssetInfo locate(AssetManager manager, AssetKey key) {
if(key instanceof ContentTextureKey) {
String name = key.getName();
byte[] content = ((ContentTextureKey) key).getContent();
if(content != null) {
return new ContentAssetInfo(manager, key, content);
} else {
logger.log(Level.WARNING, "No content for " + name);
return null;
}
} else {
logger.log(Level.SEVERE, "AssetKey should be TextureContentKey instance");
return null;
}
}
private class ContentAssetInfo extends AssetInfo {
private InputStream stream;
@SuppressWarnings("rawtypes")
public ContentAssetInfo(AssetManager assetManager, AssetKey key, byte[] content) {
super(assetManager, key);
this.stream = (content != null) ? new ByteArrayInputStream(content) : null;
}
@Override
public InputStream openStream() {
return stream;
}
}
}

@ -0,0 +1,54 @@
/*
* Copyright (c) 2009-2014 jMonkeyEngine
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.jme3.scene.plugins.fbx;
import com.jme3.asset.ModelKey;
public class SceneKey extends ModelKey {
private final AnimationList animList;
public SceneKey(String name) {
super(name);
this.animList = null;
}
public SceneKey(String name, AnimationList animationList) {
super(name);
this.animList = animationList;
}
public AnimationList getAnimations() {
return this.animList;
}
}

@ -0,0 +1,64 @@
/*
* Copyright (c) 2009-2014 jMonkeyEngine
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.jme3.scene.plugins.fbx.file;
import java.util.ArrayList;
import java.util.List;
public class FBXElement {
public String id;
public List<Object> properties;
/*
* Y - signed short
* C - boolean
* I - signed integer
* F - float
* D - double
* L - long
* R - byte array
* S - string
* f - array of floats
* i - array of ints
* d - array of doubles
* l - array of longs
* b - array of boleans
* c - array of unsigned bytes (represented as array of ints)
*/
public char[] propertiesTypes;
public List<FBXElement> children = new ArrayList<FBXElement>();
public FBXElement(int propsCount) {
properties = new ArrayList<Object>(propsCount);
propertiesTypes = new char[propsCount];
}
}

@ -0,0 +1,42 @@
/*
* Copyright (c) 2009-2014 jMonkeyEngine
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.jme3.scene.plugins.fbx.file;
import java.util.ArrayList;
import java.util.List;
public class FBXFile {
public List<FBXElement> rootElements = new ArrayList<FBXElement>();
public long version;
}

@ -0,0 +1,225 @@
/*
* Copyright (c) 2009-2014 jMonkeyEngine
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of 'jMonkeyEngine' nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package com.jme3.scene.plugins.fbx.file;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.util.Arrays;
import java.util.zip.InflaterInputStream;
public class FBXReader {
public static final int BLOCK_SENTINEL_LENGTH = 13;
public static final byte[] BLOCK_SENTINEL_DATA = new byte[BLOCK_SENTINEL_LENGTH];
/**
* Majic string at start:
* "Kaydara FBX Binary\x20\x20\x00\x1a\x00"
*/
public static final byte[] HEAD_MAGIC = new byte[]{0x4b, 0x61, 0x79, 0x64, 0x61, 0x72, 0x61, 0x20, 0x46, 0x42, 0x58, 0x20, 0x42, 0x69, 0x6e, 0x61, 0x72, 0x79, 0x20, 0x20, 0x00, 0x1a, 0x00};
public static FBXFile readFBX(InputStream stream) throws IOException {
FBXFile fbxFile = new FBXFile();
// Read file to byte buffer so we can know current position in file
ByteBuffer byteBuffer = readToByteBuffer(stream);
try {
stream.close();
} catch(IOException e) {
}
// Check majic header
byte[] majic = getBytes(byteBuffer, HEAD_MAGIC.length);
if(!Arrays.equals(HEAD_MAGIC, majic))
throw new IOException("Not FBX file");
// Read version
fbxFile.version = getUInt(byteBuffer);
// Read root elements
while(true) {
FBXElement e = readFBXElement(byteBuffer);
if(e == null)
break;
fbxFile.rootElements.add(e);
}
return fbxFile;
}
private static FBXElement readFBXElement(ByteBuffer byteBuffer) throws IOException {
long endOffset = getUInt(byteBuffer);
if(endOffset == 0)
return null;
long propCount = getUInt(byteBuffer);
getUInt(byteBuffer); // Properties length unused
FBXElement element = new FBXElement((int) propCount);
element.id = new String(getBytes(byteBuffer, getUByte(byteBuffer)));
for(int i = 0; i < propCount; ++i) {
char dataType = readDataType(byteBuffer);
element.properties.add(readData(byteBuffer, dataType));
element.propertiesTypes[i] = dataType;
}
if(byteBuffer.position() < endOffset) {
while(byteBuffer.position() < (endOffset - BLOCK_SENTINEL_LENGTH))
element.children.add(readFBXElement(byteBuffer));
if(!Arrays.equals(BLOCK_SENTINEL_DATA, getBytes(byteBuffer, BLOCK_SENTINEL_LENGTH)))
throw new IOException("Failed to read block sentinel, expected 13 zero bytes");
}
if(byteBuffer.position() != endOffset)
throw new IOException("Data length not equal to expected");
return element;
}
private static Object readData(ByteBuffer byteBuffer, char dataType) throws IOException {
switch(dataType) {
case 'Y':
return byteBuffer.getShort();
case 'C':
return byteBuffer.get() == 1;
case 'I':
return byteBuffer.getInt();
case 'F':
return byteBuffer.getFloat();
case 'D':
return byteBuffer.getDouble();
case 'L':
return byteBuffer.getLong();
case 'R':
return getBytes(byteBuffer, (int) getUInt(byteBuffer));
case 'S':
return new String(getBytes(byteBuffer, (int) getUInt(byteBuffer)));
case 'f':
return readArray(byteBuffer, 'f', 4);
case 'i':
return readArray(byteBuffer, 'i', 4);
case 'd':
return readArray(byteBuffer, 'd', 8);
case 'l':
return readArray(byteBuffer, 'l', 8);
case 'b':
return readArray(byteBuffer, 'b', 1);
case 'c':
return readArray(byteBuffer, 'c', 1);
}
throw new IOException("Unknown data type: " + dataType);
}
private static Object readArray(ByteBuffer byteBuffer, char type, int bytes) throws IOException {
int count = (int) getUInt(byteBuffer);
int encoding = (int) getUInt(byteBuffer);
int length = (int) getUInt(byteBuffer);
byte[] data = getBytes(byteBuffer, length);
if(encoding == 1)
data = inflate(data);
if(data.length != count * bytes)
throw new IOException("Wrong data lenght. Expected: " + count * bytes + ", got: " + data.length);
ByteBuffer dis = ByteBuffer.wrap(data).order(ByteOrder.LITTLE_ENDIAN);
switch(type) {
case 'f':
float[] arr = new float[count];
for(int i = 0; i < count; ++i)
arr[i] = dis.getFloat();
return arr;
case 'i':
int[] arr2 = new int[count];
for(int i = 0; i < count; ++i)
arr2[i] = dis.getInt();
return arr2;
case 'd':
double[] arr3 = new double[count];
for(int i = 0; i < count; ++i)
arr3[i] = dis.getDouble();
return arr3;
case 'l':
long[] arr4 = new long[count];
for(int i = 0; i < count; ++i)
arr4[i] = dis.getLong();
return arr4;
case 'b':
boolean[] arr5 = new boolean[count];
for(int i = 0; i < count; ++i)
arr5[i] = dis.get() == 1;
return arr5;
case 'c':
int[] arr6 = new int[count];
for(int i = 0; i < count; ++i)
arr6[i] = dis.get() & 0xFF;
return arr6;
}
throw new IOException("Unknown array data type: " + type);
}
private static byte[] inflate(byte[] input) throws IOException {
InflaterInputStream gzis = new InflaterInputStream(new ByteArrayInputStream(input));
ByteArrayOutputStream out = new ByteArrayOutputStream();
byte[] buffer = new byte[1024];
while(gzis.available() > 0) {
int l = gzis.read(buffer);
if(l > 0)
out.write(buffer, 0, l);
}
return out.toByteArray();
}
private static char readDataType(ByteBuffer byteBuffer) {
return (char) byteBuffer.get();
}
private static long getUInt(ByteBuffer byteBuffer) {
return byteBuffer.getInt() & 0x00000000ffffffffL;
}
private static int getUByte(ByteBuffer byteBuffer) {
return byteBuffer.get() & 0xFF;
}
private static byte[] getBytes(ByteBuffer byteBuffer, int size) {
byte[] b = new byte[size];
byteBuffer.get(b);
return b;
}
private static ByteBuffer readToByteBuffer(InputStream input) throws IOException {
ByteArrayOutputStream out = new ByteArrayOutputStream(2048);
byte[] tmp = new byte[2048];
while(true) {
int r = input.read(tmp);
if(r == -1)
break;
out.write(tmp, 0, r);
}
return ByteBuffer.wrap(out.toByteArray()).order(ByteOrder.LITTLE_ENDIAN);
}
}
Loading…
Cancel
Save