1 files changed, 400 insertions, 0 deletions
diff --git a/rba.tool.editor.endpoint/src/rba/tool/editor/endpoint/generator/RBAModelWebGenerator.xtend b/rba.tool.editor.endpoint/src/rba/tool/editor/endpoint/generator/RBAModelWebGenerator.xtend
new file mode 100644
index 0000000..2a9d1e7
--- /dev/null
+++ b/rba.tool.editor.endpoint/src/rba/tool/editor/endpoint/generator/RBAModelWebGenerator.xtend
@@ -0,0 +1,400 @@
+package rba.tool.editor.endpoint.generator
+
+import java.util.List
+import java.io.File;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import groovy.lang.Binding;
+import groovy.util.GroovyScriptEngine;
+import org.codehaus.groovy.control.CompilerConfiguration;
+
+import org.eclipse.emf.ecore.resource.Resource
+import org.eclipse.emf.ecore.resource.ResourceSet
+import org.eclipse.xtext.generator.AbstractGenerator
+import org.eclipse.xtext.generator.IFileSystemAccess2
+import org.eclipse.xtext.generator.IGeneratorContext
+
+import rba.tool.editor.model.manager.ResourceManager
+import rba.tool.core.sort.SortValueManager;
+
+import rba.core.Allocatable;
+import rba.view.Area;
+import rba.sound.Zone;
+import rba.view.AreaSet;
+import rba.sound.ZoneSet;
+import rba.core.Content;
+import rba.view.ViewContent;
+import rba.view.ViewContentSet;
+import rba.sound.SoundContent;
+import rba.sound.SoundContentSet;
+import rba.core.State;
+import rba.core.Scene;
+import rba.view.Size;
+import rba.view.Display
+import rba.view.PositionContainer
+import rba.core.Package;
+import rba.core.Constraint;
+import rba.core.Variable;
+import rba.core.SetOfOperator;
+
+import rba.tool.editor.generator.z3.SortValueCodeGenerationSupporter;
+import rba.tool.core.sort.ISortValueCalculation;
+
+import java.io.OutputStreamWriter
+import java.io.FileOutputStream
+import java.nio.charset.StandardCharsets
+import rba.tool.core.sort.SortValue
+import java.util.Arrays
+import org.eclipse.emf.ecore.EObject
+import java.util.ArrayList
+import org.eclipse.emf.ecore.util.EContentsEList
+import org.eclipse.emf.ecore.EReference
+import org.eclipse.emf.common.util.EList
+
+class RBAModelWebGenerator extends AbstractGenerator {
+		
+	override doGenerate(Resource resource, IFileSystemAccess2 fsa, IGeneratorContext context) {
+		var isError = false
+		
+		for(r : resource.resourceSet.resources)
+		{
+			for(eo : r.allContents.toIterable)
+			{				
+				//to resolve cross reference 
+				var EContentsEList.FeatureIterator<EObject> fi = eo.eCrossReferences.iterator as EContentsEList.FeatureIterator<EObject>;
+				while( fi.hasNext() ) {
+					var EObject o = fi.next();
+					var EReference er = fi.feature() as EReference;
+					if (er.getEOpposite() !== null) {
+						//System.out.println(eo + "'s " + er.name + " --> " + o);
+						if (er.getEOpposite().isMany) {
+							var oppositeList =(o.eGet(er.getEOpposite()) as EList<EObject>);
+							if (!oppositeList.contains(eo)) {
+								oppositeList.add(eo)
+							}
+						} else {
+							
+						}
+					}
+				}
+			}
+			
+			for(w : r.warnings)
+			{
+				System.err.println(w.toString)
+			}
+			for(e : r.errors)
+			{
+				var filename = new File(r.URI.toFileString).name;
+				System.err.println("Error: " + filename + ":" + e.line + " " + e.message)
+				isError = true
+			}
+		}	
+		
+		if(isError) 
+		{
+			return;	
+		}
+		
+		runScript(WorkingDirectory.get() + "/script", "TemplateGenerator.groovy", resource.resourceSet);		
+	}	
+	
+	def void runScript(String scriptDir, String fileName, ResourceSet resourceSet) {
+		try {			
+			var dir = new File(scriptDir)
+			var file = new File(scriptDir + "/" + fileName)
+			if (!dir.exists() || !file.exists()) {
+				return;
+			}
+			var e = new GroovyScriptEngine(#[scriptDir], getClass().getClassLoader())
+			var config = new CompilerConfiguration()
+			config.setTargetDirectory(scriptDir)
+			var RunScriptInternalUtil util = new RunScriptInternalUtil()
+			util.setResourceManager(ResourceManager.INSTANCE)
+			util.setResourceSet(resourceSet)
+			util.loadZ3Lib();
+			util.calculate();
+			var Binding bind = new Binding()
+			bind.setVariable("location", scriptDir)
+			bind.setVariable("util", util)
+			e.setConfig(config)
+			var start_time = System.currentTimeMillis()
+			e.run(fileName, bind)
+			var finish_time = System.currentTimeMillis()
+			removeClassFiles(scriptDir)
+			new File(WorkingDirectory.get() + "/" + "success").createNewFile
+		} catch (Throwable e) {
+			e.printStackTrace()
+			System.err.println("error:" + e.getMessage())
+		}
+
+	}
+	
+	def static void removeClassFiles(String folderPath) {
+		var File folder = new File(folderPath)
+		for (File file : folder.listFiles()) {
+			var pattern = Pattern.compile(".class$")
+			var Matcher m = pattern.matcher(file.getName())
+			if (file.isFile() && m.find()) {
+				file.delete()
+			}
+		}
+	}
+
+}
+
+class WorkingDirectory {
+	def static String get() {
+		return new File(RBAModelWebGenerator.getProtectionDomain().getCodeSource().getLocation().toURI()).parent;
+	}
+}
+
+class RunScriptInternalUtil {
+	ResourceSet resourceSet;
+	ResourceManager resourceManager
+	private static SortValueCodeGenerationSupporter generationSupporter = new SortValueCodeGenerationSupporter();
+
+	private SortValue areaZorderSortValue = new SortValue();
+    private SortValue visibilitySortValue = new SortValue();
+    private SortValue csPrioritySortValue = new SortValue();
+    
+    static String OS_NAME = System.getProperty("os.name").toLowerCase();
+    def boolean isLinux() {
+        return OS_NAME.startsWith("linux");
+    }
+    def boolean isWindows() {
+        return OS_NAME.startsWith("windows");
+    }
+
+	def protected void setSortValueManager(SortValueManager sortValueManager) {
+		this.sortValueManager = sortValueManager
+	}
+
+	def protected void setResourceManager(ResourceManager resourceManager) {
+		this.resourceManager = resourceManager
+	}
+	
+	def protected void setResourceSet(ResourceSet resourceSet) {
+		this.resourceSet = resourceSet
+	}
+	
+	def boolean loadZ3Lib() {
+		var z3Path = WorkingDirectory.get();
+		
+		if(isLinux()) {
+			z3Path = z3Path + "/lib/linux/z3/bin";
+		}
+		else if(isWindows()) {
+			z3Path = z3Path + "/lib/windows/z3";
+		}
+		else {
+			print("error: Not supported OS");
+			return false;
+		}
+        
+        try {
+            loadLibs(z3Path);
+            addLibraryPath(z3Path);
+            return true;
+        } catch (Exception e) {
+            e.printStackTrace();
+            return false;
+        }
+    }
+    
+    def void loadLibs(String z3Path) throws Exception {
+    	if(isLinux()) {
+    		System.load(z3Path + "/libz3.so");
+        	System.load(z3Path + "/libz3java.so");
+    	}
+    	else if(isWindows()) {
+	        System.load(z3Path + "/Microsoft.Z3.dll");
+	        System.load(z3Path + "/msvcr110.dll");
+	        System.load(z3Path + "/msvcp110.dll");
+	        System.load(z3Path + "/vcomp110.dll");
+	        System.load(z3Path + "/libz3.dll");
+	        System.load(z3Path + "/libz3java.dll");
+        }
+    }
+
+    def void addLibraryPath(String pathToAdd) throws Exception {
+        val usrPathsField = ClassLoader.getDeclaredField("usr_paths");
+        usrPathsField.setAccessible(true);
+        val paths = usrPathsField.get(null) as String[];
+        for (String path : paths) {
+            if (path.equals(pathToAdd)) {
+                return;
+            }
+        }
+        val newPaths = Arrays.copyOf(paths, paths.length + 1);
+        newPaths.set(newPaths.length - 1, pathToAdd);
+        usrPathsField.set(null, newPaths);
+    }
+	
+	def void calculate() {
+		var code = resourceSet.compile.toString();
+		val createdFile = new File("./SortValueCalculation.java");
+		try { 
+			var writer = new OutputStreamWriter(new FileOutputStream(createdFile), StandardCharsets.UTF_8);
+			writer.write(code, 0, code.length());
+	    	writer.flush();
+	    	writer.close();
+		}
+		catch(Exception e) {
+			System.out.println("error: " + e.toString());
+		}
+		
+		var e = new GroovyScriptEngine("./", getClass().getClassLoader());
+        e.loadScriptByName("./SortValueCalculation.java");
+        var s = e.getGroovyClassLoader().loadClass("SortValueCalculation");
+
+        var calc = s.newInstance() as ISortValueCalculation;
+        calc.setUp();
+        areaZorderSortValue.setSortValue(calc.calculateArea_zorder());
+        visibilitySortValue.setSortValue(calc.calculateAllocatable_visibility());
+        csPrioritySortValue.setSortValue(calc.calculateContentState_priority());
+        calc.close();
+        
+        createdFile.delete();
+	}
+	
+	def compile(ResourceSet resourceSet) '''
+		«val allAllocatables = ResourceManager.INSTANCE.getRbaAllocatables(resourceSet)»
+		
+		«val allAreas = ResourceManager.INSTANCE.getRbaAreas(resourceSet)»
+«««		«FOR area : allAreas»
+«««			«area.name» : [ «FOR content : area.contents»«content.name», «ENDFOR»]
+«««		«ENDFOR»
+		«val allContents = ResourceManager.INSTANCE.getRbaContents(resourceSet)»
+«««		«FOR content : allContents»
+«««			«content.name» : [ «FOR area : content.allocatable»«area.name», «ENDFOR»]
+«««		«ENDFOR»
+		«generationSupporter.generate(allContents, allAllocatables, allAreas)»
+	'''
+
+	def void print(String message) {
+		System.out.println(message)
+	}
+
+	def List<Allocatable> getAllocatables() {
+		return resourceManager.getRbaAllocatables(resourceSet)
+	}
+	
+	def List<Area> getAreas() {
+		return resourceManager.getRbaAreas(resourceSet)
+	}
+	
+	def List<Zone> getZones() {		
+		return resourceManager.getRbaZones(resourceSet)
+	}
+	
+	def List<AreaSet> getAreaSets() {
+		return resourceManager.getRbaAreaSets(resourceSet)
+	}
+	
+	def List<ZoneSet> getZoneSets() {
+		return resourceManager.getRbaZoneSets(resourceSet)
+	}
+
+	def List<Content> getContents() {
+		return resourceManager.getRbaContents(resourceSet)
+	}
+	
+	def List<ViewContent> getViewContents() {
+		return resourceManager.getRbaViewContents(resourceSet)
+	}
+
+	def List<ViewContentSet> getViewContentSets() {
+		return resourceManager.getRbaViewContentSets(resourceSet)
+	}
+	
+	def List<SoundContent> getSoundContents() {
+		return resourceManager.getRbaSoundContents(resourceSet)
+	}
+
+	def List<SoundContentSet> getSoundContentSets() {
+		return resourceManager.getRbaSoundContentSets(resourceSet)
+	}
+	
+	def List<State> getStates() {
+		return resourceManager.getRbaState(resourceSet)
+	}
+	
+	def List<Scene> getScenes() {
+		return resourceManager.getRbaScenes(resourceSet)
+	}
+	
+	def List<Size> getSizes() {
+		return resourceManager.getRbaSizes(resourceSet)
+	} 
+	
+	def List<Display> getDisplays() {
+		return resourceManager.getRbaDisplays(resourceSet)
+	}
+
+	def List<PositionContainer> getPositionContainers() {
+		return resourceManager.getRbaPositionContainers(resourceSet)
+	}
+
+	def List<Package> getPackages() {
+		return resourceManager.getRbaPackages(resourceSet)
+	}
+
+	def List<Package> getRootPackages() {
+		return resourceManager.getRbaRootPackages(resourceSet)
+	}
+
+	def List<Constraint> getConstraints() {
+		return resourceManager.getRbaConstraints(resourceSet)
+	}
+
+	def List<Constraint> getOnlineConstraints() {
+		return resourceManager.getRbaOnlineConstraints(resourceSet)
+	}
+
+	def List<Constraint> getOfflineConstraints() {
+		return resourceManager.getRbaOfflineConstraints(resourceSet)
+	}
+	
+	def List<Constraint> getViewConstraints() {
+		return resourceManager.getRbaViewConstraints(resourceSet)
+	}
+
+	def List<Constraint> getSoundConstraints() {
+		return resourceManager.getRbaSoundConstraints(resourceSet)
+	}
+
+
+	def List<Variable> getVariables() {
+		return resourceManager.getRbaVariables(resourceSet)
+	}
+	
+	def List<SetOfOperator> getSetOfOperators() {
+		return resourceManager.getRbaSetOfOperators(resourceSet)
+	}	
+
+	def int getVisibility(String areaName) {
+		return visibilitySortValue.getValue(areaName);
+	}
+
+	def int getZorder(String areaName) {
+		return areaZorderSortValue.getValue(areaName);
+	}
+
+	def int getPriority(String contentName, String contentStateName) {
+		return csPrioritySortValue.getValue(contentName + "_" + contentStateName);
+	}
+	
+	def <T extends EObject> List<T> collect(Class<T> clazz) {
+       var results = new ArrayList<T>();
+       var ite = resourceManager.getRbaAllContents(resourceSet).iterator();
+       while (ite.hasNext()) {
+           var obj = ite.next();
+           if (clazz.isAssignableFrom(obj.getClass())) {
+               results.add(obj as T);
+           }
+       }
+       return results;
+   }
+}