From a3d209ae35b31e27ddbd10d862d656a1f2f8c009 Mon Sep 17 00:00:00 2001
From: twanvl <twanvl@0fc631ac-6414-0410-93d0-97cfa31319b6>
Date: Wed, 18 Jun 2008 23:26:41 +0000
Subject: [PATCH] Use ScriptRule also for regex functions, optimizing them by
 pre-compiling default arguments

git-svn-id: svn://svn.code.sf.net/p/magicseteditor/code/trunk@995 0fc631ac-6414-0410-93d0-97cfa31319b6
---
 src/mse.vcproj                 |  3 ++
 src/script/functions/regex.cpp | 89 ++++++++++++++++++++++++++++++++--
 2 files changed, 89 insertions(+), 3 deletions(-)

diff --git a/src/mse.vcproj b/src/mse.vcproj
index b12c11b8..c774c810 100644
--- a/src/mse.vcproj
+++ b/src/mse.vcproj
@@ -2663,6 +2663,9 @@
 							ObjectFile="$(IntDir)/$(InputName)5.obj"/>
 					</FileConfiguration>
 				</File>
+				<File
+					RelativePath=".\script\functions\regex.cpp">
+				</File>
 				<File
 					RelativePath=".\script\functions\util.hpp">
 				</File>
diff --git a/src/script/functions/regex.cpp b/src/script/functions/regex.cpp
index 97c16731..87f3f6dc 100644
--- a/src/script/functions/regex.cpp
+++ b/src/script/functions/regex.cpp
@@ -12,6 +12,38 @@
 #include <script/functions/util.hpp>
 #include <util/error.hpp>
 
