501 lines
14 KiB
Python
Raw Normal View History

2019-02-11 17:02:25 +01:00
import logging
2019-02-22 16:46:35 +01:00
import random
import string
2019-03-25 14:56:09 +01:00
import time
2019-03-27 19:11:08 +01:00
import asyncio
2019-03-28 17:14:40 +01:00
import queue
2019-04-03 15:07:35 +02:00
from operator import itemgetter
2019-04-08 17:01:02 +02:00
import subprocess
2019-04-04 18:02:51 +02:00
import uuid
import bgl
import blf
2019-03-25 14:56:09 +01:00
import bpy
2019-04-08 17:01:02 +02:00
import os
import gpu
2019-03-25 14:56:09 +01:00
import mathutils
from bpy_extras import view3d_utils
from gpu_extras.batch import batch_for_shader
2019-02-11 17:02:25 +01:00
2019-04-11 14:39:31 +02:00
from . import client, ui, draw, helpers
from .libs import umsgpack
2019-03-25 14:56:09 +01:00
2019-02-11 17:02:25 +01:00
logger = logging.getLogger(__name__)
2019-04-10 17:01:21 +02:00
client_instance = None
2019-04-11 14:39:31 +02:00
client_keys = None
2019-02-11 15:48:07 +01:00
server = None
context = None
2019-04-01 12:28:13 +02:00
drawer = None
2019-04-01 11:45:16 +02:00
update_list = {}
2019-03-25 23:31:44 +01:00
SUPPORTED_DATABLOCKS = ['collections', 'meshes', 'objects',
2019-04-01 11:45:16 +02:00
'materials', 'textures', 'lights', 'cameras', 'actions', 'armatures', 'grease_pencils']
2019-04-11 14:39:31 +02:00
SUPPORTED_TYPES = [ 'Material',
'Texture', 'Light', 'Camera','Mesh', 'Grease Pencil', 'Object', 'Action', 'Armature','Collection', 'Scene']
2019-04-10 18:01:55 +02:00
2019-03-25 14:56:09 +01:00
# UTILITY FUNCTIONS
2019-03-25 23:31:44 +01:00
2019-03-25 14:56:09 +01:00
def clean_scene(elements=SUPPORTED_DATABLOCKS):
for datablock in elements:
datablock_ref = getattr(bpy.data, datablock)
for item in datablock_ref:
datablock_ref.remove(item)
2019-02-22 16:46:35 +01:00
def randomStringDigits(stringLength=6):
"""Generate a random string of letters and digits """
lettersAndDigits = string.ascii_letters + string.digits
return ''.join(random.choice(lettersAndDigits) for i in range(stringLength))
2019-04-01 11:45:16 +02:00
def randomColor():
r = random.random()
v = random.random()
b = random.random()
2019-04-01 11:45:16 +02:00
return [r, v, b]
2019-02-25 18:05:46 +01:00
2019-02-22 16:46:35 +01:00
def refresh_window():
import bpy
bpy.ops.wm.redraw_timer(type='DRAW_WIN_SWAP', iterations=1)
2019-02-11 15:48:07 +01:00
2019-04-15 16:21:50 +02:00
2019-04-10 17:01:21 +02:00
def upload_client_instance_position():
global client_instance
2019-04-01 11:45:16 +02:00
username = bpy.context.scene.session_settings.username
2019-04-10 17:01:21 +02:00
if client_instance:
key = "Client/{}".format(username)
# key = "Object/Cube"
print(key)
try:
current_coords = draw.get_client_view_rect()
client = client_instance.get(key)
# print((client[0][1][b'location']))
# if data is None:
# data = {}
# data['location'] = current_coords
# color = bpy.context.scene.session_settings.client_instance_color
# data['color'] = (color.r, color.g, color.b, 1)
# client_instance.push_update(key, 'client_instance', data)
if current_coords != client[0][1][b'location']:
print(current_coords)
print(client[0][1][b'location'])
2019-04-15 17:08:05 +02:00
# client[0][1][b'color'] = [0,1,1,1]
client[0][1][b'location'] = current_coords
client_instance.set(key, client[0][1])
except:
pass
2019-04-15 16:21:50 +02:00
def update_selected_object(context):
2019-04-10 17:01:21 +02:00
global client_instance
session = bpy.context.scene.session_settings
# Active object bounding box
if len(context.selected_objects) > 0:
if session.active_object is not context.selected_objects[0] or session.active_object.is_evaluated:
session.active_object = context.selected_objects[0]
2019-04-10 17:01:21 +02:00
key = "net/objects/{}".format(client_instance.id.decode())
data = {}
2019-04-10 17:01:21 +02:00
data['color'] = [session.client_instance_color.r,
session.client_instance_color.g, session.client_instance_color.b]
data['object'] = session.active_object.name
2019-04-10 17:01:21 +02:00
client_instance.push_update(
key, 'client_instanceObject', data)
2019-04-03 15:07:35 +02:00
return True
elif len(context.selected_objects) == 0 and session.active_object:
session.active_object = None
data = {}
2019-04-10 17:01:21 +02:00
data['color'] = [session.client_instance_color.r,
session.client_instance_color.g, session.client_instance_color.b]
data['object'] = None
2019-04-10 17:01:21 +02:00
key = "net/objects/{}".format(client_instance.id.decode())
client_instance.push_update(key, 'client_instanceObject', data)
2019-04-01 11:45:16 +02:00
2019-04-03 15:07:35 +02:00
return True
return False
2019-04-01 15:41:28 +02:00
2019-04-11 14:39:31 +02:00
def init_datablocks():
global client_instance
for datatype in SUPPORTED_TYPES:
for item in getattr(bpy.data,helpers.CORRESPONDANCE[datatype]):
key = "{}/{}".format(datatype,item.name)
2019-04-11 15:54:15 +02:00
print(key)
2019-04-11 14:39:31 +02:00
client_instance.set(key)
2019-03-25 14:56:09 +01:00
2019-02-22 16:46:35 +01:00
2019-04-17 11:01:01 +02:00
def default_tick():
2019-04-15 16:21:50 +02:00
bpy.ops.session.refresh()
# global client_instance
2019-04-11 18:07:59 +02:00
# if not client_instance.queue.empty():
# update = client_instance.queue.get()
# helpers.load(update[0],update[1])
2019-04-11 15:54:15 +02:00
2019-04-15 16:21:50 +02:00
return 0.5
def draw_tick():
# drawing
2019-04-15 16:21:50 +02:00
global drawer
2019-04-15 16:21:50 +02:00
drawer.draw()
# Upload
2019-04-10 17:01:21 +02:00
upload_client_instance_position()
return 1
def register_ticks():
# REGISTER Updaters
bpy.app.timers.register(draw_tick)
2019-04-17 11:01:01 +02:00
2019-04-02 18:07:59 +02:00
bpy.app.timers.register(default_tick)
2019-04-04 18:02:51 +02:00
def unregister_ticks():
# REGISTER Updaters
2019-04-15 16:21:50 +02:00
global drawer
drawer.unregister_handlers()
bpy.app.timers.unregister(draw_tick)
2019-04-17 11:01:01 +02:00
2019-04-02 18:07:59 +02:00
bpy.app.timers.unregister(default_tick)
2019-03-27 19:11:08 +01:00
# OPERATORS
2019-02-11 17:31:03 +01:00
class session_join(bpy.types.Operator):
2019-03-25 14:56:09 +01:00
2019-02-11 17:31:03 +01:00
bl_idname = "session.join"
bl_label = "join"
2019-02-11 15:48:07 +01:00
bl_description = "connect to a net server"
bl_options = {"REGISTER"}
@classmethod
def poll(cls, context):
return True
def execute(self, context):
2019-04-10 17:01:21 +02:00
global client_instance, drawer
2019-02-11 15:48:07 +01:00
2019-02-22 16:46:35 +01:00
net_settings = context.scene.session_settings
2019-03-25 14:56:09 +01:00
# Scene setup
2019-04-10 18:26:39 +02:00
if net_settings.session_mode == "CONNECT" and net_settings.clear_scene:
clean_scene()
2019-02-25 18:05:46 +01:00
2019-03-25 14:56:09 +01:00
# Session setup
2019-02-22 16:46:35 +01:00
if net_settings.username == "DefaultUser":
2019-02-25 18:05:46 +01:00
net_settings.username = "{}_{}".format(
net_settings.username, randomStringDigits())
2019-02-22 16:46:35 +01:00
username = str(context.scene.session_settings.username)
2019-03-25 14:56:09 +01:00
2019-04-10 17:01:21 +02:00
client_instance = client.RCFClient()
2019-04-10 18:26:39 +02:00
client_instance.connect(net_settings.username,"127.0.0.1",5555)
2019-04-08 17:01:02 +02:00
# net_settings.is_running = True
2019-04-15 16:21:50 +02:00
drawer = draw.HUD(client_instance=client_instance)
2019-04-11 15:54:15 +02:00
register_ticks()
2019-02-11 15:48:07 +01:00
return {"FINISHED"}
2019-02-25 18:05:46 +01:00
2019-04-11 14:39:31 +02:00
class session_refresh(bpy.types.Operator):
bl_idname = "session.refresh"
bl_label = "refresh"
bl_description = "refresh client ui keys "
bl_options = {"REGISTER"}
@classmethod
def poll(cls, context):
return True
def execute(self, context):
global client_instance, client_keys
client_keys = client_instance.list()
return {"FINISHED"}
class session_add_property(bpy.types.Operator):
bl_idname = "session.add_prop"
bl_label = "add"
2019-04-10 17:01:21 +02:00
bl_description = "broadcast a property to connected client_instances"
2019-02-11 17:02:25 +01:00
bl_options = {"REGISTER"}
property_path: bpy.props.StringProperty(default="None")
2019-03-25 14:56:09 +01:00
depth: bpy.props.IntProperty(default=1)
2019-03-25 23:31:44 +01:00
2019-02-11 17:02:25 +01:00
@classmethod
def poll(cls, context):
return True
def execute(self, context):
2019-04-10 17:01:21 +02:00
global client_instance
2019-02-11 17:02:25 +01:00
2019-04-10 18:01:55 +02:00
client_instance.set(self.property_path)
2019-02-11 17:02:25 +01:00
return {"FINISHED"}
2019-02-25 18:05:46 +01:00
class session_get_property(bpy.types.Operator):
bl_idname = "session.get_prop"
bl_label = "get"
bl_description = "broadcast a property to connected client_instances"
bl_options = {"REGISTER"}
@classmethod
def poll(cls, context):
return True
def execute(self, context):
global client_instance
client_instance.get("client")
return {"FINISHED"}
class session_remove_property(bpy.types.Operator):
bl_idname = "session.remove_prop"
bl_label = "remove"
2019-04-10 17:01:21 +02:00
bl_description = "broadcast a property to connected client_instances"
bl_options = {"REGISTER"}
property_path: bpy.props.StringProperty(default="None")
@classmethod
def poll(cls, context):
return True
def execute(self, context):
2019-04-10 17:01:21 +02:00
global client_instance
try:
2019-04-10 17:01:21 +02:00
del client_instance.property_map[self.property_path]
return {"FINISHED"}
except:
return {"CANCELED"}
2019-02-25 18:05:46 +01:00
2019-02-11 17:31:03 +01:00
class session_create(bpy.types.Operator):
bl_idname = "session.create"
bl_label = "create"
bl_description = "create to a net session"
2019-02-11 15:48:07 +01:00
bl_options = {"REGISTER"}
@classmethod
def poll(cls, context):
return True
def execute(self, context):
global server
2019-04-10 17:01:21 +02:00
global client_instance
2019-02-25 18:05:46 +01:00
2019-04-08 17:01:02 +02:00
server = subprocess.Popen(['python','server.py'], shell=False, stdout=subprocess.PIPE)
2019-02-22 16:46:35 +01:00
time.sleep(0.1)
2019-02-11 17:02:25 +01:00
2019-02-22 16:46:35 +01:00
bpy.ops.session.join()
2019-02-25 18:05:46 +01:00
2019-04-11 14:39:31 +02:00
if context.scene.session_settings.init_scene:
init_datablocks()
2019-02-11 17:02:25 +01:00
return {"FINISHED"}
2019-02-11 17:31:03 +01:00
class session_stop(bpy.types.Operator):
bl_idname = "session.stop"
bl_label = "close"
bl_description = "stop net service"
2019-02-11 17:02:25 +01:00
bl_options = {"REGISTER"}
@classmethod
def poll(cls, context):
return True
def execute(self, context):
global server
2019-04-11 15:54:15 +02:00
global client_instance, client_keys
net_settings = context.scene.session_settings
2019-02-25 18:05:46 +01:00
if server:
2019-04-08 17:01:02 +02:00
server.kill()
2019-02-11 17:49:28 +01:00
del server
server = None
2019-04-10 17:01:21 +02:00
if client_instance:
client_instance.exit()
del client_instance
client_instance = None
2019-04-11 15:54:15 +02:00
del client_keys
client_keys = None
net_settings.is_running = False
2019-04-01 11:45:16 +02:00
2019-04-11 15:54:15 +02:00
unregister_ticks()
2019-02-11 17:02:25 +01:00
else:
2019-04-10 17:01:21 +02:00
logger.debug("No server/client_instance running.")
2019-02-11 15:48:07 +01:00
return {"FINISHED"}
2019-02-08 18:34:10 +01:00
2019-02-11 17:02:25 +01:00
2019-02-11 18:10:56 +01:00
class session_settings(bpy.types.PropertyGroup):
username = bpy.props.StringProperty(
name="Username", default="user_{}".format(randomStringDigits()))
2019-03-25 14:56:09 +01:00
ip = bpy.props.StringProperty(name="ip")
2019-02-11 18:10:56 +01:00
port = bpy.props.IntProperty(name="5555")
2019-03-25 14:56:09 +01:00
2019-03-25 23:31:44 +01:00
add_property_depth = bpy.props.IntProperty(
name="add_property_depth", default=1)
buffer = bpy.props.StringProperty(name="None")
2019-02-25 18:05:46 +01:00
is_running = bpy.props.BoolProperty(name="is_running", default=False)
2019-03-25 14:56:09 +01:00
load_data = bpy.props.BoolProperty(name="load_data", default=True)
2019-04-01 11:45:16 +02:00
init_scene = bpy.props.BoolProperty(name="load_data", default=True)
clear_scene = bpy.props.BoolProperty(name="clear_scene", default=True)
2019-02-25 18:05:46 +01:00
update_frequency = bpy.props.FloatProperty(
name="update_frequency", default=0.008)
2019-03-14 11:04:41 +01:00
active_object = bpy.props.PointerProperty(
name="active_object", type=bpy.types.Object)
2019-03-18 18:21:33 +01:00
session_mode = bpy.props.EnumProperty(
name='session_mode',
description='session mode',
items={
('HOST', 'hosting', 'host a session'),
('CONNECT', 'connexion', 'connect to a session')},
default='HOST')
2019-04-10 17:01:21 +02:00
client_color = bpy.props.FloatVectorProperty(name="client_instance_color",
2019-04-01 11:45:16 +02:00
subtype='COLOR',
default=randomColor())
2019-03-13 17:02:53 +01:00
2019-03-14 11:04:41 +01:00
class session_snapview(bpy.types.Operator):
bl_idname = "session.snapview"
2019-04-10 17:01:21 +02:00
bl_label = "draw client_instances"
2019-03-14 11:04:41 +01:00
bl_description = "Description that shows in blender tooltips"
bl_options = {"REGISTER"}
2019-04-10 17:01:21 +02:00
target_client_instance = bpy.props.StringProperty()
2019-03-14 11:04:41 +01:00
@classmethod
def poll(cls, context):
return True
def execute(self, context):
2019-04-10 17:01:21 +02:00
global client_instance
2019-03-14 11:04:41 +01:00
2019-04-02 14:42:39 +02:00
area, region, rv3d = net_draw.view3d_find()
2019-03-14 11:04:41 +01:00
2019-04-10 17:01:21 +02:00
for k, v in client_instance.property_map.items():
if v.mtype == 'client_instance' and v.id.decode() == self.target_client_instance:
rv3d.view_location = v.body['location'][1]
2019-03-14 13:45:42 +01:00
rv3d.view_distance = 30.0
2019-03-14 11:04:41 +01:00
return {"FINISHED"}
return {"CANCELLED"}
pass
2019-03-14 12:09:33 +01:00
2019-03-25 14:56:09 +01:00
2019-02-13 12:11:05 +01:00
# TODO: Rename to match official blender convention
classes = (
2019-02-11 17:31:03 +01:00
session_join,
2019-04-11 14:39:31 +02:00
session_refresh,
session_add_property,
session_get_property,
2019-02-11 17:31:03 +01:00
session_stop,
session_create,
2019-02-11 18:10:56 +01:00
session_settings,
session_remove_property,
2019-03-14 11:04:41 +01:00
session_snapview,
)
2019-04-03 15:07:35 +02:00
def ordered(updates):
# sorted = sorted(updates, key=lambda tup: SUPPORTED_TYPES.index(tup[1].id.bl_rna.name))
uplist = [(SUPPORTED_TYPES.index(item[1].id.bl_rna.name),item[1].id.bl_rna.name,item[1].id.name) for item in updates.items()]
uplist.sort(key=itemgetter(0))
return uplist
2019-04-11 15:54:15 +02:00
def is_dirty(updates):
global client_keys
if client_keys:
if len(client_keys) > 0:
for u in updates:
key = "{}/{}".format(u.id.bl_rna.name,u.id.name)
if key not in client_keys:
return True
return False
2019-04-01 11:45:16 +02:00
def depsgraph_update(scene):
2019-04-10 17:01:21 +02:00
global client_instance
2019-04-04 18:02:51 +02:00
2019-04-11 15:54:15 +02:00
if client_instance and client_instance.agent.is_alive():
2019-04-02 18:07:59 +02:00
updates = bpy.context.depsgraph.updates
2019-04-17 11:01:01 +02:00
2019-04-11 18:07:59 +02:00
if is_dirty(updates):
2019-04-03 17:52:45 +02:00
for update in ordered(updates):
if update[2] == "Master Collection":
pass
elif update[1] in SUPPORTED_TYPES:
2019-04-11 15:54:15 +02:00
client_instance.set("{}/{}".format(update[1], update[2]))
2019-04-03 15:07:35 +02:00
2019-04-11 18:07:59 +02:00
if len(bpy.context.selected_objects)>0:
2019-04-11 18:07:59 +02:00
updated_data = updates[0]
if updated_data.id.name == bpy.context.selected_objects[0].name:
if updated_data.is_updated_transform or updated_data.is_updated_geometry:
client_instance.set("{}/{}".format(updated_data.id.bl_rna.name, updated_data.id.name))
2019-04-01 11:45:16 +02:00
2019-02-08 18:34:10 +01:00
def register():
from bpy.utils import register_class
for cls in classes:
register_class(cls)
2019-04-11 15:54:15 +02:00
bpy.types.ID.is_dirty = bpy.props.BoolProperty(default=False)
2019-02-25 18:05:46 +01:00
bpy.types.Scene.session_settings = bpy.props.PointerProperty(
type=session_settings)
2019-04-11 15:54:15 +02:00
bpy.app.handlers.depsgraph_update_post.append(depsgraph_update)
2019-02-25 18:05:46 +01:00
2019-04-01 11:45:16 +02:00
def unregister():
global server
2019-04-11 15:54:15 +02:00
global client_instance, client_keys
2019-04-11 15:54:15 +02:00
try:
bpy.app.handlers.depsgraph_update_post.remove(depsgraph_update)
except:
pass
2019-03-28 15:02:21 +01:00
if server:
2019-04-11 15:54:15 +02:00
server.kill()
server = None
2019-04-11 15:54:15 +02:00
del server
2019-04-10 17:01:21 +02:00
if client_instance:
2019-04-11 15:54:15 +02:00
client_instance.exit()
2019-04-10 17:01:21 +02:00
client_instance = None
2019-04-11 15:54:15 +02:00
del client_instance
del client_keys
2019-02-08 18:34:10 +01:00
from bpy.utils import unregister_class
for cls in reversed(classes):
unregister_class(cls)
2019-02-11 18:10:56 +01:00
del bpy.types.Scene.session_settings
2019-04-11 15:54:15 +02:00
del bpy.types.ID.is_dirty
if __name__ == "__main__":
2019-02-08 18:34:10 +01:00
register()