DaoArray* DaoProcess_PutVectorComplex( DaoProcess *self, dao_complex *array, daoint N )
{
DaoArray *res = DaoProcess_PutArray( self );
DaoArray_SetNumType( res, DAO_COMPLEX );
if( array ) DaoArray_SetVectorFloat64( res, (double*)array, N );
return res;
}
DaoArray* DaoProcess_PutVectorFloat64( DaoProcess *self, double *array, daoint N )
{
DaoArray *res = DaoProcess_PutArray( self );
DaoArray_SetNumType( res, DAO_FLOAT );
if( array ) DaoArray_SetVectorFloat64( res, array, N );
return res;
}
DaoArray* DaoProcess_PutVectorInt64( DaoProcess *self, long long *array, daoint N )
{
DaoArray *res = DaoProcess_PutArray( self );
DaoArray_SetNumType( res, DAO_INTEGER );
if( array ) DaoArray_SetVectorInt64( res, array, N );
return res;
}
DaoArray* DaoProcess_PutVectorUInt32( DaoProcess *self, unsigned int *array, daoint N )
{
DaoArray *res = DaoProcess_PutArray( self );
DaoArray_SetNumType( res, DAO_INTEGER );
if( array ) DaoArray_SetVectorUInt32( res, array, N );
return res;
}
DaoArray* DaoProcess_PutVectorSInt16( DaoProcess *self, signed short *array, daoint N )
{
DaoArray *res = DaoProcess_PutArray( self );
DaoArray_SetNumType( res, DAO_INTEGER );
if( array ) DaoArray_SetVectorSInt16( res, array, N );
return res;
}
static void STD_Array( DaoProcess *proc, DaoValue *p[], int N )
{
DaoInteger idint = {DAO_INTEGER,0,0,0,0,0};
DaoValue *res, *index = (DaoValue*)(void*)&idint;
DaoVmCode *sect = DaoGetSectionCode( proc->activeCode );
DaoArray *array = DaoProcess_PutArray( proc );
DaoArray *first = NULL;
DaoArray *sub = NULL;
daoint i, j, k, entry, size = 1;
/* if multi-dimensional array is disabled, DaoProcess_PutArray() will raise exception. */
#ifdef DAO_WITH_NUMARRAY
for(i=0; i<N; i++){
daoint d = p[i]->xInteger.value;
if( d < 0 ){
DaoProcess_RaiseException( proc, DAO_ERROR_PARAM, NULL );
break;
}
size *= d;
}
if( size == 0 ) return;
if( sect == NULL ) return; // TODO exception
if( DaoProcess_PushSectionFrame( proc ) == NULL ) return;
entry = proc->topFrame->entry;
DaoProcess_AcquireCV( proc );
for(i=0; i<size; i++){
idint.value = i;
if( sect->b >0 ) DaoProcess_SetValue( proc, sect->a, index );
proc->topFrame->entry = entry;
DaoProcess_Execute( proc );
if( proc->status == DAO_PROCESS_ABORTED ) break;
res = proc->stackValues[0];
if( i == 0 ){
int D = N;
DaoArray_SetDimCount( array, N + (res->type == DAO_ARRAY ? res->xArray.ndim : 0) );
for(j=0; j<N; j++) array->dims[j] = p[j]->xInteger.value;
if( res->type == DAO_ARRAY ){
first = DaoArray_Copy( (DaoArray*) res );
if( first->ndim == 2 && (first->dims[0] == 1 || first->dims[1] == 1) ){
D += 1;
array->dims[N] = first->dims[ first->dims[0] == 1 ];
}else{
D += first->ndim;
memmove( array->dims + N, first->dims, first->ndim*sizeof(daoint) );
}
}
DaoArray_ResizeArray( array, array->dims, D );
}
if( res->type == DAO_ARRAY ){
sub = (DaoArray*) res;
if( first == NULL || DaoArray_AlignShape( sub, NULL, first->dims, first->ndim ) ==0 ){
DaoProcess_RaiseException( proc, DAO_ERROR, "inconsistent elements or subarrays" );
break;
}
k = i * sub->size;
for(j=0; j<sub->size; j++){
switch( array->etype ){
case DAO_INTEGER : array->data.i[k+j] = DaoArray_GetInteger( sub, j ); break;
case DAO_FLOAT : array->data.f[k+j] = DaoArray_GetFloat( sub, j ); break;
case DAO_DOUBLE : array->data.d[k+j] = DaoArray_GetDouble( sub, j ); break;
case DAO_COMPLEX : array->data.c[k+j] = DaoArray_GetComplex( sub, j ); break;
}
}
}else{
switch( array->etype ){
case DAO_INTEGER : array->data.i[i] = DaoValue_GetInteger( res ); break;
case DAO_FLOAT : array->data.f[i] = DaoValue_GetFloat( res ); break;
case DAO_DOUBLE : array->data.d[i] = DaoValue_GetDouble( res ); break;
case DAO_COMPLEX : array->data.c[i] = DaoValue_GetComplex( res ); break;
}
}
}
DaoProcess_ReleaseCV( proc );
DaoProcess_PopFrame( proc );
if( first ) DaoArray_Delete( first );
#endif
}
static void DaoMT_Functional( DaoProcess *proc, DaoValue *P[], int N, int F )
{
DMutex mutex;
DCondVar condv;
DaoTaskData *tasks;
DaoValue *param = P[0];
DaoValue *result = NULL;
DaoList *list = NULL;
DaoArray *array = NULL;
DaoVmCode *sect = DaoGetSectionCode( proc->activeCode );
int i, entry, threads = P[1]->xInteger.value;
daoint index = -1, status = 0, joined = 0;
DNode *node = NULL;
switch( F ){
case DVM_FUNCT_MAP :
if( param->type == DAO_ARRAY ){
array = DaoProcess_PutArray( proc );
result = (DaoValue*) array;
}else{
list = DaoProcess_PutList( proc );
result = (DaoValue*) list;
}
break;
case DVM_FUNCT_APPLY : DaoProcess_PutValue( proc, param ); break;
case DVM_FUNCT_FIND : DaoProcess_PutValue( proc, dao_none_value ); break;
}
if( threads <= 0 ) threads = 2;
if( sect == NULL || DaoMT_PushSectionFrame( proc ) == 0 ) return;
if( list ){
DArray_Clear( & list->items );
if( param->type == DAO_LIST ) DArray_Resize( & list->items, param->xList.items.size, NULL );
if( param->type == DAO_MAP ) DArray_Resize( & list->items, param->xMap.items->size, NULL );
#ifdef DAO_WITH_NUMARRAY
}else if( array && F == DVM_FUNCT_MAP ){
DaoArray_GetSliceShape( (DaoArray*) param, & array->dims, & array->ndim );
DaoArray_ResizeArray( array, array->dims, array->ndim );
#endif
}
DMutex_Init( & mutex );
DCondVar_Init( & condv );
entry = proc->topFrame->entry;
tasks = (DaoTaskData*) dao_calloc( threads, sizeof(DaoTaskData) );
DaoProcess_PopFrame( proc );
for(i=0; i<threads; i++){
DaoTaskData *task = tasks + i;
task->param = param;
task->result = result;
task->proto = proc;
task->sect = sect;
task->funct = F;
task->entry = entry;
task->first = i;
task->step = threads;
task->index = & index;
task->node = & node;
task->joined = & joined;
task->condv = & condv;
task->mutex = & mutex;
task->clone = DaoVmSpace_AcquireProcess( proc->vmSpace );
task->clone->mutex = & mutex;
if( i ) DaoCallServer_AddTask( DaoMT_RunFunctional, task, 1 );
}
DaoMT_RunFunctional( tasks );
DMutex_Lock( & mutex );
while( joined < threads ) DCondVar_TimedWait( & condv, & mutex, 0.01 );
DMutex_Unlock( & mutex );
for(i=0; i<threads; i++){
DaoTaskData *task = tasks + i;
DaoVmSpace_ReleaseProcess( proc->vmSpace, task->clone );
status |= task->status;
}
if( F == DVM_FUNCT_FIND ){
DaoTuple *tuple = DaoProcess_PutTuple( proc, 0 );
if( param->type == DAO_LIST && index != -1 ){
DaoValue **items = param->xList.items.items.pValue;
GC_ShiftRC( items[index], tuple->items[1] );
tuple->items[1] = items[index];
tuple->items[0]->xInteger.value = index;
}else if( param->type == DAO_MAP && node ){
GC_ShiftRC( node->key.pValue, tuple->items[0] );
GC_ShiftRC( node->value.pValue, tuple->items[1] );
tuple->items[0] = node->key.pValue;
tuple->items[1] = node->value.pValue;
}
}
if( status ) DaoProcess_RaiseException( proc, DAO_ERROR, "code section execution failed!" );
DMutex_Destroy( & mutex );
DCondVar_Destroy( & condv );
dao_free( tasks );
}
