1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
|
/*******************************************************
* 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 <gtest/gtest.h>
#include <arrayfire.h>
#include <af/dim4.hpp>
#include <af/traits.hpp>
#include <vector>
#include <iostream>
#include <string>
#include <testHelpers.hpp>
using std::vector;
using std::string;
using std::cout;
using std::endl;
template<typename T>
class TransformCoordinates : public ::testing::Test
{
public:
virtual void SetUp() {}
};
typedef ::testing::Types<float, double> TestTypes;
TYPED_TEST_CASE(TransformCoordinates, TestTypes);
template<typename T>
void transformCoordinatesTest(string pTestFile)
{
if (noDoubleTests<T>()) return;
vector<af::dim4> inDims;
vector<vector<T> > in;
vector<vector<float> > gold;
readTests<T, float, float>(pTestFile, inDims, in, gold);
af_array tfArray = 0;
af_array outArray = 0;
ASSERT_EQ(AF_SUCCESS, af_create_array(&tfArray, &(in[0].front()), inDims[0].ndims(), inDims[0].get(), (af_dtype)af::dtype_traits<T>::af_type));
size_t nTests = in.size();
for (int test = 1; test < nTests; test++) {
dim_t d0 = (dim_t)in[test][0];
dim_t d1 = (dim_t)in[test][1];
ASSERT_EQ(AF_SUCCESS, af_transform_coordinates(&outArray, tfArray, d0, d1));
// Get result
dim_t outEl = 0;
ASSERT_EQ(AF_SUCCESS, af_get_elements(&outEl, outArray));
T* outData = new T[outEl];
ASSERT_EQ(AF_SUCCESS, af_get_data_ptr((void*)outData, outArray));
const float thr = 1.f;
for (size_t elIter = 0; elIter < outEl; elIter++) {
ASSERT_LE(fabs(outData[elIter] - gold[test-1][elIter]), thr) << "at: " << elIter << std::endl;
}
delete[] outData;
}
if(tfArray != 0) af_release_array(tfArray);
if(outArray != 0) af_release_array(outArray);
}
TYPED_TEST(TransformCoordinates, RotateMatrix)
{
transformCoordinatesTest<TypeParam>(string(TEST_DIR"/transformCoordinates/rotate_matrix.test"));
}
TYPED_TEST(TransformCoordinates, 3DMatrix)
{
transformCoordinatesTest<TypeParam>(string(TEST_DIR"/transformCoordinates/3d_matrix.test"));
}
///////////////////////////////////// CPP ////////////////////////////////
//
TEST(TransformCoordinates, CPP)
{
vector<af::dim4> inDims;
vector<vector<float> > in;
vector<vector<float> > gold;
readTests<float, float, float>(TEST_DIR"/transformCoordinates/3d_matrix.test",inDims,in,gold);
af::array tf = af::array(inDims[0][0], inDims[0][1], &(in[0].front()));
float d0 = in[1][0];
float d1 = in[1][1];
af::array out = af::transformCoordinates(tf, d0, d1);
af::dim4 outDims = out.dims();
float* h_out = new float[outDims[0] * outDims[1]];
out.host(h_out);
const size_t n = gold[0].size();
const float thr = 1.f;
for (size_t elIter = 0; elIter < n; elIter++) {
ASSERT_LE(fabs(h_out[elIter] - gold[0][elIter]), thr) << "at: " << elIter << std::endl;
}
delete[] h_out;
}
|