Add Apply overloads to Spark*Config

- Fixes #68

Author: virtuald

gen/SparkFlexConfig.yml

@@ -1,5 +1,8 @@
 ---
 
+extra_includes:
+- sparkbaseconfig_apply.h
+
 classes:
   SparkFlexConfig:
     attributes:
@@ -11,3 +14,6 @@ classes:
           SparkFlexConfig&:
           ExternalEncoderConfig&:
       Flatten:
+inline_code: |
+  bind_sparkbaseconfig_apply<SparkFlexConfig, decltype(cls_SparkFlexConfig)>(cls_SparkFlexConfig);
+

gen/SparkMaxConfig.yml

@@ -1,5 +1,8 @@
 ---
 
+extra_includes:
+- sparkbaseconfig_apply.h
+
 classes:
   SparkMaxConfig:
     attributes:
@@ -13,3 +16,5 @@ classes:
           SparkMaxConfig&:
           AlternateEncoderConfig&:
       Flatten:
+inline_code: |
+  bind_sparkbaseconfig_apply<SparkMaxConfig, decltype(cls_SparkMaxConfig)>(cls_SparkMaxConfig);

rev/sparkbaseconfig_apply.h

@@ -0,0 +1,43 @@
+#pragma once
+
+template <typename T, typename Cls>
+void bind_sparkbaseconfig_apply(Cls &cls) {
+    using namespace rev::spark;
+
+    auto sparkBaseConfigType = py::type::of<SparkBaseConfig>();
+
+    cls
+        .def("apply", [=](T &self, SparkBaseConfig &config) -> T& {
+            sparkBaseConfigType.attr("apply")(self, config);
+            return self;
+        }, py::arg("config"))
+        .def("apply", [=](T &self, AbsoluteEncoderConfig &config) -> T& {
+            sparkBaseConfigType.attr("apply")(self, config);
+            return self;
+        }, py::arg("config"))
+        .def("apply", [=](T &self, AnalogSensorConfig &config) -> T& {
+            sparkBaseConfigType.attr("apply")(self, config);
+            return self;
+        }, py::arg("config"))
+        .def("apply", [=](T &self, EncoderConfig &config) -> T& {
+            sparkBaseConfigType.attr("apply")(self, config);
+            return self;
+        }, py::arg("config"))
+        .def("apply", [=](T &self, LimitSwitchConfig &config) -> T& {
+            sparkBaseConfigType.attr("apply")(self, config);
+            return self;
+        }, py::arg("config"))
+        .def("apply", [=](T &self, SoftLimitConfig &config) -> T& {
+            sparkBaseConfigType.attr("apply")(self, config);
+            return self;
+        }, py::arg("config"))
+        .def("apply", [=](T &self, ClosedLoopConfig &config) -> T& {
+            sparkBaseConfigType.attr("apply")(self, config);
+            return self;
+        }, py::arg("config"))
+        .def("apply", [=](T &self, SignalsConfig &config) -> T& {
+            sparkBaseConfigType.attr("apply")(self, config);
+            return self;
+        }, py::arg("config"))
+    ;
+}

tests/test_config.py

@@ -0,0 +1,11 @@
+import rev
+
+
+def test_spark_flex_config():
+    config = rev.SparkFlexConfig()
+    config.apply(rev.EncoderConfig()).apply(rev.AbsoluteEncoderConfig())
+
+
+def test_spark_max_config():
+    config = rev.SparkMaxConfig()
+    config.apply(rev.EncoderConfig()).apply(rev.AbsoluteEncoderConfig())