void getLocalToGlobalMap(int (*local2global)[2],
int &off_global,
const Basis<Scalar,ArrayType> &basis,
const int *element) {
const int local = 0, global = 1;
const int nbf = basis.getCardinality();
const shards::CellTopology cell = basis.getBaseCellTopology();
const int dim = cell.getDimension();
int cnt = 0, off_element = 0;
int subcell_verts[4], nids;
const int nvert = cell.getVertexCount();
for (int i=0;i<nvert;++i) {
const int ord_vert = (off_element < nbf ? basis.getDofOrdinal(0, i, 0) : 0);
const int dof_vert = (off_element < nbf ? basis.getDofTag(ord_vert)[3] : 0);
local2global[cnt][local] = off_element;
off_element += dof_vert;
Orientation::getElementNodeMap(subcell_verts, nids,
cell, element,
0, i);
if (!findNode(local2global[cnt][global], subcell_verts, nids, true)) {
addNode(subcell_verts, nids, off_global);
local2global[cnt][global] = off_global;
off_global += dof_vert;
}
++cnt;
}
const int nedge = cell.getEdgeCount();
for (int i=0;i<nedge;++i) {
const int ord_edge = (off_element < nbf ? basis.getDofOrdinal(1, i, 0) : 0);
const int dof_edge = (off_element < nbf ? basis.getDofTag(ord_edge)[3] : 0);
local2global[cnt][local] = off_element;
off_element += dof_edge;
Orientation::getElementNodeMap(subcell_verts, nids,
cell, element,
1, i);
if (!findNode(local2global[cnt][global], subcell_verts, nids, true)) {
addNode(subcell_verts, nids, off_global);
local2global[cnt][global] = off_global;
off_global += dof_edge;
}
++cnt;
}
const int nface = cell.getFaceCount();
for (int i=0;i<nface;++i) {
const int ord_face = (off_element < nbf ? basis.getDofOrdinal(2, i, 0) : 0);
const int dof_face = (off_element < nbf ? basis.getDofTag(ord_face)[3] : 0);
local2global[cnt][local] = off_element;
off_element += dof_face;
Orientation::getElementNodeMap(subcell_verts, nids,
cell, element,
2, i);
if (!findNode(local2global[cnt][global], subcell_verts, nids, true)) {
addNode(subcell_verts, nids, off_global);
local2global[cnt][global] = off_global;
off_global += dof_face;
}
++cnt;
}
{
const int i = 0;
const int ord_intr = (off_element < nbf ? basis.getDofOrdinal(dim, i, 0) : 0);
const int dof_intr = (off_element < nbf ? basis.getDofTag(ord_intr)[3] : 0);
local2global[cnt][local] = off_element;
off_element += dof_intr;
Orientation::getElementNodeMap(subcell_verts, nids,
cell, element,
dim, i);
if (!findNode(local2global[cnt][global], subcell_verts, nids, true)) {
addNode(subcell_verts, nids, off_global);
local2global[cnt][global] = off_global;
off_global += dof_intr;
}
++cnt;
}
// add the last offset
local2global[cnt][local] = off_element;
local2global[cnt][global] = -1; // invalid values
}
void
OrientationTools<SpT>::
modifyBasisByOrientation(/**/ Kokkos::DynRankView<outputValueType,outputProperties...> output,
const Kokkos::DynRankView<inputValueType, inputProperties...> input,
const Kokkos::DynRankView<ortValueType, ortProperties...> orts,
const BasisPtrType basis ) {
#ifdef HAVE_INTREPID2_DEBUG
{
INTREPID2_TEST_FOR_EXCEPTION( input.rank() != output.rank(), std::invalid_argument,
">>> ERROR (OrientationTools::modifyBasisByOrientation): Input and output rank are not 3.");
for (ordinal_type i=0;i<input.rank();++i)
INTREPID2_TEST_FOR_EXCEPTION( input.dimension(i) != output.dimension(i), std::invalid_argument,
">>> ERROR (OrientationTools::modifyBasisByOrientation): Input and output dimension does not match.");
INTREPID2_TEST_FOR_EXCEPTION( input.dimension(1) != basis->getCardinality(), std::invalid_argument,
">>> ERROR (OrientationTools::modifyBasisByOrientation): Field dimension of input/output does not match to basis cardinality.");
INTREPID2_TEST_FOR_EXCEPTION( input.dimension(3) != basis->getBaseCellTopology().getDimension(), std::invalid_argument,
">>> ERROR (OrientationTools::modifyBasisByOrientation): Space dimension of input/output does not match to topology dimension.");
}
#endif
if (basis->requireOrientation()) {
auto ordinalToTag = Kokkos::create_mirror_view(typename SpT::memory_space(), basis->getAllDofTags());
auto tagToOrdinal = Kokkos::create_mirror_view(typename SpT::memory_space(), basis->getAllDofOrdinal());
Kokkos::deep_copy(ordinalToTag, basis->getAllDofTags());
Kokkos::deep_copy(tagToOrdinal, basis->getAllDofOrdinal());
const ordinal_type
numCells = output.dimension(0),
//numBasis = output.dimension(1),
numPoints = output.dimension(2),
dimBasis = output.dimension(3);
const CoeffMatrixDataViewType matData = createCoeffMatrix(basis);
const shards::CellTopology cellTopo = basis->getBaseCellTopology();
const ordinal_type
numVerts = cellTopo.getVertexCount(),
numEdges = cellTopo.getEdgeCount(),
numFaces = cellTopo.getFaceCount();
const ordinal_type intrDim = ( numEdges == 0 ? 1 :
numFaces == 0 ? 2 :
/**/ 3 );
for (auto cell=0;cell<numCells;++cell) {
auto out = Kokkos::subview(output, cell, Kokkos::ALL(), Kokkos::ALL(), Kokkos::ALL());
auto in = Kokkos::subview(input, cell, Kokkos::ALL(), Kokkos::ALL(), Kokkos::ALL());
// vertex copy (no orientation)
for (auto vertId=0;vertId<numVerts;++vertId) {
const auto i = tagToOrdinal(0, vertId, 0);
if (i != -1) // if dof does not exist i returns with -1
for (auto j=0;j<numPoints;++j)
for (auto k=0;k<dimBasis;++k)
out(i, j, k) = in(i, j, k);
}
// interior copy
{
const auto ordIntr = tagToOrdinal(intrDim, 0, 0);
if (ordIntr != -1) {
const auto ndofIntr = ordinalToTag(ordIntr, 3);
for (auto i=0;i<ndofIntr;++i) {
const auto ii = tagToOrdinal(intrDim, 0, i);
for (auto j=0;j<numPoints;++j)
for (auto k=0;k<dimBasis;++k)
out(ii, j, k) = in(ii, j, k);
}
}
}
// edge transformation
if (numEdges > 0) {
ordinal_type ortEdges[12];
orts(cell).getEdgeOrientation(ortEdges, numEdges);
// apply coeff matrix
for (auto edgeId=0;edgeId<numEdges;++edgeId) {
const auto ordEdge = tagToOrdinal(1, edgeId, 0);
if (ordEdge != -1) {
const auto ndofEdge = ordinalToTag(ordEdge, 3);
const auto mat = Kokkos::subview(matData,
edgeId, ortEdges[edgeId],
Kokkos::ALL(), Kokkos::ALL());
for (auto j=0;j<numPoints;++j)
for (auto i=0;i<ndofEdge;++i) {
const auto ii = tagToOrdinal(1, edgeId, i);
for (auto k=0;k<dimBasis;++k) {
double temp = 0.0;
for (auto l=0;l<ndofEdge;++l) {
const auto ll = tagToOrdinal(1, edgeId, l);
temp += mat(i,l)*in(ll, j, k);
}
out(ii, j, k) = temp;
}
}
}
}
}
// face transformation
if (numFaces > 0) {
ordinal_type ortFaces[12];
orts(cell).getFaceOrientation(ortFaces, numFaces);
// apply coeff matrix
for (auto faceId=0;faceId<numFaces;++faceId) {
const auto ordFace = tagToOrdinal(2, faceId, 0);
if (ordFace != -1) {
const auto ndofFace = ordinalToTag(ordFace, 3);
const auto mat = Kokkos::subview(matData,
numEdges+faceId, ortFaces[faceId],
Kokkos::ALL(), Kokkos::ALL());
for (auto j=0;j<numPoints;++j)
for (auto i=0;i<ndofFace;++i) {
const auto ii = tagToOrdinal(2, faceId, i);
for (auto k=0;k<dimBasis;++k) {
double temp = 0.0;
for (auto l=0;l<ndofFace;++l) {
const auto ll = tagToOrdinal(2, faceId, l);
temp += mat(i,l)*in(ll, j, k);
}
out(ii, j, k) = temp;
}
}
}
}
}
}
} else {
Kokkos::deep_copy(output, input);
}
}
/** \brief Tests for experimental assembly procedure matching basis values.
\param argc [in] - number of command-line arguments
\param argv [in] - command-line arguments
*/
int main(int argc, char *argv[]) {
Teuchos::GlobalMPISession mpiSession(&argc, &argv);
Kokkos::initialize();
// This little trick lets us print to std::cout only if a (dummy) command-line argument is provided.
int iprint = argc - 1;
bool verbose = false;
int maxp = INTREPID2_MAX_ORDER;
for (int i=0;i<argc;++i) {
if ((strcmp(argv[i],"--verbose") == 0)) { verbose = atoi(argv[++i]); continue;}
if ((strcmp(argv[i],"--maxp") == 0)) { maxp = atoi(argv[++i]); continue;}
}
Teuchos::RCP<std::ostream> outStream;
Teuchos::oblackholestream bhs; // outputs nothing
if (iprint > 0)
outStream = Teuchos::rcp(&std::cout, false);
else
outStream = Teuchos::rcp(&bhs, false);
// Save the format state of the original std::cout.
Teuchos::oblackholestream oldFormatState;
oldFormatState.copyfmt(std::cout);
*outStream \
<< "===============================================================================\n" \
<< "| |\n" \
<< "| Unit Test HGRAD_TET_Cn_FEM |\n" \
<< "| |\n" \
<< "| 1) High order assembly |\n" \
<< "| |\n" \
<< "| Questions? Contact Pavel Bochev ([email protected]) or |\n" \
<< "| Denis Ridzal ([email protected]) or |\n" \
<< "| Robert Kirby ([email protected]) or |\n" \
<< "| Kyungjoo Kim ([email protected]) |\n" \
<< "| |\n" \
<< "| Intrepid's website: http://trilinos.sandia.gov/packages/intrepid |\n" \
<< "| Trilinos website: http://trilinos.sandia.gov |\n" \
<< "| |\n" \
<< "===============================================================================\n";
int r_val = 0;
#if defined( INTREPID_USING_EXPERIMENTAL_HIGH_ORDER )
typedef double value_type;
// Let's instantiate a basis
try {
OrientationTools<value_type>::verboseStreamPtr = outStream.get();
//for (int test_order=1;test_order<=10;++test_order) {
for (int test_order=1;test_order<=maxp;++test_order) {
// Step 0 : construct basis function set
const int order = test_order;
BasisSet_HGRAD_TET_Cn_FEM<double,FieldContainer<double> > basis_set(order , POINTTYPE_EQUISPACED);
const auto& cell_basis = basis_set.getCellBasis();
const auto& face_basis = basis_set.getTriangleBasis();
const shards::CellTopology cell_topo = cell_basis.getBaseCellTopology();
const shards::CellTopology face_topo = face_basis.getBaseCellTopology();
const int nbf_cell = cell_basis.getCardinality();
const int nbf_face = face_basis.getCardinality();
const int ndim_cell = cell_topo.getDimension();
const int ndim_face = face_topo.getDimension();
const int npts = PointTools::getLatticeSize(face_topo, order, 1);
for (int test_face=0;test_face<4;++test_face) {
// tricky part
const bool left_handed = cell_topo.getNodeMap(2, test_face, 1) > cell_topo.getNodeMap(2, test_face, 2);
for (int test_ort=0;test_ort<6;++test_ort) {
*outStream << "\n" \
<< "===============================================================================\n" \
<< " Order = " << test_order << " , Face = " << test_face << " , Orientation = " << test_ort << "\n" \
<< "===============================================================================\n";
// Step 1 : create reference and modified triangle points
// reference triangle points
FieldContainer<value_type> ref_face_pts(npts, ndim_face);
PointTools::getLattice<value_type>(ref_face_pts,
face_topo,
order, 1);
// modified triangle points
const int left_ort[] = { 0, 2, 1, 3, 5, 4 };
FieldContainer<value_type> ort_face_pts(npts, ndim_face);
OrientationTools<value_type>::mapToModifiedReference(ort_face_pts,
ref_face_pts,
face_topo,
(left_handed ? left_ort[test_ort] : test_ort));
// Step 2 : map face points to cell points appropriately
const int nface = cell_topo.getFaceCount();
// create orientation object
int orts[4] = {};
orts[test_face] = test_ort;
Orientation ort;
ort.setFaceOrientation(nface, orts);
// map triangle points and modified points to reference coordinates
FieldContainer<value_type> ref_cell_pts(npts, ndim_cell);
CellTools<value_type>::mapToReferenceSubcell(ref_cell_pts,
ref_face_pts,
ndim_face,
test_face,
cell_topo);
// Step 3 : evaluate modified basis functions with orientation for reference cell points
FieldContainer<double> ort_cell_vals(nbf_cell, npts);
{
// temporary cell workspace
FieldContainer<double> tmp_cell_vals(nbf_cell, npts);
cell_basis.getValues(tmp_cell_vals, ref_cell_pts, OPERATOR_VALUE);
OrientationTools<value_type>::getBasisFunctionsByTopology(ort_cell_vals,
tmp_cell_vals,
cell_basis);
for (int i=0;i<nbf_cell;++i)
for (int j=0;j<npts;++j)
tmp_cell_vals(i, j) = ort_cell_vals(i, j);
OrientationTools<value_type>::verbose = verbose;
OrientationTools<value_type>::reverse = true; // for point matching only
OrientationTools<value_type>::getModifiedBasisFunctions(ort_cell_vals,
tmp_cell_vals,
basis_set,
ort);
OrientationTools<value_type>::verbose = false;
}
// Step 4 : evaluate reference face basis functions for modified face points
FieldContainer<double> ref_face_vals(nbf_face, npts);
{
// temporary face workspace
FieldContainer<double> tmp_face_vals(nbf_face, npts);
face_basis.getValues(tmp_face_vals, ort_face_pts, OPERATOR_VALUE);
OrientationTools<value_type>::getBasisFunctionsByTopology(ref_face_vals,
tmp_face_vals,
face_basis);
}
// Step 5 : compare the basis functions to face functions
{
// strip the range of cell DOFs
int off_cell = 0;
{
const int nvert = cell_topo.getVertexCount();
for (int i=0;i<nvert;++i) {
const int ord_vert = cell_basis.getDofOrdinal(0, i, 0);
off_cell += cell_basis.getDofTag(ord_vert)[3];
}
if (off_cell < nbf_cell) {
const int nedge = cell_topo.getEdgeCount();
for (int i=0;i<nedge;++i) {
const int ord_edge = cell_basis.getDofOrdinal(1, i, 0);
off_cell += cell_basis.getDofTag(ord_edge)[3];
}
}
if (off_cell < nbf_cell) {
for (int i=0;i<test_face;++i) {
const int ord_face = cell_basis.getDofOrdinal(2, i, 0);
off_cell += cell_basis.getDofTag(ord_face)[3];
}
}
}
// strip the range of face DOFs
int off_face = 0;
{
const int nvert = face_topo.getVertexCount();
for (int i=0;i<nvert;++i) {
const int ord_vert = face_basis.getDofOrdinal(0, i, 0);
off_face += face_basis.getDofTag(ord_vert)[3];
}
if (off_face < nbf_face) {
const int nedge = face_topo.getEdgeCount();
for (int i=0;i<nedge;++i) {
const int ord_edge = face_basis.getDofOrdinal(1, i, 0);
off_face += face_basis.getDofTag(ord_edge)[3];
}
}
}
const int ndof = nbf_face - off_face;
for (int i=0;i<ndof;++i) {
for (int j=0;j<npts;++j) {
const value_type diff = std::abs(ort_cell_vals(i+off_cell,j) - ref_face_vals(i+off_face,j));
if (diff > INTREPID_TOL) {
++r_val;
*outStream << std::setw(70) << "^^^^----FAILURE!" << "\n";
*outStream << " Basis function " << i << " at point " << j << " does not match each other\n";
}
}
}
}
} // test ort
} // test face
} // test order
}
catch (std::exception err) {
std::cout << " Exeption\n";
*outStream << err.what() << "\n\n";
r_val = -1000;
}
#else
*outStream << "\t This test is for high order element assembly. \n"
<< "\t Use -D INTREPID_USING_EXPERIMENTAL_HIGH_ORDER in CMAKE_CXX_FLAGS \n";
#endif
if (r_val != 0)
std::cout << "End Result: TEST FAILED, r_val = " << r_val << "\n";
else
std::cout << "End Result: TEST PASSED\n";
// reset format state of std::cout
std::cout.copyfmt(oldFormatState);
Kokkos::finalize();
return r_val;
}