Update the LED pattern helper to use a more robust testing method
[led-pattern-ed] / src / led-pattern-helper.vala
index cda11da..5bd07c6 100644 (file)
  * along with LED Pattern Editor. If not, see <http://www.gnu.org/licenses/>.
  */
 
+bool sysfs_write (string filename, string contents) {
+       var f = FileStream.open (filename, "w");
+       if (f == null) {
+               stderr.printf ("failed to open '%s' for writing\n", filename);
+               return false;
+       }
+       f.printf ("%s", contents);
+       return true;
+}
+
 public bool test_pattern (string pattern, bool apply) {
        string[] key_value = pattern.split ("=");
 
@@ -39,44 +49,72 @@ public bool test_pattern (string pattern, bool apply) {
        // screen_on = p[1].to_int ();
        // timeout = p[2].to_int ();
 
-       int[] led_currents = { 2, 2, 2 };
-       string led_map1 = "000";
-       switch (p[3]) {
-       case "r":
-               led_map1 = "001";
-               led_currents = { 8, 0, 2 };
+       int led_map1 = 0x000;
+       int led_map2 = 0x000;
+       if ("r" in p[3])
+               led_map1 |= 0x001;
+       if ("R" in p[3])
+               led_map2 |= 0x001;
+       if ("g" in p[3])
+               led_map1 |= 0x010;
+       if ("G" in p[3])
+               led_map2 |= 0x010;
+       if ("b" in p[3])
+               led_map1 |= 0x100;
+       if ("B" in p[3])
+               led_map2 |= 0x100;
+
+       if ((led_map1 & led_map2) != 0) {
+               stderr.printf ("pattern assigns a led to both channels: '%s'\n", p[3]);
+               return false;
+       }
+
+       string led_map = "";
+       if (0x001 in led_map1)
+               led_map += "r";
+       if (0x001 in led_map2)
+               led_map += "R";
+       if (0x010 in led_map1)
+               led_map += "g";
+       if (0x010 in led_map2)
+               led_map += "G";
+       if (0x100 in led_map1)
+               led_map += "b";
+       if (0x100 in led_map2)
+               led_map += "B";
+
+       if (led_map != p[3]) {
+               stderr.printf ("pattern contains invalid led map: '%s\n", p[3]);
+               return false;
+       }
+
+       string[] led_currents = { "2", "2", "2" };
+       switch (led_map1 | led_map2) {
+       case 0x001:
+               led_currents = { "8", "0", "2" };
                break;
-       case "g":
-               led_map1 = "010";
-               led_currents = { 2, 2, 2 };
+       case 0x010:
+               led_currents = { "2", "2", "2" };
                break;
-       case "b":
-               led_map1 = "100";
-               led_currents = { 2, 2, 2 };
+       case 0x100:
+               led_currents = { "2", "2", "2" };
                break;
-       case "rg":
-               led_map1 = "011";
-               led_currents = { 20, 2, 0 };
+       case 0x011:
+               led_currents = { "20", "2", "0" };
                break;
-       case "rb":
-               led_map1 = "101";
+       case 0x101:
                // TODO: led_currents?
                break;
-       case "gb":
-               led_map1 = "110";
+       case 0x110:
                // TODO: led_currents?
                break;
-       case "rgb":
-               led_map1 = "111";
-               led_currents = { 8, 2, 2 };
+       case 0x111:
+               led_currents = { "8", "2", "2" };
                break;
-       default:
-               stderr.printf ("only single-engine patterns supported for now\n");
-               return false;
        }
 
        if (p[4].length > 16*4 || p[5].length > 16*4) {
-               stderr.printf ("pattern too long!\n");
+               stderr.printf ("engine1 pattern too long!\n");
                return false;
        }
 
@@ -90,70 +128,47 @@ public bool test_pattern (string pattern, bool apply) {
                return false;
        }
 
-       if (p[5] != "9d800000") {
-               // FIXME
-               stderr.printf ("only single-engine patterns supported for now\n");
-               return false;
-       }
-
-       if (apply == false)
-               return true;
-
-       string i2c_dev = "/sys/bus/i2c/devices/2-0032";
-
-       var f = FileStream.open ("/sys/class/leds/lp5523:r/led_current", "w");
-       if (f == null) {
-               stderr.printf ("failed to set red led current\n");
-               return false;
-       }
-       f.printf ("%d\n", led_currents[0]);
-
-       f = FileStream.open ("/sys/class/leds/lp5523:g/led_current", "w");
-       if (f == null) {
-               stderr.printf ("failed to set green led current\n");
-               return false;
-       }
-       f.printf ("%d\n", led_currents[1]);
-
-       f = FileStream.open ("/sys/class/leds/lp5523:b/led_current", "w");
-       if (f == null) {
-               stderr.printf ("failed to set blue led current\n");
+       if (p[5].length > 16*4 || p[5].length > 16*4) {
+               stderr.printf ("engine2 pattern too long!\n");
                return false;
        }
-       f.printf ("%d\n", led_currents[2]);
 
-       f = FileStream.open (Path.build_filename (i2c_dev, "engine1_mode"), "w");
-       if (f == null) {
-               stderr.printf ("failed to set engine1 to load mode\n");
+       if (!(p[5].has_prefix ("9d80"))) {
+               stderr.printf ("engine2 pattern doesn't start with reset mux command\n");
                return false;
        }
-       f.printf ("load\n");
 
-       int retries = 100;
-       for (int i = 0; i < retries; i++) {
-               f = FileStream.open (Path.build_filename (i2c_dev, "engine1_leds"), "w");
-               if (f != null)
-                       break;
-       }
-       if (f == null) {
-               stderr.printf ("failed to set engine1 mux\n");
+       if (!(p[5].has_suffix ("0000")) && !(p[5].has_suffix ("c000"))) {
+               stderr.printf ("engine2 pattern doesn't end with repeat or stop command\n");
                return false;
        }
-       f.printf ("0000%s00\n", led_map1);
 
-       f = FileStream.open (Path.build_filename (i2c_dev, "engine1_load"), "w");
-       if (f == null) {
-               stderr.printf ("failed to load engine1 pattern\n");
-               return false;
-       }
-       f.printf ("%s\n", p[4]);
+       if (apply == false)
+               return true;
 
-       f = FileStream.open (Path.build_filename (i2c_dev, "engine1_mode"), "w");
-       if (f == null) {
-               stderr.printf ("failed to set engine1 to run mode\n");
-               return false;
-       }
-       f.printf ("run\n");
+       string mux1 = "0000%03x00\n".printf (led_map1);
+       string mux2 = "0000%03x00\n".printf (led_map2);
+       string pattern1 = p[4];
+       string pattern2 = p[5];
+
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine1_mode", "disabled");
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine2_mode", "disabled");
+       sysfs_write ("/sys/class/leds/lp5523:r/brightness", "0");
+       sysfs_write ("/sys/class/leds/lp5523:g/brightness", "0");
+       sysfs_write ("/sys/class/leds/lp5523:b/brightness", "0");
+
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine1_mode", "load");
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine1_leds", mux1);
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine1_load", pattern1);
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine2_mode", "load");
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine2_leds", mux2);
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine2_load", pattern2);
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine2_mode", "run");
+       sysfs_write ("/sys/class/i2c-adapter/i2c-2/2-0032/engine1_mode", "run");
+
+       sysfs_write ("/sys/class/leds/lp5523:r/led_current", led_currents[0]);
+       sysfs_write ("/sys/class/leds/lp5523:g/led_current", led_currents[1]);
+       sysfs_write ("/sys/class/leds/lp5523:b/led_current", led_currents[2]);
 
        return true;
 }