NonlinearElasticityAssembler.hpp

00001 /*
00002 
00003 Copyright (C) University of Oxford, 2005-2010
00004 
00005 University of Oxford means the Chancellor, Masters and Scholars of the
00006 University of Oxford, having an administrative office at Wellington
00007 Square, Oxford OX1 2JD, UK.
00008 
00009 This file is part of Chaste.
00010 
00011 Chaste is free software: you can redistribute it and/or modify it
00012 under the terms of the GNU Lesser General Public License as published
00013 by the Free Software Foundation, either version 2.1 of the License, or
00014 (at your option) any later version.
00015 
00016 Chaste is distributed in the hope that it will be useful, but WITHOUT
00017 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
00018 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
00019 License for more details. The offer of Chaste under the terms of the
00020 License is subject to the License being interpreted in accordance with
00021 English Law and subject to any action against the University of Oxford
00022 being under the jurisdiction of the English Courts.
00023 
00024 You should have received a copy of the GNU Lesser General Public License
00025 along with Chaste. If not, see <http://www.gnu.org/licenses/>.
00026 
00027 */
00028 #ifndef NONLINEARELASTICITYASSEMBLER_HPP_
00029 #define NONLINEARELASTICITYASSEMBLER_HPP_
00030 
00031 /*
00032  * NOTE ON COMPILATION ERRORS:
00033  *
00034  * This file won't compile with Intel icpc version 9.1.039, with error message:
00035  * "Terminate with:
00036   (0): internal error: backend signals"
00037  *
00038  * Try recompiling with icpc version 10.0.025.
00039  */
00040 
00041 
00043 
00044 #include "AbstractNonlinearElasticityAssembler.hpp"
00045 //#include "LinearBasisFunction.hpp"
00046 //#include "QuadraticBasisFunction.hpp"
00047 #include "QuadraticMesh.hpp"
00048 #include "GaussianQuadratureRule.hpp"
00049 
00060 template<size_t DIM>
00061 class NonlinearElasticityAssembler : public AbstractNonlinearElasticityAssembler<DIM>
00062 {
00063     friend class TestNonlinearElasticityAssembler;
00064 
00065 protected:
00066 
00068     static const size_t NUM_VERTICES_PER_ELEMENT = DIM+1;
00070     static const size_t NUM_NODES_PER_ELEMENT = (DIM+1)*(DIM+2)/2; // assuming quadratic
00072     static const size_t STENCIL_SIZE = DIM*NUM_NODES_PER_ELEMENT + NUM_VERTICES_PER_ELEMENT;
00074     static const size_t NUM_NODES_PER_BOUNDARY_ELEMENT = DIM*(DIM+1)/2;
00076     static const size_t BOUNDARY_STENCIL_SIZE = DIM*NUM_NODES_PER_BOUNDARY_ELEMENT + DIM;
00077 
00082     QuadraticMesh<DIM>* mpQuadMesh;
00083 
00085     std::vector<BoundaryElement<DIM-1,DIM>*> mBoundaryElements;
00086 
00088     GaussianQuadratureRule<DIM>* mpQuadratureRule;
00089 
00091     GaussianQuadratureRule<DIM-1>* mpBoundaryQuadratureRule;
00092 
00110     virtual void AssembleOnElement(Element<DIM, DIM>& rElement,
00111                                    c_matrix<double, STENCIL_SIZE, STENCIL_SIZE >& rAElem,
00112                                    c_matrix<double, STENCIL_SIZE, STENCIL_SIZE >& rAElemPrecond,
00113                                    c_vector<double, STENCIL_SIZE>& rBElem,
00114                                    bool assembleResidual,
00115                                    bool assembleJacobian);
00116 
00133     virtual void AssembleOnBoundaryElement(BoundaryElement<DIM-1, DIM>& rBoundaryElement,
00134                                            c_matrix<double, BOUNDARY_STENCIL_SIZE, BOUNDARY_STENCIL_SIZE>& rAelem,
00135                                            c_vector<double, BOUNDARY_STENCIL_SIZE>& rBelem,
00136                                            c_vector<double, DIM>& rTraction,
00137                                            bool assembleResidual,
00138                                            bool assembleJacobian);
00139 
00156     void FormInitialGuess();
00157 
00167     void AssembleSystem(bool assembleResidual, bool assembleJacobian);
00168 
00174     void Initialise(std::vector<c_vector<double,DIM> >* pFixedNodeLocations);
00175 
00180     void AllocateMatrixMemory();
00181 
00182 public:
00183 
00197     NonlinearElasticityAssembler(QuadraticMesh<DIM>* pQuadMesh,
00198                                  AbstractIncompressibleMaterialLaw<DIM>* pMaterialLaw,
00199                                  c_vector<double,DIM> bodyForce,
00200                                  double density,
00201                                  std::string outputDirectory,
00202                                  std::vector<unsigned>& fixedNodes,
00203                                  std::vector<c_vector<double,DIM> >* pFixedNodeLocations = NULL);
00204 
00216     NonlinearElasticityAssembler(QuadraticMesh<DIM>* pQuadMesh,
00217                                  std::vector<AbstractIncompressibleMaterialLaw<DIM>*>& rMaterialLaws,
00218                                  c_vector<double,DIM> bodyForce,
00219                                  double density,
00220                                  std::string outputDirectory,
00221                                  std::vector<unsigned>& fixedNodes,
00222                                  std::vector<c_vector<double,DIM> >* pFixedNodeLocations = NULL);
00223 
00225     ~NonlinearElasticityAssembler();
00226 
00235     void SetSurfaceTractionBoundaryConditions(std::vector<BoundaryElement<DIM-1,DIM>*>& rBoundaryElements,
00236                                               std::vector<c_vector<double,DIM> >& rSurfaceTractions);
00237 
00245     void SetFunctionalTractionBoundaryCondition(std::vector<BoundaryElement<DIM-1,DIM>*> rBoundaryElements,
00246                                                 c_vector<double,DIM> (*pFunction)(c_vector<double,DIM>&));
00247 
00248 
00252     std::vector<double>& rGetPressures();
00253 
00257     std::vector<c_vector<double,DIM> >& rGetDeformedPosition();
00258 };
00259 
00260 #endif /*NONLINEARELASTICITYASSEMBLER_HPP_*/