/Users/jack/Code/basso_dev/inc/DirichletBC.h

Go to the documentation of this file.

#ifndef _DIRICHLET_BC_BASSO_H_
#define _DIRICHLET_BC_BASSO_H_

// std includes
#include <list>

// Basso includes
#include "basso.h"
#include "DofMap.h"

namespace basso {


class DirichletBC {

public:
        
        // PUBLIC DEFS
        
        // CONSTRUCTORS 
        DirichletBC( const set<int> &nodeids, const Array<Dof> &dofs, Numeric sf=0.0 ) 
                {
                        fNodeIds=&nodeids;
                        mScaleFactor=sf; 
                        mDofs=dofs; 
                }
                
        DirichletBC( const set<int> &nodeids, const Dof &ldof, Numeric sf=0.0 ) : mDofs(1)
                { 
                        fNodeIds=&nodeids;
                        mScaleFactor=sf;
                        mDofs[0]=ldof;
                }
        

        virtual ~DirichletBC() {  }
        
        // ACCESSORS
        
        // MEMBER FUNCTIONS
        virtual  void AddFixedDofs( const DofMap &dofmap, list<int> &fdofs, list<Numeric> &fvals ) const;
        
        virtual  void AddFixedDofs( const DofMap &dofmap, list<int> &fdofs ) const;
        
protected:
        
        virtual Numeric DofValue() const { return mScaleFactor; }

protected:

        const set<int> *fNodeIds;
        Numeric mScaleFactor;
        Array<Dof> mDofs;
        
};

void DirichletBC::AddFixedDofs( const DofMap &dofmap, list<int> &fdofs, list<Numeric> &fvals ) const
{
        set<int>::const_iterator nItr;
        Array<Dof>::const_iterator dofItr;
        for ( nItr=fNodeIds->begin(); nItr!=fNodeIds->end(); ++nItr )
                for ( dofItr=mDofs.begin(); dofItr!=mDofs.end(); ++dofItr )
                {
                        fdofs.push_back( dofmap.GlobalDof( *nItr, *dofItr ) );
                        fvals.push_back( DofValue() );
                }
}

void DirichletBC::AddFixedDofs( const DofMap &dofmap, list<int> &fdofs ) const
{
        set<int>::const_iterator nItr;
        Array<Dof>::const_iterator dofItr;
        for ( nItr=fNodeIds->begin(); nItr!=fNodeIds->end(); ++nItr )
                for ( dofItr=mDofs.begin(); dofItr!=mDofs.end(); ++dofItr )
                        fdofs.push_back( dofmap.GlobalDof( *nItr, *dofItr ) );
}


} // end namespace

#endif

---