/*******************************************************

* 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

********************************************************/

#include <assign.hpp>

#include <kernel/assign.hpp>

#include <Array.hpp>

#include <Param.hpp>

#include <common/half.hpp>

#include <handle.hpp>

#include <platform.hpp>

#include <types.hpp>

#include <af/defines.h>

#include <af/dim4.hpp>

#include <af/index.h>

#include <af/seq.h>

#include <utility>

#include <vector>

using af::dim4;

using std::vector;

namespace cpu {

template<typename T>

void assign(Array<T>& out, const af_index_t idxrs[], const Array<T>& rhs) {

vector<bool> isSeq(4);

vector<af_seq> seqs(4, af_span);

// create seq vector to retrieve output dimensions, offsets & offsets

for (dim_t x = 0; x < 4; ++x) {

if (idxrs[x].isSeq) { seqs[x] = idxrs[x].idx.seq; }

isSeq[x] = idxrs[x].isSeq;

}

vector<Array<uint>> idxArrs(4, createEmptyArray<uint>(dim4()));

// look through indexs to read af_array indexs

for (dim_t x = 0; x < 4; ++x) {

if (!isSeq[x]) {

idxArrs[x] = castArray<uint>(idxrs[x].idx.arr);

idxArrs[x].eval();

}

}

vector<CParam<uint>> idxParams(idxArrs.begin(), idxArrs.end());

getQueue().enqueue(kernel::assign<T>, out, out.getDataDims(), rhs,

move(isSeq), move(seqs), move(idxParams));

}

#define INSTANTIATE(T) \

template void assign<T>(Array<T> & out, const af_index_t idxrs[], \

const Array<T>& rhs);

INSTANTIATE(cdouble)

INSTANTIATE(double)

INSTANTIATE(cfloat)

INSTANTIATE(float)

INSTANTIATE(uintl)

INSTANTIATE(uint)

INSTANTIATE(intl)

INSTANTIATE(int)

INSTANTIATE(uchar)

INSTANTIATE(char)

INSTANTIATE(ushort)

INSTANTIATE(short)

INSTANTIATE(common::half)

} // namespace cpu