/*

 * This file implements assignment from a scalar to an ndarray.

 *

 * Written by Mark Wiebe (mwwiebe@gmail.com)

 * Copyright (c) 2011 by Enthought, Inc.

 *

 * See LICENSE.txt for the license.

 */

#define PY_SSIZE_T_CLEAN

#include <Python.h>

#define NPY_NO_DEPRECATED_API NPY_API_VERSION

#define _MULTIARRAYMODULE

#include <numpy/ndarraytypes.h>

#include "npy_config.h"

#include "npy_pycompat.h"

#include "convert_datatype.h"

#include "methods.h"

#include "shape.h"

#include "lowlevel_strided_loops.h"

#include "array_assign.h"

/*

 * Assigns the scalar value to every element of the destination raw array.

 *

 * Returns 0 on success, -1 on failure.

 */

NPY_NO_EXPORT int

        PyArray_Descr *dst_dtype, char *dst_data, npy_intp const *dst_strides,

        PyArray_Descr *src_dtype, char *src_data)

{

    int idim;

    npy_intp shape_it[NPY_MAXDIMS], dst_strides_it[NPY_MAXDIMS];

    npy_intp coord[NPY_MAXDIMS];

    /* Check both uint and true alignment */

    /* Use raw iteration with no heap allocation */

                    ndim, shape,

                    dst_data, dst_strides,

                    &ndim, shape_it,

                    &dst_data, dst_strides_it) < 0) {

        return -1;

    }

    /* Get the function to do the casting */

                        0, dst_strides_it[0],

                        src_dtype, dst_dtype,

                        0,

                        &stransfer, &transferdata,

                        &needs_api) != NPY_SUCCEED) {

        return -1;

    }

        npy_intp nitems = 1, i;

        }

    }

        /* Process the innermost dimension */

                dst_data, dst_strides_it[0], src_data, 0,

                shape_it[0], src_itemsize, transferdata) < 0) {

            goto fail;

        }

                            shape_it, dst_data, dst_strides_it);

    return 0;

    return -1;

}

/*

 * Assigns the scalar value to every element of the destination raw array

 * where the 'wheremask' value is True.

 *

 * Returns 0 on success, -1 on failure.

 */

NPY_NO_EXPORT int

        PyArray_Descr *dst_dtype, char *dst_data, npy_intp const *dst_strides,

        PyArray_Descr *src_dtype, char *src_data,

        PyArray_Descr *wheremask_dtype, char *wheremask_data,

        npy_intp const *wheremask_strides)

{

    int idim;

    npy_intp shape_it[NPY_MAXDIMS], dst_strides_it[NPY_MAXDIMS];

    npy_intp wheremask_strides_it[NPY_MAXDIMS];

    npy_intp coord[NPY_MAXDIMS];

    /* Check both uint and true alignment */

    /* Use raw iteration with no heap allocation */

                    ndim, shape,

                    dst_data, dst_strides,

                    wheremask_data, wheremask_strides,

                    &ndim, shape_it,

                    &dst_data, dst_strides_it,

                    &wheremask_data, wheremask_strides_it) < 0) {

        return -1;

    }

    /* Get the function to do the casting */

                        0, dst_strides_it[0], wheremask_strides_it[0],

                        src_dtype, dst_dtype, wheremask_dtype,

                        0,

                        &stransfer, &transferdata,

                        &needs_api) != NPY_SUCCEED) {

        return -1;

    }

        npy_intp nitems = 1, i;

        }

    }

        /* Process the innermost dimension */

                    (npy_bool *)wheremask_data, wheremask_strides_it[0],

                    shape_it[0], src_itemsize, transferdata);

                            dst_data, dst_strides_it,

                            wheremask_data, wheremask_strides_it);

}

/*

 * Assigns a scalar value specified by 'src_dtype' and 'src_data'

 * to elements of 'dst'.

 *

 * dst: The destination array.

 * src_dtype: The data type of the source scalar.

 * src_data: The memory element of the source scalar.

 * wheremask: If non-NULL, a boolean mask specifying where to copy.

 * casting: An exception is raised if the assignment violates this

 *          casting rule.

 *

 * This function is implemented in array_assign_scalar.c.

 *

 * Returns 0 on success, -1 on failure.

 */

NPY_NO_EXPORT int

                        PyArray_Descr *src_dtype, char *src_data,

                        PyArrayObject *wheremask,

                        NPY_CASTING casting)

{

    npy_longlong scalarbuffer[4];

        return -1;

    }

    /* Check the casting rule */

                            PyArray_DESCR(dst), casting)) {

                src_dtype, PyArray_DESCR(dst), casting, NPY_TRUE);

    }

    /*

     * Make a copy of the src data if it's a different dtype than 'dst'

     * or isn't aligned, and the destination we're copying to has

     * more than one element. To avoid having to manage object lifetimes,

     * we also skip this if 'dst' has an object dtype.

     */

        char *tmp_src_data;

        /*

         * Use a static buffer to store the aligned/cast version,

         * or allocate some memory if more space is needed.

         */

            tmp_src_data = (char *)&scalarbuffer[0];

        }

        else {

                goto fail;

            }

            allocated_src_data = 1;

        }

        }

                            src_dtype, PyArray_DESCR(dst), 0) != NPY_SUCCEED) {

            src_data = tmp_src_data;

            goto fail;

        }

        /* Replace src_data/src_dtype */

    }

        /* A straightforward value assignment */

        /* Do the assignment with raw array iteration */

                src_dtype, src_data) < 0) {

            goto fail;

        }

    }

    else {

        npy_intp wheremask_strides[NPY_MAXDIMS];

        /* Broadcast the wheremask to 'dst' for raw iteration */

                    wheremask_strides) < 0) {

            goto fail;

        }

        /* Do the masked assignment with raw array iteration */

                src_dtype, src_data,

                wheremask_strides) < 0) {

            goto fail;

        }

    }

    }

    return 0;

    }

    return -1;

}