Backups/multi-user
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

reformat: code cleanup

Browse Source
This commit is contained in:
Swann Martinez
2019-05-02 17:58:37 +02:00
parent 824c70eaff
commit c1119976cd
3 changed files with 71 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
client.py
View File

@ -1,25 +1,22 @@
import binascii import binascii
import collections import collections
import copy
import logging import logging
import os import os
import queue
import sys import sys
import threading import threading
import time import time
from enum import Enum from enum import Enum
from random import randint from random import randint
import copy
import queue from . import draw, helpers, message
from .libs import dump_anything, umsgpack, zmq
# import zmq # import zmq
lock = threading.Lock() lock = threading.Lock()
from .libs import umsgpack
from .libs import zmq
from .libs import dump_anything
from . import helpers
from . import message
from . import draw
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -487,4 +484,3 @@ def serial_worker(product, feed):
elif command == 'LOAD': elif command == 'LOAD':
if value: if value:
helpers.load(key, value) helpers.load(key, value)

39
helpers.py
View File

@ -1,10 +1,12 @@
import bpy
import sys
import mathutils
from .libs import dump_anything
from uuid import uuid4
import logging import logging
import sys
from uuid import uuid4
import bpy
import mathutils
from . import draw from . import draw
from .libs import dump_anything
CORRESPONDANCE = {'Collection': 'collections', 'Mesh': 'meshes', 'Object': 'objects', 'Material': 'materials', CORRESPONDANCE = {'Collection': 'collections', 'Mesh': 'meshes', 'Object': 'objects', 'Material': 'materials',
'Texture': 'textures', 'Scene': 'scenes', 'Light': 'lights', 'Camera': 'cameras', 'Action': 'actions', 'Armature': 'armatures', 'Grease Pencil': 'grease_pencils'} 'Texture': 'textures', 'Scene': 'scenes', 'Light': 'lights', 'Camera': 'cameras', 'Action': 'actions', 'Armature': 'armatures', 'Grease Pencil': 'grease_pencils'}
@ -15,6 +17,8 @@ SUPPORTED_TYPES = ['Material',
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
# UTILITY FUNCTIONS # UTILITY FUNCTIONS
def refresh_window(): def refresh_window():
import bpy import bpy
bpy.ops.wm.redraw_timer(type='DRAW_WIN_SWAP', iterations=1) bpy.ops.wm.redraw_timer(type='DRAW_WIN_SWAP', iterations=1)
@ -28,6 +32,7 @@ def get_selected_objects(scene):
return selected_objects return selected_objects
def get_all_datablocks(): def get_all_datablocks():
datas = [] datas = []
for datatype in SUPPORTED_TYPES: for datatype in SUPPORTED_TYPES:
@ -38,6 +43,8 @@ def get_all_datablocks():
return datas return datas
# LOAD HELPERS # LOAD HELPERS
def load(key, value): def load(key, value):
target = resolve_bpy_path(key) target = resolve_bpy_path(key)
target_type = key.split('/')[0] target_type = key.split('/')[0]
@ -94,25 +101,10 @@ def load_client(client=None, data=None):
D = bpy.data D = bpy.data
net_settings = C.scene.session_settings net_settings = C.scene.session_settings
if client and data: if client and data:
if net_settings.enable_draw: if net_settings.enable_draw:
draw.renderer.draw_client(data) draw.renderer.draw_client(data)
draw.renderer.draw_client_selected_objects(data) draw.renderer.draw_client_selected_objects(data)
# localy_selected = get_selected_objects(C.scene)
# Draw client
# Load selected object
# for obj in C.scene.objects:
# if obj.id == client:
# D.objects[obj.name].hide_select = True
# else:
# D.objects[obj.name].hide_select = False
# if client_data['active_objects'] and obj.name in client_data['active_objects']:
# D.objects[obj.name].hide_select = True
# else:
# D.objects[obj.name].hide_select = False
def load_mesh(target=None, data=None, create=False): def load_mesh(target=None, data=None, create=False):
@ -354,7 +346,6 @@ def load_light(target=None, data=None, create=False, type=None):
if target is None and create: if target is None and create:
target = bpy.data.lights.new(data["name"], data["type"]) target = bpy.data.lights.new(data["name"], data["type"])
dump_anything.load(target, data) dump_anything.load(target, data)
target.id = data['id'] target.id = data['id']
@ -374,12 +365,13 @@ def load_default(target=None, data=None, create=False, type=None):
logger.error("default loading error {}".format(e)) logger.error("default loading error {}".format(e))
# DUMP HELPERS # DUMP HELPERS
def dump(key): def dump(key):
target = resolve_bpy_path(key) target = resolve_bpy_path(key)
target_type = key.split('/')[0] target_type = key.split('/')[0]
data = None data = None
if target_type == 'Material': if target_type == 'Material':
data = dump_datablock_attibute(target, ['name', 'node_tree', 'id'], 7) data = dump_datablock_attibute(target, ['name', 'node_tree', 'id'], 7)
elif target_type == 'Grease Pencil': elif target_type == 'Grease Pencil':
@ -397,7 +389,8 @@ def dump(key):
elif target_type == 'Collection': elif target_type == 'Collection':
data = dump_datablock(target, 4) data = dump_datablock(target, 4)
elif target_type == 'Scene': elif target_type == 'Scene':
data = dump_datablock_attibute(target,['name','collection','id','camera','grease_pencil'], 4) data = dump_datablock_attibute(
target, ['name', 'collection', 'id', 'camera', 'grease_pencil'], 4)
return data return data

66
operators.py
View File

@ -1,21 +1,22 @@
import asyncio
import logging import logging
import os
import queue
import random import random
import string import string
import time
import asyncio
import queue
from operator import itemgetter
import subprocess import subprocess
import time
from operator import itemgetter
import bgl import bgl
import blf import blf
import bpy import bpy
import os
import gpu import gpu
import mathutils import mathutils
from bpy_extras import view3d_utils from bpy_extras import view3d_utils
from gpu_extras.batch import batch_for_shader from gpu_extras.batch import batch_for_shader
from . import client, ui, draw, helpers from . import client, draw, helpers, ui
from .libs import umsgpack from .libs import umsgpack
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -116,37 +117,6 @@ def update_selected_object(context):
client_instance.set(client_key, client_data[0][1]) client_instance.set(client_key, client_data[0][1])
# for update in local_updates:
# client_instance.get('')
# if session.active_object is not context.selected_objects[0] or session.active_object.is_evaluated:
# session.active_object = context.selected_objects[0]
# key = "net/objects/{}".format(client_instance.id.decode())
# data = {}
# data['color'] = [session.client_instance_color.r,
# session.client_instance_color.g, session.client_instance_color.b]
# data['object'] = session.active_object.name
# client_instance.push_update(
# key, 'client_instanceObject', data)
# return True
# elif len(context.selected_objects) == 0 and session.active_object:
# session.active_object = None
# data = {}
# data['color'] = [session.client_instance_color.r,
# session.client_instance_color.g, session.client_instance_color.b]
# data['object'] = None
# key = "net/objects/{}".format(client_instance.id.decode())
# client_instance.push_update(key, 'client_instanceObject', data)
# return True
# return False
def init_datablocks(): def init_datablocks():
global client_instance global client_instance
@ -160,11 +130,6 @@ def init_datablocks():
def default_tick(): def default_tick():
bpy.ops.session.refresh() bpy.ops.session.refresh()
upload_client_instance_position() upload_client_instance_position()
# global client_instance
# if not client_instance.queue.empty():
# update = client_instance.queue.get()
# helpers.load(update[0],update[1])
return 1 return 1
@ -180,15 +145,16 @@ def sync():
key = "{}/{}".format(datatype, item.name) key = "{}/{}".format(datatype, item.name)
client_instance.add(key) client_instance.add(key)
return .2 return .2
def register_ticks(): def register_ticks():
# REGISTER Updaters # REGISTER Updaters
bpy.app.timers.register(sync) bpy.app.timers.register(sync)
bpy.app.timers.register(default_tick) bpy.app.timers.register(default_tick)
pass pass
def unregister_ticks(): def unregister_ticks():
# REGISTER Updaters # REGISTER Updaters
bpy.app.timers.unregister(sync) bpy.app.timers.unregister(sync)
@ -196,6 +162,8 @@ def unregister_ticks():
pass pass
# OPERATORS # OPERATORS
class session_join(bpy.types.Operator): class session_join(bpy.types.Operator):
bl_idname = "session.join" bl_idname = "session.join"
@ -373,7 +341,6 @@ class session_stop(bpy.types.Operator):
net_settings = context.scene.session_settings net_settings = context.scene.session_settings
if server: if server:
server.kill() server.kill()
time.sleep(0.25) time.sleep(0.25)
@ -419,8 +386,6 @@ class session_rights(bpy.types.Operator):
col = layout.column() col = layout.column()
col.prop(net_settings, "clients") col.prop(net_settings, "clients")
def execute(self, context): def execute(self, context):
global server global server
global client_instance, client_keys, client_state global client_instance, client_keys, client_state
@ -431,10 +396,10 @@ class session_rights(bpy.types.Operator):
val = client_instance.get(self.key) val = client_instance.get(self.key)
val[0][1]['id'] = net_settings.clients val[0][1]['id'] = net_settings.clients
client_instance.set(key=self.key, value=val[0][1], override=True) client_instance.set(key=self.key, value=val[0][1], override=True)
print("Updating {} rights to {}".format(self.key,net_settings.clients)) print("Updating {} rights to {}".format(
self.key, net_settings.clients))
else: else:
print("Not admin") print("Not admin")
@ -475,7 +440,9 @@ class session_settings(bpy.types.PropertyGroup):
description="client enum", description="client enum",
items=client_list_callback items=client_list_callback
) )
enable_draw = bpy.props.BoolProperty(name="enable_draw", description='Enable overlay drawing module', default=True) enable_draw = bpy.props.BoolProperty(
name="enable_draw", description='Enable overlay drawing module', default=True)
class session_snapview(bpy.types.Operator): class session_snapview(bpy.types.Operator):
bl_idname = "session.snapview" bl_idname = "session.snapview"
@ -553,7 +520,6 @@ def depsgraph_update(scene):
username = bpy.context.scene.session_settings.username username = bpy.context.scene.session_settings.username
update_selected_object(bpy.context) update_selected_object(bpy.context)
selected_objects = helpers.get_selected_objects(scene) selected_objects = helpers.get_selected_objects(scene)
if len(selected_objects) > 0: if len(selected_objects) > 0:
for updated_data in updates: for updated_data in updates: