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

Go to the documentation of this file.

#ifndef _FEINPUT_BASSO_H_
#define _FEINPUT_BASSO_H_

// std includes 
 
// Basso includes
#include "basso.h" 
#include "Input.h"
#include "Node.h"
#include "Element.h"

namespace basso {
        
        class FEInput : public Input 
        {
        public:
                FEInput( ) : Input ( )  { }
                
                FEInput( const String &file ) : Input(file) {  }

                virtual void ReadNodes( Array<Node> &nodes ) = 0;
                
                virtual void AddElements( list<Element> &elem, int pid=-999 ) const = 0;
                
                virtual void AddElements( list<Element> &elem, iArray &pids ) const;
                
                virtual void AddNodeSet( set<int> &nid, int sid ) const = 0;
                
                virtual void AddNodeSet( set<int> &nid, const iArray &sids ) const;
                
        protected:
                
                
                void SetNodeMap( const iArray &nids )
                {
                        for ( int i=0; i<vect_size(nids); ++i )
                                mNodeMap[ nids[i] ] = i;
                }
                
                void RenumberNodeIds( list<Element> &elem ) const 
                {
                        list<Element>::iterator eItr;
                        Element::iterator nItr;
                        map<int,int>::const_iterator mItr;
                        for ( eItr=elem.begin(); eItr!=elem.end(); ++eItr )
                                for ( nItr=eItr->begin(); nItr!=eItr->begin(); ++nItr )
                                {
                                        mItr = mNodeMap.find( *nItr );
                                        *nItr = mItr->second;
                                }
                }
                
                void RenumberNodeIds( set<int> &nids ) const 
                {
                        set<int> newSet;
                        set<int>::iterator nItr;
                        map<int,int>::const_iterator mItr;
                        for ( nItr=nids.begin(); nItr!=nids.end(); ++nItr )
                        {
                                mItr = mNodeMap.find( *nItr );
                                newSet.insert( mItr->second );
                        }
                        nids=newSet;
                }               
                
        protected:      
                map<int,int> mNodeMap;
                
        };
        
        void FEInput::AddElements( list<Element> &elem, iArray &pids ) const
        {
                iArray::const_iterator pItr;
                for ( pItr=pids.begin(); pItr!=pids.end(); ++pItr )
                        AddElements( elem, *pItr );
        }
        
        void FEInput::AddNodeSet( set<int> &nid, const iArray &sids ) const
        {
                iArray::const_iterator pItr;
                for ( pItr=sids.begin(); pItr!=sids.end(); ++pItr )
                        AddNodeSet( nid, *pItr );
        }
        
} // end namespace

#endif

---