Add to NoteDataUtil a way to quickly segment a file's Notedata

poco0317 · Oct 4, 2019 · 373e034 · 373e034
1 parent d403b18
commit 373e034
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/Etterna/Models/NoteData/NoteDataUtil.cpp b/src/Etterna/Models/NoteData/NoteDataUtil.cpp
@@ -3104,6 +3104,15 @@ NoteDataUtil::DeleteRows(NoteData& nd, int iStartIndex, int iRowsToDelete)
 	nd.RevalidateATIs(vector<int>(), false);
 }
 
+void
+NoteDataUtil::RemoveAllButRange(NoteData& nd, int iStartIndex, int iEndIndex)
+{
+	ASSERT(abs(iStartIndex - iEndIndex) > 0);
+	nd.ClearRange(0, iStartIndex - 1);
+	nd.ClearRange(iEndIndex, MAX_NOTE_ROW);
+	nd.RevalidateATIs(vector<int>(), false);
+}
+
 void
 NoteDataUtil::RemoveAllTapsOfType(NoteData& ndInOut, TapNoteType typeToRemove)
 {

diff --git a/src/Etterna/Models/NoteData/NoteDataUtil.h b/src/Etterna/Models/NoteData/NoteDataUtil.h
@@ -1,4 +1,4 @@
-#ifndef NOTEDATAUTIL_H
+#ifndef NOTEDATAUTIL_H
 #define NOTEDATAUTIL_H
 
 #include "Etterna/Models/Misc/GameConstantsAndTypes.h"
@@ -296,6 +296,9 @@ InsertRows(NoteData& nd, int iStartIndex, int iRowsToShift);
 void
 DeleteRows(NoteData& nd, int iStartIndex, int iRowsToShift);
 
+void
+RemoveAllButRange(NoteData& nd, int iStartIndex, int iEndIndex);
+
 void
 RemoveAllTapsOfType(NoteData& ndInOut, TapNoteType typeToRemove);
 void