Allow some methods of charm object to be remotely callable

The charm object is present in every process and represents the
Charm4py runtime. This patch effectively turns it into a chare, by
creating a Group of CharmRemote objects which act as an interface
to the charm object in every process. The new chare can be
accessed via the 'thisProxy' attribute of 'charm'.

Author: juanjgalvez

charmpy/charm.py

@@ -118,6 +118,7 @@ def buildMainchare(self, onPe, objPtr, ep, args):
         assert cid not in self.chares, "Chare " + str(cid) + " already instantiated"
         em = self.entryMethods[ep]
         assert em.isCtor, "Specified mainchare entry method is not constructor"
+        self._createInternalChares()
         obj = object.__new__(em.C)  # create object but don't call __init__
         Mainchare.initMember(obj, cid)
         super(em.C, obj).__init__()  # call Chare class __init__ first
@@ -403,6 +404,12 @@ def register(self, C, collections=(GROUP, ARRAY)):
             self.registerAs(C, charm_type_id)
         self.register_order.append(C)
 
+    def _registerInternalChares(self):
+        self.register(CharmRemote, (GROUP,))
+
+    def _createInternalChares(self):
+        Group(CharmRemote)
+
     def start(self, entry, classes=[], modules=[]):
         """
         Start Charmpy program.
@@ -429,6 +436,8 @@ def start(self, entry, classes=[], modules=[]):
         if "++quiet" in sys.argv:
             Options.QUIET = True
 
+        self._registerInternalChares()
+
         if hasattr(entry, 'mro') and Chare in entry.mro():
             if entry.__init__.__code__.co_argcount != 2:
                 raise CharmPyError("Mainchare constructor must have only one parameter")
@@ -612,6 +621,18 @@ def LBTurnInstrumentOff(self):
         self.lib.LBTurnInstrumentOff()
 
 
+class CharmRemote(Chare):
+
+    def __init__(self):
+        charm.thisProxy = self.thisProxy
+
+    def exit(self, exit_code=0):
+        charm.exit(exit_code)
+
+    def myPe(self):
+        return charm.myPe()
+
+
 def load_charm_library(charm):
     args = sys.argv
     libcharm_path = os.path.join(os.path.dirname(__file__), '.libs')

test_config.json

@@ -1,4 +1,8 @@
 [
+    {
+        "force_min_processes": 4,
+        "path": "tests/charm_remote.py"
+    },
     {
         "force_min_processes": 4,
         "path": "tests/array_maps/test1.py"

tests/array_maps/test1.py

@@ -16,10 +16,7 @@ class MyChare(Chare):
 
     def __init__(self, last):
         assert charm.myPe() == index_to_pe(self.thisIndex), "ArrayMap failed"
-        if last: self.contribute(None, None, self.thisProxy.done)
-
-    def done(self):
-        exit()
+        if last: self.contribute(None, None, charm.thisProxy[0].exit)
 
 
 def main(args):

tests/charm_remote.py

@@ -0,0 +1,30 @@
+from charmpy import charm, Chare, Group, threaded
+import random
+
+
+class Controller(Chare):
+
+    @threaded
+    def start(self):
+        # print('Controller running on PE', charm.myPe())
+        for i in range(charm.numPes()):
+            assert i == charm.thisProxy[i].myPe(ret=True).get()
+
+        g = Group(Test)
+
+
+class Test(Chare):
+
+    def __init__(self):
+        self.contribute(None, None, charm.thisProxy[0].exit)
+
+
+def main(args):
+    if charm.numPes() > 1:
+        pes = list(range(1, charm.numPes()))
+    else:
+        pes = [0]
+    Chare(Controller, onPE=random.choice(pes)).start()
+
+
+charm.start(main)

tests/migration/chare_migration.py

@@ -22,10 +22,7 @@ def test(self):
         if self.thisIndex == (0,):
             print("Test called on PE ", charm.myPe())
         assert charm.myPe() == self.toPe
-        self.contribute(None, None, self.thisProxy[0].done)
-
-    def done(self):
-        exit()
+        self.contribute(None, None, charm.thisProxy[0].exit)
 
     def start(self):
         """

tests/migration/test_migrate.py

@@ -18,9 +18,6 @@ def arrayElemsCreated(self, home_pes):
         for i in range(len(home_pes)): arrayElemHomeMap[i] = home_pes[i]
         self.contribute(None, None, ro.array.start)
 
-    def done(self):
-        exit()
-
 
 class Test(Chare):
 
@@ -40,7 +37,7 @@ def start(self):
         for i in range(work): A += 1.33
         self.iteration += 1
         if self.iteration == MAX_ITER:
-            self.contribute(None, None, ro.controllers[0].done)
+            self.contribute(None, None, charm.thisProxy[0].exit)
         elif self.iteration % 20 == 0:
             self.AtSync()
         else: