Mercurial > code > home > repos > light9
annotate flax/Submaster.py @ 205:3905d3c92aaa
twisted mainloop, more row-change keys, xmlrpc fadesub command on port 8050
| author | drewp |
|---|---|
| date | Sun, 10 Apr 2005 15:03:24 +0000 |
| parents | 79bc84310e80 |
| children |
| rev | line source |
|---|---|
| 0 | 1 from __future__ import division |
| 2 from TLUtility import dict_scale, dict_max | |
| 3 | |
| 4 import sys | |
| 5 sys.path.append('../light8') | |
| 6 | |
| 7 import Patch | |
| 8 | |
| 9 class Submaster: | |
| 10 "Contain a dictionary of levels, but you didn't need to know that" | |
| 167 | 11 def __init__(self, name, leveldict=None, temporary=0): |
| 0 | 12 self.name = name |
| 167 | 13 self.temporary = temporary |
| 0 | 14 if leveldict: |
| 15 self.levels = leveldict | |
| 16 else: | |
| 17 self.levels = {} | |
| 18 self.reload() | |
| 19 def reload(self): | |
| 167 | 20 if self.temporary: |
| 21 return | |
| 0 | 22 try: |
| 23 self.levels.clear() | |
| 24 subfile = file("subs/%s" % self.name) | |
| 25 for line in subfile.readlines(): | |
| 26 if not line.strip(): # if line is only whitespace | |
| 27 continue # "did i say newspace?" | |
| 28 | |
| 29 try: | |
| 30 name, val = line.split(':') | |
| 31 name = name.strip() | |
| 32 self.levels[name] = float(val) | |
| 33 except ValueError: | |
| 34 print "(%s) Error with this line: %s" % (self.name, | |
| 35 line[:-1]) | |
| 36 except IOError: | |
| 37 print "Can't read file for sub: %s" % self.name | |
| 38 def save(self): | |
| 167 | 39 if self.temporary: |
|
205
3905d3c92aaa
twisted mainloop, more row-change keys, xmlrpc fadesub command on port 8050
drewp
parents:
167
diff
changeset
|
40 print "not saving temporary sub named",self.name |
| 167 | 41 return |
| 42 | |
| 0 | 43 subfile = file("subs/%s" % self.name, 'w') |
| 44 names = self.levels.keys() | |
| 45 names.sort() | |
| 46 for name in names: | |
| 47 val = self.levels[name] | |
| 48 subfile.write("%s : %s\n" % (name, val)) | |
| 49 def set_level(self, channelname, level, save=1): | |
| 50 self.levels[Patch.resolve_name(channelname)] = level | |
| 51 if save: | |
| 52 self.save() | |
| 53 def set_all_levels(self, leveldict): | |
| 54 self.levels.clear() | |
| 55 for k, v in leveldict.items(): | |
| 56 self.set_level(k, v, save=0) | |
| 57 self.save() | |
| 58 def get_levels(self): | |
| 59 return self.levels | |
| 60 def __mul__(self, scalar): | |
| 61 return Submaster("%s*%s" % (self.name, scalar), | |
| 167 | 62 dict_scale(self.levels, scalar), temporary=1) |
| 0 | 63 __rmul__ = __mul__ |
| 64 def max(self, *othersubs): | |
| 65 return sub_maxes(self, *othersubs) | |
| 66 def __repr__(self): | |
| 138 | 67 levels = ' '.join(["%s:%.2f" % item for item in self.levels.items()]) |
| 68 return "<'%s': [%s]>" % (self.name, levels) | |
| 0 | 69 def get_dmx_list(self): |
| 70 leveldict = self.get_levels() # gets levels of sub contents | |
| 71 | |
| 72 levels = [0] * 68 | |
| 73 for k, v in leveldict.items(): | |
| 138 | 74 dmxchan = Patch.get_dmx_channel(k) - 1 |
| 75 levels[dmxchan] = max(v, levels[dmxchan]) | |
| 0 | 76 |
| 77 return levels | |
| 138 | 78 def normalize_patch_names(self): |
| 167 | 79 """Use only the primary patch names.""" |
| 138 | 80 # possibly busted -- don't use unless you know what you're doing |
| 81 self.set_all_levels(self.levels.copy()) | |
| 167 | 82 def get_normalized_copy(self): |
| 83 """Get a copy of this sumbaster that only uses the primary patch | |
| 84 names. The levels will be the same.""" | |
| 85 newsub = Submaster("%s (normalized)" % self.name, temporary=1) | |
| 86 newsub.set_all_levels(self.levels) | |
| 87 return newsub | |
| 88 def crossfade(self, othersub, amount): | |
| 89 """Returns a new sub that is a crossfade between this sub and | |
| 90 another submaster. | |
| 91 | |
| 92 NOTE: You should only crossfade between normalized submasters.""" | |
| 93 otherlevels = othersub.get_levels() | |
| 94 keys_set = {} | |
| 95 for k in self.levels.keys() + otherlevels.keys(): | |
| 96 keys_set[k] = 1 | |
| 97 all_keys = keys_set.keys() | |
| 98 | |
| 99 xfaded_sub = Submaster("xfade", temporary=1) | |
| 100 for k in all_keys: | |
| 101 xfaded_sub.set_level(k, | |
| 102 linear_fade(self.levels.get(k, 0), | |
| 103 otherlevels.get(k, 0), | |
| 104 amount)) | |
| 105 | |
| 106 return xfaded_sub | |
| 107 | |
| 108 def linear_fade(start, end, amount): | |
| 109 """Fades between two floats by an amount. amount is a float between | |
| 110 0 and 1. If amount is 0, it will return the start value. If it is 1, | |
| 111 the end value will be returned.""" | |
| 112 level = start + (amount * (end - start)) | |
| 113 return level | |
| 0 | 114 |
| 115 def sub_maxes(*subs): | |
| 116 return Submaster("max(%r)" % (subs,), | |
| 167 | 117 dict_max(*[sub.levels for sub in subs]), temporary=1) |
| 0 | 118 |
| 119 class Submasters: | |
| 120 "Collection o' Submaster objects" | |
| 121 def __init__(self): | |
| 122 self.submasters = {} | |
| 123 | |
| 124 import os | |
| 125 files = os.listdir('subs') | |
| 126 | |
| 127 for filename in files: | |
| 138 | 128 # we don't want these files |
| 129 if filename.startswith('.') or filename.endswith('~') or \ | |
| 130 filename.startswith('CVS'): | |
| 0 | 131 continue |
| 132 self.submasters[filename] = Submaster(filename) | |
| 133 def get_all_subs(self): | |
| 134 "All Submaster objects" | |
|
165
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
135 l = self.submasters.items() |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
136 l.sort() |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
137 l = [x[1] for x in l] |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
138 songs = [] |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
139 notsongs = [] |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
140 for s in l: |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
141 if s.name.startswith('song'): |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
142 songs.append(s) |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
143 else: |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
144 notsongs.append(s) |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
145 combined = notsongs + songs |
|
1fe54442db38
get_all_subs sorts, and puts some elements at the end
drewp
parents:
138
diff
changeset
|
146 return combined |
| 0 | 147 def get_sub_by_name(self, name): |
| 148 "Makes a new sub if there isn't one." | |
| 149 return self.submasters.get(name, Submaster(name)) | |
| 150 __getitem__ = get_sub_by_name | |
| 151 | |
| 152 if __name__ == "__main__": | |
| 153 Patch.reload_data() | |
| 154 s = Submasters() | |
| 155 print s.get_all_subs() | |
| 138 | 156 if 0: # turn this on to normalize all subs |
| 157 for sub in s.get_all_subs(): | |
| 158 print "before", sub | |
| 159 sub.normalize_patch_names() | |
| 160 sub.save() | |
| 161 print "after", sub |