resizablepmcarray.pmc

/*
Copyright (C) 2001-2012, Parrot Foundation.

=head1 NAME

src/pmc/resizablepmcarray.pmc - resizable array for PMCs only

=head1 DESCRIPTION

This class, ResizablePMCArray, implements an resizable array which stores PMCs.
It puts things into Integer, Float, or String PMCs as appropriate.

=head2 Vtable Functions

=over 4

=cut

*/

#define PMC_size(x)      ((Parrot_ResizablePMCArray_attributes *)PMC_data(x))->size
#define PMC_array(x)     ((Parrot_ResizablePMCArray_attributes *)PMC_data(x))->pmc_array
#define PMC_threshold(x) ((Parrot_ResizablePMCArray_attributes *)PMC_data(x))->resize_threshold

/* HEADERIZER HFILE: none */
/* HEADERIZER BEGIN: static */
/* Don't modify between HEADERIZER BEGIN / HEADERIZER END.  Your changes will be lost. */

PARROT_INLINE
static void do_shift(ARGIN(PMC *arr))
        __attribute__nonnull__(1);

PARROT_INLINE
static void do_unshift(PARROT_INTERP, ARGIN(PMC *arr), ARGIN(PMC *val))
        __attribute__nonnull__(1)
        __attribute__nonnull__(2)
        __attribute__nonnull__(3);

PARROT_DOES_NOT_RETURN
static void throw_pop_empty(PARROT_INTERP)
        __attribute__nonnull__(1);

PARROT_DOES_NOT_RETURN
static void throw_shift_empty(PARROT_INTERP)
        __attribute__nonnull__(1);

#define ASSERT_ARGS_do_shift __attribute__unused__ int _ASSERT_ARGS_CHECK = (\
       PARROT_ASSERT_ARG(arr))
#define ASSERT_ARGS_do_unshift __attribute__unused__ int _ASSERT_ARGS_CHECK = (\
       PARROT_ASSERT_ARG(interp) \
    , PARROT_ASSERT_ARG(arr) \
    , PARROT_ASSERT_ARG(val))
#define ASSERT_ARGS_throw_pop_empty __attribute__unused__ int _ASSERT_ARGS_CHECK = (\
       PARROT_ASSERT_ARG(interp))
#define ASSERT_ARGS_throw_shift_empty __attribute__unused__ int _ASSERT_ARGS_CHECK = (\
       PARROT_ASSERT_ARG(interp))
/* Don't modify between HEADERIZER BEGIN / HEADERIZER END.  Your changes will be lost. */
/* HEADERIZER END: static */

PARROT_INLINE
static void
do_shift(ARGIN(PMC *arr))
{
    ASSERT_ARGS(do_shift)
    INTVAL    size = PMC_size(arr);
    PMC ** const item = PMC_array(arr);

    PMC_size(arr) = --size;
    memmove(item, item + 1, size * sizeof (PMC *));
    item[size] = PMCNULL;
}

PARROT_INLINE
static void
do_unshift(PARROT_INTERP, ARGIN(PMC *arr), ARGIN(PMC *val))
{
    ASSERT_ARGS(do_unshift)
    const INTVAL size = PMC_size(arr);
    PMC        **item;

    VTABLE_set_integer_native(interp, arr, size + 1);
    item = PMC_array(arr);
    memmove(item + 1, item, size * sizeof (PMC *));
    item[0] = val;
}


