[300] | 1 | #ifndef __CONTEXT_CLIENT_HPP__ |
---|
| 2 | #define __CONTEXT_CLIENT_HPP__ |
---|
| 3 | |
---|
[591] | 4 | #include "xios_spl.hpp" |
---|
[300] | 5 | #include "buffer_out.hpp" |
---|
[697] | 6 | #include "buffer_in.hpp" |
---|
[300] | 7 | #include "buffer_client.hpp" |
---|
| 8 | #include "event_client.hpp" |
---|
[697] | 9 | #include "event_server.hpp" |
---|
[382] | 10 | #include "mpi.hpp" |
---|
[697] | 11 | #include "registry.hpp" |
---|
[300] | 12 | |
---|
[335] | 13 | namespace xios |
---|
[300] | 14 | { |
---|
[595] | 15 | class CContext; |
---|
[509] | 16 | |
---|
[512] | 17 | /*! |
---|
| 18 | \class CContextClient |
---|
| 19 | A context can be both on client and on server side. In order to differenciate the role of |
---|
| 20 | context on each side, e.x client sending events, server receiving and processing events, there is a need of |
---|
| 21 | concrete "context" classes for both sides. |
---|
| 22 | CContextClient processes and sends events from client to server where CContextServer receives these events |
---|
| 23 | and processes them. |
---|
| 24 | */ |
---|
[300] | 25 | class CContextClient |
---|
| 26 | { |
---|
| 27 | public: |
---|
[595] | 28 | // Contructor |
---|
| 29 | CContextClient(CContext* parent, MPI_Comm intraComm, MPI_Comm interComm, CContext* parentServer = 0); |
---|
[300] | 30 | |
---|
[595] | 31 | // Send event to server |
---|
| 32 | void sendEvent(CEventClient& event); |
---|
[1033] | 33 | bool sendTemporarilyBufferedEvent(); |
---|
[595] | 34 | void waitEvent(list<int>& ranks); |
---|
[300] | 35 | |
---|
[1033] | 36 | // Functions to set/get buffers |
---|
| 37 | bool getBuffers(const list<int>& serverList, const list<int>& sizeList, list<CBufferOut*>& retBuffers, bool nonBlocking = false); |
---|
[595] | 38 | void newBuffer(int rank); |
---|
| 39 | bool checkBuffers(list<int>& ranks); |
---|
| 40 | bool checkBuffers(void); |
---|
| 41 | void releaseBuffers(void); |
---|
[512] | 42 | |
---|
[595] | 43 | bool isServerLeader(void) const; |
---|
[988] | 44 | bool isServerNotLeader(void) const; |
---|
[595] | 45 | const std::list<int>& getRanksServerLeader(void) const; |
---|
[988] | 46 | const std::list<int>& getRanksServerNotLeader(void) const; |
---|
[512] | 47 | |
---|
[704] | 48 | bool isAttachedModeEnabled() const; |
---|
| 49 | |
---|
[1033] | 50 | bool hasTemporarilyBufferedEvent() const { return !tmpBufferedEvent.isEmpty(); }; |
---|
| 51 | |
---|
[595] | 52 | // Close and finalize context client |
---|
| 53 | void closeContext(void); |
---|
| 54 | void finalize(void); |
---|
[300] | 55 | |
---|
[917] | 56 | void setBufferSize(const std::map<int,StdSize>& mapSize, const std::map<int,StdSize>& maxEventSize); |
---|
[726] | 57 | |
---|
[512] | 58 | public: |
---|
[595] | 59 | CContext* context; //!< Context for client |
---|
[509] | 60 | |
---|
[595] | 61 | size_t timeLine; //!< Timeline of each event |
---|
[509] | 62 | |
---|
[595] | 63 | int clientRank; //!< Rank of current client |
---|
[300] | 64 | |
---|
[595] | 65 | int clientSize; //!< Size of client group |
---|
[300] | 66 | |
---|
[595] | 67 | int serverSize; //!< Size of server group |
---|
[300] | 68 | |
---|
[595] | 69 | MPI_Comm interComm; //!< Communicator of server group |
---|
[300] | 70 | |
---|
[595] | 71 | MPI_Comm intraComm; //!< Communicator of client group |
---|
[300] | 72 | |
---|
[595] | 73 | map<int,CClientBuffer*> buffers; //!< Buffers for connection to servers |
---|
[300] | 74 | |
---|
[512] | 75 | private: |
---|
| 76 | //! Mapping of server and buffer size for each connection to server |
---|
[595] | 77 | std::map<int,StdSize> mapBufferSize_; |
---|
[917] | 78 | //! Maximum number of events that can be buffered |
---|
| 79 | StdSize maxBufferedEvents; |
---|
[300] | 80 | |
---|
[1033] | 81 | struct { |
---|
| 82 | std::list<int> ranks, sizes; |
---|
| 83 | std::list<CBufferOut*> buffers; |
---|
| 84 | |
---|
| 85 | bool isEmpty() const { return ranks.empty(); }; |
---|
| 86 | void clear() { |
---|
| 87 | ranks.clear(); |
---|
| 88 | sizes.clear(); |
---|
| 89 | |
---|
| 90 | for (std::list<CBufferOut*>::iterator it = buffers.begin(); it != buffers.end(); it++) |
---|
| 91 | delete *it; |
---|
| 92 | |
---|
| 93 | buffers.clear(); |
---|
| 94 | }; |
---|
| 95 | } tmpBufferedEvent; //! Event temporarily buffered (used only on the server) |
---|
| 96 | |
---|
[512] | 97 | //! Context for server (Only used in attached mode) |
---|
| 98 | CContext* parentServer; |
---|
| 99 | |
---|
[595] | 100 | //! List of server ranks for which the client is leader |
---|
| 101 | std::list<int> ranksServerLeader; |
---|
| 102 | |
---|
[988] | 103 | //! List of server ranks for which the client is not leader |
---|
| 104 | std::list<int> ranksServerNotLeader; |
---|
| 105 | |
---|
[512] | 106 | public: // Some function should be removed in the future |
---|
[595] | 107 | // void registerEvent(CEventClient& event); |
---|
| 108 | // list<CBufferOut*> newEvent(CEventClient& event,list<int>& sizes); |
---|
| 109 | // bool locked; |
---|
| 110 | // set<int> connectedServer; |
---|
[512] | 111 | |
---|
[595] | 112 | }; |
---|
[512] | 113 | } |
---|
| 114 | |
---|
| 115 | #endif // __CONTEXT_CLIENT_HPP__ |
---|