1 files changed, 6 insertions, 0 deletions
diff --git a/src/labeller.cc b/src/labeller.cc
index 32f69c9..f7fa69e 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;
@@ -30,6 +31,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;
         if(std::filesystem::directory_entry(savepath).exists()) {
             load(savepath, *data);
@@ -56,6 +58,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);
     }