summaryrefslogtreecommitdiffstats
path: root/src/adhoc.py
blob: 517aca8531fa179699ab6aac82e382dbe477ad81 (plain) (blame)
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
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
#!/usr/bin/env python3

# Copyright © 2014, 2015, 2016, 2017  Mattias Andrée (maandree@kth.se)
# 
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# This module contains the implementions of the policies for ad-hoc mode.

import sys
import time
import signal
import datetime


## Warn if we are using configuration script arguments
if len(parser.files) > 0:
    print('%s: warning: configuration script arguments are not supported in ad-hoc mode' % sys.argv[0])

## Determine whether we should run in continuous mode
continuous = any(map(lambda a : (a is not None) and (len(a) == 2), settings)) or (location is not None)

## Select default settings when not specified
d = lambda a, default : [default, default] if a is None else (a * 2 if len(a) == 1 else a)
# Set gamma and brightness to 1 (unmodified) if not specified
gammas = d(gammas, '1:1:1')
rgb_brightnesses = d(rgb_brightnesses, '1')
cie_brightnesses = d(cie_brightnesses, '1')
if (rgb_temperatures is None) and (cie_temperatures is None):
    # If temperature is not specified, set the temperature
    # to 3700 K during the day, and 6500 K (neutral) during
    # the night. Do not use CIE xyY, hence set cie_temperatures
    # to 6500 K (neutral).
    rgb_temperatures = ['3500', '5500']
    cie_temperatures = ['6500', '6500']
else:
    # If cie_temperatures is specified but not rgb_temperatures,
    # then set rgb_temperatures to 6500 K. But if rgb_temperatures
    # is indeed specified but only with one value, duplicate that
    # value so that it is used both during day and night.
    if rgb_temperatures is None:
        rgb_temperatures = ['6500', '6500']
    elif len(rgb_temperatures) == 1:
        rgb_temperatures *= 2
    # And vice versa.
    if cie_temperatures is None:
        cie_temperatures = ['6500', '6500']
    elif len(cie_temperatures) == 1:
        cie_temperatures *= 2

## Parse string arrays into floating point matrices
# Pack all settings into an unregulare matrix
settings = [gammas, rgb_brightnesses, cie_brightnesses, rgb_temperatures, cie_temperatures, [location]]
# Parse into a vector-matrix
s = lambda f, v : f(v) if v is not None else None
settings = [s(lambda c : [s(lambda x : [float(y) for y in x.split(':')], x) for x in c], c) for c in settings]
# Unpack settings
[gammas, rgb_brightnesses, cie_brightnesses, rgb_temperatures, cie_temperatures, location] = settings
location = None if location is None else location[0]

## Select method for calculating to what degree the adjustments should be applied
# Assume it is day if not running in continuous mode
alpha = lambda : 1
if continuous:
    if location is not None:
        # If we have a geographical location, determine to
        # what degree it is day from the Sun's elecation
        alpha = lambda : sun(*location)
    else:
        # If we do not have a location, determine to what
        # degree it is day from local time,
        def alpha():
            '''
            This algorithm is very crude.
            It places 100 % day at 12:00 and 100 % night at
            22:00, and only at those exact time points.
            
            @return  :float  [0, 1] floating point of the degree to which it is day
            '''
            now = datetime.datetime.now()
            hh, mm = now.hour, now.minute + now.second / 60
            if 12 <= hh <= 22:
                return 1 - (hh - 12) / (22 - 12) - mm / 60
            return (hh + (10 if hh <= 12 else 0) - 22) / 14 + mm / 60

## Set monitor control
output_ = []
for o in output:
    output_ += [int(x) for x in o.split(',')]
# Use selected CRTC:s (all if none are selected)
# in the first screen or graphics card.
monitor_controller = lambda : (drm if ttymode else randr)(*output_)

# Interpolation between day and night and between pure and adjusted
interpol_ = lambda d, p, a, r : d * r + (p[0] * a + p[1] * (1 - a)) * (1 - r)

def apply(dayness, pureness):
    '''
    Apply adjustments
    
    @param  dayness:float   The visibility of the sun
    @param  pureness:float  Transitioning progress, 1 for at clean state, 0 for at adjusted state
    '''
    # Clean up adjustments from last run
    start_over()
    # Interpolation between day and night and between pure and adjusted
    interpol = lambda d, p : [interpol_(d, [p[0][i], p[1][i]], dayness, pureness) for i in range(len(p[0]))]
    # Apply temperature adjustment
    temperature_algorithm = lambda t : clip_whitepoint(divide_by_maximum(cmf_10deg(t)))
    rgb_temperature(*interpol(6500, rgb_temperatures), algorithm = temperature_algorithm)
    cie_temperature(*interpol(6500, cie_temperatures), algorithm = temperature_algorithm)
    # Apply white point brightness adjustment
    rgb_brightness(*interpol(1, rgb_brightnesses))
    cie_brightness(*interpol(1, cie_brightnesses))
    # Clip values, specifically those under zero to
    # avoid complex numbers, and apply gamma correction
    clip()
    gamma(*interpol(1, gammas))
    # Clip values to avoid unwanted effects on oversaturation
    clip()
    # Apply settings to all selected monitors
    monitor_controller()

if continuous and not doreset:
    ## Continuous mode
    def periodically(_year, _month, _day, _hour, _minute, _second, _weekday, fade):
        '''
        Invoked periodically
        
        @param  fade:float?  Blueshift can use this function to fade into a state when it start
                             or exits. `fade` can either be negative, zero or positive or `None`,
                             but the magnitude of value cannot exceed 1. When Blueshift starts,
                             this function will be invoked multiple with the time parameters
                             of the time it is invoked and each time `fade` will increase towards
                             1, starting at 0, when the value is 1, the settings should be applied
                             to 100 %. After this this function will be invoked once again with
                             `fade` being `None`. When Blueshift exits the same behaviour is used
                             except, `fade` decrease towards -1 but start slightly below 0, when
                             -1 is reached all settings should be normal. Then Blueshift will NOT
                             invoke this function with `fade` being `None`, instead it will by
                             itself revert all settings and quit.
        '''
        apply(alpha(), 0 if fade is None else 1 - abs(fade))
else:
    ## One shot mode
    if not panicgate:
        ## Fade in to settings
        signal.signal(signal.SIGTERM, signal_SIGTERM)
        trans = 0
        while running and (trans < 1):
            try:
                apply(alpha(), trans if doreset else 1 - trans)
                trans += 0.05
                time.sleep(0.1)
            except KeyboardInterrupt:
                signal_SIGTERM(0, None)
    ## Apply or revert settings and exit
    apply(alpha(), 1 if doreset else 0)