File: TrajectoryServer.py

package info (click to toggle)
mmtk 2.7.9-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 11,788 kB
  • ctags: 6,600
  • sloc: python: 18,050; ansic: 12,400; makefile: 129; csh: 3
file content (70 lines) | stat: -rw-r--r-- 1,971 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# Pyro-based trajectory server
#
# Written by Konrad Hinsen
#

from TrajectoryInspector import TrajectoryInspector
from Scientific import N as Numeric # to enable pickling of arrays
import Pyro.core, Pyro.naming
import socket, sys

class PyroTrajectoryInspector(Pyro.core.ObjBase, TrajectoryInspector):

    def __init__(self, filename):
        Pyro.core.ObjBase.__init__(self)
        TrajectoryInspector.__init__(self, filename)

    def close(self):
        TrajectoryInspector.close(self)
        PyroDaemon.disconnect(self)


class TrajectoryServer(Pyro.core.ObjBase):

    def __init__(self, hostname, ip_address):
        self.hostname = hostname
        self.ip_address = ip_address
        self.files = {}
        self.exit = 0
        Pyro.core.ObjBase.__init__(self)

    def publishTrajectory(self, filename):
        inspector = PyroTrajectoryInspector(filename)
        self.files[filename] = inspector
        PyroDaemon.connect(inspector,
                           ip_address + ':MMTK:trajectory:' + filename)

    def unpublishTrajectory(self, filename):
        try:
            inspector = self.files[filename]
        except KeyError:
            raise OSError("file not published")
        del self.files[filename]
        inspector.close()

    def stop(self):
        for filename, inspector in self.files.items():
            del self.files[filename]
            inspector.close()
        PyroDaemon.disconnect(self)
        self.exit = 1

Pyro.core.initServer()
PyroDaemon = Pyro.core.Daemon()
hostname = socket.gethostname()
ip_address = socket.gethostbyname(hostname)

locator = Pyro.naming.NameServerLocator()
try:
    pyro_ns = locator.getNS()
except (PyroError,socket.error),x:
    pyro_ns = locator.getNS(host=hostname)
PyroDaemon.useNameServer(pyro_ns)

server = TrajectoryServer(hostname, ip_address)
PyroDaemon.connect(server, ip_address + ':MMTK:server')

while 1:
    PyroDaemon.handleRequests(10.)
    if server.exit:
        break