pmclass ResizablePMCArray extends FixedPMCArray auto_attrs provides array {
    ATTR INTVAL resize_threshold; /* max size before array needs resizing */

/*

=item C<void init_int(INTVAL size)>

Initializes the array.

=cut

*/

    VTABLE void init_int(INTVAL size) {
        if (size < 0)
            Parrot_ex_throw_from_c_args(INTERP, NULL, EXCEPTION_OUT_OF_BOUNDS,
                    _("ResizeblePMCArray: Cannot set array size to a negative number (%d)"), size);
        SUPER(size);
        if (size > 0)
            PMC_threshold(SELF) = size;
    }

/*

=item C<void set_integer_native(INTVAL size)>

Resizes the array to C<size> elements.

=cut

*/

    VTABLE void set_integer_native(INTVAL size) {
        if (size < 0)
            Parrot_ex_throw_from_c_args(INTERP, NULL, EXCEPTION_OUT_OF_BOUNDS,
                    "ResizablePMCArray: Can't resize!");

        if (!PMC_array(SELF)) {
            /* empty - used fixed routine */
            if (size < 8) {

                SUPER(8);
                PMC_size(SELF)      = size;
                PMC_threshold(SELF) = 8;

            }
            else {
                SUPER(size);
                PMC_threshold(SELF) = size;
            }
        }
        else if (size == PMC_size(SELF)) {
            return;
        }
        else if (size <= PMC_threshold(SELF)) {

            PMC_size(SELF) = size;
            /* we could shrink here if necessary */
            return;
        }
        else {
            INTVAL i, cur;
            i = cur = PMC_threshold(SELF);
            if (cur < 8192)
                cur = (size < 2 * cur) ? (2 * cur) : size;
            else {
                const INTVAL needed = size - cur;
                cur   += needed + 4096;
                cur   &= ~0xfff;
            }

            if (cur < 8)
                cur = 8;

            PMC_array(SELF) = mem_gc_realloc_n_typed_zeroed(INTERP,
                    PMC_array(SELF), cur, i, PMC *);

            for (; i < cur; ++i) {
                (PMC_array(SELF))[i] = PMCNULL;
            }

            PMC_threshold(SELF) = cur;
            PMC_size(SELF)      = size;
        }
    }

/*

=item C<FLOATVAL shift_float()>

=item C<INTVAL shift_integer()>

=item C<PMC *shift_pmc()>

=item C<STRING *shift_string()>

Removes and returns an item from the start of the array.

=cut

TODO: This always moves the array memory, which is not very performant
TODO: Check whether there is already an element that can be shifted

*/

    VTABLE FLOATVAL shift_float() {
        const INTVAL size = PMC_size(SELF);
        FLOATVAL value;

        if (0 == size)
            throw_shift_empty(INTERP);

        value = VTABLE_get_number(INTERP, PMC_array(SELF)[0]);
        do_shift(SELF);
        return value;
    }

    VTABLE INTVAL shift_integer() {
        const INTVAL size = PMC_size(SELF);
        INTVAL value;

        if (0 == size)
            throw_shift_empty(INTERP);

        value = VTABLE_get_integer(INTERP, PMC_array(SELF)[0]);
        do_shift(SELF);
        return value;
    }

    VTABLE PMC *shift_pmc() {
        const INTVAL size = PMC_size(SELF);
        PMC   *data;

        if (0 == size)
            throw_shift_empty(INTERP);

        data = PMC_array(SELF)[0];
        do_shift(SELF);
        return data;
    }

    VTABLE STRING *shift_string() {
        const INTVAL size = PMC_size(SELF);
        STRING  *value;

        if (0 == size)
            throw_shift_empty(INTERP);

        value = VTABLE_get_string(INTERP, PMC_array(SELF)[0]);
        do_shift(SELF);
        return value;
    }


/*

=item C<PMC *get_pmc_keyed_int(INTVAL key)>

Returns the PMC value of the element at index C<key>.

=cut

*/

    VTABLE PMC *get_pmc_keyed_int(INTVAL key) {
        PMC **data;

        if (key < 0)
            key += PMC_size(SELF);

        if (key < 0)
            Parrot_ex_throw_from_c_args(INTERP, NULL, EXCEPTION_OUT_OF_BOUNDS,
                "ResizablePMCArray: index out of bounds!");

        if (key >= PMC_size(SELF))
            return PMCNULL;

        data = PMC_array(SELF);

        if (PMC_IS_NULL(data[key]))
            return PMCNULL;

        return data[key];
    }

/*

=item C<void set_pmc_keyed_int(INTVAL key, PMC *src)>

Sets the PMC value of the element at index C<key> to C<*src>.

=item C<void set_pmc_keyed(PMC *key, PMC *src)>

Sets the PMC value of the element keyed by C<key> to C<*src>.

=cut

*/

    VTABLE void set_pmc_keyed_int(INTVAL key, PMC *src) {
        PMC **data;

        if (key < 0)
            key += PMC_size(SELF);

        if (key < 0)
            Parrot_ex_throw_from_c_args(INTERP, NULL, EXCEPTION_OUT_OF_BOUNDS,
                "ResizablePMCArray: index out of bounds!");

        if (key >= PMC_size(SELF))
            SELF.set_integer_native(key+1);

        data      = PMC_array(SELF);

        data[key] = src;
    }

    VTABLE void set_pmc_keyed(PMC *key, PMC *src) {
        SUPER(key, src);
    }

    VTABLE void set_pmc(PMC *value) {
        INTVAL size;
        INTVAL i;

        if (SELF == value)
            return;

        if (!VTABLE_does(INTERP, value, CONST_STRING(INTERP, "array")))
            Parrot_ex_throw_from_c_args(INTERP, NULL, EXCEPTION_OUT_OF_BOUNDS,
                    _("Can't set self from this type"));

        if (PMC_array(SELF))
            mem_gc_free(INTERP, PMC_array(SELF));

        size            = PMC_size(SELF) = VTABLE_elements(INTERP, value);
        PMC_array(SELF) = mem_gc_allocate_n_typed(INTERP, size, PMC *);

        for (i = 0; i < size; ++i)
            (PMC_array(SELF))[i] = VTABLE_get_pmc_keyed_int(INTERP, value, i);

        PObj_custom_mark_destroy_SETALL(SELF);
    }

/*

=item C<void delete_keyed_int(INTVAL key)>

=item C<void delete_keyed(PMC *key)>

Delete the element at index C<key>.

=cut

*/

    VTABLE void delete_keyed(PMC *key) {
        const INTVAL idx  = VTABLE_get_integer(INTERP, key);
        SELF.delete_keyed_int(idx);
    }

    VTABLE void delete_keyed_int(INTVAL key) {
        UNUSED(INTERP)
        PMC ** const data = PMC_array(SELF);
        const INTVAL  n   = PMC_size(SELF);
        INTVAL  i;

        for (i = key; i < n - 1; ++i)
            data[i] = data[i + 1];

        --PMC_size(SELF);
    }

/*

=item C<INTVAL exists_keyed_int(INTVAL key)>

=item C<INTVAL exists_keyed(PMC *key)>

Returns TRUE is the element at C<key> exists; otherwise returns false.

=cut

*/
    VTABLE INTVAL exists_keyed_int(INTVAL key) {
        UNUSED(INTERP)
        PMC **data;

        if (key < 0)
            key += PMC_size(SELF);

        if (key < 0 || key >= PMC_size(SELF))
            return 0;

        data = PMC_array(SELF);
        return !PMC_IS_NULL(data[key]);
    }

    VTABLE INTVAL exists_keyed(PMC *key) {
        const INTVAL ix = VTABLE_get_integer(INTERP, key);
        return SELF.exists_keyed_int(ix);
    }

/*

=item C<INTVAL defined_keyed_int(INTVAL key)>

Returns TRUE is the element at C<key> is defined; otherwise returns false.

=cut

*/

    VTABLE INTVAL defined_keyed_int(INTVAL key) {
        PMC *val;

        if (key < 0)
            key += PMC_size(SELF);

        if (key < 0 || key >= PMC_size(SELF))
            return 0;

        val = SELF.get_pmc_keyed_int(key);

        if (PMC_IS_NULL(val))
            return 0;

        return VTABLE_defined(INTERP, val);
    }

/*

=item C<void push_float(FLOATVAL value)>

=item C<void push_integer(INTVAL value)>

=item C<void push_pmc(PMC *value)>

=item C<void push_string(STRING *value)>

Extends the array by adding an element of value C<*value> to the end of
the array.

=cut

*/

    VTABLE void push_float(FLOATVAL value) {

        const INTVAL size = PMC_size(SELF);
        PMC   * const val = Parrot_pmc_new(INTERP, enum_class_Float);

        VTABLE_set_number_native(INTERP, val, value);
        SELF.set_pmc_keyed_int(size, val);

        return;
    }

    VTABLE void push_integer(INTVAL value) {

        const INTVAL size = PMC_size(SELF);
        PMC   * const val = Parrot_pmc_new_init_int(INTERP, enum_class_Integer, value);
        SELF.set_pmc_keyed_int(size, val);

        return;
    }

    VTABLE void push_pmc(PMC *value) {
        const INTVAL size   = PMC_size(SELF);
        const INTVAL thresh = PMC_threshold(SELF);

        if (PMC_array(SELF) && size < thresh)
            PMC_size(SELF) = size + 1;
        else {
            SELF.set_integer_native(size + 1);
        }
        ((PMC **)PMC_array(SELF))[size] = value;

        return;
    }

    VTABLE void push_string(STRING *value) {

        const INTVAL size = PMC_size(SELF);
        PMC   * const val = Parrot_pmc_new(INTERP, enum_class_String);

        VTABLE_assign_string_native(INTERP, val, value);
        SELF.set_pmc_keyed_int(size, val);

        return;
    }

/*

Removes and returns the last element in the array.

=item C<INTVAL pop_float()>

=item C<INTVAL pop_integer()>

=item C<PMC *pop_pmc()>

=item C<STRING *pop_string()>

=cut

*/

    VTABLE FLOATVAL pop_float() {

        INTVAL   size = PMC_size(SELF);
        PMC     *data;

        if (0 == size)
            throw_pop_empty(INTERP);

        data           = PMC_array(SELF)[--size];
        PMC_size(SELF) = size;

        return VTABLE_get_number(INTERP, data);
    }

    VTABLE INTVAL pop_integer() {

        INTVAL  size = PMC_size(SELF);
        PMC    *data;

        if (0 == size)
            throw_pop_empty(INTERP);

        data           = PMC_array(SELF)[--size];
        PMC_size(SELF) = size;

        return VTABLE_get_integer(INTERP, data);
    }

    VTABLE PMC *pop_pmc() {

        INTVAL size = PMC_size(SELF);
        PMC   *data;

        if (0 == size)
            throw_pop_empty(INTERP);

        data           = PMC_array(SELF)[--size];
        PMC_size(SELF) = size;

        return data;
    }

    VTABLE STRING *pop_string() {

        INTVAL  size = PMC_size(SELF);
        PMC    *data;

        if (0 == size)
            throw_pop_empty(INTERP);

        data           = PMC_array(SELF)[--size];
        PMC_size(SELF) = size;

        return VTABLE_get_string(INTERP, data);
    }

/*

=item C<void unshift_float(FLOATVAL value)>

=item C<void unshift_integer(INTVAL value)>

=item C<void unshift_pmc(PMC *value)>

=item C<void unshift_string(STRING *value)>

Extends the array by adding an element of value C<*value> to the begin of
the array.

=cut

*/

    VTABLE void unshift_float(FLOATVAL value) {

        PMC * const val = Parrot_pmc_new(INTERP, enum_class_Float);
        VTABLE_set_number_native(INTERP, val, value);
        do_unshift(INTERP, SELF, val);
    }

    VTABLE void unshift_integer(INTVAL value) {

        PMC * const val = Parrot_pmc_new_init_int(INTERP, enum_class_Integer, value);
        do_unshift(INTERP, SELF, val);
    }

    VTABLE void unshift_pmc(PMC *value) {

        do_unshift(INTERP, SELF, value);
    }

    VTABLE void unshift_string(STRING *value) {

        PMC * const val = Parrot_pmc_new(INTERP, enum_class_String);
        VTABLE_set_string_native(INTERP, val, value);
        do_unshift(INTERP, SELF, val);
    }

/*

=item C<PMC *clone()>

Creates and returns a copy of the array.

=cut

*/

    VTABLE PMC *clone() {
        PMC * const copy = SUPER();

        /* copy trimmed extra space */
        PMC_threshold(copy) = PMC_size(SELF);

        return copy;
    }

/*

=item C<STRING *get_repr()>

Returns the Parrot string representation C<ResizablePMCArray>.

=cut

*/

    VTABLE STRING *get_repr() {

        INTVAL  j;
        const INTVAL n = VTABLE_elements(INTERP, SELF);
        STRING *res = CONST_STRING(INTERP, "[ ");

        for (j = 0; j < n; ++j) {
            PMC * const val = SELF.get_pmc_keyed_int(j);
            res      = Parrot_str_concat(INTERP, res, VTABLE_get_repr(INTERP, val));

            if (j < n - 1)
                res = Parrot_str_concat(INTERP, res, CONST_STRING(INTERP, ", "));
        }

        return Parrot_str_concat(INTERP, res, CONST_STRING(INTERP, " ]"));
    }

/*

=item C<void splice(PMC *from, INTVAL offset, INTVAL count)>

Replaces C<count> elements starting at C<offset> with the elements in
C<from>.

Note that the C<from> PMC can be of any of the various array types.

=cut

*/

    VTABLE void splice(PMC *from, INTVAL offset, INTVAL count) {
        const INTVAL elems0 = VTABLE_elements(INTERP, SELF);
        const INTVAL elems1 = VTABLE_elements(INTERP, from);
        PMC **item = 0;
        INTVAL tail;
        INTVAL i;

        /* start from end? */
        if (offset < 0)
            offset += elems0;

        if (offset < 0)
            Parrot_ex_throw_from_c_args(INTERP, NULL, EXCEPTION_OUT_OF_BOUNDS,
                "illegal splice offset\n");

        /* number of elements to the right of the splice (the "tail") */
        tail = elems0 - offset - count;
        if (tail < 0) tail = 0;

        item = PMC_array(SELF);
        if (tail > 0 && count > elems1) {
            /* we're shrinking the array, so first move the tail left */
            memmove(item + offset + elems1, item + offset + count,
                 tail * sizeof (PMC *));
        }

        /* now resize the array */
        SELF.set_integer_native(offset + elems1 + tail);

        item = PMC_array(SELF);
        if (tail > 0 && count < elems1) {
            /* the array grew, so move the tail to the right */
            memmove(item + offset + elems1, item + offset + count,
                 tail * sizeof (PMC *));
        }

        /* now copy C<from>'s elements into SELF */
        if (elems1 > 0) {
            PMC *iter = VTABLE_get_iter(INTERP, from);
            for (i = 0; i < elems1; i++)
                item[offset + i] = VTABLE_shift_pmc(INTERP, iter);
        }
    }

/*

=back

=head2 Methods

=over 4

=item METHOD append(PMC *other)

Append the other array to this array.

=cut

*/

    METHOD append(PMC *other) {

        const INTVAL n = VTABLE_elements(INTERP, SELF);
        const INTVAL m = VTABLE_elements(INTERP, other);

        if (!m)
            RETURN(void);

        /* pre-size it */
        VTABLE_set_integer_native(INTERP, SELF, n + m);

        if (other->vtable->base_type == SELF->vtable->base_type
        ||  other->vtable->base_type == enum_class_FixedPMCArray) {
            PMC ** const other_data = PMC_array(other);
            PMC ** const this_data  = PMC_array(SELF);

            /* libc is faster at copying data than a manual loop here */
            memmove(this_data + n, other_data, m  * sizeof (PMC *));
        }
        else {
            PMC ** const this_data = PMC_array(SELF);
            INTVAL i;

            for (i = 0; i < m; ++i)
                this_data[n + i] = VTABLE_get_pmc_keyed_int(INTERP, other, i);
        }
    }

/*

=item METHOD PMC* shift()

=item METHOD PMC* pop()

Method forms to remove and return a PMC from the beginning or
end of the array.

=cut

*/

    METHOD shift() {
        PMC * const value = VTABLE_shift_pmc(INTERP, SELF);
        RETURN(PMC *value);
    }

    METHOD pop() {
        PMC * const value = VTABLE_pop_pmc(INTERP, SELF);
        RETURN(PMC *value);
    }

/*

=item METHOD unshift(PMC* value)

=item METHOD push(PMC* value)

Method forms to add a PMC to the beginning or end of the array.

=cut

*/

    METHOD unshift(PMC* value) {
        VTABLE_unshift_pmc(INTERP, SELF, value);
    }

    METHOD push(PMC* value) {
        VTABLE_push_pmc(INTERP, SELF, value);
    }

}

