peter-nordlund-3.cpp

来自「A C++ class library for scientific compu」· C++ 代码 · 共 47 行

#include "testsuite.h"
#include <blitz/array.h>

BZ_USING_NAMESPACE(blitz)

int main()
{
    Array<int,2> A(3,3);
    A = 3, 5, 8,
        2, 4, 9,
        1, 0, 1;

    // workaround for MIPSpro compiler bug
#if defined(__sgi)
    Array<int,2> Asqrd(3,3);
    Asqrd = sqr(A);
#endif

    TinyVector<int,2> minpos = minIndex(A);

    BZTEST(minpos(0) == 2);
    BZTEST(minpos(1) == 1);

    // workaround for MIPSpro compiler bug
#if defined(__sgi)
    minpos = minIndex(Asqrd);
#else
    minpos = minIndex(sqr(A));
#endif
    BZTEST(minpos(0) == 2);
    BZTEST(minpos(1) == 1);

    TinyVector<int,2> maxpos = maxIndex(A);
    BZTEST(maxpos(0) == 1);
    BZTEST(maxpos(1) == 2);

    // workaround for MIPSpro compiler bug
#if defined(__sgi)
    maxpos = maxIndex(Asqrd);
#else
    maxpos = maxIndex(sqr(A));
#endif
    BZTEST(maxpos(0) == 1);
    BZTEST(maxpos(1) == 2);
}