1 files changed, 6 insertions, 0 deletions
diff --git a/src/labeller.cc b/src/labeller.cc
index 64da2d7..024b720 100644
--- a/src/labeller.cc
+++ b/src/labeller.cc
@@ -6,6 +6,7 @@
 namespace backend {
     struct labeller_impl {
         std::vector<std::string> labels;
+        std::vector<std::string> rectangleLabels;
         std::vector<label> annotations;
         // Bool in undo/redo buffers is whether was applied or deleted
         std::vector<std::pair<bool, label>> undoBuffer;
@@ -16,6 +17,7 @@ namespace backend {
     labeller::labeller(const std::filesystem::path& savepath) {
         data = std::shared_ptr<labeller_impl>(new labeller_impl);
         data->labels = settings::getLabels();
+        data->rectangleLabels = settings::getRectangleLabels();
         data->savepath = savepath;
     }
 
@@ -39,6 +41,10 @@ namespace backend {
         return data->labels;
     }
 
+    std::vector<std::string> labeller::getRectangleLabels() const {
+        return data->rectangleLabels;
+    }
+
     void appLab(std::vector<label>& annotations, label lab) {
         annotations.push_back(lab);
     }