index.h

Go to the documentation of this file.

/*******************************************************
 * Copyright (c) 2014, ArrayFire
 * All rights reserved.
 *
 * This file is distributed under 3-clause BSD license.
 * The complete license agreement can be obtained at:
 * http://arrayfire.com/licenses/BSD-3-Clause
 ********************************************************/
 
#pragma once
#include <af/defines.h>
#include <af/seq.h>
 
typedef struct af_index_t {
    union {
        af_array arr;   
        af_seq   seq;   
    } idx;
 
    bool     isSeq;     
    bool     isBatch;   
} af_index_t;
 
 
#if __cplusplus
namespace af
{
 
class dim4;
class array;
class seq;
 
class AFAPI index {
 
    af_index_t impl;
    public:
    index();
    ~index();
 
    index(const int idx);
 
    index(const af::seq& s0);
 
    index(const af_seq& s0);
 
    index(const af::array& idx0);
 
#if AF_API_VERSION >= 31
    index(const index& idx0);
#endif
 
    bool isspan() const;
 
    const af_index_t& get() const;
 
#if AF_API_VERSION >= 31
    index & operator=(const index& idx0);
 
#if AF_COMPILER_CXX_RVALUE_REFERENCES
    index(index &&idx0);
    index& operator=(index &&idx0);
#endif
#endif // AF_API_VERSION
};
 
AFAPI array lookup(const array &in, const array &idx, const int dim = -1);
 
#if AF_API_VERSION >= 31
AFAPI void copy(array &dst, const array &src,
                const index &idx0,
                const index &idx1 = span,
                const index &idx2 = span,
                const index &idx3 = span);
#endif
 
}
#endif
 
#ifdef __cplusplus
extern "C" {
#endif
 
    AFAPI af_err af_index(  af_array *out,
                            const af_array in,
                            const unsigned ndims, const af_seq* const index);
 
 
    AFAPI af_err af_lookup( af_array *out,
                            const af_array in, const af_array indices,
                            const unsigned dim);
 
    AFAPI af_err af_assign_seq( af_array *out,
                                const af_array lhs,
                                const unsigned ndims, const af_seq* const indices,
                                const af_array rhs);
 
    AFAPI af_err af_index_gen(  af_array *out,
                                const af_array in,
                                const dim_t ndims, const af_index_t* indices);
 
    AFAPI af_err af_assign_gen( af_array *out,
                                const af_array lhs,
                                const dim_t ndims, const af_index_t* indices,
                                const af_array rhs);
 
#if AF_API_VERSION >= 32
    AFAPI af_err af_create_indexers(af_index_t** indexers);
#endif
 
#if AF_API_VERSION >= 32
    AFAPI af_err af_set_array_indexer(af_index_t* indexer, const af_array idx, const dim_t dim);
#endif
 
#if AF_API_VERSION >= 32
    AFAPI af_err af_set_seq_indexer(af_index_t* indexer, const af_seq* idx,
                                  const dim_t dim, const bool is_batch);
#endif
 
#if AF_API_VERSION >= 32
    AFAPI af_err af_set_seq_param_indexer(af_index_t* indexer,
                                        const double begin, const double end, const double step,
                                        const dim_t dim, const bool is_batch);
#endif
 
#if AF_API_VERSION >= 32
    //  \returns \ref af_err error code
    AFAPI af_err af_release_indexers(af_index_t* indexers);
#endif
 
#ifdef __cplusplus
}
#endif