Update SparseBinaryMatrix to use Serializable

numenta · Dec 4, 2015 · e221e3d · e221e3d
1 parent 8917eae
commit e221e3d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/nupic/math/SparseBinaryMatrix.hpp b/src/nupic/math/SparseBinaryMatrix.hpp
@@ -34,6 +34,7 @@
 #include <nupic/math/StlIo.hpp>
 #include <nupic/math/ArrayAlgo.hpp>
 #include <nupic/proto/SparseBinaryMatrixProto.capnp.h>
+#include <nupic/types/Serializable.hpp>
 
 namespace nupic {
 
@@ -45,7 +46,7 @@ namespace nupic {
  *
  */
 template <typename UI1 = nupic::UInt32, typename UI2 = nupic::UInt32>
-class SparseBinaryMatrix {
+class SparseBinaryMatrix : Serializable<SparseBinaryMatrixProto> {
 public:
   typedef UI1 size_type;
   typedef UI2 nz_index_type;
@@ -1294,6 +1295,8 @@ class SparseBinaryMatrix {
     }
   }
 
+  using Serializable::write;
+
   inline void write(SparseBinaryMatrixProto::Builder &proto) const {
     proto.setNumRows(nRows());
     proto.setNumColumns(nCols());
@@ -1307,6 +1310,8 @@ class SparseBinaryMatrix {
     }
   }
 
+  using Serializable::read;
+
   inline void read(SparseBinaryMatrixProto::Reader &proto) {
     auto rows = proto.getNumRows();
     auto columns = proto.getNumColumns();