1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
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
|
