91 lines
2.8 KiB
C++
91 lines
2.8 KiB
C++
/*
|
|
* Copyright (C) 2011 Thorsten Liebig (Thorsten.Liebig@gmx.de)
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program 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 General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef OPERATOR_MPI_H
|
|
#define OPERATOR_MPI_H
|
|
|
|
#include "operator_sse_compressed.h"
|
|
|
|
class Operator_MPI : public Operator_SSE_Compressed
|
|
{
|
|
friend class Engine_MPI;
|
|
public:
|
|
//! Create a new operator
|
|
static Operator_MPI* New();
|
|
virtual ~Operator_MPI();
|
|
|
|
bool GetMPIEnabled() const {return m_MPI_Enabled;}
|
|
|
|
virtual void SetBoundaryCondition(int* BCs);
|
|
|
|
virtual Engine* CreateEngine() const;
|
|
|
|
virtual void SetTag(int tag) {m_MyTag=tag;}
|
|
|
|
//! Set the number of splits for a given direction. This also defines the size of the process table. \sa SetProcessTable
|
|
virtual void SetSplitNumbers(int ny, unsigned int splits) {m_SplitNumber[ny]=splits;}
|
|
//! Set the table containing a list of all MPI rank ID's and there mesh affiliation. \sa SetProcessTablePosition
|
|
virtual void SetProcessTable(unsigned int*** procTable) {m_ProcTable=procTable;}
|
|
//! Save the position for this rank in the process table. \sa SetProcessTable
|
|
virtual void SetProcessTablePosition(int ny, unsigned int pos) {m_ProcTablePos[ny]=pos;}
|
|
|
|
virtual void SetNeighborUp(int ny, int id);
|
|
virtual void SetNeighborDown(int ny, int id);
|
|
|
|
//! Set the lower original mesh position
|
|
virtual void SetSplitPos(int ny, unsigned int pos) {m_SplitPos[ny]=pos;}
|
|
virtual void SetOriginalMesh(CSRectGrid* orig_Mesh);
|
|
|
|
virtual unsigned int GetNumberOfLines(int ny, bool fullMesh=false) const;
|
|
|
|
virtual void AddExtension(Operator_Extension* op_ext);
|
|
|
|
protected:
|
|
Operator_MPI();
|
|
bool m_MPI_Enabled;
|
|
virtual void Init();
|
|
void Delete();
|
|
virtual void Reset();
|
|
|
|
virtual double CalcTimestep();
|
|
|
|
unsigned int m_MyID;
|
|
unsigned int m_NumProc;
|
|
int m_MyTag;
|
|
char* m_Processor_Name;
|
|
|
|
//the up and down neighbors, -1 if non for the given direction
|
|
int m_NeighborUp[3];
|
|
int m_NeighborDown[3];
|
|
|
|
unsigned int m_SplitNumber[3];
|
|
unsigned int m_ProcTablePos[3];
|
|
unsigned int*** m_ProcTable;
|
|
|
|
double* m_OrigDiscLines[3];
|
|
unsigned int m_OrigNumLines[3];
|
|
unsigned int m_SplitPos[3];
|
|
|
|
string PrependRank(string name);
|
|
|
|
virtual void DumpOperator2File(string filename);
|
|
virtual void DumpMaterial2File(string filename);
|
|
virtual void DumpPEC2File( string filename, unsigned int *range = NULL );
|
|
};
|
|
|
|
#endif // OPERATOR_MPI_H
|