PowerToys/Pythonnet.Runtime/arrayobject.cs

// ==========================================================================
// This software is subject to the provisions of the Zope Public License,
// Version 2.0 (ZPL).  A copy of the ZPL should accompany this distribution.
// THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
// WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
// WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
// FOR A PARTICULAR PURPOSE.
// ==========================================================================

using System;
using System.Collections;
using System.Reflection;

namespace Python.Runtime {

    /// <summary>
    /// Implements a Python type for managed arrays. This type is essentially
    /// the same as a ClassObject, except that it provides sequence semantics 
    /// to support natural array usage (indexing) from Python.
    /// </summary>

    internal class ArrayObject : ClassBase {

        internal ArrayObject(Type tp) : base(tp) {}

        internal override bool CanSubclass() {
            return false;
        }

        public static IntPtr tp_new(IntPtr tp, IntPtr args, IntPtr kw) {
            ArrayObject self = GetManagedObject(tp) as ArrayObject;
            if (Runtime.PyTuple_Size(args) != 1) {
                return Exceptions.RaiseTypeError("array expects 1 argument");
            }
            IntPtr op = Runtime.PyTuple_GetItem(args, 0);
            Object result;

            if (!Converter.ToManaged(op, self.type, out result, true)) {
                return IntPtr.Zero;
            }
            return CLRObject.GetInstHandle(result, tp);
        }


        //====================================================================
        // Implements __getitem__ for array types.
        //====================================================================

        public static IntPtr mp_subscript(IntPtr ob, IntPtr idx) {
            CLRObject obj = (CLRObject)ManagedType.GetManagedObject(ob);
            Array items = obj.inst as Array;
            Type itemType = obj.inst.GetType().GetElementType();
            int rank = items.Rank;
            int index = 0;
            object value;

            // Note that CLR 1.0 only supports int indexes - methods to
            // support long indices were introduced in 1.1. We could
            // support long indices automatically, but given that long
            // indices are not backward compatible and a relative edge 
            // case, we won't bother for now.

            // Single-dimensional arrays are the most common case and are
            // cheaper to deal with than multi-dimensional, so check first.

            if (rank == 1) {
                index = (int)Runtime.PyInt_AsLong(idx);

                if (Exceptions.ErrorOccurred()) {
                    return Exceptions.RaiseTypeError("invalid index value");
                }

                if (index < 0) {
                    index = items.Length + index;
                }

                try {
                    value = items.GetValue(index);
                }
                catch (IndexOutOfRangeException) {
                    Exceptions.SetError(Exceptions.IndexError,
                                        "array index out of range"
                                        );
                    return IntPtr.Zero; 
                }

                return Converter.ToPython(items.GetValue(index), itemType);
            }

            // Multi-dimensional arrays can be indexed a la: list[1, 2, 3].

            if (!Runtime.PyTuple_Check(idx)) {
                Exceptions.SetError(Exceptions.TypeError,
                                    "invalid index value"
                                    );
                return IntPtr.Zero;
            }

            int count = Runtime.PyTuple_Size(idx);

            Array args = Array.CreateInstance(typeof(Int32), count);

            for (int i = 0; i < count; i++) {
                IntPtr op = Runtime.PyTuple_GetItem(idx, i);
                index = (int)Runtime.PyInt_AsLong(op);

                if (Exceptions.ErrorOccurred()) {
                    return Exceptions.RaiseTypeError("invalid index value");
                }

                if (index < 0) {
                    index = items.GetLength(i) + index;
                }

                args.SetValue(index, i);
            }

            try {
                value = items.GetValue((int[]) args);
            }
            catch (IndexOutOfRangeException) {
                Exceptions.SetError(Exceptions.IndexError,
                                    "array index out of range"
                                    );
                return IntPtr.Zero; 
            }

            return Converter.ToPython(value, itemType);
        }


        //====================================================================
        // Implements __setitem__ for array types.
        //====================================================================

        public static int mp_ass_subscript(IntPtr ob, IntPtr idx, IntPtr v) {
            CLRObject obj = (CLRObject)ManagedType.GetManagedObject(ob);
            Array items = obj.inst as Array;
            Type itemType = obj.inst.GetType().GetElementType();
            int rank = items.Rank;
            int index = 0;
            object value;

            if (items.IsReadOnly) {
                Exceptions.RaiseTypeError("array is read-only");
                return -1;
            }

            if (!Converter.ToManaged(v, itemType, out value, true)) {
                return -1;
            }

            if (rank == 1) {
                index = (int)Runtime.PyInt_AsLong(idx);

                if (Exceptions.ErrorOccurred()) {
                    Exceptions.RaiseTypeError("invalid index value");
                    return -1;
                }

                if (index < 0) {
                    index = items.Length + index;
                }

                try {
                    items.SetValue(value, index);
                }
                catch (IndexOutOfRangeException) {
                    Exceptions.SetError(Exceptions.IndexError,
                                        "array index out of range"
                                        );
                    return -1; 
                }

                return 0;
            }

            if (!Runtime.PyTuple_Check(idx)) {
                Exceptions.RaiseTypeError("invalid index value");
                return -1;
            }

            int count = Runtime.PyTuple_Size(idx);

            Array args = Array.CreateInstance(typeof(Int32), count);

            for (int i = 0; i < count; i++) {
                IntPtr op = Runtime.PyTuple_GetItem(idx, i);
                index = (int)Runtime.PyInt_AsLong(op);

                if (Exceptions.ErrorOccurred()) {
                    Exceptions.RaiseTypeError("invalid index value");
                    return -1;
                }

                if (index < 0) {
                    index = items.GetLength(i) + index;
                }

                args.SetValue(index, i);
            }

            try {
                items.SetValue(value, (int[])args);
            }
            catch (IndexOutOfRangeException) {
                Exceptions.SetError(Exceptions.IndexError,
                                    "array index out of range"
                                    );
                return -1;
            }

            return 0;
        }


        //====================================================================
        // Implements __contains__ for array types.
        //====================================================================

        public static int sq_contains(IntPtr ob, IntPtr v) {
            CLRObject obj = (CLRObject)ManagedType.GetManagedObject(ob);
            Type itemType = obj.inst.GetType().GetElementType();
            IList items = obj.inst as IList;
            object value;

            if (!Converter.ToManaged(v, itemType, out value, false)) {
                return 0;
            }

            if (items.Contains(value)) {
                return 1;
            }

            return 0;
        }


        //====================================================================
        // Implements __len__ for array types.
        //====================================================================

        public static int mp_length(IntPtr ob) {
            CLRObject self = (CLRObject)ManagedType.GetManagedObject(ob);
            Array items = self.inst as Array;
            return items.Length;
        }


    }        

}