getting_started/integer.cpp

/*******************************************************
 * 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 <arrayfire.h>
#include <stdio.h>
#include <af/util.h>
#include <cstdlib>
 
using namespace af;
 
int main(int argc, char** argv) {
    try {
        int device = argc > 1 ? atoi(argv[1]) : 0;
        af::setDevice(device);
        af::info();
 
        printf(
            "\n=== ArrayFire signed(s32) / unsigned(u32) Integer Example "
            "===\n");
 
        int h_A[] = {1, 2, 4, -1, 2, 0, 4, 2, 3};
        int h_B[] = {2, 3, -5, 6, 0, 10, -12, 0, 1};
        array A   = array(3, 3, h_A);
        array B   = array(3, 3, h_B);
 
        printf("--\nSub-refencing and Sub-assignment\n");
        af_print(A);
        af_print(A.col(0));
        af_print(A.row(0));
        A(0) = 11;
        A(1) = 100;
        af_print(A);
        af_print(B);
        A(1, span) = B(2, span);
        af_print(A);
 
        printf("--Bit-wise operations\n");
        // Returns an array of type s32
        af_print(A & B);
        af_print(A | B);
        af_print(A ^ B);
 
        printf("\n--Logical operations\n");
        // Returns an array of type b8
        af_print(A && B);
        af_print(A || B);
 
        printf("\n--Transpose\n");
        af_print(A);
        af_print(A.T());
 
        printf("\n--Flip Vertically / Horizontally\n");
        af_print(A);
        af_print(flip(A, 0));
        af_print(flip(A, 1));
 
        printf("\n--Sum along columns\n");
        af_print(A);
        af_print(sum(A));
 
        printf("\n--Product along columns\n");
        af_print(A);
        af_print(product(A));
 
        printf("\n--Minimum along columns\n");
        af_print(A);
        af_print(min(A));
 
        printf("\n--Maximum along columns\n");
        af_print(A);
        af_print(max(A));
 
        printf("\n--Minimum along columns with index\n");
        af_print(A);
 
        array out, idx;
        min(out, idx, A);
        af_print(out);
        af_print(idx);
 
    } catch (af::exception& e) {
        fprintf(stderr, "%s\n", e.what());
        throw;
    }
 
    return 0;
}