2 files changed, 177 insertions, 0 deletions
diff --git a/test_fw/include/i2c_scan.h b/test_fw/include/i2c_scan.h
new file mode 100644
index 0000000..1fc81c1
--- /dev/null
+++ b/test_fw/include/i2c_scan.h
@@ -0,0 +1,37 @@
+#pragma once
+
+#include <Arduino.h>
+#include <Wire.h>
+
+namespace ocularium {
+    inline void scan(TwoWire& wire, bool post_delay = true)
+    {
+        for(auto address = 1; address < 127; address++)
+        {
+            wire.beginTransmission(address);
+            const uint8_t error = wire.endTransmission();
+
+            if (error == 0)
+            {
+                Serial.print("I2C device found at address 0x");
+                if (address<16)
+                    Serial.print("0");
+                Serial.print(address,HEX);
+                Serial.println();
+            }
+            else if (error==4)
+            {
+                Serial.print("Unknown error at address 0x");
+                if (address<16)
+                    Serial.print("0");
+                Serial.println(address,HEX);
+            }
+        }
+
+        if (post_delay)
+        {
+            // Let the bus settle before doing anything else.
+            delay(10);
+        }
+    }
+}
+\ No newline at end of file
diff --git a/test_fw/include/veml.h b/test_fw/include/veml.h
new file mode 100644
index 0000000..89282c8
--- /dev/null
+++ b/test_fw/include/veml.h
@@ -0,0 +1,140 @@
+#pragma once
+
+#include <Wire.h>
+
+namespace ocularium
+{
+    namespace veml
+    {
+        enum class Reg: uint8_t {
+            CONFIG = 0x0,
+            THRS_HIGH = 0x1,
+            THRS_LOW = 0x2,
+            POWER = 0x3,
+            LUX = 0x4,
+            WHITE = 0x5,
+            INT_STATUS = 0x6,
+            CHIPID = 0x7,
+        };
+
+        enum class AmbientLightGain: uint8_t
+        {
+            Unity = 0b00,
+            TwoX = 0b01,
+            Quarter = 0b11,
+            Eighth = 0b10,
+        };
+
+        enum class IntegrationTime: uint8_t
+        {
+            Ms25 = 0b1100,
+            Ms50 = 0b1000,
+            Ms100 = 0b0000,
+            Ms200 = 0b0001,
+            Ms400 = 0b0010,
+            Ms800 = 0b0011,
+        };
+
+        enum class PowerMode: uint8_t
+        {
+            Mode1 = 0b00,
+            Mode2 = 0b01,
+            Mode3 = 0b10,
+            Mode4 = 0b11,
+        };
+        
+        struct Config
+        {
+            bool shutdown = false;
+            bool interrupt = false;
+            AmbientLightGain gain = AmbientLightGain::Eighth;
+            IntegrationTime integration = IntegrationTime::Ms100;
+
+            uint16_t to_reg() const;
+            
+            int32_t gain_factor() const
+            {
+                int32_t gain = 0;
+                
+                switch (this->gain)
+                {
+                    case AmbientLightGain::TwoX:
+                        gain = 1;
+                        break;
+                    
+                    case AmbientLightGain::Unity:
+                        gain = 2;
+                        break;
+                    
+                    case AmbientLightGain::Eighth:
+                        gain = 8;
+                        break;
+                    
+                    case AmbientLightGain::Quarter:
+                        gain = 16;
+                        break;
+
+                    default:
+                        return -1;
+                }
+
+                int32_t integ = 0;
+
+                switch (integration)
+                {
+                    case IntegrationTime::Ms800:
+                        integ = 1;
+                        break;
+                        
+                    case IntegrationTime::Ms400:
+                        integ = 2;
+                        break;
+
+                    case IntegrationTime::Ms200:
+                        integ = 4;
+                        break;
+
+                    case IntegrationTime::Ms100:
+                        integ = 8;
+                        break;
+                        
+                    case IntegrationTime::Ms50:
+                        integ = 16;
+                        break;
+                    
+                    case IntegrationTime::Ms25:
+                        integ = 32;
+                        break;
+                    
+                    default:
+                        return -1;
+                }
+
+                return integ * gain;
+            }
+        };
+        
+        struct VEML {
+            static constexpr uint8_t ADDR = 0x10;
+            static constexpr uint16_t CHIPID = 0xc481;
+
+            explicit VEML(TwoWire& wire) : wire(wire) {}
+
+            bool init(const Config& config = Config{});
+        
+            uint16_t chipid() const;
+            float lux() const;
+
+            void set_config(const Config& config);
+
+        private:
+            TwoWire& wire;
+            Config config = Config{};
+            
+            void write_reg(Reg reg, uint16_t val) const;
+            uint16_t read_reg(Reg reg) const;
+        };
+    }
+
+    using veml::VEML;
+}
+\ No newline at end of file