mesh_field_function.h

Go to the documentation of this file.

/*
 * MAST: Multidisciplinary-design Adaptation and Sensitivity Toolkit
 * Copyright (C) 2013-2019  Manav Bhatia
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

#ifndef __mast__mesh_field_function__
#define __mast__mesh_field_function__

// MAST includes
#include "base/field_function_base.h"


// libMesh includes
#include "libmesh/numeric_vector.h"
#include "libmesh/mesh_function.h"
#include "libmesh/system.h"



namespace MAST {

    // Forward declerations
    class SystemInitialization;
    
    
    class MeshFieldFunction:
    public MAST::FieldFunction<RealVectorX> {
        
    public:
        MeshFieldFunction(MAST::SystemInitialization& sys,
                          const std::string& nm);

        
        MeshFieldFunction(libMesh::System& sys, const std::string& nm);

        
        virtual ~MeshFieldFunction();

       
        virtual void operator() (const libMesh::Point& p,
                                 const Real t,
                                 RealVectorX& v) const;
        
        virtual void gradient(const libMesh::Point& p,
                              const Real t,
                              RealMatrixX& g) const;

        virtual void perturbation (const libMesh::Point& p,
                                   const Real t,
                                   RealVectorX& v) const;

        
        virtual void derivative (const MAST::FunctionBase& f,
                                 const libMesh::Point& p,
                                 const Real t,
                                 RealVectorX& v) const;
        
        
        void init(const libMesh::NumericVector<Real>& sol,
                  const libMesh::NumericVector<Real>* dsol = nullptr);

        
        libMesh::MeshFunction& get_function() {
            
            libmesh_assert(_function);
            return *_function;
        }

        libMesh::MeshFunction& get_perturbed_function() {
            
            libmesh_assert(_perturbed_function);
            return *_perturbed_function;
        }

        
        virtual void set_element_quadrature_point_solution(RealVectorX& sol);

        
        virtual void clear_element_quadrature_point_solution();

        
        void clear();

    protected:

        bool _use_qp_sol;
        
        
        RealVectorX _qp_sol;
        
        libMesh::System* _sys;
        
        libMesh::NumericVector<Real> *_sol, *_dsol;
        
        libMesh::MeshFunction *_function, *_perturbed_function;
    };
}

#endif // __mast__mesh_field_function__