Add preprocessor instruction #sortreallocate

Enable the reallocation for a single module. If specified in the same
module as "Off sortreallocate;", the reallocation will still happen in
that module.

Author: jodavies

sources/compcomm.c

@@ -628,6 +628,14 @@ int CoOff(UBYTE *s)
 				AC.TestValue = 0;
 			}
 		}
+		else if ( StrICont(t,(UBYTE *)"sortreallocate") == 0 ) {
+			if ( AC.SortReallocateFlag == 2 ) {
+				/* The flag has been set by #sortreallocate, and it was off before. Leave it as 2,
+				   so that the reallocation still happens in the current module. It will be turned
+				   off after the reallocation is done. */
+				return(0);
+			}
+		}
 		*s = c;
 	 	*onoffoptions[i].var = onoffoptions[i].flags; 
 		AR.SortType = AC.SortType;

sources/declare.h

@@ -957,6 +957,7 @@ extern int    DoPrePrintTimes(UBYTE *);
 extern int    DoPreWrite(UBYTE *);
 extern int    DoPreClose(UBYTE *);
 extern int    DoPreRemove(UBYTE *);
+extern int    DoPreSortReallocate(UBYTE *);
 extern int    DoCommentChar(UBYTE *);
 extern int    DoPrcExtension(UBYTE *);
 extern int    DoPreReset(UBYTE *);

sources/execute.c

@@ -1078,6 +1078,11 @@ if ( AC.SwitchInArray > 0 ) {
 		}
 #endif
 	}
+	if ( AC.SortReallocateFlag == 2 ) {
+		/* The Flag was set for a single module by the preprocessor #sortreallocate,
+		   so turn it off again. */
+		AC.SortReallocateFlag = 0;
+	}
 
 	return(RetCode);
 }

sources/pre.c

@@ -104,6 +104,7 @@ static KEYWORD precommands[] = {
 	,{"setrandom"    , DoSetRandom    , 0, 0}
 	,{"show"         , DoPreShow      , 0, 0}
 	,{"skipextrasymbols" , DoSkipExtraSymbols , 0, 0}
+	,{"sortreallocate", DoPreSortReallocate , 0, 0}
 #ifdef WITHFLOAT
     ,{"startfloat"   , DoStartFloat   , 0, 0}
 #endif
@@ -3777,6 +3778,23 @@ int DoPrePrintTimes(UBYTE *s)
 
 /*
  		#] DoPrePrintTimes : 
+		#[ DoPreSortReallocate :
+*/
+
+int DoPreSortReallocate(UBYTE *s)
+{
+	DUMMYUSE(s);
+	if ( AC.SortReallocateFlag == 0 ) {
+		/* Currently off, so set to 2. Then the reallocation code knows the flag was
+		   set here, since "On sortreallocate;" sets it to 1. */
+		AC.SortReallocateFlag = 2;
+	}
+	/* If the flag is already on, do nothing. */
+	return(0);
+}
+
+/*
+		#] DoPreSortReallocate :
  		#[ DoPreAppend :
 
 		Syntax:

sources/structs.h

@@ -1867,7 +1867,10 @@ struct C_const {
     int     MemDebugFlag;          /* Only used when MALLOCDEBUG in tools.c */
     int     OldGCDflag;
     int     WTimeStatsFlag;
-    int     SortReallocateFlag;    /* Controls reallocation of large+small buffer at module end */
+    int     SortReallocateFlag;    /* Controls reallocation of large+small buffer at module end.
+                                        0 : Off
+                                        1 : On, every module (set by On sortreallocate;)
+                                        2 : On, single module (set by #sortreallocate) */
 	int     doloopstacksize;
 	int     dolooplevel;
     int     CheckpointFlag;        /**< Tells preprocessor whether checkpoint code must executed.