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
143
144
145
146
147
|
#!/usr/bin/env python
import math
interactive = False
def main(args):
"""Calculates PWM levels for visually-linear steps.
"""
# Get parameters from the user
questions_main = [
(int, 'num_channels', 1, 'How many power channels?'),
(int, 'num_levels', 4, 'How many total levels do you want?'),
]
questions_per_channel = [
(int, 'pwm_min', 6, 'Lowest visible PWM level:'),
(float, 'lm_min', 0.25, 'How bright is the lowest level, in lumens?'),
#(int, 'pwm_max', 255, 'Highest PWM level:'),
(float, 'lm_max', 1000, 'How bright is the highest level, in lumens?'),
]
def ask(questions, ans):
for typ, name, default, text in questions:
value = get_value(text, default, args)
if not value:
value = default
else:
value = typ(value)
setattr(ans, name, value)
answers = Empty()
ask(questions_main, answers)
channels = []
if not args:
print('Describe the channels in order of lowest to highest power.')
for chan_num in range(answers.num_channels):
if not args:
print('===== Channel %s =====' % (chan_num+1))
chan = Empty()
chan.pwm_max = 255
ask(questions_per_channel, chan)
channels.append(chan)
multi_pwm(answers, channels)
if interactive: # Wait on exit, in case user invoked us by clicking an icon
print 'Press Enter to exit:'
raw_input()
class Empty:
pass
def multi_pwm(answers, channels):
lm_min = channels[0].lm_min
lm_max = channels[-1].lm_max
visual_min = invpower(lm_min)
visual_max = invpower(lm_max)
step_size = (visual_max - visual_min) / (answers.num_levels-1)
# Determine ideal lumen levels
goals = []
goal_vis = visual_min
for i in range(answers.num_levels):
goal_lm = power(goal_vis)
goals.append((goal_vis, goal_lm))
goal_vis += step_size
# Calculate each channel's output for each level
for cnum, channel in enumerate(channels):
prev_channel = Empty() ; prev_channel.lm_max = 0.0
if cnum > 0:
prev_channel = channels[cnum-1]
channel.modes = []
for i in range(answers.num_levels):
goal_vis, goal_lm = goals[i]
# This channel already is maxed out
if goal_lm >= channel.lm_max:
# Handle turbo specially, enable only biggest channel
if (i == (answers.num_levels - 1)) and (cnum < (len(channels)-1)):
channel.modes.append(0.0)
else:
channel.modes.append(channel.pwm_max)
# This channel's active ramp-up range
elif goal_lm > prev_channel.lm_max:
# FIXME: This produces somewhat different values than the
# dual_pwm() algorithm, and I'm not sure which one is "right"
diff = channel.lm_max - prev_channel.lm_max
needed = goal_lm - prev_channel.lm_max
ratio = needed / diff * (channel.pwm_max-channel.pwm_min)
channel.modes.append(ratio + channel.pwm_min)
# This channel isn't active yet, output too low
else:
channel.modes.append(0)
# Show individual levels in detail
for i in range(answers.num_levels):
goal_vis, goal_lm = goals[i]
pwms = []
for channel in channels:
pwms.append('%.2f/%i' % (channel.modes[i], channel.pwm_max))
print('%i: visually %.2f (%.2f lm): %s' %
(i+1, goal_vis, goal_lm, ', '.join(pwms)))
# Show values we can paste into source code
for cnum, channel in enumerate(channels):
print('PWM%s values: %s' %
(cnum+1,
','.join([str(int(round(i))) for i in channel.modes])))
def get_value(text, default, args):
"""Get input from the user, or from the command line args."""
if args:
result = args[0]
del args[0]
else:
global interactive
interactive = True
print text, '(%s)' % (default),
result = raw_input()
result = result.strip()
return result
def power(x):
#return x**5
return x**3
#return x**2
#return math.e**x
#return 2.0**x
def invpower(x):
#return math.pow(x, 1/5.0)
return math.pow(x, 1/3.0)
#return math.pow(x, 1/2.0)
#return math.log(x, math.e)
#return math.log(x, 2.0)
if __name__ == "__main__":
import sys
main(sys.argv[1:])
|
