Chart fix module complete

Author: AXIS5hacker

Dynamix_chart_width_control/chart_store.cpp

@@ -9,12 +9,15 @@
 #include<sstream>
 #include<climits>
 #include<map>
+#include<set>
 
 using std::vector;
 using std::string;
 using std::ofstream;
 using std::pair;
 using std::map;
+using std::set;
+
 //added default note type: NULLTP
 enum types { NORMAL = 1, CHAIN, HOLD, SUB , NULLTP};
 //added default sides type: UNKNOWN
@@ -42,10 +45,19 @@ class chart_store
 	//now we use map to store the notes, in order to deal with discrete note id
 	//<note id,note>
 	map<int, note> m_notes, m_left, m_right;//note list(the key is the note's id)
+	//hold list and sub list(used for hold-sub checking)
+	map<int, int> hold_mid, hold_left, hold_right;
+	set<int> sub_mid, sub_left, sub_right;
+	//mismatch note list(used for hold-sub autofix)
+	vector<pair<int, string> > mismatched_notes;
 
 	bool to_file(string f);//print to XML
 	int readfile(string fn);//read XML
 	string chart_filename;//filename of the chart
+	void set_barpm(double f) { barpm = f; }
+	void set_lside(sides x) { ltype = x; }
+	void set_rside(sides x) { rtype = x; }
+
 private:
 
 	string name;//song name

Dynamix_chart_width_control/chart_store.h

@@ -8,6 +8,7 @@
 
 #include<iomanip>
 #include<fstream>
+#include<algorithm>
 #include<sys/stat.h>
 #if defined(_WIN64)||defined(WIN32)||defined(_WIN32)
 #include<direct.h>
@@ -17,10 +18,17 @@
 #endif
 
 #include<cstdio>
+//side change flags
 #define MID_CHANGE 0x4
 #define LEFT_CHANGE 0x2
 #define RIGHT_CHANGE 0x1
 
+//chart read flags
+#define BARPM_MISSING 0x1
+#define LEFT_SIDE_MISSING 0x2
+#define RIGHT_SIDE_MISSING 0x4
+#define HOLD_SUB_MISMATCH 0x8
+
 #include "chart_store.h"
 #include<string>
 using std::string;

Dynamix_chart_width_control/defs.h

@@ -231,8 +231,109 @@ int main(int argc, char* argv[])
 		chart_store cs;//store the chart
 		try {
 			int fail_read = cs.readfile(filename);//open file
+			//fixing stage
+			//missing barpm
+			if (fail_read & BARPM_MISSING) {
+				double new_barpm = 0;
+				string ll;
+				istringstream istr;
+				cout << "Illegal Barpm! Please enter a valid Barpm:" << endl;;
+				getline(cin, ll);
+				istr.str(ll);
+				istr >> new_barpm;
 
-			if (fail_read != 1) {
+				while (new_barpm <= 0) {
+					cout << "Illegal Barpm! Please re-enter a valid Barpm:" << endl;
+					getline(cin, ll);
+					istr.clear();
+					istr.str(ll);
+					istr >> new_barpm;
+				}
+				char new_barpm_string[64];
+				sprintf_s(new_barpm_string, "%f", new_barpm);
+				cs.set_barpm(new_barpm);
+				cout << "Barpm is set to " + string(new_barpm_string) + "." << endl;
+				cout << "===============================" << endl;
+
+				fail_read &= (~(int)BARPM_MISSING);
+			}
+			//missing left side
+			if (fail_read & LEFT_SIDE_MISSING) {
+				sides s = sides::UNKNOWN;
+				string side_string;
+				cout << "Left side type is not specified! Please enter a valid side type:" << endl;
+				while (s == sides::UNKNOWN) {
+					cin >> side_string;
+					//lowercase
+					transform(side_string.begin(), side_string.end(), side_string.begin(), tolower);
+					if (side_string == "pad") {
+						s = sides::PAD;
+					}
+					else if (side_string == "mixer") {
+						s = sides::MIXER;
+					}
+					else if (side_string == "multi") {
+						s = sides::MULTI;
+					}
+					else {
+						cout << "Invalid side type! Please enter again!" << endl;
+					}
+				}
+				cs.set_lside(s);
+				cout << "Left side fixed." << endl;
+				cout << "===============================" << endl;
+				fail_read &= (~(int)LEFT_SIDE_MISSING);
+			}
+			//missing right side
+			if (fail_read & RIGHT_SIDE_MISSING) {
+				sides s = sides::UNKNOWN;
+				string side_string;
+				cout << "Right side type is not specified! Please enter a valid side type:" << endl;
+				while (s == sides::UNKNOWN) {
+					cin >> side_string;
+					//lowercase
+					transform(side_string.begin(), side_string.end(), side_string.begin(), tolower);
+					if (side_string == "pad") {
+						s = sides::PAD;
+					}
+					else if (side_string == "mixer") {
+						s = sides::MIXER;
+					}
+					else if (side_string == "multi") {
+						s = sides::MULTI;
+					}
+					else {
+						cout << "Invalid side type! Please enter again!" << endl;
+					}
+				}
+				cs.set_rside(s);
+				cout << "Right side fixed." << endl;
+				cout << "===============================" << endl;
+				fail_read &= (~(int)RIGHT_SIDE_MISSING);
+			}
+			//Hold-sub mismatch autofix
+			if (fail_read & HOLD_SUB_MISMATCH) {
+				cout << "mismatched notes found:" << endl;
+				for (auto& ii : cs.mismatched_notes) {
+					cout << ii.first << '\t' << ii.second << endl;
+					//fix
+					if (ii.second == "middle") {
+						cs.m_notes.erase(ii.first);
+					}
+					else if (ii.second == "left") {
+						cs.m_left.erase(ii.first);
+					}
+					else if (ii.second == "right") {
+						cs.m_right.erase(ii.first);
+					}
+				}
+				cout << "\nThe mismatching Holds and Subs have been fixed automatically." << endl;
+				cout << "===============================" << endl;
+				//set status to success
+				fail_read &= (~(int)HOLD_SUB_MISMATCH);
+			}
+
+			if (fail_read == 0) {
 
 				//this function now only processes chart store class
 				int success = width_change(cs, width, start_time, end_time, side_mask, random_trigger);//width=1 as default width multiplier
@@ -290,6 +391,9 @@ int main(int argc, char* argv[])
 					}
 				}
 			}
+			else {
+				cout << "Unknown error" << endl;
+			}
 		}
 		catch (exception& ex) {
 			cout << ex.what() << endl;