tests/lldb/java/Allocations/src/rs/allocs.rs

#pragma version(1)
#pragma rs java_package_name(com.android.rs.allocations)

struct simpleStruct {
   int i;
   unsigned int j;
};

struct complexStruct {
    struct simpleStruct s;
    uchar4 c;
    float f[2];
};

struct complexStruct *g_complexStruct_in;
struct complexStruct *g_complexStruct_out;

// Kernel performs basic vector swizzle
uchar4 __attribute__((kernel)) swizzle_kernel(uchar4 in)
{
    return in.wzyx;
}

// Kernel squares every element in allocation
uint __attribute__((kernel)) square_kernel(ushort in)
{
    uint result = (uint)(in) * (uint)in;
    return result;
}

// Helper function adding 1/2 to passed in double
static double half_helper(double in)
{
    return (in + 0.5);
}

// Kernel returns first 3 elements of a double4 plus 1/2
double3 __attribute__((kernel)) add_half_kernel(double4 in)
{
    double3 result;
    result.x = half_helper(in.x);
    result.y = half_helper(in.y);
    result.z = half_helper(in.z);
    return result;
}

// Kernel for testing structs
struct complexStruct __attribute__((kernel))
struct_kernel(struct complexStruct in, uint32_t x)
{
   struct complexStruct complex_out;
   struct simpleStruct simple_out;
   simple_out.i = (int) x;
   simple_out.j = x;
   complex_out.s = simple_out;

   complex_out.f[0] = (float) x;
   complex_out.f[1] = (float) x + 0.5;

   complex_out.c.x = (uchar) (x % 128);
   complex_out.c.y = 'A';
   complex_out.c.z = 'B';
   complex_out.c.w = 'C';

   return complex_out;
}