+DECLARE_POINTER_TYPE(ScriptRegex);
+DECLARE_TYPEOF_COLLECTION(pair<Variable COMMA ScriptValueP>);
+
+// ----------------------------------------------------------------------------- : Regex type
+
+/// A regular expression for use in a script
+class ScriptRegex : public ScriptValue {
+  public:
+	virtual ScriptType type() const { return SCRIPT_REGEX; }
+	virtual String typeName() const { return _("regex"); }
+	
+	wxRegEx regex; ///< The regular expression
+};
+
+ScriptRegexP regex_from_script(const ScriptValueP& value) {
+	// is it a regex already?
+	ScriptRegexP regex = dynamic_pointer_cast<ScriptRegex>(value);
+	if (!regex) {
+		// compile string
+		regex = new_intrusive<ScriptRegex>();
+		if (!regex->regex.Compile(*value, wxRE_ADVANCED)) {
+			throw ScriptError(_("Error while compiling regular expression: '") + value->toString() + _("'"));
+		}
+		assert(regex->regex.IsValid());
+	}
+	return regex;
+}
+
+template <> inline ScriptRegexP from_script<ScriptRegexP>(const ScriptValueP& value) {
+	return regex_from_script(value);
+}
+
 // ----------------------------------------------------------------------------- : Rules : regex replace
 
 class ScriptReplaceRule : public ScriptValue {
@@ -143,6 +175,7 @@ ScriptValueP filter_rule(Context& ctx) {
 	SCRIPT_PARAM_DEFAULT_C(String, in_context, String());
 	
 	// cache
+	/*
 	const int CACHE_SIZE = 6;
 	struct CacheItem{
 		String match, in_context;
@@ -162,6 +195,9 @@ ScriptValueP filter_rule(Context& ctx) {
 	cache[cache_pos].rule = intrusive_ptr<ScriptFilterRule>(new ScriptFilterRule);
 	intrusive_ptr<ScriptFilterRule>& ret = cache[cache_pos].rule;
 	cache_pos = (cache_pos+1) % CACHE_SIZE;
+	/*/
+	intrusive_ptr<ScriptFilterRule> ret(new ScriptFilterRule); 
+	//*/
 	
 	// match
 	if (!ret->regex.Compile(match, wxRE_ADVANCED)) {
@@ -185,6 +221,7 @@ SCRIPT_FUNCTION(filter_text) {
 
 // ----------------------------------------------------------------------------- : Rules : regex break
 
+/*
 class ScriptBreakRule : public ScriptValue {
   public:
 	virtual ScriptType type() const { return SCRIPT_FUNCTION; }
@@ -234,10 +271,42 @@ SCRIPT_FUNCTION(break_rule) {
 }
 SCRIPT_FUNCTION(break_text) {
 	return break_rule(ctx)->eval(ctx);
+}*/
+
+SCRIPT_FUNCTION_WITH_SIMPLIFY(break_text) {
+	SCRIPT_PARAM_C(String, input);
+	SCRIPT_PARAM_C(ScriptRegexP, match);
+	SCRIPT_OPTIONAL_PARAM_C_(ScriptRegexP, in_context);
+	intrusive_ptr<ScriptCustomCollection> ret(new ScriptCustomCollection);
+	// find all matches
+	while (match->regex.Matches(input)) {
+		// match, append to result
+		size_t start, len;
+		bool ok = match->regex.GetMatch(&start, &len, 0);
+		assert(ok);
+		String inside     = input.substr(start, len);  // the match
+		String next_input = input.substr(start + len); // everything after the match
+		if (!in_context || in_context->regex.Matches(input.substr(0,start) + _("<match>") + next_input)) {
+			// no context or context match
+			ret->value.push_back(to_script(inside));
+		}
+		input = next_input;
+	}
+	return ret;
+}
+SCRIPT_FUNCTION_SIMPLIFY_CLOSURE(break_text) {
+	FOR_EACH(b, closure.bindings) {
+		if (b.first == SCRIPT_VAR_match || b.first == SCRIPT_VAR_in_context) {
+			b.second = regex_from_script(b.second); // pre-compile
+		}
+	}
+	return ScriptValueP();
 }
 
 // ----------------------------------------------------------------------------- : Rules : regex match
 
+
+/*
 class ScriptMatchRule : public ScriptValue {
   public:
 	virtual ScriptType type() const { return SCRIPT_FUNCTION; }
@@ -276,6 +345,20 @@ SCRIPT_FUNCTION_SIMPLIFY_CLOSURE(match) {
 	}
 	return ScriptValueP();
 }
+*/
+SCRIPT_FUNCTION_WITH_SIMPLIFY(match) {
+	SCRIPT_PARAM_C(String, input);
+	SCRIPT_PARAM_C(ScriptRegexP, match);
+	SCRIPT_RETURN(match->regex.Matches(input));
+}
+SCRIPT_FUNCTION_SIMPLIFY_CLOSURE(match) {
+	FOR_EACH(b, closure.bindings) {
+		if (b.first == SCRIPT_VAR_match) {
+			b.second = regex_from_script(b.second); // pre-compile
+		}
+	}
+	return ScriptValueP();
+}
 
 // ----------------------------------------------------------------------------- : Init
 
@@ -283,9 +366,9 @@ void init_script_regex_functions(Context& ctx) {
 	ctx.setVariable(_("replace"),              script_replace);
 	ctx.setVariable(_("filter text"),          script_filter_text);
 	ctx.setVariable(_("break text"),           script_break_text);
-	ctx.setVariable(_("match"),                script_match);;
+	ctx.setVariable(_("match"),                script_match);
 	ctx.setVariable(_("replace rule"),         script_replace_rule);
 	ctx.setVariable(_("filter rule"),          script_filter_rule);
-	ctx.setVariable(_("break rule"),           script_break_rule);
-	ctx.setVariable(_("match rule"),           script_match_rule);
+	ctx.setVariable(_("break rule"),           new_intrusive1<ScriptRule>(script_break_text));
+	ctx.setVariable(_("match rule"),           new_intrusive1<ScriptRule>(script_match));
 }