void VariableSuite_Setup( VariableSuiteData* data ) {
Particle tmpParticle;
Name pNames[] = { "mass", "force", "info" };
SizeT pOffsets[] = { 0, 0, 0 }; /* Init later... */
Variable_DataType pDataTypes[] = {
Variable_DataType_Int,
Variable_DataType_Float,
Variable_DataType_Pointer,
};
Index pDtCounts[] = { 1, 3, 1 };
static SizeT pSize = sizeof(Particle);
data->aSize[0] = 16;
data->aSize[1] = 16;
data->aSize[2] = 16;
pOffsets[0] = (ArithPointer)&tmpParticle.mass - (ArithPointer)&tmpParticle;
pOffsets[1] = (ArithPointer)&tmpParticle.force - (ArithPointer)&tmpParticle;
pOffsets[2] = (ArithPointer)&tmpParticle.info - (ArithPointer)&tmpParticle;
/* Construction phase --------------------------------------------------------------------------------------------*/
data->vr = Variable_Register_New();
Variable_NewScalar( "temperature", NULL, Variable_DataType_Double, &data->aSize[0], NULL, (void**)&data->temperature, data->vr );
Variable_NewVector( "velocity", NULL, Variable_DataType_Double, 3, &data->aSize[1], NULL, (void**)&data->velocity, data->vr, "vx", "vy", "vz" );
Variable_New( "particle", NULL, 3, pOffsets, pDataTypes, pDtCounts, pNames, &pSize, &data->aSize[2], NULL, (void**)&data->particle, data->vr );
/* Build phase ---------------------------------------------------------------------------------------------------*/
data->temperature = Memory_Alloc_Array( double, data->aSize[0], "temperature" );
data->velocity = Memory_Alloc_Array( VectorD, data->aSize[1], "velocity" );
data->particle = Memory_Alloc_Array( Particle, data->aSize[2], "array" );
Variable_Register_BuildAll( data->vr );
}
void ParticleCoordsSuite_TestLineParticle( ParticleCoordsSuiteData* data ) {
ExtensionManager_Register* extensionMgr_Register;
Variable_Register* variable_Register;
Swarm* swarm;
Stream* stream;
Dictionary* dictionary;
DomainContext* context;
int procToWatch = data->nProcs > 1 ? 1 : 0;
char input_file[PCU_PATH_MAX];
char expected_file[PCU_PATH_MAX];
Stg_ComponentFactory* cf;
if( data->rank == procToWatch ) {
Journal_Enable_AllTypedStream( False );
/* Registers */
extensionMgr_Register = ExtensionManager_Register_New();
variable_Register = Variable_Register_New();
/* read in the xml input file */
pcu_filename_input( "testLineParticleLayout.xml", input_file );
cf = stgMainInitFromXML( input_file, data->comm, NULL );
stgMainBuildAndInitialise( cf );
context = (DomainContext*)LiveComponentRegister_Get( cf->LCRegister, (Name)"context" );
dictionary = context->dictionary;
swarm = (Swarm* ) LiveComponentRegister_Get( context->CF->LCRegister, (Name)"swarm" );
pcu_check_true( swarm );
Journal_Enable_AllTypedStream( True );
stream = Journal_Register( Info_Type, (Name)"LinearParticleStream" );
Stream_RedirectFile( stream, "linearParticle.dat" );
Swarm_PrintParticleCoords( swarm, stream );
Journal_Enable_AllTypedStream( False );
pcu_filename_expected( "testLineParticleLayoutOutput.expected", expected_file );
pcu_check_fileEq( "linearParticle.dat", expected_file );
/* Destroy stuff */
Stg_Class_Delete( extensionMgr_Register );
Stg_Class_Delete( variable_Register );
remove( "linearParticle.dat" );
}
stgMainDestroy( cf );
}
void Variable_RegisterSuite_Setup( Variable_RegisterSuiteData* data ) {
data->reg = Variable_Register_New();
}
void DofLayoutSuite_TestBasic( DofLayoutSuiteData* data ) {
char expected_file[PCU_PATH_MAX];
int procToWatch;
Stream* stream = Journal_Register( Info_Type, (Name)"DofLayoutBasic" );
procToWatch = data->nProcs >=2 ? 1 : 0;
if( data->rank == procToWatch ) {
DofLayout* dof;
DofLayout* destDof;
Variable_Register* variableRegister;
StgVariable* var[6];
char* varName[] = {"x", "y", "z", "vx", "vy", "vz"};
Index ii, dof_I, var_I;
Index arraySize = 27;
double* varArrays[6];
int counts[27];
Stream_RedirectFile( stream, "testBasic.dat" );
/* Create variable register */
variableRegister = Variable_Register_New();
/* Create variables */
for (var_I = 0; var_I < 6; var_I++) {
varArrays[var_I] = Memory_Alloc_Array_Unnamed( double, arraySize );
var[var_I] = StgVariable_NewScalar( varName[var_I], NULL, StgVariable_DataType_Double, (Index*)&arraySize, NULL, (void**)&(varArrays[var_I]), variableRegister );
Stg_Component_Build( var[var_I], 0, False );
Stg_Component_Initialise( var[var_I], 0, False );
}
for (ii = 0; ii < arraySize; ii++) {
for (var_I = 0; var_I < 6; var_I++) {
StgVariable_SetValueDouble( var[var_I], ii, 0.0 );
}
}
/* Simple test */
dof = DofLayout_New( "dofLayout", variableRegister, arraySize, NULL );
for (ii = 0; ii < arraySize; ii++)
for (var_I = 0; var_I < 6; var_I++)
DofLayout_AddDof_ByVarName(dof, varName[var_I], ii);
Stg_Component_Build(dof, 0, False);
Journal_Printf( stream, "Simple test:\n" );
for (ii = 0; ii < arraySize; ii++) {
Journal_Printf( stream, "\t%u\n", dof->dofCounts[ii] );
pcu_check_true( dof->dofCounts[ii] == 6 );
}
Stg_Class_Delete(dof);
/* Advanced test */
for (ii = 0; ii < 27; ii++) counts[ii] = 0;
dof = DofLayout_New( "dofLayout1", variableRegister, arraySize, NULL );
for (ii = 0; ii < 12; ii++) {
for (var_I = 0; var_I < 2; var_I++) {
DofLayout_AddDof_ByVarName(dof, varName[var_I], ii);
counts[ii]++;
}
}
for (ii = 9; ii < 20; ii++) {
for (var_I = 2; var_I < 6; var_I++) {
DofLayout_AddDof_ByVarName(dof, varName[var_I], ii);
counts[ii]++;
}
}
Stg_Component_Build(dof, 0, False);
Journal_Printf( stream, "\nAdvanced test:\n" );
for (ii = 0; ii < arraySize; ii++) {
Journal_Printf( stream, "\t%u\n", dof->dofCounts[ii] );
pcu_check_true( counts[ii] == dof->dofCounts[ii] );
}
Stg_Class_Delete(dof);
/* Copy test */
dof = DofLayout_New( "dofLayout2", variableRegister, arraySize, NULL );
destDof = DofLayout_New( "dofLayout3", variableRegister, arraySize, NULL );
for (ii = 0; ii < arraySize; ii++) {
for (var_I = 0; var_I < 3; var_I++) {
DofLayout_AddDof_ByVarName(dof, varName[var_I], ii);
}
for (var_I = 3; var_I < 6; var_I++) {
DofLayout_AddDof_ByVarName(destDof, varName[var_I], ii);
}
}
Stg_Component_Build(dof, NULL, False);
Stg_Component_Build(destDof, NULL, False);
for (ii = 0; ii < arraySize; ii++) {
for (dof_I = 0; dof_I < 3; dof_I++) {
DofLayout_SetValueDouble( dof, ii, dof_I, ii*10 );
DofLayout_SetValueDouble( destDof, ii, dof_I, 0 );
}
}
Journal_Printf( stream, "Copy Test: pre copy:\n" );
for (ii = 0; ii < arraySize; ii++) {
Journal_Printf( stream, "\tIndex %d - src %2g,%2g,%2g - dest %2g, %2g, %2g\n", ii,
DofLayout_GetValueDouble( dof, ii, 0 ),
DofLayout_GetValueDouble( dof, ii, 1 ),
DofLayout_GetValueDouble( dof, ii, 2 ),
DofLayout_GetValueDouble( destDof, ii, 0 ),
DofLayout_GetValueDouble( destDof, ii, 1 ),
DofLayout_GetValueDouble( destDof, ii, 2 ) );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 0 ) == ii * 10 );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 1 ) == ii * 10 );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 2 ) == ii * 10 );
pcu_check_true( DofLayout_GetValueDouble( destDof, ii, 0 ) == 0 );
pcu_check_true( DofLayout_GetValueDouble( destDof, ii, 1 ) == 0 );
pcu_check_true( DofLayout_GetValueDouble( destDof, ii, 2 ) == 0 );
}
DofLayout_CopyValues( dof, destDof );
Journal_Printf( stream, "Copy Test: post copy:\n" );
for (ii = 0; ii < arraySize; ii++) {
Journal_Printf( stream, "\tIndex %d - src %2g,%2g,%2g - dest %2g, %2g, %2g\n", ii,
DofLayout_GetValueDouble( dof, ii, 0 ),
DofLayout_GetValueDouble( dof, ii, 1 ),
DofLayout_GetValueDouble( dof, ii, 2 ),
DofLayout_GetValueDouble( destDof, ii, 0 ),
DofLayout_GetValueDouble( destDof, ii, 1 ),
DofLayout_GetValueDouble( destDof, ii, 2 ) );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 0 ) == DofLayout_GetValueDouble( destDof, ii, 0 ) );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 1 ) == DofLayout_GetValueDouble( destDof, ii, 1 ) );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 2 ) == DofLayout_GetValueDouble( destDof, ii, 2 ) );
}
Stg_Class_Delete(destDof);
Journal_Printf( stream, "Zero Test: all values in src dof should be zero again\n" );
DofLayout_SetAllToZero( dof );
for (ii = 0; ii < arraySize; ii++) {
Journal_Printf( stream, "\tIndex %d - src %2g,%2g,%2g\n", ii,
DofLayout_GetValueDouble( dof, ii, 0 ),
DofLayout_GetValueDouble( dof, ii, 1 ),
DofLayout_GetValueDouble( dof, ii, 2 ) );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 0 ) == 0 );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 1 ) == 0 );
pcu_check_true( DofLayout_GetValueDouble( dof, ii, 2 ) == 0 );
}
Stg_Class_Delete(dof);
/* Cleanup */
Stg_Class_Delete(variableRegister);
for (var_I = 0; var_I < 6; var_I++) {
if (var[var_I]) Stg_Class_Delete(var[var_I]);
Memory_Free( varArrays[var_I] );
}
pcu_filename_expected( "testDofLayoutBasicOutput.expected", expected_file );
pcu_check_fileEq( "testBasic.dat", expected_file );
remove( "testBasic.dat" );
}
void WallVCSuite_TestWallVC( WallVCSuiteData* data ) {
unsigned nDomains;
unsigned nDims = 3;
unsigned meshSize[3] = {3, 3, 3};
int procToWatch;
double minCrds[3] = {0.0, 0.0, 0.0};
double maxCrds[3] = {1.0, 1.0, 1.0};
char* vcKey[] = {"WallVC_Front", "WallVC_Back", "WallVC_Left", "WallVC_Right",
"WallVC_Top", "WallVC_Bottom"};
char* vcKeyName[] = {"WallVC_FrontName", "WallVC_BackName", "WallVC_LeftName", "WallVC_RightName",
"WallVC_TopName", "WallVC_BottomName"};
char* varName[] = {"x", "y", "z", "vx", "vy", "vz", "temp"};
char input_file[PCU_PATH_MAX];
char expected_file[PCU_PATH_MAX];
Mesh* mesh;
Variable_Register* variable_Register;
ConditionFunction* quadCF;
ConditionFunction* expCF;
ConditionFunction_Register* conFunc_Register;
ExtensionManager_Register* extensionMgr_Register;
Dictionary* dictionary;
Dictionary* sources;
Stream* stream;
XML_IO_Handler* io_handler;
Variable* var[7];
double* array[7];
VariableCondition* vc;
Index i;
procToWatch = data->nProcs >=2 ? 1 : 0;
io_handler = XML_IO_Handler_New();
stream = Journal_Register( Info_Type, (Name)"WallVCStream" );
Stream_RedirectFile( stream, "testWallVC.dat" );
dictionary = Dictionary_New();
sources = Dictionary_New();
Dictionary_Add( dictionary, (Dictionary_Entry_Key)"outputPath", Dictionary_Entry_Value_FromString("./output") );
/* Input file */
pcu_filename_input( "wallVC.xml", input_file );
IO_Handler_ReadAllFromFile( io_handler, input_file, dictionary, sources );
fflush( stdout );
extensionMgr_Register = ExtensionManager_Register_New();
/* Create a mesh. */
mesh = (Mesh*) WallVCSuite_buildMesh( nDims, meshSize, minCrds, maxCrds, extensionMgr_Register );
nDomains = Mesh_GetDomainSize( mesh, MT_VERTEX );
/* Create CF stuff */
quadCF = ConditionFunction_New( WallVCSuite_quadratic, (Name)"quadratic", NULL );
expCF = ConditionFunction_New( WallVCSuite_exponential, (Name)"exponential", NULL);
conFunc_Register = ConditionFunction_Register_New( );
ConditionFunction_Register_Add(conFunc_Register, quadCF);
ConditionFunction_Register_Add(conFunc_Register, expCF);
/* Create variable register */
variable_Register = Variable_Register_New();
/* Create variables */
for (i = 0; i < 6; i++) {
array[i] = Memory_Alloc_Array( double, nDomains, "array[i]" );
var[i] = Variable_NewScalar( varName[i], NULL, Variable_DataType_Double, (Index*)&nDomains, NULL, (void**)&array[i], 0 );
Variable_Register_Add(variable_Register, var[i]);
}
array[6] = Memory_Alloc_Array( double, nDomains * 5, "array[6]" );
var[6] = Variable_NewVector( varName[6], NULL, Variable_DataType_Double, 5, &nDomains, NULL, (void**)&array[6], 0 );
Variable_Register_Add(variable_Register, var[6]);
Variable_Register_BuildAll(variable_Register);
for (i = 0; i < 6; i++) {
Index j, k;
vc = (VariableCondition*) WallVC_New( vcKeyName[i], NULL, vcKey[i], variable_Register, conFunc_Register, dictionary, mesh );
Stg_Component_Build( vc, 0, False );
for (j = 0; j < 6; j++)
memset(array[j], 0, sizeof(double)* nDomains );
memset(array[6], 0, sizeof(double)* nDomains * 5);
VariableCondition_Apply(vc, NULL);
if (data->rank == procToWatch) {
Journal_Printf( stream,"Testing for %s\n", vcKey[i]);
for (j = 0; j < 6; j++) {
Journal_Printf( stream,"\nvar[%u]: %.2lf", j, array[j][0]);
for (k = 1; k < nDomains; k++)
Journal_Printf( stream,", %.2lf", array[j][k]);
}
Journal_Printf( stream,"\nvar[6]: %.2lf", array[6][0]);
for (j = 1; j < nDomains*5; j++)
Journal_Printf( stream,", %.2lf", array[6][j]);
Journal_Printf( stream,"\n\n");
for (j = 0; j < 7; j++) {
for (k = 0; k < nDomains; k++)
Journal_Printf( stream,"%s ", VariableCondition_IsCondition(vc, k, j) ? "True " : "False");
Journal_Printf( stream,"\n");
} Journal_Printf( stream,"\n");
for (j = 0; j < 7; j++) {
for (k = 0; k < nDomains; k++) {
VariableCondition_ValueIndex valIndex;
valIndex = VariableCondition_GetValueIndex(vc, k, j);
if (valIndex != (unsigned)-1)
Journal_Printf( stream,"%03u ", valIndex);
else
Journal_Printf( stream,"XXX ");
} Journal_Printf( stream,"\n");
} Journal_Printf( stream,"\n");
}
Stg_Class_Delete(vc);
}
if (data->rank == procToWatch) {
pcu_filename_expected( "testWallVC.expected", expected_file );
pcu_check_fileEq( "testWallVC.dat", expected_file );
remove( "testWallVC.dat" );
}
Stg_Class_Delete(variable_Register);
for (i = 0; i < 7; i++) {
Stg_Class_Delete(var[i]);
if (array[i]) Memory_Free(array[i]);
}
Stg_Class_Delete(extensionMgr_Register);
Stg_Class_Delete(io_handler);
Stg_Class_Delete(conFunc_Register);
Stg_Class_Delete(quadCF);
Stg_Class_Delete(expCF);
Stg_Class_Delete(dictionary);
Stg_Class_Delete(sources);
FreeObject( mesh );
}
int main( int argc, char *argv[] ) {
MPI_Comm CommWorld;
int rank;
int procCount;
int procToWatch;
/* Initialise MPI, get world info */
MPI_Init( &argc, &argv );
MPI_Comm_dup( MPI_COMM_WORLD, &CommWorld );
MPI_Comm_size( MPI_COMM_WORLD, &procCount );
MPI_Comm_rank( MPI_COMM_WORLD, &rank );
Base_Init( &argc, &argv );
DiscretisationGeometry_Init( &argc, &argv );
DiscretisationShape_Init( &argc, &argv );
DiscretisationMesh_Init( &argc, &argv );
DiscretisationUtils_Init( &argc, &argv );
MPI_Barrier( CommWorld ); /* Ensures copyright info always come first in output */
if( argc >= 2 ) {
procToWatch = atoi( argv[1] );
}
else {
procToWatch = 0;
}
if( rank == procToWatch ) {
DofLayout* dof;
Variable_Register* variableRegister;
Variable* var[6];
char* varName[] = {"x", "y", "z", "vx", "vy", "vz"};
Index ii, var_I;
Index arraySize = 27;
double* varArrays[6];
/* Create variable register */
variableRegister = Variable_Register_New();
/* Create variables */
for (var_I = 0; var_I < 6; var_I++) {
varArrays[var_I] = Memory_Alloc_Array_Unnamed( double, arraySize );
var[var_I] = Variable_NewScalar( varName[var_I], Variable_DataType_Double, &arraySize,
(void**)&(varArrays[var_I]), variableRegister );
Build( var[var_I], 0, False );
Initialise( var[var_I], 0, False );
}
for (ii = 0; ii < arraySize; ii++) {
for (var_I = 0; var_I < 6; var_I++) {
Variable_SetValueDouble( var[var_I], ii, (ii*10 + var_I) );
}
}
/* Simple test */
dof = DofLayout_New( "dofLayout", variableRegister, arraySize );
for (ii = 0; ii < arraySize; ii++) {
for (var_I = 0; var_I < 6; var_I++) {
DofLayout_AddDof_ByVarName(dof, varName[var_I], ii);
}
}
Build(dof, 0, False);
DofLayout_SaveAllVariablesToFiles( dof, "output/testDofSave", rank );
DofLayout_SetAllToZero( dof );
DofLayout_LoadAllVariablesFromFiles( dof, "output/testDofSave", rank );
for (ii = 0; ii < arraySize; ii++) {
printf("\tIndex %d - %2g,%2g,%2g,%2g,%2g,%2g\n", ii,
DofLayout_GetValueDouble( dof, ii, 0 ),
DofLayout_GetValueDouble( dof, ii, 1 ),
DofLayout_GetValueDouble( dof, ii, 2 ),
DofLayout_GetValueDouble( dof, ii, 3 ),
DofLayout_GetValueDouble( dof, ii, 4 ),
DofLayout_GetValueDouble( dof, ii, 5 ) );
}
/* Cleanup */
Stg_Class_Delete(dof);
Stg_Class_Delete(variableRegister);
for (var_I = 0; var_I < 6; var_I++) {
if (var[var_I]) Stg_Class_Delete(var[var_I]);
Memory_Free( varArrays[var_I] );
}
}
void LiveComponentRegisterSuite_TestGet( LiveComponentRegisterSuiteData* data ) {
typedef float Triple[3];
float* array;
Triple* structArray;
StgVariable* var;
StgVariable* vec;
StgVariable* vecVar[3];
StgVariable* tempVar = NULL;
Index length = 10;
Variable_Register* reg;
array = Memory_Alloc_Array( float, length, "test" );
structArray = Memory_Alloc_Array( Triple, length, "test" );
reg = Variable_Register_New();
var = StgVariable_NewScalar(
"Scalar",
NULL,
StgVariable_DataType_Float,
&length,
NULL,
(void**)&array,
reg );
vec = StgVariable_NewVector(
"Three",
NULL,
StgVariable_DataType_Float,
3,
&length,
NULL,
(void**)&structArray,
reg,
"a",
"b",
"c" );
vecVar[0] = Variable_Register_GetByName( reg, "a" );
vecVar[1] = Variable_Register_GetByName( reg, "b" );
vecVar[2] = Variable_Register_GetByName( reg, "c" );
Variable_Register_BuildAll( reg );
LiveComponentRegister_Add( data->lcRegister, (Stg_Component*) var );
pcu_check_true( LiveComponentRegister_IfRegThenAdd( (Stg_Component*) vec ) );
LiveComponentRegister_Add( data->lcRegister, (Stg_Component*) vecVar[0] );
LiveComponentRegister_Add( data->lcRegister, (Stg_Component*) vecVar[1] );
LiveComponentRegister_Add( data->lcRegister, (Stg_Component*) vecVar[2] );
tempVar = (StgVariable*) LiveComponentRegister_Get( data->lcRegister, (Name)"Scalar" );
pcu_check_true( tempVar == var );
tempVar = (StgVariable* ) LiveComponentRegister_Get( LiveComponentRegister_GetLiveComponentRegister(), (Name)"Three" );
pcu_check_true( tempVar == vec );
tempVar = (StgVariable* ) LiveComponentRegister_Get( data->lcRegister, (Name)"a" );
pcu_check_true( tempVar == vecVar[0] );
tempVar = (StgVariable* ) LiveComponentRegister_Get( data->lcRegister, (Name)"b" );
pcu_check_true( tempVar == vecVar[1] );
tempVar = (StgVariable* ) LiveComponentRegister_Get( data->lcRegister, (Name)"c" );
pcu_check_true( tempVar == vecVar[2] );
}
int main(int argc, char *argv[])
{
MPI_Comm CommWorld;
int rank;
int procCount;
int procToWatch;
Stream* stream;
Dictionary* dictionary;
XML_IO_Handler* io_handler;
Topology* nTopology;
ElementLayout* eLayout;
NodeLayout* nLayout;
MeshDecomp* decomp;
MeshLayout* layout;
Mesh* mesh;
Variable* var[7];
Variable_Register* variable_Register;
WallVC* vc;
ConditionFunction* quadCF;
ConditionFunction* expCF;
ConditionFunction_Register* conFunc_Register;
ExtensionManager_Register* extensionMgr_Register;
double* array[7];
char* vcKey[] = {"WallVC_Front", "WallVC_Back", "WallVC_Left", "WallVC_Right",
"WallVC_Top", "WallVC_Bottom"};
char* vcKeyName[] = {"WallVC_FrontName", "WallVC_BackName", "WallVC_LeftName", "WallVC_RightName",
"WallVC_TopName", "WallVC_BottomName"};
char* varName[] = {"x", "y", "z", "vx", "vy", "vz", "temp"};
Index i;
/* Initialise MPI, get world info */
MPI_Init(&argc, &argv);
MPI_Comm_dup( MPI_COMM_WORLD, &CommWorld );
MPI_Comm_size(CommWorld, &procCount);
MPI_Comm_rank(CommWorld, &rank);
Base_Init( &argc, &argv );
DiscretisationGeometry_Init( &argc, &argv );
DiscretisationShape_Init( &argc, &argv );
DiscretisationMesh_Init( &argc, &argv );
DiscretisationUtils_Init( &argc, &argv );
MPI_Barrier( CommWorld ); /* Ensures copyright info always come first in output */
io_handler = XML_IO_Handler_New();
stream = Journal_Register (Info_Type, "myStream");
procToWatch = argc >= 2 ? atoi(argv[1]) : 0;
dictionary = Dictionary_New();
IO_Handler_ReadAllFromFile(io_handler, "data/wallVC.xml", dictionary);
fflush(stdout);
MPI_Barrier(MPI_COMM_WORLD);
Dictionary_Add(dictionary, "rank", Dictionary_Entry_Value_FromUnsignedInt(rank));
Dictionary_Add(dictionary, "numProcessors", Dictionary_Entry_Value_FromUnsignedInt(procCount));
Dictionary_Add(dictionary, "meshSizeI", Dictionary_Entry_Value_FromUnsignedInt(4));
Dictionary_Add(dictionary, "meshSizeJ", Dictionary_Entry_Value_FromUnsignedInt(4));
Dictionary_Add(dictionary, "meshSizeK", Dictionary_Entry_Value_FromUnsignedInt(4));
Dictionary_Add(dictionary, "allowUnbalancing", Dictionary_Entry_Value_FromBool(True));
extensionMgr_Register = ExtensionManager_Register_New();
nTopology = (Topology*)IJK6Topology_New( "IJK6Topology", dictionary );
eLayout = (ElementLayout*)ParallelPipedHexaEL_New( "PPHexaEL", 3, dictionary );
nLayout = (NodeLayout*)CornerNL_New( "CornerNL", dictionary, eLayout, nTopology );
decomp = (MeshDecomp*)HexaMD_New( "HexaMD", dictionary, MPI_COMM_WORLD, eLayout, nLayout );
layout = MeshLayout_New( "MeshLayout", eLayout, nLayout, decomp );
mesh = Mesh_New( "Mesh", layout, 0, 0, extensionMgr_Register, dictionary );
/* Create CF stuff */
quadCF = ConditionFunction_New(quadratic, "quadratic");
expCF = ConditionFunction_New(exponential, "exponential");
conFunc_Register = ConditionFunction_Register_New();
ConditionFunction_Register_Add(conFunc_Register, quadCF);
ConditionFunction_Register_Add(conFunc_Register, expCF);
/* Create variable register */
variable_Register = Variable_Register_New();
/* Create variables */
for (i = 0; i < 6; i++) {
array[i] = Memory_Alloc_Array( double, decomp->nodeLocalCount, "array[i]" );
var[i] = Variable_NewScalar( varName[i], Variable_DataType_Double, &decomp->nodeLocalCount, (void**)&array[i], 0 );
Variable_Register_Add(variable_Register, var[i]);
}
array[6] = Memory_Alloc_Array( double, decomp->nodeLocalCount * 5, "array[6]" );
var[6] = Variable_NewVector( varName[6], Variable_DataType_Double, 5, &decomp->nodeLocalCount, (void**)&array[6], 0 );
Variable_Register_Add(variable_Register, var[6]);
Variable_Register_BuildAll(variable_Register);
/* Create WallVC */
for (i = 0; i < 6; i++)
{
Index j, k;
vc = WallVC_New( vcKeyName[i], vcKey[i], variable_Register, conFunc_Register, dictionary, mesh );
Build( vc, 0, False );
for (j = 0; j < 6; j++)
memset(array[j], 0, sizeof(double)* decomp->nodeLocalCount );
memset(array[6], 0, sizeof(double)* decomp->nodeLocalCount * 5);
VariableCondition_Apply(vc, NULL);
if (rank == procToWatch)
{
printf("Testing for %s\n", vcKey[i]);
Print(vc, stream);
printf("\n");
for (j = 0; j < 6; j++)
{
printf("\nvar[%u]: %.2lf", j, array[j][0]);
for (k = 1; k < decomp->nodeLocalCount; k++)
printf(", %.2lf", array[j][k]);
}
printf("\nvar[6]: %.2lf", array[6][0]);
for (j = 1; j < decomp->nodeLocalCount*5; j++)
printf(", %.2lf", array[6][j]);
printf("\n\n");
for (j = 0; j < 7; j++)
{
for (k = 0; k < decomp->nodeLocalCount; k++)
printf("%s ", VariableCondition_IsCondition(vc, k, j) ? "True " : "False");
printf("\n");
}
printf("\n");
for (j = 0; j < 7; j++)
{
for (k = 0; k < decomp->nodeLocalCount; k++)
{
VariableCondition_ValueIndex valIndex;
valIndex = VariableCondition_GetValueIndex(vc, k, j);
if (valIndex != (unsigned)-1)
printf("%03u ", valIndex);
else
printf("XXX ");
}
printf("\n");
}
printf("\n");
}
Stg_Class_Delete(vc);
}
Stg_Class_Delete(variable_Register);
for (i = 0; i < 7; i++)
{
Stg_Class_Delete(var[i]);
if (array[i]) Memory_Free(array[i]);
}
Stg_Class_Delete(conFunc_Register);
Stg_Class_Delete(quadCF);
Stg_Class_Delete(expCF);
Stg_Class_Delete(layout);
Stg_Class_Delete(decomp);
Stg_Class_Delete(nLayout);
Stg_Class_Delete(eLayout);
Stg_Class_Delete( nTopology );
Stg_Class_Delete(dictionary);
DiscretisationUtils_Finalise();
DiscretisationMesh_Finalise();
DiscretisationShape_Finalise();
DiscretisationGeometry_Finalise();
Base_Finalise();
/* Close off MPI */
MPI_Finalize();
return 0; /* success */
}
