openEMS/FDTD/engine_multithread.h

85 lines
2.3 KiB
C++

/*
* Copyright (C) 2010 Sebastian Held (sebastian.held@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 ENGINE_MULTITHREAD_H
#define ENGINE_MULTITHREAD_H
#include "operator.h"
#include "engine.h"
#include <boost/thread.hpp>
class Engine_Multithread : public Engine
{
friend class Processing;
friend class ProcessVoltage;
friend class ProcessCurrent;
friend class ProcessFields;
friend class ProcessFieldsTD;
friend class thread;
friend class thread_e_excitation;
public:
static Engine_Multithread* createEngine(Operator* op);
virtual ~Engine_Multithread();
virtual void Init();
virtual void Reset();
//!Iterate a number of timesteps
virtual bool IterateTS(unsigned int iterTS);
virtual unsigned int GetNumberOfTimesteps() {return m_numTS_times_threads / m_numThreads;}
protected:
Engine_Multithread(Operator* op);
boost::thread_group m_thread_group;
boost::barrier *m_barrier1, *m_barrier2, *m_barrier3, *m_startBarrier, *m_stopBarrier;
volatile unsigned int m_iterTS;
volatile unsigned int m_numTS_times_threads; //!< numTS times the number of worker threads
unsigned int m_numThreads; //!< number of worker threads
};
class thread {
public:
thread( Engine_Multithread* ptr, unsigned int start, unsigned int stop );
void operator()();
protected:
unsigned int m_start, m_stop;
volatile bool m_stopThread;
Engine_Multithread *m_enginePtr;
Operator *Op;
FDTD_FLOAT**** volt;
FDTD_FLOAT**** curr;
};
class thread_e_excitation {
public:
thread_e_excitation( Engine_Multithread* ptr);
void operator()();
protected:
volatile bool m_stopThread;
Engine_Multithread *m_enginePtr;
Operator *Op;
FDTD_FLOAT**** volt;
FDTD_FLOAT**** curr;
};
#endif // ENGINE_MULTITHREAD_H