source: src/tests/polymorphism.c @ 8a78dd3

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// polymorphism.c --
//
// Author           : Rob Schluntz
// Created On       : Tue Oct 17 12:19:48 2017
// Last Modified By : Rob Schluntz
// Last Modified On : Tue Oct 17 12:21:07 2017
// Update Count     : 1
//

#include <assert.h>
#include <inttypes.h>

forall(otype T)
T f(T x, T y) {
        x = y;
        return x;
}

forall(otype T) T ident(T x) {
        return x;
}

forall( otype T, otype U )
size_t struct_size( T i, U j ) {
        struct S { T i; U j; };
        return sizeof(S);
}

forall( otype T, otype U )
size_t union_size( T i, U j ) {
        union B { T i; U j; };
        return sizeof(B);
}

// perform some simple operations on aggregates of T and U
forall( otype T | { void print(T); int ?==?(T, T); }, otype U | { void print(U); U ?=?(U&, zero_t); } )
U foo(T i, U j) {
        struct S { T i; U j; };
        union B { T i; U j; };

        S s;
        s.i = i;
        assertf(s.i == i, "struct operation fails in polymorphic context.");

        B b;
        b.j = 0;
        b.i = s.i;
        return b.j;
}

int main() {
        {
                // ensure that x is not changed by the invocation of a polymorphic function
                int x = 123;
                int y = 456;
                int z = f(x, y);
                printf("%d %d %d\n", x, y, z);
        }

        {
                // explicitly specialize function
                int (*f)(int) = ident;
                ((int(*)(int))ident);
                printf("%d %d\n", f(5), ((int(*)(int))ident)(5));
        }

        {
                // test aggregates with polymorphic members
                typedef __attribute__((aligned(8))) uint32_t x_type;
                typedef __attribute__((aligned(8))) uint64_t y_type;

                x_type x = 3;
                y_type y = 3;

                struct S {
                        x_type f1;
                        y_type f2;
                };
                union U {
                        x_type f1;
                        y_type f2;
                };
                // ensure that the size of aggregates with polymorphic members
                // matches the size of the aggregates in a monomorphic context
                assertf( struct_size(x, y) == sizeof(S), "struct size differs in polymorphic context." );
                assertf( union_size(x, y) == sizeof(U), "union size differs in polymorphic context." );

                y_type ?=?(y_type & this, zero_t) {
                        this = (int)0;
                        return this;
                }

                void print(x_type x) {
                        printf("%"PRIu32"\n", x);
                }

                void print(y_type y) {
                        printf("%"PRIu64"\n", y);
                }

                y_type ret = foo(x, y);

                // duplicate logic from inside of foo to ensure the same results
                U u;
                u.f2 = 0;
                u.f1 = x;
                assertf(ret == u.f2, "union operation fails in polymorphic context.");
        }
}

// Local Variables: //
// tab-width: 4 //
// End: //