1 files changed, 184 insertions, 0 deletions
diff --git a/rba.tool.editor/src/rba/tool/editor/generator/RBAModelGeneratorDelegate.xtend b/rba.tool.editor/src/rba/tool/editor/generator/RBAModelGeneratorDelegate.xtend
new file mode 100644
index 0000000..5d0b582
--- /dev/null
+++ b/rba.tool.editor/src/rba/tool/editor/generator/RBAModelGeneratorDelegate.xtend
@@ -0,0 +1,184 @@
+package rba.tool.editor.generator
+
+import com.google.inject.Inject
+import java.util.Set
+import org.eclipse.emf.ecore.resource.Resource
+import org.eclipse.emf.ecore.resource.ResourceSet
+import org.eclipse.xtext.generator.GeneratorContext
+import org.eclipse.xtext.generator.IFileSystemAccess
+import org.eclipse.xtext.generator.IFileSystemAccess2
+import org.eclipse.xtext.generator.IGenerator
+import org.eclipse.xtext.generator.IGenerator2
+import org.eclipse.xtext.generator.IGeneratorContext
+import org.eclipse.xtext.generator.OutputConfiguration
+import org.eclipse.xtext.util.CancelIndicator
+import rba.tool.editor.generator.json.IJSONGenerator
+import rba.tool.editor.generator.rtmodel.IRTModelUpdater
+import rba.tool.editor.generator.z3.IConstraintGenerator
+import rba.tool.editor.generator.z3.ISortValueGenerator
+
+class RBAModelGeneratorDelegate implements IGenerator, IGenerator2 {
+
+	@Inject(optional=true)
+	private ISortValueGenerator sortValueGenerator;
+
+	private ThreadLocal<Boolean> sortValueGeneratorInBuilding = new ThreadLocal<Boolean>() {
+		override protected initialValue() {
+			return false;
+		}
+	};
+
+	@Inject(optional=true)
+	private IRTModelUpdater rtModelUpdater;
+
+	private ThreadLocal<Boolean> rtModelUpdaterInBuilding = new ThreadLocal<Boolean>() {
+		override protected initialValue() {
+			return false;
+		}
+	};
+
+	@Inject(optional=true)
+	private IConstraintGenerator constraintGenerator;
+
+	private ThreadLocal<Boolean> constraintGeneratorInBuilding = new ThreadLocal<Boolean>() {
+		override protected initialValue() {
+			return false;
+		}
+	};
+
+	@Inject(optional=true)
+	private IJSONGenerator jsonGenerator;
+
+	private ThreadLocal<Boolean> jsonGeneratorInBuilding = new ThreadLocal<Boolean>() {
+		override protected initialValue() {
+			return false;
+		}
+	};
+
+	def public void generate(Resource input, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		generate(input.resourceSet, fsa, context);
+	}
+
+	def public void generate(ResourceSet input, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		if (sortValueGenerator !== null && !sortValueGeneratorInBuilding.get.booleanValue) {
+			sortValueGeneratorInBuilding.set(true);
+		}
+		if (rtModelUpdater !== null && !rtModelUpdaterInBuilding.get.booleanValue) {
+			rtModelUpdaterInBuilding.set(true);
+		}
+		if (constraintGenerator !== null && !constraintGeneratorInBuilding.get.booleanValue) {
+			constraintGeneratorInBuilding.set(true);
+		}
+		if (jsonGenerator !== null && !jsonGeneratorInBuilding.get.booleanValue) {
+			jsonGeneratorInBuilding.set(true);
+		}
+
+		try {
+			beforeGenerate(input, fsa, context);
+			doGenerate(input, fsa, context);
+		} finally {
+			afterGenerate(input, fsa, context);
+		}
+	}
+
+	override doGenerate(Resource input, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		doGenerate(input.resourceSet, fsa, context);
+	}
+
+	def void doGenerate(ResourceSet input, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		if (sortValueGenerator !== null && sortValueGeneratorInBuilding.get.booleanValue) {
+			sortValueGenerator.doGenerate(input, fsa, context);
+		}
+		if (rtModelUpdater !== null && rtModelUpdaterInBuilding.get.booleanValue) {
+			rtModelUpdater.doGenerate(input, fsa, context);
+		}
+		if (constraintGenerator !== null && constraintGeneratorInBuilding.get.booleanValue) {
+			constraintGenerator.doGenerate(input, fsa, context);
+		}
+		if (jsonGenerator !== null && jsonGeneratorInBuilding.get.booleanValue) {
+			jsonGenerator.doGenerate(input, fsa, context);
+		}
+	}
+
+	override beforeGenerate(Resource input, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		beforeGenerate(input.resourceSet, fsa, context);
+	}
+
+	def void beforeGenerate(ResourceSet input, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		if (sortValueGenerator !== null && sortValueGeneratorInBuilding.get.booleanValue) {
+			sortValueGenerator.beforeGenerate(input, fsa, context);
+		}
+		if (rtModelUpdater !== null && rtModelUpdaterInBuilding.get.booleanValue) {
+			rtModelUpdater.beforeGenerate(input, fsa, context);
+		}
+		if (constraintGenerator !== null && constraintGeneratorInBuilding.get.booleanValue) {
+			constraintGenerator.beforeGenerate(input, fsa, context);
+		}
+		if (jsonGenerator !== null && jsonGeneratorInBuilding.get.booleanValue) {
+			jsonGenerator.beforeGenerate(input, fsa, context);
+		}
+	}
+
+	override afterGenerate(Resource input, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		afterGenerate(input.resourceSet, fsa, context);
+	}
+
+	def void afterGenerate(ResourceSet input, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		if (sortValueGenerator !== null && sortValueGeneratorInBuilding.get.booleanValue) {
+			sortValueGenerator.afterGenerate(input, fsa, context);
+		}
+		if (rtModelUpdater !== null && rtModelUpdaterInBuilding.get.booleanValue) {
+			rtModelUpdater.afterGenerate(input, fsa, context);
+		}
+		if (constraintGenerator !== null && constraintGeneratorInBuilding.get.booleanValue) {
+			constraintGenerator.afterGenerate(input, fsa, context);
+		}
+		if (jsonGenerator !== null && jsonGeneratorInBuilding.get.booleanValue) {
+			jsonGenerator.afterGenerate(input, fsa, context);
+		}
+	}
+
+	override doGenerate(Resource input, IFileSystemAccess fsa) {
+		doGenerate(input.resourceSet, fsa);
+	}
+
+	def void doGenerate(ResourceSet input, IFileSystemAccess fsa) {
+		val casted = fsa as IFileSystemAccess2;
+		val context = new GeneratorContext();
+		context.setCancelIndicator(CancelIndicator.NullImpl);
+		try {
+			beforeGenerate(input, casted, context);
+			doGenerate(input, casted, context);
+		} finally {
+			afterGenerate(input, casted, context);
+		}
+	}
+
+	def public void setRBAModelInBuilding(boolean value) {
+		sortValueGeneratorInBuilding.set(value);
+		rtModelUpdaterInBuilding.set(value);
+		constraintGeneratorInBuilding.set(value);
+		jsonGeneratorInBuilding.set(value);
+	}
+
+	def public boolean isRBAModelInBuilding() {
+		if (sortValueGeneratorInBuilding.get().booleanValue) {
+			return true;
+		}
+		if (rtModelUpdaterInBuilding.get().booleanValue) {
+			return true;
+		}
+		if (constraintGeneratorInBuilding.get().booleanValue) {
+			return true;
+		}
+		if (jsonGeneratorInBuilding.get().booleanValue) {
+			return true;
+		}
+		return false;
+	}
+	
+	def public Set<OutputConfiguration> getRBAModelGeneratorOutputConfigurations() {
+		return newHashSet(IJSONGenerator.OUTPUT_CONFIGURATION);
+	}
+
+}