Added language support

Author: Fran314

SmashDataCollector.py

@@ -1,6 +1,13 @@
 data_path = r'C:\Users\franc\Documents\VSCode\SmashDataCollector\data'
 res_path = r'C:\Users\franc\Documents\VSCode\SmashDataCollector\res'
-output_path = r'C:\Users\franc\Documents\VSCode\SmashDataCollector'
+output_path = r'C:\Users\franc\Documents\VSCode\SmashDataCollector\output.tsv'
+
+# Set the language variable to 0, 1 or 2 according to the following:
+#   0 for Japanese or Korean
+#   1 for English, Italian, Dutch or Russian
+#   2 for French, Spanish or German
+#   3 for Chinese (traditional or simplified)
+LANGUAGE = 1
 
 LIVES = 3
 TAKEN_GIVEN_DMG_THRESHOLD = 30

customizable.py

@@ -80,7 +80,6 @@ def getClosestCharacterByCloseup(image, character_closeups, mask, background_bgr
         distances.append(numpy.min(character_distances))
     return numpy.argmin(distances)
 
-
 def getClosestPlayerByIcon(image, icons, background_bgr):
     empty_icon = numpy.full_like(image, background_bgr[0])
     empty_icon[:,:,1] = numpy.full((image.shape[0], image.shape[1]), background_bgr[1])
@@ -96,7 +95,7 @@ def getClosestPlayerByIcon(image, icons, background_bgr):
 
 def getClosestDigit(image, digit_images):
     distances = []
-    for i in range(11):
+    for i in range(len(digit_images)):
         distances.append(numpy.sum(cv2.absdiff(digit_images[i], image)))
     digit = numpy.argmin(distances)
     return digit-1
@@ -190,15 +189,19 @@ def isValidFirstData(positions, times):
 
 
 def time2int(arg):
-    return 60 * int(arg[0]) + int(arg[2:4])
+    if(len(arg) == 4):
+        return 60 * int(arg[0]) + int(arg[2:4])
+    else:
+        return 60 * int(arg[0:2]) + int(arg[3:5])
 def time2string(arg):
     if(arg == ""):
         return ""
     else:
         return str(time2int(arg))
 
 
-def convertMatchToString(file_name, players, characters, positions, times, falls, given_damages, taken_damages):
+def convertMatchToString(file_name, characters, positions, times, falls, given_damages, taken_damages):
+    players = len(characters)
     match_string = file_name[6:8] + "/" + file_name[4:6] + "/" + file_name[0:4] + "\t"
     match_string += str(players) + "\t"
     for i in range(players):

data/2020121110511900_c.jpg

@@ -1 +1,3 @@
+
+
 11/12/2020	3	Donkey Kong	2	3,2,2	69	69	83	Meta Knight	1	1,3	76	72		Young Link	3	1,2,2	57	63	77