Also dump stats on display requests

src/compiler/compiler.ml

@@ -398,7 +398,7 @@ let compile ctx actx callbacks =
 		if is_compilation then Generate.check_auxiliary_output com actx;
 		enter_stage com CGenerationStart;
 		ServerMessage.compiler_stage com;
-		Generate.maybe_generate_dump ctx tctx;
+		Generate.maybe_generate_dump tctx;
 		if not actx.no_output then Generate.generate ctx tctx ext actx;
 		enter_stage com CGenerationDone;
 		ServerMessage.compiler_stage com;

src/compiler/generate.ml

@@ -104,7 +104,7 @@ let parse_swf_header ctx h = match ExtString.String.nsplit h ":" with
 
 let delete_file f = try Sys.remove f with _ -> ()
 
-let maybe_generate_dump ctx tctx =
+let maybe_generate_dump tctx =
 	let com = tctx.Typecore.com in
 	if Common.defined com Define.Dump then begin
 		Codegen.Dump.dump_types com;
@@ -115,8 +115,10 @@ let maybe_generate_dump ctx tctx =
 		if not com.is_macro_context then match tctx.Typecore.g.Typecore.macros with
 			| None -> ()
 			| Some(_,ctx) -> Codegen.Dump.dump_dependencies ~target_override:(Some "macro") ctx.Typecore.com
-	end;
-	if Common.defined ctx.com Define.DumpInvalidationStats then begin
+	end
+
+let maybe_generate_stats_dump com =
+	if Common.defined com Define.DumpInvalidationStats then begin
 		Codegen.Dump.dump_invalidation_stats com;
 		Option.may Codegen.Dump.dump_invalidation_stats (com.get_macros())
 	end

src/compiler/server.ml

@@ -640,7 +640,7 @@ let after_save sctx ctx =
 		maybe_cache_context sctx ctx.com
 
 let after_compilation sctx ctx =
-	()
+	Generate.maybe_generate_stats_dump ctx.com
 
 let mk_length_prefixed_communication allow_nonblock chin chout =
 	let sin = Unix.descr_of_in_channel chin in