Intrepid: http://trilinos.sandia.gov/packages/docs/r10.0/packages/intrepid/test/Discretization/Basis/HDIV_HEX_I1_FEM/test

00001 // @HEADER
00002 // ************************************************************************
00003 //
00004 //                           Intrepid Package
00005 //                 Copyright (2007) Sandia Corporation
00006 //
00007 // Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
00008 // license for use of this work by or on behalf of the U.S. Government.
00009 //
00010 // This library is free software; you can redistribute it and/or modify
00011 // it under the terms of the GNU Lesser General Public License as
00012 // published by the Free Software Foundation; either version 2.1 of the
00013 // License, or (at your option) any later version.
00014 //
00015 // This library is distributed in the hope that it will be useful, but
00016 // WITHOUT ANY WARRANTY; without even the implied warranty of
00017 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00018 // Lesser General Public License for more details.
00019 //
00020 // You should have received a copy of the GNU Lesser General Public
00021 // License along with this library; if not, write to the Free Software
00022 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
00023 // USA
00024 // Questions? Contact Pavel Bochev  (pbboche@sandia.gov), 
00025 //                    Denis Ridzal  (dridzal@sandia.gov),
00026 //                    Kara Peterson (kjpeter@sandia.gov).
00027 //
00028 // ************************************************************************
00029 // @HEADER
00030 
00035 #include "Intrepid_FieldContainer.hpp"
00036 #include "Intrepid_HDIV_HEX_I1_FEM.hpp"
00037 #include "Teuchos_oblackholestream.hpp"
00038 #include "Teuchos_RCP.hpp"
00039 #include "Teuchos_GlobalMPISession.hpp"
00040 
00041 using namespace std;
00042 using namespace Intrepid;
00043 
00044 #define INTREPID_TEST_COMMAND( S , throwCounter, nException )                                                              \
00045 {                                                                                                                          \
00046   ++nException;                                                                                                            \
00047   try {                                                                                                                    \
00048     S ;                                                                                                                    \
00049   }                                                                                                                        \
00050   catch (std::logic_error err) {                                                                                           \
00051       ++throwCounter;                                                                                                      \
00052       *outStream << "Expected Error " << nException << " -------------------------------------------------------------\n"; \
00053       *outStream << err.what() << '\n';                                                                                    \
00054       *outStream << "-------------------------------------------------------------------------------" << "\n\n";           \
00055   };                                                                                                                       \
00056 }
00057 
00058 int main(int argc, char *argv[]) {
00059   
00060   Teuchos::GlobalMPISession mpiSession(&argc, &argv);
00061 
00062   // This little trick lets us print to std::cout only if
00063   // a (dummy) command-line argument is provided.
00064   int iprint     = argc - 1;
00065   Teuchos::RCP<std::ostream> outStream;
00066   Teuchos::oblackholestream bhs; // outputs nothing
00067   if (iprint > 0)
00068     outStream = Teuchos::rcp(&std::cout, false);
00069   else
00070     outStream = Teuchos::rcp(&bhs, false);
00071   
00072   // Save the format state of the original std::cout.
00073   Teuchos::oblackholestream oldFormatState;
00074   oldFormatState.copyfmt(std::cout);
00075   
00076   *outStream \
00077     << "===============================================================================\n" \
00078     << "|                                                                             |\n" \
00079     << "|                 Unit Test (Basis_HDIV_HEX_I1_FEM)                           |\n" \
00080     << "|                                                                             |\n" \
00081     << "|     1) Conversion of Dof tags into Dof ordinals and back                    |\n" \
00082     << "|     2) Basis values for VALUE and DIV operators                             |\n" \
00083     << "|                                                                             |\n" \
00084     << "|  Questions? Contact  Pavel Bochev  (pbboche@sandia.gov),                    |\n" \
00085     << "|                      Denis Ridzal  (dridzal@sandia.gov),                    |\n" \
00086     << "|                      Kara Peterson (kjpeter@sandia.gov).                    |\n" \
00087     << "|                                                                             |\n" \
00088     << "|  Intrepid's website: http://trilinos.sandia.gov/packages/intrepid           |\n" \
00089     << "|  Trilinos website:   http://trilinos.sandia.gov                             |\n" \
00090     << "|                                                                             |\n" \
00091     << "===============================================================================\n"\
00092     << "| TEST 1: Basis creation, exception testing                                   |\n"\
00093     << "===============================================================================\n";
00094   
00095   // Define basis and error flag
00096   Basis_HDIV_HEX_I1_FEM<double, FieldContainer<double> > hexBasis;
00097   int errorFlag = 0;
00098 
00099   // Initialize throw counter for exception testing
00100   int nException     = 0;
00101   int throwCounter   = 0;
00102 
00103   // Define array containing the 8 vertices of the reference HEX, its center and 6 face centers
00104   FieldContainer<double> hexNodes(15, 3);
00105   hexNodes(0,0) = -1.0;  hexNodes(0,1) = -1.0;  hexNodes(0,2) = -1.0;
00106   hexNodes(1,0) =  1.0;  hexNodes(1,1) = -1.0;  hexNodes(1,2) = -1.0;
00107   hexNodes(2,0) =  1.0;  hexNodes(2,1) =  1.0;  hexNodes(2,2) = -1.0;
00108   hexNodes(3,0) = -1.0;  hexNodes(3,1) =  1.0;  hexNodes(3,2) = -1.0;
00109   
00110   hexNodes(4,0) = -1.0;  hexNodes(4,1) = -1.0;  hexNodes(4,2) =  1.0;
00111   hexNodes(5,0) =  1.0;  hexNodes(5,1) = -1.0;  hexNodes(5,2) =  1.0;
00112   hexNodes(6,0) =  1.0;  hexNodes(6,1) =  1.0;  hexNodes(6,2) =  1.0;
00113   hexNodes(7,0) = -1.0;  hexNodes(7,1) =  1.0;  hexNodes(7,2) =  1.0;  
00114   
00115   hexNodes(8,0) =  0.0;  hexNodes(8,1) =  0.0;  hexNodes(8,2) =  0.0;
00116   
00117   hexNodes(9,0) =  1.0;  hexNodes(9,1) =  0.0;  hexNodes(9,2) =  0.0;
00118   hexNodes(10,0)= -1.0;  hexNodes(10,1)=  0.0;  hexNodes(10,2)=  0.0;
00119   
00120   hexNodes(11,0)=  0.0;  hexNodes(11,1)=  1.0;  hexNodes(11,2)=  0.0;
00121   hexNodes(12,0)=  0.0;  hexNodes(12,1)= -1.0;  hexNodes(12,2)=  0.0;
00122   
00123   hexNodes(13,0)=  0.0;  hexNodes(13,1)=  0.0;  hexNodes(13,2)=  1.0;
00124   hexNodes(14,0)=  0.0;  hexNodes(14,1)=  0.0;  hexNodes(14,2)= -1.0;
00125 
00126   
00127   // Generic array for the output values; needs to be properly resized depending on the operator type
00128   FieldContainer<double> vals;
00129 
00130   try{
00131     // exception #1: GRAD cannot be applied to HDIV functions 
00132     // resize vals to rank-3 container with dimensions (num. basis functions, num. points, arbitrary)
00133     vals.resize(hexBasis.getCardinality(), hexNodes.dimension(0), 3 );
00134     INTREPID_TEST_COMMAND( hexBasis.getValues(vals, hexNodes, OPERATOR_GRAD), throwCounter, nException );
00135 
00136     // exception #2: CURL cannot be applied to HDIV functions
00137     INTREPID_TEST_COMMAND( hexBasis.getValues(vals, hexNodes, OPERATOR_CURL), throwCounter, nException );
00138 
00139     // Exceptions 3-7: all bf tags/bf Ids below are wrong and should cause getDofOrdinal() and 
00140     // getDofTag() to access invalid array elements thereby causing bounds check exception
00141     // exception #3
00142     INTREPID_TEST_COMMAND( hexBasis.getDofOrdinal(3,0,0), throwCounter, nException );
00143     // exception #4
00144     INTREPID_TEST_COMMAND( hexBasis.getDofOrdinal(1,1,1), throwCounter, nException );
00145     // exception #5
00146     INTREPID_TEST_COMMAND( hexBasis.getDofOrdinal(0,4,1), throwCounter, nException );
00147     // exception #6
00148     INTREPID_TEST_COMMAND( hexBasis.getDofTag(12), throwCounter, nException );
00149     // exception #7
00150     INTREPID_TEST_COMMAND( hexBasis.getDofTag(-1), throwCounter, nException );
00151 
00152 #ifdef HAVE_INTREPID_DEBUG
00153     // Exceptions 8- test exception handling with incorrectly dimensioned input/output arrays
00154     // exception #8: input points array must be of rank-2
00155     FieldContainer<double> badPoints1(4, 5, 3);
00156     INTREPID_TEST_COMMAND( hexBasis.getValues(vals, badPoints1, OPERATOR_VALUE), throwCounter, nException );
00157     
00158     // exception #9 dimension 1 in the input point array must equal space dimension of the cell
00159     FieldContainer<double> badPoints2(4, 2);
00160     INTREPID_TEST_COMMAND( hexBasis.getValues(vals, badPoints2, OPERATOR_VALUE), throwCounter, nException );
00161     
00162     // exception #10 output values must be of rank-3 for OPERATOR_VALUE
00163     FieldContainer<double> badVals1(4, 3);
00164     INTREPID_TEST_COMMAND( hexBasis.getValues(badVals1, hexNodes, OPERATOR_VALUE), throwCounter, nException );
00165  
00166     // exception #11 output values must be of rank-2 for OPERATOR_DIV
00167     FieldContainer<double> badVals2(4, 3, 3);
00168     INTREPID_TEST_COMMAND( hexBasis.getValues(badVals2, hexNodes, OPERATOR_DIV), throwCounter, nException );
00169     
00170     // exception #12 incorrect 0th dimension of output array (must equal number of basis functions)
00171     FieldContainer<double> badVals3(hexBasis.getCardinality() + 1, hexNodes.dimension(0), 3);
00172     INTREPID_TEST_COMMAND( hexBasis.getValues(badVals3, hexNodes, OPERATOR_VALUE), throwCounter, nException );
00173     
00174     // exception #13 incorrect 0th dimension of output array (must equal number of basis functions)
00175     FieldContainer<double> badVals4(hexBasis.getCardinality() + 1, hexNodes.dimension(0));
00176     INTREPID_TEST_COMMAND( hexBasis.getValues(badVals4, hexNodes, OPERATOR_DIV), throwCounter, nException );
00177 
00178     // exception #14 incorrect 1st dimension of output array (must equal number of points)
00179     FieldContainer<double> badVals5(hexBasis.getCardinality(), hexNodes.dimension(0) + 1, 3);
00180     INTREPID_TEST_COMMAND( hexBasis.getValues(badVals5, hexNodes, OPERATOR_VALUE), throwCounter, nException );
00181 
00182     // exception #15 incorrect 1st dimension of output array (must equal number of points)
00183     FieldContainer<double> badVals6(hexBasis.getCardinality(), hexNodes.dimension(0) + 1);
00184     INTREPID_TEST_COMMAND( hexBasis.getValues(badVals6, hexNodes, OPERATOR_DIV), throwCounter, nException );
00185 
00186     // exception #16: incorrect 2nd dimension of output array (must equal the space dimension)
00187     FieldContainer<double> badVals7(hexBasis.getCardinality(), hexNodes.dimension(0), 4);
00188     INTREPID_TEST_COMMAND( hexBasis.getValues(badVals7, hexNodes, OPERATOR_VALUE), throwCounter, nException );
00189 #endif
00190     
00191   }
00192   catch (std::logic_error err) {
00193     *outStream << "UNEXPECTED ERROR !!! ----------------------------------------------------------\n";
00194     *outStream << err.what() << '\n';
00195     *outStream << "-------------------------------------------------------------------------------" << "\n\n";
00196     errorFlag = -1000;
00197   };
00198   
00199   // Check if number of thrown exceptions matches the one we expect 
00200   if (throwCounter != nException) {
00201     errorFlag++;
00202     *outStream << std::setw(70) << "^^^^----FAILURE!" << "\n";
00203   }
00204   
00205   *outStream \
00206     << "\n"
00207     << "===============================================================================\n"\
00208     << "| TEST 2: correctness of tag to enum and enum to tag lookups                  |\n"\
00209     << "===============================================================================\n";
00210   
00211   try{
00212     std::vector<std::vector<int> > allTags = hexBasis.getAllDofTags();
00213     
00214     // Loop over all tags, lookup the associated dof enumeration and then lookup the tag again
00215     for (unsigned i = 0; i < allTags.size(); i++) {
00216       int bfOrd  = hexBasis.getDofOrdinal(allTags[i][0], allTags[i][1], allTags[i][2]);
00217       
00218       std::vector<int> myTag = hexBasis.getDofTag(bfOrd);
00219        if( !( (myTag[0] == allTags[i][0]) &&
00220               (myTag[1] == allTags[i][1]) &&
00221               (myTag[2] == allTags[i][2]) &&
00222               (myTag[3] == allTags[i][3]) ) ) {
00223         errorFlag++;
00224         *outStream << std::setw(70) << "^^^^----FAILURE!" << "\n";
00225         *outStream << " getDofOrdinal( {" 
00226           << allTags[i][0] << ", " 
00227           << allTags[i][1] << ", " 
00228           << allTags[i][2] << ", " 
00229           << allTags[i][3] << "}) = " << bfOrd <<" but \n";   
00230         *outStream << " getDofTag(" << bfOrd << ") = { "
00231           << myTag[0] << ", " 
00232           << myTag[1] << ", " 
00233           << myTag[2] << ", " 
00234           << myTag[3] << "}\n";        
00235       }
00236     }
00237     
00238     // Now do the same but loop over basis functions
00239     for( int bfOrd = 0; bfOrd < hexBasis.getCardinality(); bfOrd++) {
00240       std::vector<int> myTag  = hexBasis.getDofTag(bfOrd);
00241       int myBfOrd = hexBasis.getDofOrdinal(myTag[0], myTag[1], myTag[2]);
00242       if( bfOrd != myBfOrd) {
00243         errorFlag++;
00244         *outStream << std::setw(70) << "^^^^----FAILURE!" << "\n";
00245         *outStream << " getDofTag(" << bfOrd << ") = { "
00246           << myTag[0] << ", " 
00247           << myTag[1] << ", " 
00248           << myTag[2] << ", " 
00249           << myTag[3] << "} but getDofOrdinal({" 
00250           << myTag[0] << ", " 
00251           << myTag[1] << ", " 
00252           << myTag[2] << ", " 
00253           << myTag[3] << "} ) = " << myBfOrd << "\n";
00254       }
00255     }
00256   }
00257   catch (std::logic_error err){
00258     *outStream << err.what() << "\n\n";
00259     errorFlag = -1000;
00260   };
00261   
00262   *outStream \
00263     << "\n"
00264     << "===============================================================================\n"\
00265     << "| TEST 3: correctness of basis function values                                |\n"\
00266     << "===============================================================================\n";
00267   
00268   outStream -> precision(20);
00269   
00270   // VALUE: Each row pair gives the 6x3 correct basis set values at an evaluation point: (P,F,D) layout
00271   double basisValues[] = {
00272     // bottom 4 vertices
00273       0.,-0.25,0.,  0.,0.,0.,  0.,0.,0., -0.25,0.,0.,  0.,0.,-0.25,  0.,0.,0.,
00274       0.,-0.25,0.,  0.25,0.,0.,  0.,0.,0.,  0.,0.,0.,  0.,0.,-0.25,  0.,0.,0.,  
00275       0.,0.,0.,  0.25,0.,0.,  0.,0.25,0.,  0.,0.,0.,  0.,0.,-0.25,  0.,0.,0.,
00276       0.,0.,0.,  0.,0.,0.,  0.,0.25,0., -0.25,0.,0.,  0.,0.,-0.25,  0.,0.,0.,
00277     // top 4 vertices
00278       0.,-0.25,0.,  0.,0.,0.,  0.,0.,0.,  -0.25,0.,0.,  0.,0.,0.,  0.,0.,0.25,
00279       0.,-0.25,0.,  0.25,0.,0.,  0.,0.,0.,  0.,0.,0.,  0.,0.,0.,  0.,0.,0.25,
00280       0.,0.,0.,  0.25,0.,0.,  0.,0.25,0.,  0.,0.,0.,  0.,0.,0.,  0.,0.,0.25,
00281       0.,0.,0.,  0.,0.,0.,  0.,0.25,0.,  -0.25,0.,0.,  0.,0.,0.,  0.,0.,0.25,
00282     // center {0, 0, 0}
00283       0.,-0.125,0.,  0.125,0.,0.,  0.,0.125,0.,  -0.125,0.,0.,  0.,0.,-0.125,  0.,0.,0.125,
00284     // faces { 1, 0, 0} and {-1, 0, 0}
00285       0.,-0.125,0.,  0.25,0.,0.,  0.,0.125,0.,  0.,0.,0.,  0.,0.,-0.125,  0.,0.,0.125, 
00286       0.,-0.125,0.,  0.,0.,0.,  0.,0.125,0., -0.25,0.,0.,  0.,0.,-0.125,  0.,0.,0.125,
00287     // faces { 0, 1, 0} and { 0,-1, 0}
00288       0.,0.,0.,  0.125,0.,0.,  0.,0.25,0.,  -0.125,0.,0.,  0.,0.,-0.125,  0.,0.,0.125,
00289       0.,-0.25,0.,  0.125,0.,0.,  0.,0.,0.,  -0.125,0.,0.,  0.,0.,-0.125,  0.,0.,0.125,
00290     // faces {0, 0, 1} and {0, 0, -1}
00291       0.,-0.125,0.,  0.125,0.,0.,  0.,0.125,0.,  -0.125,0.,0.,  0.,0.,0.,  0.,0.,0.25,
00292       0.,-0.125,0.,  0.125,0.,0.,  0.,0.125,0.,  -0.125,0.,0.,  0.,0.,-0.25,  0.,0.,0.
00293   };
00294   
00295   // DIV: each row gives the 6 correct values of the divergence of the 6 basis functions: (P,F) layout
00296   double basisDivs[] = {   
00297     // bottom 4 vertices
00298       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00299       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00300       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00301       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00302     // top 4 vertices
00303       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00304       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00305       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00306       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00307     // center {0, 0, 0}
00308       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00309     // faces { 1, 0, 0} and {-1, 0, 0}
00310       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00311       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00312     // faces { 0, 1, 0} and { 0,-1, 0}
00313       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00314       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00315     // faces {0, 0, 1} and {0, 0, -1}
00316       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00317       0.125, 0.125, 0.125, 0.125, 0.125, 0.125,  
00318   };
00319   
00320   try{
00321         
00322     // Dimensions for the output arrays:
00323     int numPoints = hexNodes.dimension(0);
00324     int numFields = hexBasis.getCardinality();
00325     int spaceDim  = hexBasis.getBaseCellTopology().getDimension();
00326     
00327     // Generic array for values and curls that will be properly sized before each call
00328     FieldContainer<double> vals;
00329     
00330     // Check VALUE of basis functions: resize vals to rank-3 container:
00331     vals.resize(numFields, numPoints, spaceDim);
00332     hexBasis.getValues(vals, hexNodes, OPERATOR_VALUE);
00333     for (int i = 0; i < numFields; i++) {
00334       for (int j = 0; j < numPoints; j++) {
00335         for (int k = 0; k < spaceDim; k++) {
00336 
00337           // compute offset for (P,F,D) data layout: indices are P->j, F->i, D->k
00338            int l = k + i * spaceDim + j * spaceDim * numFields;
00339            if (std::abs(vals(i,j,k) - basisValues[l]) > INTREPID_TOL) {
00340              errorFlag++;
00341              *outStream << std::setw(70) << "^^^^----FAILURE!" << "\n";
00342 
00343              // Output the multi-index of the value where the error is:
00344              *outStream << " At multi-index { ";
00345              *outStream << i << " ";*outStream << j << " ";*outStream << k << " ";
00346              *outStream << "}  computed value: " << vals(i,j,k)
00347                << " but reference value: " << basisValues[l] << "\n";
00348             }
00349          }
00350       }
00351     }
00352 
00353     // Check DIV of basis function: resize vals to rank-2 container
00354     vals.resize(numFields, numPoints);
00355     hexBasis.getValues(vals, hexNodes, OPERATOR_DIV);
00356     for (int i = 0; i < numFields; i++) {
00357       for (int j = 0; j < numPoints; j++) {
00358           int l =  i + j * numFields;
00359            if (std::abs(vals(i,j) - basisDivs[l]) > INTREPID_TOL) {
00360              errorFlag++;
00361              *outStream << std::setw(70) << "^^^^----FAILURE!" << "\n";
00362 
00363              // Output the multi-index of the value where the error is:
00364              *outStream << " At multi-index { ";
00365              *outStream << i << " ";*outStream << j << " ";
00366              *outStream << "}  computed divergence component: " << vals(i,j)
00367                << " but reference divergence component: " << basisDivs[l] << "\n";
00368          }
00369       }
00370     }
00371     
00372    }    
00373   
00374   // Catch unexpected errors
00375   catch (std::logic_error err) {
00376     *outStream << err.what() << "\n\n";
00377     errorFlag = -1000;
00378   };
00379   
00380   if (errorFlag != 0)
00381     std::cout << "End Result: TEST FAILED\n";
00382   else
00383     std::cout << "End Result: TEST PASSED\n";
00384   
00385   // reset format state of std::cout
00386   std::cout.copyfmt(oldFormatState);
00387   
00388   return errorFlag;
00389 }
http://trilinos.sandia.gov/packages/docs/r10.0/packages/intrepid/test/Discretization/Basis/HDIV_HEX_I1_FEM/test_01.cpp
