refactoring

1 files changed, 11 insertions, 0 deletions

diff --git a/src/iscanner.h b/src/iscanner.h
index 75300e1..e09197e 100644
--- a/src/iscanner.h
+++ b/src/iscanner.h
@@ -4,12 +4,16 @@
 #include <memory>
 #include <vector>
 
+// TODO: remove this include
+#include "calibration.h"
+
 class IProtocol;
 class ICamera;
 
 class IScanner
 {
 public:
+    // TODO: get rid of constructor or rename class
     explicit IScanner(std::shared_ptr<ICamera> camera,
                       std::vector<std::shared_ptr<IProtocol>> protocols);
     virtual ~IScanner() = default;
@@ -18,6 +22,13 @@ public:
     virtual bool startAllProtocols() = 0;
     virtual void stopAllProtocols() = 0;
 
+    // TODO: think about more flexible calibration interface
+    virtual CalibrationTablePtr calibrationTableX() const = 0;
+    virtual CalibrationTablePtr calibrationTableZ() const = 0;
+
+    // TODO: add nullptr check everywhere where this function is used
+    std::shared_ptr<ICamera> camera() const;
+
 protected:
     std::shared_ptr<ICamera> m_camera;
     std::vector<std::shared_ptr<IProtocol>> m_protocols;