From 54a555640f3d3cb908b79139a0f2900a7cdaa0f9 Mon Sep 17 00:00:00 2001
From: mrbesen <y.g.2@gmx.de>
Date: Mon, 14 Jun 2021 12:28:52 +0200
Subject: [PATCH] getLevel

---
 include/beatsaber-impl/beatsetimpl.h | 2 ++
 include/beatsaber/beatset.h          | 2 ++
 src/beatset.cpp                      | 7 ++++++-
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/include/beatsaber-impl/beatsetimpl.h b/include/beatsaber-impl/beatsetimpl.h
index dde68e7..411f19c 100644
--- a/include/beatsaber-impl/beatsetimpl.h
+++ b/include/beatsaber-impl/beatsetimpl.h
@@ -27,6 +27,8 @@ public:
 
 	virtual void printDebug() const override;
 
+	virtual const std::vector<std::shared_ptr<BeatLevel>>& getLevel() const override;
+
 	virtual BeatmapCharacteristic::BeatmapCharacteristic getCharacteristic() const override;
 	virtual std::shared_ptr<BeatMap> getBeatMap() const override;
 };
diff --git a/include/beatsaber/beatset.h b/include/beatsaber/beatset.h
index 05c2314..7051eb4 100644
--- a/include/beatsaber/beatset.h
+++ b/include/beatsaber/beatset.h
@@ -16,6 +16,8 @@ public:
 
 	virtual void printDebug() const = 0;
 
+	virtual const std::vector<std::shared_ptr<BeatLevel>>& getLevel() const = 0;
+
 	virtual BeatmapCharacteristic::BeatmapCharacteristic getCharacteristic() const = 0;
 	virtual std::shared_ptr<BeatMap> getBeatMap() const = 0;
 };
diff --git a/src/beatset.cpp b/src/beatset.cpp
index a47f156..705b682 100644
--- a/src/beatset.cpp
+++ b/src/beatset.cpp
@@ -36,13 +36,18 @@ bool BeatSetImpl::load() {
 }
 
 void BeatSetImpl::printDebug() const {
-	std::cout << "  Characteristic: " << BeatmapCharacteristic::toString(characteristic) << std::endl;
+	std::cout << "  Characteristic: " << BeatmapCharacteristic::toString(characteristic)
+		<< "\n  Levels: " << level.size() << std::endl;
 
 	for(auto it : level) {
 		it->printDebug();
 	}
 }
 
+const std::vector<std::shared_ptr<BeatLevel>>& BeatSetImpl::getLevel() const {
+	return level;
+}
+
 BeatmapCharacteristic::BeatmapCharacteristic BeatSetImpl::getCharacteristic() const {
 	return characteristic;
 }