mirror of
https://github.com/schrodinger/pymol-open-source.git
synced 2026-06-06 20:24:23 +08:00
2dd9a34ded
- Make `_self` a keyword-only argument - Fix `get_viewport` logging - Error propagation - Remove unused `set_colorection_name`
119 lines
3.7 KiB
Python
119 lines
3.7 KiB
Python
from pymol import cmd as global_cmd
|
|
import pymol
|
|
import inspect
|
|
import itertools
|
|
import weakref
|
|
|
|
|
|
class Cmd:
|
|
'''
|
|
Proxy for the pymol.cmd module for multiple instances support.
|
|
'''
|
|
|
|
def __init__(self, _pymol, _COb):
|
|
self._weakref = weakref.ref(self)
|
|
self._weakrefproxy = weakref.proxy(self)
|
|
|
|
# store parent
|
|
|
|
self._pymol = weakref.proxy(_pymol)
|
|
|
|
# store C object for easy access
|
|
|
|
self._COb = _COb
|
|
|
|
# private data
|
|
|
|
self.color_sc = None
|
|
self.reaper = None
|
|
|
|
# deferred initiailization
|
|
|
|
global_cmd._deferred_init_pymol_internals(_pymol)
|
|
|
|
# PRIVATE FUNCTIONS (requiring '_self' as a keyword argument)
|
|
|
|
# locking.py
|
|
|
|
self.reaper = None
|
|
|
|
if 1:
|
|
# use own locks (for performance)
|
|
self.lock_api = _pymol.lock_api
|
|
self.lock_api_data = _pymol.lock_api_data
|
|
self.lock_api_glut = _pymol.lock_api_glut
|
|
self.lock_api_status = _pymol.lock_api_status
|
|
else:
|
|
# use global locks (for debugging)
|
|
self.lock_api = global_cmd._pymol.lock_api
|
|
self.lock_api_data = global_cmd._pymol.lock_api_data
|
|
self.lock_api_glut = global_cmd._pymol.lock_api_glut
|
|
self.lock_api_status = global_cmd._pymol.lock_api_status
|
|
|
|
self.lock_api_allow_flush = 1
|
|
self.lockcm = global_cmd.LockCM(self)
|
|
|
|
# now we create the command langauge
|
|
|
|
from pymol import keywords
|
|
|
|
self.keyword = keywords.get_command_keywords()
|
|
self.kw_list = list(self.keyword)
|
|
|
|
keywords.fix_list(self.kw_list)
|
|
self.kwhash = self.Shortcut(self.kw_list)
|
|
keywords.fix_dict(self.keyword)
|
|
|
|
self.help_only = keywords.get_help_only_keywords()
|
|
self.help_sc = self.Shortcut(
|
|
itertools.chain(self.keyword, self.help_only))
|
|
|
|
|
|
self.selection_sc = lambda sc=self.Shortcut,gn=self.get_names:sc(gn('public')+['all'])
|
|
self.object_sc = lambda sc=self.Shortcut,gn=self.get_names:sc(gn('objects'))
|
|
self.map_sc = lambda sc=self.Shortcut,gnot=self.get_names_of_type:sc(gnot('object:map'))
|
|
self.contour_sc = lambda sc=self.Shortcut,gnot=self.get_names_of_type:sc(gnot('object:mesh')+gnot('object:surface'))
|
|
self.group_sc = lambda sc=self.Shortcut,gnot=self.get_names_of_type:sc(gnot('object:group'))
|
|
|
|
self.fb_action_sc = pymol.feedingback.fb_action_sc
|
|
self.fb_module_sc = pymol.feedingback.fb_module_sc
|
|
self.fb_mask_sc = pymol.feedingback.fb_mask_sc
|
|
|
|
self.auto_arg = pymol.completing.get_auto_arg_list(self)
|
|
self.color_sc = None
|
|
|
|
# keyboard configuration
|
|
|
|
from pymol import keyboard
|
|
|
|
self.key_mappings = keyboard.get_default_keys(self)
|
|
|
|
# PUBLIC API METHODS which expect "self" as the first argument
|
|
|
|
def __getattr__(self, key):
|
|
v = getattr(global_cmd, key)
|
|
# determine whether `v` is a callable and has a `_self` argument or
|
|
# a keywords dictionary
|
|
i = -1
|
|
try:
|
|
argspec = inspect.getfullargspec(v)
|
|
if '_self' not in argspec.kwonlyargs:
|
|
try:
|
|
i = argspec.args.index('_self')
|
|
except ValueError:
|
|
if argspec.varkw is None:
|
|
raise TypeError
|
|
except TypeError:
|
|
setattr(self, key, v)
|
|
return v
|
|
# don't bind a circular reference into the wrapper
|
|
# namespace, use a weak reference instead
|
|
cmdref = self._weakref
|
|
def wrapper(*a, **k):
|
|
if i == -1 or len(a) <= i:
|
|
k['_self'] = cmdref()
|
|
return v(*a, **k)
|
|
wrapper.__name__ = key
|
|
setattr(self, key, wrapper)
|
|
return wrapper
|