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

Go to the documentation of this file.

#ifndef _GMSH_VER1_INPUT_BASSO_H_
#define _GMSH_VER1_INPUT_BASSO_H_

// I/O includes
#include <iostream>
#include <iomanip>
#include <fstream>

// STL includes
#include <vector>
#include <list>
#include <set>
#include <map>

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

namespace basso {

class GmshVer1Input : public FEInput {
        
public:
        

                GmshVer1Input( ) : FEInput( ) { }
                GmshVer1Input( const String &file ) : FEInput(file) {  }
        virtual ~GmshVer1Input() {}
        
        // MEMEBER FUNCTIONS
        
        virtual void ReadNodes( Array<Node> &nodes );
        
        virtual void AddElements( list<Element> &elem, int pid=-999 ) const;
        
        virtual void AddNodeSet( set<int> &nid, int pid ) const;

protected:
        
        BasisType GetBasisType( int etype ) const;
        
protected:

                
};


void GmshVer1Input::ReadNodes( Array<Node> &node )  
{

        // open msh file
        std::ifstream inFile;
        if ( OpenFile(inFile) ) 
                WarningMessage("GmshVer1Input::ReadNodes","could not open file");

        // look for node section 
        GotoSection(inFile,"$NOD");

        //  get the number of nodes
        int nn;
        inFile >> nn;

        if ( node.size()!=nn )
                resize( node, nn );

        // fill node
        for ( int i=0; i<nn; ++i ) 
        {
                int id;
                Numeric x, y, z;
                inFile >> id >> x >> y >> z;
                node[i]=Node(i,x,y,z);
                mNodeMap[id]=i;
        }
        CloseFile(inFile);
        
}

void GmshVer1Input::AddElements( list<Element> &elem, int pid ) const
{
        // open msh file
        std::ifstream inFile;
        if ( OpenFile(inFile) ) 
                WarningMessage("GmshVer1Input::ReadElements","could not open file");
        

        // look for element section 
        GotoSection(inFile,"$ELM");

        //  get the number of elements
        int ne;
        inFile >> ne;

        // fill element
        for ( int e=0; e<ne; ++e ) 
        {
                int eid, etype, nne, epid, egid, nid;
                map<int,int>::const_iterator mItr;
                inFile >> eid >> etype >> epid >> egid >> nne;
                iArray econn(nne);
                for ( int n=0; n<nne; ++n )  
                        inFile >> econn[n];
                if ( pid==epid || pid==-999 ) 
                {
                        for ( int n=0; n<nne; ++n ) 
                        {
                                mItr = mNodeMap.find( econn[n] );
                                econn[n] = mItr->second;   // renumber conn
                        }
                                
                        elem.push_back( Element( eid, epid, GetBasisType(etype), econn ) );
                }
        }
        CloseFile(inFile);
         
}

void GmshVer1Input::AddNodeSet( set<int> &nset, int pid ) const
{
        // open msh file
        std::ifstream inFile;
        if ( OpenFile(inFile) ) 
                WarningMessage("GmshVer1Input::ReadNodeSet","could not open file");
         

        // look for element section 
        GotoSection(inFile,"$ELM");     

        //  get the number of elements
        int ne;
        inFile >> ne;

        // fill element
        for ( int e=0; e<ne; ++e ) 
        {
                int eid, etype, nne, epid, egid;
                map<int,int>::const_iterator mItr;
                inFile >> eid >> etype >> epid >> egid >> nne;
                iArray econn(nne);
                for ( int n=0; n<nne; ++n )
                        inFile >> econn[n];
                if ( pid==epid )
                        for ( int n=0; n<nne; ++n )
                        {
                                mItr = mNodeMap.find( econn[n] );
                                nset.insert( mItr->second );
                        }
        }

        CloseFile(inFile);
}

        

BasisType GmshVer1Input::GetBasisType( int etype ) const
{
        switch (etype) {
                
                case 1:
                return kLINE2;
                
                case 2:
                return kTRIA3;
                
                case 3:
                return kQUAD4;
                
                case 4:
                return kTETRA4;
                
                case 5:
                return kHEXA8;
                
                case 6:
                return kPRISM6;
                
                case 7:
                return kPYRAMID5;
                
                case 8:
                return kLINE3;
                
                case 9:
                return kTRIA6;
                
                case 10:
                return kQUAD9;
                
                case 11:
                return kTETRA10;
                
                case 12:
                return kHEXA27;
                
                case 13:
                return kPRISM18;
                
                case 14:
                return kPYRAMID14;
                
                case 15:
                return kPOINT1;
                
                case 16:
                return kQUAD8;
                
                case 17:
                return kHEXA20;
                
                case 18:
                return kPRISM15;
                
                case 19:
                return kPYRAMID13;
                
                default:
                WarningMessage("BasisType GmshVer1Input::GetBasisType( int etype )","Unknown etype");
                return kNONE;
        }
}





} // end Basso namepsace 
#endif  

---