Added PinArray class.

1 files changed, 42 insertions, 0 deletions

diff --git a/gpio/gpio.h b/gpio/gpio.h
index 9d95b1d..705f3cf 100644
--- a/gpio/gpio.h
+++ b/gpio/gpio.h
@@ -143,14 +143,56 @@ class GPIO_t {
 				}
 		};
 		
+		class PinArray {
+			private:
+				const GPIO_t& g;
+				int f;
+				int l;
+				
+				constexpr uint32_t mask1() {
+					return ((2 << l) - 1) ^ ((1 << f) - 1);
+				}
+				
+				constexpr uint32_t mask2() {
+					return ((4 << (l * 2)) - 1) ^ ((1 << (f * 2)) - 1);
+				}
+			
+			public:
+				constexpr PinArray(const GPIO_t& gpio, int first, int last) : g(gpio), f(first), l(last) {}
+				
+				#if defined(STM32F3) || defined(STM32F4)
+				void set_mode(Pin::Mode m) {
+					g.reg.MODER = (g.reg.MODER & ~mask2()) | ((0x55555555 * m) & mask2());
+				}
+				
+				void set_pull(Pin::Pull p) {
+					g.reg.PUPDR = (g.reg.PUPDR & ~mask2()) | ((0x55555555 * p) & mask2());
+				}
+				#endif
+				
+				void set(uint16_t value) {
+					value <<= f;
+					g.reg.BSRR = ((~value & mask1()) << 16) | (value & mask1());
+				}
+				
+				uint16_t get() {
+					return g.reg.IDR & mask1();
+				}
+		};
+		
 		constexpr GPIO_t(uint32_t reg_addr) : reg(*(GPIO_reg_t*)reg_addr) {}
 		
 		constexpr Pin operator[](int pin) {
 			return Pin(*this, pin);
 		}
+		
+		constexpr PinArray array(int first, int last) {
+			return PinArray(*this, first, last);
+		}
 };
 
 typedef GPIO_t::Pin Pin;
+typedef GPIO_t::PinArray PinArray;
 
 #if defined(STM32F1)
 static GPIO_t GPIOA(0x40010800);