From 96d63e0a614fb7f67eb95715beb585940afb005c Mon Sep 17 00:00:00 2001
From: Jan Grewe <jan.grewe@g-node.org>
Date: Fri, 20 Nov 2020 09:46:29 +0100
Subject: [PATCH] [wrapper] add gain access functions

---
 pylonwrapper.cpp | 20 ++++++++++++++++++++
 pylonwrapper.h   |  2 ++
 2 files changed, 22 insertions(+)

diff --git a/pylonwrapper.cpp b/pylonwrapper.cpp
index f9b8913..c8ca30a 100644
--- a/pylonwrapper.cpp
+++ b/pylonwrapper.cpp
@@ -82,6 +82,26 @@ bool PylonWrapper::exposureTime(double exposure_time) {
     return false;
 }
 
+double PylonWrapper::gain() {
+  double gain = -1.;
+  if (valid) {
+      GenApi::INodeMap& nodemap = camera->GetNodeMap();
+      GenApi::INode* n = nodemap.GetNode( "Gain" );
+      Pylon::CFloatParameter detector_gain( n );
+      gain = detector_gain.GetValue();
+    }
+  return gain;
+}
+
+bool PylonWrapper::gain(double gain_db) {
+    if (valid) {
+        GenAPI::INodeMap& nodemap = camera.GetNodeMap();
+        GenAPI::CFloatPtr(nodemap.GetNode("Gain"))->SetValue(gain_db);
+        return true;
+      }
+    return false;
+}
+
 ImageSettings PylonWrapper::getImageSettings() {
   ImageSettings settings;
   if (valid) {
diff --git a/pylonwrapper.h b/pylonwrapper.h
index f2e5e24..e4d6e7b 100644
--- a/pylonwrapper.h
+++ b/pylonwrapper.h
@@ -29,6 +29,8 @@ public:
     double maxFrameRate();
     double exposureTime();
     bool exposureTime(double exposure_time);
+    double gain();
+    bool gain(double gain_db);
     Pylon::CInstantCamera *getCamera();
 
 private: