Initial commit

2 files changed, 198 insertions, 0 deletions

diff --git a/src/bin/main.rs b/src/bin/main.rs
new file mode 100644
index 0000000..7e69ae1
--- /dev/null
+++ b/src/bin/main.rs
@@ -0,0 +1,78 @@
+extern crate chrono;
+extern crate spa;
+
+use chrono::prelude::*;
+use spa::{SunriseAndSet, calc_sunrise_and_set};
+use std::os::raw::{c_ushort, c_void};
+use std::ptr;
+use std::thread;
+use std::time::Duration;
+use x11::xlib::{XDefaultScreen, XFree, XOpenDisplay, XRootWindow};
+use x11::xrandr::{
+    XRRAllocGamma, XRRCrtcGamma, XRRGetCrtcGammaSize, XRRGetScreenResourcesCurrent, XRRSetCrtcGamma,
+};
+
+fn set_temp(temp: u32) {
+    let ratio: f64 = (temp % 500) as f64 / 500f64;
+    unsafe {
+        let display = XOpenDisplay(ptr::null_mut());
+        let screen = XDefaultScreen(display);
+        let root = XRootWindow(display, screen);
+        let resource = XRRGetScreenResourcesCurrent(display, root);
+
+        for x in 0..(*resource).ncrtc {
+            let crtcxid = (*resource).crtcs.offset(x as isize);
+            let size = XRRGetCrtcGammaSize(display, *crtcxid);
+            let crtc_gamma: *mut XRRCrtcGamma = XRRAllocGamma(size);
+            let gamma = sctd::avg(temp, ratio);
+
+            for i in 0..size {
+                let g = (65535f64 * i as f64) / size as f64;
+                *((*crtc_gamma).red as *mut c_ushort).offset(i as isize) = (g * gamma.red) as u16;
+                *((*crtc_gamma).green as *mut c_ushort).offset(i as isize) =
+                    (g * gamma.green) as u16;
+                *((*crtc_gamma).blue as *mut c_ushort).offset(i as isize) = (g * gamma.blue) as u16;
+            }
+            XRRSetCrtcGamma(display, *crtcxid, crtc_gamma);
+            XFree(crtc_gamma as *mut c_void);
+        }
+    }
+}
+
+fn get_temp(utc: DateTime<Utc>, ss: &SunriseAndSet) -> u32 {
+    let low_temp = 2500;
+    let high_temp = 5500;
+
+    match *ss {
+        SunriseAndSet::Daylight(sunrise, sunset) => {
+            let since_sunrise = utc.signed_duration_since(sunrise);
+            let since_sunset = utc.signed_duration_since(sunset);
+
+            // this is where gradual increase/decrease should happen
+            if since_sunrise.num_seconds() < 0 {
+                return low_temp;
+            } else if since_sunset.num_seconds() < 0 {
+                return high_temp;
+            } else {
+                return low_temp;
+            }
+        },
+        SunriseAndSet::PolarDay => return high_temp,
+        SunriseAndSet::PolarNight => return low_temp,
+    }
+}
+
+fn main() {
+    // Dublin
+    let lat: f64 = 53.3498;
+    let lon: f64 = 6.2603;
+
+    loop {
+        let utc: DateTime<Utc> = Utc::now();
+        match calc_sunrise_and_set(utc, lat, lon) {
+            Ok(ss) => set_temp(get_temp(utc, &ss)),
+            Err(e) => println!("Error calculating sunrise and sunset: {:?}", e),
+        }
+        thread::sleep(Duration::from_secs(300));
+    }
+}
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..d3af757
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,120 @@
+#[derive(Debug)]
+pub struct WhitePoint {
+    pub red: f64,
+    pub green: f64,
+    pub blue: f64,
+}
+
+pub fn avg(temp: u32, ratio: f64) -> WhitePoint {
+    const WPS: [WhitePoint; 20] = [
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.18172716,
+            blue: 0.00000000,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.42322816,
+            blue: 0.00000000,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.54360078,
+            blue: 0.08679949,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.64373109,
+            blue: 0.28819679,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.71976951,
+            blue: 0.42860152,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.77987699,
+            blue: 0.54642268,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.82854786,
+            blue: 0.64816570,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.86860704,
+            blue: 0.73688797,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.90198230,
+            blue: 0.81465502,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.93853986,
+            blue: 0.88130458,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 0.97107439,
+            blue: 0.94305985,
+        },
+        WhitePoint {
+            red: 1.00000000,
+            green: 1.00000000,
+            blue: 1.00000000,
+        },
+        WhitePoint {
+            red: 0.95160805,
+            green: 0.96983355,
+            blue: 1.00000000,
+        },
+        WhitePoint {
+            red: 0.91194747,
+            green: 0.94470005,
+            blue: 1.00000000,
+        },
+        WhitePoint {
+            red: 0.87906581,
+            green: 0.92357340,
+            blue: 1.00000000,
+        },
+        WhitePoint {
+            red: 0.85139976,
+            green: 0.90559011,
+            blue: 1.00000000,
+        },
+        WhitePoint {
+            red: 0.82782969,
+            green: 0.89011714,
+            blue: 1.00000000,
+        },
+        WhitePoint {
+            red: 0.80753191,
+            green: 0.87667891,
+            blue: 1.00000000,
+        },
+        WhitePoint {
+            red: 0.78988728,
+            green: 0.86491137,
+            blue: 1.00000000,
+        },
+        WhitePoint {
+            red: 0.77442176,
+            green: 0.85453121,
+            blue: 1.00000000,
+        },
+    ];
+
+    WhitePoint {
+        red: WPS[(temp / 500) as usize].red * (1f64 - ratio)
+            + WPS[(temp / 500 + 1) as usize].red * ratio,
+        green: WPS[(temp / 500) as usize].green * (1f64 - ratio)
+            + WPS[(temp / 500 + 1) as usize].green * ratio,
+        blue: WPS[(temp / 500) as usize].blue * (1f64 - ratio)
+            + WPS[(temp / 500 + 1) as usize].blue * ratio,
+    }
+}