source: libcfa/src/concurrency/cofor.cfa @ 9c2ac95

#include <cofor.hfa>

//////////////////////////////////////////////////////////////////////////////////////////
// cofor ( uC++ COFOR )

thread cofor_task {
        ssize_t low, high;
        __cofor_body_t loop_body;
};

static void ?{}( cofor_task & this, ssize_t low, ssize_t high, __cofor_body_t loop_body ) {
        this.low = low;
        this.high = high;
        this.loop_body = loop_body;
}

void main( cofor_task & this ) with( this ) {
        for ( ssize_t i = low; i < high; i++ )
                loop_body(i);
}

void __Cofor__( ssize_t low, ssize_t high, __cofor_body_t loop_body ) libcfa_public {
        ssize_t range = high - low;
  if ( range <= 0 ) return;
        ssize_t nprocs = get_proc_count( *active_cluster() );
  if ( nprocs == 0 ) return;
        ssize_t threads = range < nprocs ? range : nprocs;
        ssize_t stride = range / threads + 1, extras = range % threads;
        ssize_t i = 0;
        ssize_t stride_iter = low;
        cofor_task * runners[ threads ];
        for ( i; threads ) {
                runners[i] = alloc();
        }
        for ( i = 0; i < extras; i += 1, stride_iter += stride ) {
                (*runners[i]){ stride_iter, stride_iter + stride, loop_body };
        }
        stride -= 1;
        for ( ; i < threads; i += 1, stride_iter += stride ) {
                (*runners[i]){ stride_iter, stride_iter + stride, loop_body };
        }
        for ( i; threads ) {
                delete( runners[i] );
        }
}