multi-user/operators.py

418 lines
10 KiB
Python
Raw Normal View History

2019-05-02 23:58:37 +08:00
import asyncio
2019-02-12 00:02:25 +08:00
import logging
2019-05-02 23:58:37 +08:00
import os
import queue
2019-02-22 23:46:35 +08:00
import random
import string
2019-05-02 23:58:37 +08:00
import subprocess
2019-03-25 21:56:09 +08:00
import time
2019-04-03 21:07:35 +08:00
from operator import itemgetter
from pathlib import Path
2019-03-25 21:56:09 +08:00
import bpy
import mathutils
from bpy_extras.io_utils import ExportHelper
2019-05-09 20:20:42 +08:00
from . import environment, presence, ui, utils
from .libs import umsgpack
2019-08-06 00:14:05 +08:00
from .libs.replication.data import ReplicatedDataFactory
from .libs.replication.interface import Client
2019-08-08 19:41:53 +08:00
from . import bl_types
2019-03-25 21:56:09 +08:00
2019-02-12 00:02:25 +08:00
logger = logging.getLogger(__name__)
client = None
2019-02-11 22:48:07 +08:00
context = None
2019-05-09 21:56:30 +08:00
2019-08-06 17:34:39 +08:00
# def upload_client_instance_position():
# username = bpy.context.window_manager.session.username
# if client:
2019-04-18 21:05:48 +08:00
2019-08-06 17:34:39 +08:00
# key = "Client/{}".format(username)
2019-04-17 20:22:56 +08:00
2019-08-06 17:34:39 +08:00
# current_coords = presence.get_client_view_rect()
# client_list = client.get(key)
2019-08-06 17:34:39 +08:00
# if current_coords and client_list:
# if current_coords != client_list[0][1]['location']:
# client_list[0][1]['location'] = current_coords
# client.set(key, client_list[0][1])
2019-04-15 22:21:50 +08:00
2019-08-06 17:34:39 +08:00
# def update_client_selected_object(context):
# session = bpy.context.window_manager.session
# username = bpy.context.window_manager.session.username
# client_key = "Client/{}".format(username)
# client_data = client.get(client_key)
2019-04-19 21:46:54 +08:00
2019-08-06 17:34:39 +08:00
# selected_objects = utils.get_selected_objects(context.scene)
# if len(selected_objects) > 0 and len(client_data) > 0:
2019-05-02 23:58:37 +08:00
2019-08-06 17:34:39 +08:00
# for obj in selected_objects:
# # if obj not in client_data[0][1]['active_objects']:
# client_data[0][1]['active_objects'] = selected_objects
2019-08-06 17:34:39 +08:00
# client.set(client_key, client_data[0][1])
# break
2019-04-19 21:46:54 +08:00
2019-08-06 17:34:39 +08:00
# elif client_data and client_data[0][1]['active_objects']:
# client_data[0][1]['active_objects'] = []
# client.set(client_key, client_data[0][1])
2019-04-11 20:39:31 +08:00
2019-08-09 00:12:13 +08:00
def add_datablock(datablock):
global client
child=[]
2019-08-09 00:12:13 +08:00
if hasattr(datablock,"data"):
child.append(add_datablock(datablock.data))
2019-08-09 00:12:13 +08:00
if datablock.uuid and client.exist(datablock.uuid) :
return datablock.uuid
else:
new_uuid = client.add(datablock,childs=child)
datablock.uuid = new_uuid
2019-08-09 22:47:15 +08:00
return new_uuid
2019-08-09 00:12:13 +08:00
# TODO: cleanup
2019-08-08 19:41:53 +08:00
def init_supported_datablocks(supported_types_id):
global client
for type_id in supported_types_id:
if hasattr(bpy.data,type_id):
for item in getattr(bpy.data,type_id):
2019-08-09 00:12:13 +08:00
add_datablock(item)
2019-05-08 23:06:52 +08:00
2019-03-25 21:56:09 +08:00
2019-08-09 20:39:59 +08:00
2019-08-06 17:34:39 +08:00
# def default_tick():
# upload_client_instance_position()
2019-08-06 17:34:39 +08:00
# return .1
2019-05-10 22:22:14 +08:00
2019-08-06 17:34:39 +08:00
# def register_ticks():
# # REGISTER Updaters
# bpy.app.timers.register(default_tick)
# bpy.app.timers.register(execute_queued_functions)
2019-04-05 00:02:51 +08:00
2019-08-06 17:34:39 +08:00
# def unregister_ticks():
# REGISTER Updaters
2019-08-06 17:34:39 +08:00
# try:
# bpy.app.timers.unregister(default_tick)
# bpy.app.timers.unregister(execute_queued_functions)
# except:
# pass
2019-08-06 00:14:05 +08:00
# OPERATORS
class SessionStartOperator(bpy.types.Operator):
bl_idname = "session.start"
bl_label = "start"
2019-02-11 22:48:07 +08:00
bl_description = "connect to a net server"
bl_options = {"REGISTER"}
host: bpy.props.BoolProperty(default=False)
2019-02-11 22:48:07 +08:00
@classmethod
def poll(cls, context):
return True
def execute(self, context):
2019-08-09 22:47:15 +08:00
global client
settings = context.window_manager.session
2019-07-02 23:44:59 +08:00
# save config
settings.save(context)
2019-07-02 23:44:59 +08:00
2019-03-25 21:56:09 +08:00
# Scene setup
2019-08-08 20:46:59 +08:00
if settings.start_empty:
utils.clean_scene()
2019-02-26 01:05:46 +08:00
2019-08-08 19:41:53 +08:00
bpy_factory = ReplicatedDataFactory()
supported_bl_types = []
2019-08-06 00:14:05 +08:00
2019-08-08 19:41:53 +08:00
# init the factory with supported types
for type in bl_types.types_to_register():
_type = getattr(bl_types,type)
supported_bl_types.append(_type.bl_id)
bpy_factory.register_type(_type.bl_class,_type.bl_rep_class)
2019-08-06 00:14:05 +08:00
client = Client(factory=bpy_factory)
2019-08-06 00:14:05 +08:00
if self.host:
client.host(
2019-08-08 19:41:53 +08:00
id=settings.username,
address=settings.ip,
port=settings.port
2019-08-06 00:14:05 +08:00
)
2019-08-08 19:41:53 +08:00
if settings.init_scene:
init_supported_datablocks(supported_bl_types)
2019-08-06 00:14:05 +08:00
else:
client.connect(
id=settings.username,
address=settings.ip,
port=settings.port
2019-08-08 19:41:53 +08:00
)
usr = presence.User(
username=settings.username,
color=list(settings.client_color),
)
client.add(usr)
2019-08-09 20:39:59 +08:00
client.push()
# settings.is_running = True
# bpy.ops.session.refresh()
2019-08-06 00:14:05 +08:00
#register_ticks()
# Launch drawing module
if settings.enable_presence:
2019-08-05 22:48:46 +08:00
presence.renderer.run()
2019-02-11 22:48:07 +08:00
return {"FINISHED"}
2019-02-26 01:05:46 +08:00
2019-08-06 17:34:39 +08:00
class SessionStopOperator(bpy.types.Operator):
bl_idname = "session.stop"
bl_label = "close"
bl_description = "stop net service"
bl_options = {"REGISTER"}
@classmethod
def poll(cls, context):
return True
def execute(self, context):
global client
assert(client)
client.disconnect()
# del client_instance
# unregister_ticks()
# presence.renderer.stop()
return {"FINISHED"}
class SessionPropertyAddOperator(bpy.types.Operator):
bl_idname = "session.add_prop"
bl_label = "add"
2019-04-10 23:01:21 +08:00
bl_description = "broadcast a property to connected client_instances"
2019-02-12 00:02:25 +08:00
bl_options = {"REGISTER"}
property_path: bpy.props.StringProperty(default="None")
2019-03-25 21:56:09 +08:00
depth: bpy.props.IntProperty(default=1)
2019-03-26 06:31:44 +08:00
2019-02-12 00:02:25 +08:00
@classmethod
def poll(cls, context):
return True
def execute(self, context):
global client
2019-02-12 00:02:25 +08:00
client.add(self.property_path)
2019-02-12 00:02:25 +08:00
return {"FINISHED"}
2019-02-26 01:05:46 +08:00
class SessionPropertyRemoveOperator(bpy.types.Operator):
bl_idname = "session.remove_prop"
bl_label = "remove"
2019-04-10 23:01:21 +08: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):
global client
try:
del client.property_map[self.property_path]
return {"FINISHED"}
except:
return {"CANCELED"}
2019-02-26 01:05:46 +08:00
class SessionPropertyRightOperator(bpy.types.Operator):
2019-04-24 23:42:23 +08:00
bl_idname = "session.right"
2019-05-09 21:56:30 +08:00
bl_label = "Change owner to"
2019-04-24 23:42:23 +08:00
bl_description = "stop net service"
bl_options = {"REGISTER"}
key: bpy.props.StringProperty(default="None")
@classmethod
def poll(cls, context):
return True
def invoke(self, context, event):
wm = context.window_manager
return wm.invoke_props_dialog(self)
def draw(self, context):
layout = self.layout
settings = context.window_manager.session
2019-04-24 23:42:23 +08:00
col = layout.column()
col.prop(settings, "clients")
2019-04-24 23:42:23 +08:00
def execute(self, context):
settings = context.window_manager.session
2019-04-24 23:42:23 +08:00
if settings.is_admin:
val = client.get(self.key)
val[0][1]['id'] = settings.clients
2019-04-24 23:42:23 +08:00
client.set(key=self.key, value=val[0][1], override=True)
item = utils.resolve_bpy_path(self.key)
if item:
item.id = settings.clients
logger.info("Updating {} rights to {}".format(
self.key, settings.clients))
2019-04-24 23:42:23 +08:00
else:
print("Not admin")
return {"FINISHED"}
class SessionSnapUserOperator(bpy.types.Operator):
2019-03-14 18:04:41 +08:00
bl_idname = "session.snapview"
2019-04-10 23:01:21 +08:00
bl_label = "draw client_instances"
2019-03-14 18:04:41 +08:00
bl_description = "Description that shows in blender tooltips"
bl_options = {"REGISTER"}
2019-04-17 20:22:56 +08:00
target_client = bpy.props.StringProperty()
2019-03-14 18:04:41 +08:00
@classmethod
def poll(cls, context):
return True
def execute(self, context):
2019-08-05 22:48:46 +08:00
area, region, rv3d = presence.view3d_find()
2019-03-14 18:04:41 +08:00
target_client = client.get(
"Client/{}".format(self.target_client))
2019-05-23 22:49:32 +08:00
if target_client:
rv3d.view_location = target_client[0][1]['location'][0]
2019-04-22 21:58:32 +08:00
rv3d.view_distance = 30.0
2019-05-02 23:58:37 +08:00
2019-04-22 21:58:32 +08:00
return {"FINISHED"}
2019-03-14 18:04:41 +08:00
return {"CANCELLED"}
pass
2019-03-14 19:09:33 +08:00
2019-08-09 22:47:15 +08:00
class SessionDumpDatabase(bpy.types.Operator, ExportHelper):
bl_idname = "session.dump"
bl_label = "dump json data"
bl_description = "dump session stored data to a json file"
bl_options = {"REGISTER"}
# ExportHelper mixin class uses this
filename_ext = ".json"
filter_glob: bpy.props.StringProperty(
default="*.json",
options={'HIDDEN'},
maxlen=255, # Max internal buffer length, longer would be clamped.
)
@classmethod
def poll(cls, context):
return True
def execute(self, context):
print(self.filepath)
if client and client.state() == 3:
client.dump(self.filepath)
return {"FINISHED"}
return {"CANCELLED"}
pass
2019-07-02 00:04:35 +08:00
class SessionSaveConfig(bpy.types.Operator):
bl_idname = "session.save"
bl_label = "Save session configuration"
bl_description = "Save session configuration"
bl_options = {"REGISTER"}
@classmethod
def poll(cls, context):
return True
def execute(self, context):
context.window_manager.session.save()
2019-08-10 00:14:32 +08:00
class SessionApply(bpy.types.Operator):
bl_idname = "session.apply"
bl_label = "apply the target item into the blender data"
bl_description = "Apply target object into blender data"
bl_options = {"REGISTER"}
target = bpy.props.StringProperty()
@classmethod
def poll(cls, context):
return True
def execute(self, context):
global client
client.apply(uuid=self.target)
return {"FINISHED"}
2019-07-02 00:04:35 +08:00
classes = (
SessionStartOperator,
SessionPropertyAddOperator,
SessionStopOperator,
SessionPropertyRemoveOperator,
SessionSnapUserOperator,
SessionPropertyRightOperator,
SessionDumpDatabase,
2019-07-02 00:04:35 +08:00
SessionSaveConfig,
2019-08-10 00:14:32 +08:00
SessionApply,
)
2019-02-09 01:34:10 +08:00
def register():
from bpy.utils import register_class
for cls in classes:
register_class(cls)
2019-08-06 22:26:19 +08:00
2019-08-05 22:48:46 +08:00
presence.register()
2019-02-26 01:05:46 +08:00
2019-04-01 17:45:16 +08:00
def unregister():
global client
2019-08-05 22:48:46 +08:00
presence.unregister()
2019-08-08 19:41:53 +08:00
if client and client.state == 2:
2019-08-06 17:34:39 +08:00
client.disconnect()
client = None
2019-02-09 01:34:10 +08:00
from bpy.utils import unregister_class
for cls in reversed(classes):
unregister_class(cls)
2019-04-18 21:05:48 +08:00
if __name__ == "__main__":
2019-02-09 01:34:10 +08:00
register()