Use different FSTObjectInput constructor which passes in the FSTConfi…

…guration
numenta · May 2, 2016 · 6693834 · 6693834
1 parent 0b3630f
commit 6693834
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/src/main/java/org/numenta/nupic/serialize/HTMObjectInput.java b/src/main/java/org/numenta/nupic/serialize/HTMObjectInput.java
@@ -4,11 +4,12 @@
 import java.io.InputStream;
 
 import org.numenta.nupic.Persistable;
+import org.nustaq.serialization.FSTConfiguration;
 import org.nustaq.serialization.FSTObjectInput;
 
 public class HTMObjectInput extends FSTObjectInput {
-    public HTMObjectInput(InputStream in) throws IOException {
-        super(in);
+    public HTMObjectInput(InputStream in, FSTConfiguration config) throws IOException {
+        super(in, config);
     }
 
     @SuppressWarnings("rawtypes")

diff --git a/src/main/java/org/numenta/nupic/serialize/HTMObjectOutput.java b/src/main/java/org/numenta/nupic/serialize/HTMObjectOutput.java
@@ -4,11 +4,12 @@
 import java.io.OutputStream;
 
 import org.numenta.nupic.Persistable;
+import org.nustaq.serialization.FSTConfiguration;
 import org.nustaq.serialization.FSTObjectOutput;
 
 public class HTMObjectOutput extends FSTObjectOutput {
-    public HTMObjectOutput(OutputStream out) {
-        super(out);
+    public HTMObjectOutput(OutputStream out, FSTConfiguration config) {
+        super(out, config);
     }
 
     @SuppressWarnings("rawtypes")

diff --git a/src/main/java/org/numenta/nupic/serialize/SerializerCore.java b/src/main/java/org/numenta/nupic/serialize/SerializerCore.java
@@ -104,7 +104,7 @@ private void readObject(ObjectInputStream in) throws IOException, ClassNotFoundE
      * @throws IOException 
      */
     public HTMObjectInput getObjectInput(InputStream is) throws IOException {
-        return new HTMObjectInput(is);
+        return new HTMObjectInput(is, fastSerialConfig);
     }
 
     /**
@@ -113,7 +113,7 @@ public HTMObjectInput getObjectInput(InputStream is) throws IOException {
      * @return  the HTMObjectOutput
      */
     public <T extends Persistable> HTMObjectOutput getObjectOutput(OutputStream os) {
-        return new HTMObjectOutput(os);
+        return new HTMObjectOutput(os, fastSerialConfig);
     }
 
     /**