Add improved chapter_selection

game/src/chapter_select.rpy

@@ -26,7 +26,7 @@ default selected_tuple = None
 init python:
 
     def select_chapter():
-        global selected_tuple, current_chapter
+        global selected_tuple, current_chapter, is_end_reached
 
         while True:
             selected_tuple = renpy.display_menu(tuples_index + [("Go Back", "go_back")])
@@ -39,17 +39,35 @@ init python:
             else:
                 break
 
+        is_end_reached = False # Reset this for when the tool is used more than once
+
         set_stats()
+        toggle_debug()
+
+        renpy.call(current_chapter)
 
 
     def set_stats():
-        global chapter_list_length, chapter_list_index, ending_route_number, is_end_reached
+        global chapter_list_length, ending_route_number, is_end_reached, lock_scores
 
-        chapter_list_length = len(chapter_list) - 1
+        # chapter_list_length = get_chapter_list_length()
+        ending_route_number = get_ending()
+
+        lock_scores = True # Prevents scores from increasing when using the chapter selection tool
 
         find_chapter_in_array()
+        
+        if not is_end_reached:
+            add_ending_chapters() # From storyline
+        
+        update_ending_variables() # From storyline
 
-        renpy.call(current_chapter)
+
+    def set_scores(anon_score, fang_score):
+        global anonscore, fangscore
+
+        anonscore = anon_score
+        fangscore = fang_score
 
 
     def find_chapter_in_array():
@@ -62,6 +80,14 @@ init python:
             current_chapter = ""
 
 
+label reset_chapter_list:
+    $ chapter_list = [
+        "chapter_1", "chapter_2", "chapter_3", "chapter_4", "chapter_5",
+        "chapter_6", "chapter_7", "chapter_8", "chapter_9", "chapter_10", "chapter_11"
+    ]
+    return
+
+
 label chapter_select:
     scene black
     stop sound
@@ -71,27 +97,19 @@ label chapter_select:
         "Initialize scores:"
 
         "Ending 1":
-            $ anon_score = 0
-            $ fang_score = 0
-            $ ending_route_number = 1
+            $ set_scores(0, 0)
         "Ending 2":
-            $ anon_score = 0
-            $ fang_score = 4
-            $ ending_route_number = 2
+            $ set_scores(0, 4)
         "Ending 3":
-            $ anon_score = 4
-            $ fang_score = 0
-            $ ending_route_number = 3
+            $ set_scores(4, 0)
         "Ending 4":
-            $ anon_score = 4
-            $ fang_score = 4
-            $ ending_route_number = 4
+            $ set_scores(4, 4)
         "Exit to main menu":
             return
-    
-    $ lock_scores = True
-    call initstats(anon_score, fang_score) from _call_initstats_2
-        
+
+    if not is_end_reached:
+        call reset_chapter_list from _call_reset_chapter_list
+
     $ select_chapter()