/*

=back

=head2 Auxiliar functions

=over 4

=item C<static void do_shift(PMC *arr)>

Common part for shift operations.

=item C<static void do_unshift(PARROT_INTERP, PMC *arr, PMC *val)>

Common part for unshift operations.

=item C<static void throw_shift_empty(PARROT_INTERP)>

=item C<static void throw_pop_empty(PARROT_INTERP)>

Throws with the appropriate message.

=cut

*/

PARROT_DOES_NOT_RETURN
static void
throw_shift_empty(PARROT_INTERP)
{
    ASSERT_ARGS(throw_shift_empty)
    Parrot_ex_throw_from_c_args(interp, NULL, EXCEPTION_OUT_OF_BOUNDS,
            "ResizablePMCArray: Can't shift from an empty array!");
}

PARROT_DOES_NOT_RETURN
static void
throw_pop_empty(PARROT_INTERP)
{
    ASSERT_ARGS(throw_pop_empty)
    Parrot_ex_throw_from_c_args(interp, NULL, EXCEPTION_OUT_OF_BOUNDS,
            "ResizablePMCArray: Can't pop from an empty array!");
}

/*

=back

=head1 See also

F<docs/pdds/pdd17_basic_types.pod>.

=cut

*/

/*
 * Local variables:
 *   c-file-style: "parrot"
 * End:
 * vim: expandtab shiftwidth=4 cinoptions='\:2=2' :
 */