feat(net_component): added early work on data factory concept
This commit is contained in:
@ -5,50 +5,82 @@ from .libs import umsgpack
|
|||||||
import time
|
import time
|
||||||
import random
|
import random
|
||||||
import struct
|
import struct
|
||||||
|
import collections
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
logging.basicConfig(level=logging.INFO)
|
logging.basicConfig(level=logging.INFO)
|
||||||
|
|
||||||
|
|
||||||
class RCFTranslation():
|
class RCFFactory(object):
|
||||||
def get(self, data):
|
|
||||||
"""
|
"""
|
||||||
local program > rcf
|
Abstract layer used to bridge external and inter
|
||||||
|
|
||||||
"""
|
"""
|
||||||
raise NotImplementedError
|
def init(self, data):
|
||||||
|
|
||||||
def set(self, data):
|
|
||||||
"""
|
"""
|
||||||
rcf > local program
|
set the RCFMessage pointer to local data
|
||||||
"""
|
"""
|
||||||
raise NotImplementedError
|
#Setup data accessor
|
||||||
|
# data.get = load_getter(data)
|
||||||
|
# data.set = load_setter(data)
|
||||||
class RCFMsgFactory():
|
|
||||||
"""
|
|
||||||
Abstract basic data bridge
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
pass
|
pass
|
||||||
raise NotImplementedError
|
# TODO: Setup local pointer
|
||||||
|
|
||||||
def load(self, data):
|
pass
|
||||||
|
|
||||||
|
def load_getter(self, data):
|
||||||
"""
|
"""
|
||||||
local program > rcf
|
local program > rcf
|
||||||
|
|
||||||
"""
|
"""
|
||||||
raise NotImplementedError
|
pass
|
||||||
|
|
||||||
def unload(self, data):
|
def load_setter(self, data):
|
||||||
"""
|
"""
|
||||||
rcf > local program
|
rcf > local program
|
||||||
"""
|
"""
|
||||||
raise NotImplementedError
|
pass
|
||||||
|
|
||||||
|
def apply(self,data):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def diff(self, data):
|
||||||
|
"""
|
||||||
|
Verify data integrity
|
||||||
|
"""
|
||||||
|
pass
|
||||||
|
|
||||||
|
class RCFStore(collections.MutableMapping,dict):
|
||||||
|
def __init__(self, custom_factory=RCFFactory()):
|
||||||
|
super().__init__()
|
||||||
|
self.factory = custom_factory
|
||||||
|
print("Init a store")
|
||||||
|
|
||||||
|
def __getitem__(self,key):
|
||||||
|
# if self[key]:
|
||||||
|
# Get dict data from external
|
||||||
|
# print("getting item from store")
|
||||||
|
return dict.__getitem__(self,key)
|
||||||
|
|
||||||
|
def __setitem__(self, key, value):
|
||||||
|
#test
|
||||||
|
# TODO: test with try - except KeyError
|
||||||
|
if key in self:
|
||||||
|
# Set dict data from external
|
||||||
|
dict.__setitem__(self,key,value)
|
||||||
|
else:
|
||||||
|
print("need init")
|
||||||
|
self.factory.init(value.body)
|
||||||
|
dict.__setitem__(self,key,value)
|
||||||
|
|
||||||
|
def __delitem__(self, key):
|
||||||
|
dict.__delitem__(self,key)
|
||||||
|
def __iter__(self):
|
||||||
|
return dict.__iter__(self)
|
||||||
|
def __len__(self):
|
||||||
|
return dict.__len__(self)
|
||||||
|
def __contains__(self, x):
|
||||||
|
return dict.__contains__(self,x)
|
||||||
|
|
||||||
# TODO: Add message time and author stamp for reliabilty
|
|
||||||
class RCFMessage(object):
|
class RCFMessage(object):
|
||||||
"""
|
"""
|
||||||
Message is formatted on wire as 2 frames:
|
Message is formatted on wire as 2 frames:
|
||||||
@ -63,17 +95,16 @@ class RCFMessage(object):
|
|||||||
mtype = None # data mtype (string)
|
mtype = None # data mtype (string)
|
||||||
body = None # data blob
|
body = None # data blob
|
||||||
|
|
||||||
def __init__(self, key=None, id=None, mtype=None, body=None, factory=None):
|
def __init__(self, key=None, id=None, mtype=None, body=None, pointer=None):
|
||||||
self.key = key
|
self.key = key
|
||||||
self.mtype = mtype
|
self.mtype = mtype
|
||||||
self.body = body
|
self.body = body
|
||||||
self.id = id
|
self.id = id
|
||||||
|
self.pointer = pointer
|
||||||
self.factory = factory
|
self.load = None
|
||||||
|
self.unload = None
|
||||||
def store(self, dikt):
|
def store(self, dikt):
|
||||||
"""Store me in a dict if I have anything to store"""
|
"""Store me in a dict if I have anything to store"""
|
||||||
# this seems weird to check, but it's what the C example does
|
|
||||||
# this currently erasing old value
|
# this currently erasing old value
|
||||||
if self.key is not None and self.body is not None:
|
if self.key is not None and self.body is not None:
|
||||||
dikt[self.key] = self
|
dikt[self.key] = self
|
||||||
@ -115,9 +146,8 @@ class RCFMessage(object):
|
|||||||
data=data,
|
data=data,
|
||||||
))
|
))
|
||||||
|
|
||||||
|
|
||||||
class Client():
|
class Client():
|
||||||
def __init__(self, context=zmq.Context(), id="default", on_recv=None, on_post_init=None, is_admin=False):
|
def __init__(self, context=zmq.Context(), id="default", on_recv=None, on_post_init=None, is_admin=False, factory=RCFFactory()):
|
||||||
self.is_admin = is_admin
|
self.is_admin = is_admin
|
||||||
|
|
||||||
# 0MQ vars
|
# 0MQ vars
|
||||||
@ -134,7 +164,7 @@ class Client():
|
|||||||
# Main client loop registration
|
# Main client loop registration
|
||||||
self.task = asyncio.ensure_future(self.main())
|
self.task = asyncio.ensure_future(self.main())
|
||||||
|
|
||||||
self.property_map = {}
|
self.property_map = RCFStore()
|
||||||
|
|
||||||
logger.info("{} client initialized".format(id))
|
logger.info("{} client initialized".format(id))
|
||||||
|
|
||||||
@ -186,6 +216,7 @@ class Client():
|
|||||||
|
|
||||||
for f in self.on_post_init:
|
for f in self.on_post_init:
|
||||||
f()
|
f()
|
||||||
|
|
||||||
logger.info("{} client running".format(id))
|
logger.info("{} client running".format(id))
|
||||||
|
|
||||||
# Main loop
|
# Main loop
|
||||||
@ -196,8 +227,10 @@ class Client():
|
|||||||
if self.pull_sock in socks:
|
if self.pull_sock in socks:
|
||||||
rcfmsg = RCFMessage.recv(self.pull_sock)
|
rcfmsg = RCFMessage.recv(self.pull_sock)
|
||||||
|
|
||||||
|
# if rcfmsg.pointer:
|
||||||
rcfmsg.store(self.property_map)
|
rcfmsg.store(self.property_map)
|
||||||
|
|
||||||
|
|
||||||
for f in self.on_recv:
|
for f in self.on_recv:
|
||||||
f(rcfmsg)
|
f(rcfmsg)
|
||||||
else:
|
else:
|
||||||
@ -226,7 +259,7 @@ class Server():
|
|||||||
self.collector_sock = None
|
self.collector_sock = None
|
||||||
self.poller = None
|
self.poller = None
|
||||||
|
|
||||||
self.property_map = {}
|
self.property_map = RCFStore()
|
||||||
self.id = id
|
self.id = id
|
||||||
self.bind_ports()
|
self.bind_ports()
|
||||||
# Main client loop registration
|
# Main client loop registration
|
||||||
@ -261,16 +294,16 @@ class Server():
|
|||||||
logger.info("{} server launched".format(id))
|
logger.info("{} server launched".format(id))
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
# TODO: Listener on anoter process linked with PAIR/PAIR ?
|
# Non blocking poller
|
||||||
socks = dict(self.poller.poll(1))
|
socks = dict(self.poller.poll(1))
|
||||||
|
|
||||||
# Snapshot system for late join
|
# Snapshot system for late join (Server - Client)
|
||||||
if self.request_sock in socks:
|
if self.request_sock in socks:
|
||||||
msg = self.request_sock.recv_multipart(zmq.DONTWAIT)
|
msg = self.request_sock.recv_multipart(zmq.DONTWAIT)
|
||||||
|
|
||||||
identity = msg[0]
|
identity = msg[0]
|
||||||
request = msg[1]
|
request = msg[1]
|
||||||
print("reveived snapshot request from {}".format(identity.decode()))
|
|
||||||
if request == b"SNAPSHOT_REQUEST":
|
if request == b"SNAPSHOT_REQUEST":
|
||||||
pass
|
pass
|
||||||
else:
|
else:
|
||||||
@ -287,13 +320,15 @@ class Server():
|
|||||||
self.request_sock.send(identity, zmq.SNDMORE)
|
self.request_sock.send(identity, zmq.SNDMORE)
|
||||||
msg_end_snapshot.send(self.request_sock)
|
msg_end_snapshot.send(self.request_sock)
|
||||||
logger.info("done")
|
logger.info("done")
|
||||||
|
|
||||||
|
# Regular update routing (Clients / Client)
|
||||||
elif self.collector_sock in socks:
|
elif self.collector_sock in socks:
|
||||||
msg = RCFMessage.recv(self.collector_sock)
|
msg = RCFMessage.recv(self.collector_sock)
|
||||||
# Update all clients
|
# Update all clients
|
||||||
msg.store(self.property_map)
|
msg.store(self.property_map)
|
||||||
msg.send(self.pub_sock)
|
msg.send(self.pub_sock)
|
||||||
else:
|
else:
|
||||||
await asyncio.sleep(0.016)
|
await asyncio.sleep(0.001)
|
||||||
|
|
||||||
def stop(self):
|
def stop(self):
|
||||||
logger.debug("Stopping server")
|
logger.debug("Stopping server")
|
||||||
|
@ -19,9 +19,10 @@ NATIVE_TYPES = (
|
|||||||
bpy.types.StringProperty,
|
bpy.types.StringProperty,
|
||||||
)
|
)
|
||||||
|
|
||||||
VECTOR_TYPES = (
|
class RNAFactory(net_components.RCFFactory):
|
||||||
'Vector'
|
def load_getter(self,data):
|
||||||
)
|
get,set = match_supported_types(data)
|
||||||
|
|
||||||
|
|
||||||
def on_scene_evalutation(scene):
|
def on_scene_evalutation(scene):
|
||||||
# TODO: viewer representation
|
# TODO: viewer representation
|
||||||
@ -35,21 +36,6 @@ def randomStringDigits(stringLength=6):
|
|||||||
return ''.join(random.choice(lettersAndDigits) for i in range(stringLength))
|
return ''.join(random.choice(lettersAndDigits) for i in range(stringLength))
|
||||||
|
|
||||||
|
|
||||||
class VectorTypeTranslation(net_components.RCFTranslation):
|
|
||||||
def set(self, data):
|
|
||||||
"""
|
|
||||||
local program > rcf
|
|
||||||
|
|
||||||
"""
|
|
||||||
return [data.x, data.y, data.z]
|
|
||||||
|
|
||||||
def get(self, data):
|
|
||||||
"""
|
|
||||||
rcf > local program
|
|
||||||
"""
|
|
||||||
return mathutils.Vector((data[0], data[1], data[2]))
|
|
||||||
|
|
||||||
|
|
||||||
def match_supported_types(value):
|
def match_supported_types(value):
|
||||||
type_factory = None
|
type_factory = None
|
||||||
|
|
||||||
@ -180,6 +166,7 @@ recv_callbacks = [patch_scene]
|
|||||||
post_init_callbacks = [refresh_window]
|
post_init_callbacks = [refresh_window]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class session_join(bpy.types.Operator):
|
class session_join(bpy.types.Operator):
|
||||||
bl_idname = "session.join"
|
bl_idname = "session.join"
|
||||||
bl_label = "join"
|
bl_label = "join"
|
||||||
@ -319,7 +306,7 @@ class session_stop(bpy.types.Operator):
|
|||||||
|
|
||||||
|
|
||||||
class session_settings(bpy.types.PropertyGroup):
|
class session_settings(bpy.types.PropertyGroup):
|
||||||
username = bpy.props.StringProperty(name="Username", default="user")
|
username = bpy.props.StringProperty(name="Username", default="user_{}".format(randomStringDigits()))
|
||||||
ip = bpy.props.StringProperty(name="localhost")
|
ip = bpy.props.StringProperty(name="localhost")
|
||||||
port = bpy.props.IntProperty(name="5555")
|
port = bpy.props.IntProperty(name="5555")
|
||||||
buffer = bpy.props.StringProperty(name="None")
|
buffer = bpy.props.StringProperty(name="None")
|
||||||
|
Reference in New Issue
Block a user