feat: expose presence draw settings

This commit is contained in:
Swann
2019-09-24 14:42:59 +02:00
parent 4ee094ec9a
commit 1e371a2e60
6 changed files with 107 additions and 26 deletions

View File

@ -18,6 +18,8 @@ import bpy
from bpy.app.handlers import persistent from bpy.app.handlers import persistent
from . import environment, utils from . import environment, utils
from .libs.replication.replication.constants import RP_COMMON
# TODO: remove dependency as soon as replication will be installed as a module # TODO: remove dependency as soon as replication will be installed as a module
DEPENDENCIES = { DEPENDENCIES = {
@ -54,7 +56,7 @@ def client_list_callback(scene, context):
from . import operators from . import operators
from .bl_types.bl_user import BlUser from .bl_types.bl_user import BlUser
items = [("Common", "Common", "")] items = [(RP_COMMON, RP_COMMON, "")]
username = bpy.context.window_manager.session.username username = bpy.context.window_manager.session.username
cli = operators.client cli = operators.client
@ -144,7 +146,20 @@ class SessionProps(bpy.types.PropertyGroup):
enable_presence: bpy.props.BoolProperty( enable_presence: bpy.props.BoolProperty(
name="Presence overlay", name="Presence overlay",
description='Enable overlay drawing module', description='Enable overlay drawing module',
default=True default=True,
update=presence.update_presence
)
presence_show_selected: bpy.props.BoolProperty(
name="Show selected objects",
description='Enable selection overlay ',
default=True,
update=presence.update_overlay_settings
)
presence_show_user: bpy.props.BoolProperty(
name="Show users",
description='Enable user overlay ',
default=True,
update=presence.update_overlay_settings
) )
supported_datablock: bpy.props.CollectionProperty( supported_datablock: bpy.props.CollectionProperty(
type=ReplicatedDatablock, type=ReplicatedDatablock,

View File

@ -29,12 +29,15 @@ class BlUser(BlDatablock):
if self.pointer: if self.pointer:
self.load(data=self.buffer, target=self.pointer) self.load(data=self.buffer, target=self.pointer)
settings = bpy.context.window_manager.session
self.state = UP self.state = UP
#TODO: refactor in order to redraw in cleaner ways #TODO: refactor in order to redraw in cleaner ways
area, region, rv3d = presence.view3d_find() area, region, rv3d = presence.view3d_find()
if presence.renderer and area and region and rv3d : if presence.renderer and area and region and rv3d and settings.enable_presence:
presence.renderer.draw_client_camera(self.buffer['name'], self.buffer['location'],self.buffer['color']) if settings.presence_show_user:
presence.renderer.draw_client_selection(self.buffer['name'], self.buffer['color'],self.buffer['selected_objects']) presence.renderer.draw_client_camera(self.buffer['name'], self.buffer['location'],self.buffer['color'])
if settings.presence_show_selected:
presence.renderer.draw_client_selection(self.buffer['name'], self.buffer['color'],self.buffer['selected_objects'])
presence.refresh_3d_view() presence.refresh_3d_view()

View File

@ -3,7 +3,7 @@ import logging
from . import operators, utils, presence from . import operators, utils, presence
from .bl_types.bl_user import BlUser from .bl_types.bl_user import BlUser
from .libs.replication.replication.constants import FETCHED from .libs.replication.replication.constants import FETCHED, RP_COMMON
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -95,7 +95,7 @@ class DynamicRightSelectTimer(Timer):
node = operators.client.get(reference=bpy.data.objects[obj]) node = operators.client.get(reference=bpy.data.objects[obj])
if node: if node:
node.owner = settings.username node.owner = settings.username
operators.client.change_owner(node.uuid, 'common') operators.client.change_owner(node.uuid, RP_COMMON)
# update our rights # update our rights
for obj in obj_ours: for obj in obj_ours:

View File

@ -110,9 +110,23 @@ class User():
def update_selected_objects(self, context): def update_selected_objects(self, context):
self.selected_objects = utils.get_selected_objects(context.scene) self.selected_objects = utils.get_selected_objects(context.scene)
def update_presence(self, context):
global renderer
if renderer and self.enable_presence:
renderer.run()
else:
renderer.stop()
def update_overlay_settings(self, context):
global renderer
if renderer and not self.presence_show_selected:
renderer.flush_selection()
if renderer and not self.presence_show_user:
renderer.flush_users()
class DrawFactory(object): class DrawFactory(object):
def __init__(self): def __init__(self):
self.d3d_items = {} self.d3d_items = {}
self.d2d_items = {} self.d2d_items = {}
@ -148,16 +162,37 @@ class DrawFactory(object):
self.d3d_items.clear() self.d3d_items.clear()
self.d2d_items.clear() self.d2d_items.clear()
def flush_selection(self):
key_to_remove = []
for k in self.d3d_items.keys():
if "select" in k:
key_to_remove.append(k)
for k in key_to_remove:
del self.d3d_items[k]
def flush_users(self):
key_to_remove = []
for k in self.d3d_items.keys():
if "select" not in k:
key_to_remove.append(k)
for k in key_to_remove:
del self.d3d_items[k]
self.d2d_items.clear()
def draw_client_selection(self, client_uuid, client_color, client_selection): def draw_client_selection(self, client_uuid, client_color, client_selection):
local_username = bpy.context.window_manager.session.username local_username = bpy.context.window_manager.session.username
key_to_remove = [] self.flush_selection()
for k in self.d3d_items.keys(): # key_to_remove = []
if "{}_select".format(client_uuid) in k: # for k in self.d3d_items.keys():
key_to_remove.append(k) # if "{}_select".format(client_uuid) in k:
# key_to_remove.append(k)
for k in key_to_remove: # for k in key_to_remove:
del self.d3d_items[k] # del self.d3d_items[k]
if client_selection: if client_selection:

50
ui.py
View File

@ -1,6 +1,6 @@
import bpy import bpy
from . import operators from . import operators
from .libs.replication.replication.constants import FETCHED, ERROR, MODIFIED, UP, ADDED from .libs.replication.replication.constants import FETCHED, ERROR, MODIFIED, UP, ADDED, RP_COMMON
from .bl_types.bl_user import BlUser from .bl_types.bl_user import BlUser
@ -36,7 +36,6 @@ class SESSION_PT_settings(bpy.types.Panel):
else: else:
# STATE ACTIVE # STATE ACTIVE
if operators.client.state == 2: if operators.client.state == 2:
row = layout.row() row = layout.row()
row.operator("session.stop", icon='QUIT', text="Exit") row.operator("session.stop", icon='QUIT', text="Exit")
row = layout.row() row = layout.row()
@ -69,9 +68,9 @@ class SESSION_PT_settings_network(bpy.types.Panel):
row = layout.row() row = layout.row()
# USER SETTINGS # USER SETTINGS
row.label(text="Presence overlay:") # row.label(text="Presence overlay:")
row.prop(settings, "enable_presence", text="") # row.prop(settings, "enable_presence", text="")
row = layout.row() # row = layout.row()
row.label(text="Own selection:") row.label(text="Own selection:")
row.prop(settings, "use_select_right", text="") row.prop(settings, "use_select_right", text="")
row = layout.row() row = layout.row()
@ -136,6 +135,7 @@ class SESSION_PT_settings_replication(bpy.types.Panel):
bl_region_type = 'UI' bl_region_type = 'UI'
bl_category = "Multiuser" bl_category = "Multiuser"
bl_parent_id = 'MULTIUSER_SETTINGS_PT_panel' bl_parent_id = 'MULTIUSER_SETTINGS_PT_panel'
bl_options = {'DEFAULT_CLOSED'}
@classmethod @classmethod
def poll(cls, context): def poll(cls, context):
@ -217,6 +217,32 @@ class SESSION_PT_user(bpy.types.Panel):
row = layout.row() row = layout.row()
class SESSION_PT_presence(bpy.types.Panel):
bl_idname = "MULTIUSER_MODULE_PT_panel"
bl_label = "Presence overlay"
bl_space_type = 'VIEW_3D'
bl_region_type = 'UI'
bl_category = "Multiuser"
bl_parent_id = 'MULTIUSER_SETTINGS_PT_panel'
@classmethod
def poll(cls, context):
return True
def draw_header(self, context):
self.layout.prop(context.window_manager.session, "enable_presence", text="")
def draw(self, context):
layout = self.layout
settings = context.window_manager.session
layout.active = settings.enable_presence
col = layout.column()
col.prop(settings,"presence_show_selected")
col.prop(settings,"presence_show_user")
row = layout.row()
def draw_property(context, parent, property_uuid, level=0): def draw_property(context, parent, property_uuid, level=0):
settings = context.window_manager.session settings = context.window_manager.session
item = operators.client.get(uuid=property_uuid) item = operators.client.get(uuid=property_uuid)
@ -240,12 +266,15 @@ def draw_property(context, parent, property_uuid, level=0):
# Operations # Operations
have_right_to_modify = settings.is_admin or item.owner == settings.username have_right_to_modify = settings.is_admin or \
item.owner == settings.username or \
item.owner == RP_COMMON
detail_item_box.operator( detail_item_box.operator(
"session.commit", "session.commit",
text="", text="",
icon='TRIA_UP').target = item.uuid icon='TRIA_UP').target = item.uuid
detail_item_box.separator() detail_item_box.separator()
if item.state in [FETCHED, UP]: if item.state in [FETCHED, UP]:
detail_item_box.operator( detail_item_box.operator(
"session.apply", "session.apply",
@ -260,7 +289,7 @@ def draw_property(context, parent, property_uuid, level=0):
detail_item_box.label(text="", icon=ICONS_PROP_STATES[item.state]) detail_item_box.label(text="", icon=ICONS_PROP_STATES[item.state])
right_icon = "DECORATE_UNLOCKED" right_icon = "DECORATE_UNLOCKED"
if item.owner != settings.username: if not have_right_to_modify:
right_icon = "DECORATE_LOCKED" right_icon = "DECORATE_LOCKED"
if have_right_to_modify: if have_right_to_modify:
@ -305,7 +334,6 @@ class SESSION_PT_outliner(bpy.types.Panel):
col = flow.column(align=True) col = flow.column(align=True)
col.prop(item, "use_as_filter", text="", icon=item.icon) col.prop(item, "use_as_filter", text="", icon=item.icon)
row = layout.row(align=True) row = layout.row(align=True)
row.prop(settings, "filter_owned", text="Show only owned") row.prop(settings, "filter_owned", text="Show only owned")
@ -334,11 +362,11 @@ class SESSION_PT_outliner(bpy.types.Panel):
classes = ( classes = (
SESSION_PT_settings, SESSION_PT_settings,
SESSION_PT_settings_user, SESSION_PT_settings_user,
SESSION_PT_presence,
SESSION_PT_settings_network, SESSION_PT_settings_network,
SESSION_PT_settings_replication, SESSION_PT_settings_replication,
SESSION_PT_user, SESSION_PT_user,
SESSION_PT_outliner, SESSION_PT_outliner
) )