[Jython-checkins] SF.net SVN: jython:[5378] trunk/jython/src/org/python/modules/PyStruct.java

[<zy...@us...>]

Revision: 5378
          http://jython.svn.sourceforge.net/jython/?rev=5378&view=rev
Author:   zyasoft
Date:     2008-10-12 00:25:07 +0000 (Sun, 12 Oct 2008)

Log Message:
-----------
Missing module for previous commit!

Added Paths:
-----------
    trunk/jython/src/org/python/modules/PyStruct.java

Added: trunk/jython/src/org/python/modules/PyStruct.java
===================================================================
--- trunk/jython/src/org/python/modules/PyStruct.java	                        (rev 0)
+++ trunk/jython/src/org/python/modules/PyStruct.java	2008-10-12 00:25:07 UTC (rev 5378)
@@ -0,0 +1,88 @@
+package org.python.modules;
+
+import org.python.core.Py;
+import org.python.core.PyArray;
+import org.python.core.PyNewWrapper;
+import org.python.core.PyObject;
+import org.python.core.PyString;
+import org.python.core.PyTuple;
+import org.python.core.PyType;
+import org.python.expose.ExposedGet;
+import org.python.expose.ExposedMethod;
+import org.python.expose.ExposedNew;
+import org.python.expose.ExposedType;
+
+@ExposedType(name = "Struct", base = PyObject.class)
+public class PyStruct extends PyObject {
+    public static final PyType TYPE = PyType.fromClass(PyStruct.class);
+    
+    @ExposedGet
+    public final String format;
+    
+    @ExposedGet
+    public final int size;
+    
+    private final struct.FormatDef[] format_def;
+
+    @ExposedGet(name = "__class__")
+    @Override
+    public PyType getType() {
+        return TYPE;
+    }
+
+    public PyStruct(PyObject[] args, String[] keywords) {
+        final int nargs = args.length;
+        if (keywords.length > 0 ) {
+            for (String keyword : keywords) {
+                if (!keyword.equals("format")) {
+                    throw Py.TypeError("'" + keyword + "' is an invalid keyword argument for this function");
+                }
+            }
+        }
+        else if (nargs < 1 || nargs > 1) {
+            throw Py.TypeError("Struct() takes exactly 1 argument (" + nargs + " given)");
+        }
+        format = args[0].toString();
+        format_def = struct.whichtable(format);
+        size = struct.calcsize(format, format_def);
+    }
+    
+    @ExposedNew
+    final static PyObject Struct___new__ (PyNewWrapper new_, boolean init,
+            PyType subtype, PyObject[] args, String[] keywords) {
+         return new PyStruct(args, keywords);
+    }
+
+    @ExposedMethod
+    public String pack(PyObject[] args, String[] kwds) {
+        return struct.pack(format, format_def, size, 0, args).toString();
+    }
+    
+    @ExposedMethod
+    final void pack_into(PyObject[] args, String[] kwds) {
+        struct.pack_into(format, format_def, size, 0, args);
+    }
+  
+    @ExposedMethod
+    public PyTuple unpack(PyObject source) {
+        String s;
+        if (source instanceof PyString)
+            s = source.toString();
+        else if (source instanceof PyArray) 
+            s = ((PyArray)source).tostring();
+        else
+            throw Py.TypeError("unpack of a str or array");
+        if (size != s.length()) 
+            throw struct.StructError("unpack str size does not match format");
+        return struct.unpack(format_def, size, format, new struct.ByteStream(s));
+    }
+    
+    // xxx - also support byte[], java.nio.(Byte)Buffer at some point?
+    @ExposedMethod(defaults = {"0"})
+    public PyTuple unpack_from(PyObject string, int offset) {
+        String s = string.toString();
+        if (size >= (s.length() - offset + 1))
+            throw struct.StructError("unpack_from str size does not match format");
+        return struct.unpack(format_def, size, format, new struct.ByteStream(s, offset));
+    }
+}
\ No newline at end of file


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.