name change of CmiNumPes => CkNumPes
[charm.git] / src / conv-com / convcomlibmanager.C
1 /* Converse ComlibManager 
2    Enables communication library strategies to be called from converse code.
3    Also called by the Charm Comlibmanager.
4    
5    Stores a strategy table. Strategies can be inserted and accessed
6    from this table.
7
8    Sameer Kumar 28/03/04
9 */
10
11 #include "convcomlibmanager.h"
12 #include "routerstrategy.h"
13
14 int comm_debug;
15
16 CkpvDeclare(ConvComlibManager *, conv_comm_ptr);
17 CkpvDeclare(int, RecvdummyHandle);
18 CkpvExtern(int, RecvmsgHandle);
19 CkpvExtern(int, RecvCombinedShortMsgHdlrIdx);
20
21
22 ConvComlibManager::ConvComlibManager(): strategyTable(10){
23     nstrats = 0;
24 }
25
26 void ConvComlibManager::insertStrategy(Strategy *s) {
27     StrategyTableEntry &st = strategyTable[nstrats];
28     st.strategy = s;
29
30     s->setInstance(nstrats);
31     // if the strategy is pure converse or pure charm the following line is a
32     // duplication, but if a charm strategy embed a converse strategy it is
33     // necessary to set the instanceID in both
34     s->getConverseStrategy()->setInstance(nstrats);
35     nstrats ++;
36 }
37
38
39 void ConvComlibManager::insertStrategy(Strategy *s, int loc) {
40
41     //For now allow insertion of any location    
42     StrategyTableEntry &st = strategyTable[loc];
43
44     st.strategy = s;
45 }
46
47 //handler for dummy messages
48 void recv_dummy(void *msg){
49     ComlibPrintf("Received Dummy %d\n", CkMyPe());    
50     CmiFree(msg);
51 }
52
53
54 //An initialization routine which does prelimnary initialization of the 
55 //Converse commlib manager. Currently also initialized krishnans code
56 void initComlibManager(){ 
57     ConvComlibManager *conv_com = new ConvComlibManager();
58     
59     CkpvInitialize(ConvComlibManager *, conv_comm_ptr);
60     CkpvAccess(conv_comm_ptr) = conv_com;
61
62     comm_debug = 0;
63     ComlibPrintf("Init Call\n");
64
65     CkpvInitialize(int, RecvmsgHandle);
66     CkpvInitialize(int, RecvCombinedShortMsgHdlrIdx);
67     
68     CkpvInitialize(int, RecvdummyHandle);
69     CkpvAccess(RecvdummyHandle) = CkRegisterHandler((CmiHandler)recv_dummy);
70
71     PUPable_reg(Strategy);
72     PUPable_reg(RouterStrategy);
73     PUPable_reg(MessageHolder);
74 }
75  
76 Strategy *ConvComlibGetStrategy(int loc) {
77     //Calling converse strategy lets Charm++ strategies one strategy
78     //table entry but multiple layers of strategies (Charm on top of Converse).
79     return (CkpvAccess(conv_comm_ptr))->getStrategy(loc)->getConverseStrategy();
80 }
81
82 void ConvComlibRegisterStrategy(Strategy *s) {
83     (CkpvAccess(conv_comm_ptr))->insertStrategy(s);    
84 }
85
86 void ConvComlibScheduleDoneInserting(int loc) {
87     (* (CkpvAccess(conv_comm_ptr))->getStrategyTable())[loc].
88         call_doneInserting++;
89 }