Welcome to mirror list, hosted at ThFree Co, Russian Federation.

textplugin_outliner.py « scripts « release - git.blender.org/blender.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 3879a2819a547c7486d34d1a82f7202d805e1736 (plain)
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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
#!BPY
"""
Name: 'Code Outline | Ctrl T'
Blender: 246
Group: 'TextPlugin'
Shortcut: 'Ctrl+T'
Tooltip: 'Provides a menu for jumping to class and functions definitions.'
"""

# Only run if we have the required modules
try:
	import bpy
	from BPyTextPlugin import *
	from Blender import Draw
except ImportError:
	OK = False
else:
	OK = True

def make_menu(items, eventoffs):
	n = len(items)
	if n < 20:
		return [(items[i], i+1+eventoffs) for i in range(len(items))]
	
	letters = []
	check = 'abcdefghijklmnopqrstuvwxyz_' # Names cannot start 0-9
	for c in check:
		for item in items:
			if item[0].lower() == c:
				letters.append(c)
				break
	
	entries = {}
	i = 0
	for item in items:
		i += 1
		c = item[0].lower()
		entries.setdefault(c, []).append((item, i+eventoffs))
	
	subs = []
	for c in letters:
		subs.append((c, entries[c]))
	
	return subs

def find_word(txt, word):
	i = 0
	txt.reset()
	while True:
		try:
			line = txt.readline()
		except StopIteration:
			break
		c = line.find(word)
		if c != -1:
			txt.setCursorPos(i, c)
			break
		i += 1

def main():
	txt = bpy.data.texts.active
	if not txt:
		return
	
	# Identify word under cursor
	if get_context(txt) == CTX_NORMAL:
		line, c = current_line(txt)
		start = c-1
		end = c
		while start >= 0:
			if not line[start].lower() in 'abcdefghijklmnopqrstuvwxyz0123456789_':
				break
			start -= 1
		while end < len(line):
			if not line[end].lower() in 'abcdefghijklmnopqrstuvwxyz0123456789_':
				break
			end += 1
		word = line[start+1:end]
		if word in KEYWORDS:
			word = None
	else:
		word = None
	
	script = get_cached_descriptor(txt)
	items = []
	desc = None
	
	tmp = script.classes.keys()
	tmp.sort(cmp = suggest_cmp)
	class_menu = make_menu(tmp, len(items))
	class_menu_length = len(tmp)
	items.extend(tmp)
	
	tmp = script.defs.keys()
	tmp.sort(cmp = suggest_cmp)
	defs_menu = make_menu(tmp, len(items))
	defs_menu_length = len(tmp)
	items.extend(tmp)
	
	tmp = script.vars.keys()
	tmp.sort(cmp = suggest_cmp)
	vars_menu = make_menu(tmp, len(items))
	vars_menu_length = len(tmp)
	items.extend(tmp)
	
	menu = [('Script %t', 0),
			('Classes', class_menu),
			('Functions', defs_menu),
			('Variables', vars_menu)]
	if word:
		menu.extend([None, ('Locate', [(word, -10)])])
	
	i = Draw.PupTreeMenu(menu)
	if i == -1:
		return
	
	# Chosen to search for word under cursor
	if i == -10:
		if script.classes.has_key(word):
			desc = script.classes[word]
		elif script.defs.has_key(word):
			desc = script.defs[word]
		elif script.vars.has_key(word):
			desc = script.vars[word]
		else:
			find_word(txt, word)
			return
	else:
		i -= 1
		if i < class_menu_length:
			desc = script.classes[items[i]]
		elif i < class_menu_length + defs_menu_length:
			desc = script.defs[items[i]]
		elif i < class_menu_length + defs_menu_length + vars_menu_length:
			desc = script.vars[items[i]]
	
	if desc:
		txt.setCursorPos(desc.lineno-1, 0)

# Check we are running as a script and not imported as a module
if __name__ == "__main__" and OK:
	main()