Added a StreamableHashSet class.

git-svn-id: svn+ssh://src.earth.threerings.net/narya/trunk@4809 542714f4-19e9-0310-aa3c-eee0fc999fb1
2007-08-10 19:44:02 +00:00
parent 1eda45d8c8
commit 2ac50f387a
1 changed files with 82 additions and 0 deletions
@@ -0,0 +1,82 @@
+//
+// $Id: StreamableHashMap.java 4602 2007-02-24 00:39:27Z mdb $
+//
+// Narya library - tools for developing networked games
+// Copyright (C) 2002-2007 Three Rings Design, Inc., All Rights Reserved
+// http://www.threerings.net/code/narya/
+//
+// This library is free software; you can redistribute it and/or modify it
+// under the terms of the GNU Lesser General Public License as published
+// by the Free Software Foundation; either version 2.1 of the License, or
+// (at your option) any later version.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+
+package com.threerings.util;
+
+import java.io.IOException;
+import java.util.HashSet;
+
+import com.threerings.io.ObjectInputStream;
+import com.threerings.io.ObjectOutputStream;
+import com.threerings.io.Streamable;
+
+/**
+ * A {@link HashSet} extension that can be streamed. The values in the set must also be of
+ * streamable types.
+ *
+ * @see Streamable
+ */
+public class StreamableHashSet<E> extends HashSet<E>
+    implements Streamable
+{
+    /**
+     * Constructs an empty hash set with the specified number of hash
+     * buckets.
+     */
+    public StreamableHashSet (int buckets, float loadFactor)
+    {
+        super(buckets, loadFactor);
+    }
+
+    /**
+     * Constructs an empty hash set with the default number of hash
+     * buckets.
+     */
+    public StreamableHashSet ()
+    {
+        super();
+    }
+
+    /**
+     * Writes our custom streamable fields.
+     */
+    public void writeObject (ObjectOutputStream out)
+        throws IOException
+    {
+        out.writeInt(size());
+        for (E value : this) {
+            out.writeObject(value);
+        }
+    }
+
+    /**
+     * Reads our custom streamable fields.
+     */
+    public void readObject (ObjectInputStream in)
+        throws IOException, ClassNotFoundException
+    {
+        int ecount = in.readInt();
+        for (int ii = 0; ii < ecount; ii++) {
+            @SuppressWarnings("unchecked") E value = (E)in.readObject();
+            add(value);
+        }
+    }
+}