source: trunk/anuga_work/anuga_cuda/src/anuga_HMPP/update.c @ 9329

// from quantity_ext.c
#include "hmpp_fun.h"



#ifdef USING_LOCAL_DIRECTIVES
#pragma hmpp update codelet, target=CUDA args[*].transfer=atcall
#endif
void update(
        int N,
        double timestep,
        double centroid_values[N],
        double explicit_update[N],
        double semi_implicit_update[N])
{
    int k;
    double denominator, x;


    // Divide semi_implicit update by conserved quantity
    #pragma hmppcg gridify(k), &
    #pragma hmppcg & private(denominator, x), &
    #pragma hmppcg & global(timestep, centroid_values, explicit_update, &
    #pragma hmppcg & semi_implicit_update)
    for (k=0; k<N; k++) {
        x = centroid_values[k];
        if (x == 0.0) {
            semi_implicit_update[k] = 0.0;
        } else {
            semi_implicit_update[k] /= x;
        }
    //}


    // Explicit updates
    //for (k=0; k<N; k++) {
        centroid_values[k] += timestep*explicit_update[k];
    //}



    // Semi implicit updates
    //for (k=0; k<N; k++) {
        denominator = 1.0 - timestep*semi_implicit_update[k];
        if (denominator <= 0.0) {
        // FIXME: can't pass, since inter-iterations dependency
//            return;
        } else {
            //Update conserved_quantities from semi implicit updates
            centroid_values[k] /= denominator;
        }
    }

    // Reset semi_implicit_update here ready for next time step
    //memset(semi_implicit_update, 0, N*sizeof(double));
}