int test_FEI_Impl::test1()
{
if (numProcs_ > 1) {
return(0);
}
#ifdef HAVE_FEI_AZTECOO
testData* testdata = new testData(localProc_, numProcs_);
fei::SharedPtr<LinearSystemCore> linSys(new fei_trilinos::Aztec_LinSysCore(comm_));
fei::SharedPtr<LibraryWrapper> wrapper(new LibraryWrapper(linSys));
fei::SharedPtr<fei::FEI_Impl> fei(new fei::FEI_Impl(wrapper, comm_, 0));
std::string param0("name test1");
FEI_OSTRINGSTREAM osstr;
osstr << "debugOutput ";
if (path_.empty()) osstr << ".";
else osstr << path_;
std::string param1 = osstr.str();
int numParams = 2;
char** params = new char*[numParams];
params[0] = const_cast<char*>(param0.c_str());
params[1] = const_cast<char*>(param1.c_str());
//call the parameters function a couple of times to test the fei's internal
//method for merging string lists when parameters is called more than once.
CHK_ERR( fei->parameters(1, ¶ms[0]) );
CHK_ERR( fei->parameters(1, ¶ms[1]) );
CHK_ERR( fei->parameters(2, params) );
delete [] params;
CHK_ERR( fei->setIDLists(1, &(testdata->ids[0]),
1, &(testdata->ids[0])) );
CHK_ERR( fei->initFields(testdata->fieldIDs.size(),
&(testdata->fieldSizes[0]),
&(testdata->fieldIDs[0])) );
int numNodesPerElem = testdata->ids.size();
std::vector<int> numFieldsPerNode(numNodesPerElem, 1);
std::vector<int*>nodalFieldIDs(numNodesPerElem, &(testdata->fieldIDs[0]));
CHK_ERR( fei->initElemBlock(0, //blockID
1, //numElements
numNodesPerElem,
&numFieldsPerNode[0],
&nodalFieldIDs[0],
0, //numElemDofFieldsPerElement
NULL, //elemDofFieldIDs
0)); //interleaveStrategy
CHK_ERR( fei->initElem(0, //blockID
0, //elemID
&(testdata->ids[0])) );
std::vector<int*> sharingProcs2D(testdata->sharedIDs.size());
int offset = 0;
for(int i=0; i<(int)testdata->numSharingProcsPerID.size(); ++i) {
sharingProcs2D[i] = &(testdata->sharingProcs[offset]);
offset += testdata->numSharingProcsPerID[i];
}
if (testdata->sharedIDs.size() > 0) {
CHK_ERR( fei->initSharedNodes(testdata->sharedIDs.size(),
&(testdata->sharedIDs[0]),
&(testdata->numSharingProcsPerID[0]),
&sharingProcs2D[0]) );
}
CHK_ERR( fei->initComplete() );
std::vector<double> rhsData(testdata->ids.size(), 1.0);
double one = 1.0;
CHK_ERR( fei->setMatScalars(1, &(testdata->ids[0]), &one) );
CHK_ERR( fei->setRHSScalars(1, &(testdata->ids[0]), &one) );
CHK_ERR( fei->setCurrentMatrix(testdata->ids[0]) );
CHK_ERR( fei->setCurrentRHS(testdata->ids[0]) );
CHK_ERR( fei->putIntoRHS(FEI_NODE, testdata->fieldIDs[0],
testdata->ids.size(),
&(testdata->ids[0]),
&rhsData[0]) );
int numBCNodes = 2;
GlobalID* BCNodeIDs = &(testdata->ids[0]);
int BCFieldID = testdata->fieldIDs[0];
double* values = new double[numBCNodes];
int* offsetsIntoField = new int[numBCNodes];
for(int ii=0; ii<numBCNodes; ++ii) {
values[ii] = 1.0;
offsetsIntoField[ii] = 0;
}
CHK_ERR( fei->loadNodeBCs(numBCNodes, BCNodeIDs, BCFieldID,
offsetsIntoField, values) );
delete [] offsetsIntoField;
delete [] values;
CHK_ERR( fei->loadComplete() );
int numActiveNodes = 0;
CHK_ERR( fei->getNumLocalNodes(numActiveNodes) );
if (numActiveNodes != (int)testdata->ids.size()) {
ERReturn(-1);
}
GlobalID* localNodes = new GlobalID[numActiveNodes];
CHK_ERR( fei->getLocalNodeIDList(numActiveNodes, localNodes, numActiveNodes) );
int totalFieldSize = 0;
for(int ii=0; ii<(int)testdata->fieldSizes.size(); ++ii) {
totalFieldSize += testdata->fieldSizes[ii];
}
double* soln = new double[numActiveNodes*totalFieldSize];
int* offsets = new int[numActiveNodes+1];
CHK_ERR( fei->getNodalSolution(numActiveNodes, localNodes,
offsets, soln) );
delete [] offsets;
delete [] soln;
delete [] localNodes;
CHK_ERR( fei->resetInitialGuess() );
int fieldSize = 0;
CHK_ERR( fei->getFieldSize(testdata->fieldIDs[0], fieldSize) );
double initTime, loadTime, solveTime, solnReturnTime;
CHK_ERR( fei->cumulative_cpu_times(initTime, loadTime, solveTime,
solnReturnTime) );
delete testdata;
#endif
return(0);
}
int test_SNL_FEI_Structure::test1()
{
testData* testdata = new testData(localProc_, numProcs_);
SNL_FEI_Structure structure(comm_);
CHK_ERR( structure.initFields(testdata->fieldIDs.size(),
&(testdata->fieldSizes[0]),
&(testdata->fieldIDs[0])) );
int numNodesPerElem = testdata->ids.size();
std::vector<int> numFieldsPerNode(numNodesPerElem, 1);
std::vector<int*>nodalFieldIDs(numNodesPerElem, &(testdata->fieldIDs[0]));
CHK_ERR( structure.initElemBlock(0, //blockID
1, //numElements
numNodesPerElem,
&numFieldsPerNode[0],
&nodalFieldIDs[0],
0, //numElemDofFieldsPerElement
NULL, //elemDofFieldIDs
0)); //interleaveStrategy
CHK_ERR( structure.initElem(0, //blockID
0, //elemID
&(testdata->ids[0])) );
std::vector<int*> sharingProcs2D(testdata->sharedIDs.size());
int i, offset = 0;
for(i=0; i<(int)testdata->numSharingProcsPerID.size(); ++i) {
sharingProcs2D[i] = &(testdata->sharingProcs[offset]);
offset += testdata->numSharingProcsPerID[i];
}
if (testdata->sharedIDs.size() > 0) {
CHK_ERR( structure.initSharedNodes(testdata->sharedIDs.size(),
testdata->sharedIDs.size() ? &(testdata->sharedIDs[0]) : 0,
testdata->numSharingProcsPerID.size() ? &(testdata->numSharingProcsPerID[0]) : 0,
&sharingProcs2D[0]) );
}
CHK_ERR( structure.initComplete() );
int numActiveNodes = structure.getNumActiveNodes();
if (numActiveNodes != (int)testdata->ids.size()) {
ERReturn(-1);
}
int fieldSize = structure.getFieldSize(testdata->fieldIDs[0]);
int numLocalEqns = fieldSize*2;
if (localProc_ == 0) numLocalEqns += 2;
int checkNumLocalEqns = structure.getNumLocalEqns();
if (numLocalEqns != checkNumLocalEqns) {
ERReturn(-1);
}
int numGlobalEqns = fieldSize*(numProcs_*2 + 2);
int checkNumGlobalEqns = structure.getNumGlobalEqns();
if (checkNumGlobalEqns != numGlobalEqns) {
ERReturn(-1);
}
std::vector<int> rowLengths;
CHK_ERR( structure.getMatrixRowLengths(rowLengths) );
int numNonzeros = 0;
for(size_t j=0; j<rowLengths.size(); ++j) {
numNonzeros += rowLengths[j];
}
std::vector<int> colIndices_1d(numNonzeros);
std::vector<int*> colIndPtrs(rowLengths.size());
offset = 0;
for(size_t j=0; j<rowLengths.size(); ++j) {
colIndPtrs[j] = &(colIndices_1d[offset]);
offset += rowLengths[j];
}
CHK_ERR( structure.getMatrixStructure(&colIndPtrs[0],
rowLengths) );
delete testdata;
return(0);
}
