Changeset 932 for trunk/modelicaml/org.openmodelica.modelicaml.simulation/src

Timestamp:

08/26/11 20:34:27 (13 years ago)

Author:

wschamai

Message:

 

File:

• trunk/modelicaml/org.openmodelica.modelicaml.simulation/src/org/openmodelica/modelicaml/simulation/handlers/CGFromEntireModelAndSimThisClassWithOMCSimCenterAction.java (moved) (moved from trunk/modelicaml/org.openmodelica.modelicaml.simulation/src/org/openmodelica/modelicaml/simulation/handlers/GenerateModelicaCodeFromEntireModelicaMLModelAndSimulationThisClassWithOMCSimCenterAction.java) (6 diffs)

trunk/modelicaml/org.openmodelica.modelicaml.simulation/src/org/openmodelica/modelicaml/simulation/handlers/CGFromEntireModelAndSimThisClassWithOMCSimCenterAction.java

@@ -36 +36 @@
 
 import java.io.IOException;
+import java.util.List;
 
 import org.eclipse.core.commands.AbstractHandler;
@@ -46 +47 @@
 import org.eclipse.core.resources.ResourcesPlugin;
 import org.eclipse.core.runtime.CoreException;
+import org.eclipse.core.runtime.IAdaptable;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.core.runtime.IStatus;
@@ -60 +62 @@
 import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
 import org.eclipse.emf.transaction.TransactionalEditingDomain;
-import org.eclipse.gmt.modisco.infra.browser.uicore.internal.model.ModelElementItem;
+import org.eclipse.jface.viewers.ISelection;
 import org.eclipse.jface.viewers.IStructuredSelection;
 import org.eclipse.jface.wizard.WizardDialog;
+import org.eclipse.papyrus.core.utils.BusinessModelResolver;
 import org.eclipse.papyrus.core.utils.EditorUtils;
-import org.eclipse.papyrus.diagram.common.editparts.IUMLEditPart;
 import org.eclipse.papyrus.resource.uml.UmlModel;
 import org.eclipse.papyrus.resource.uml.UmlUtils;
 import org.eclipse.swt.widgets.Shell;
 import org.eclipse.ui.PlatformUI;
-import org.eclipse.uml2.uml.Element;
 import org.eclipse.uml2.uml.NamedElement;
 import org.openmodelica.modelicaml.common.services.StringUtls;
@@ -93 +94 @@
  * @author rmwscham
  */
-public class GenerateModelicaCodeFromEntireModelicaMLModelAndSimulationThisClassWithOMCSimCenterAction extends AbstractHandler {
+public class CGFromEntireModelAndSimThisClassWithOMCSimCenterAction extends AbstractHandler {
 
     // private ModelManager modelManager = null;
@@ -124 +125 @@
      */
     public Object execute(ExecutionEvent event) throws ExecutionException {
-        TransactionalEditingDomain editingDomain = EditorUtils
-                .getTransactionalEditingDomain();
+        TransactionalEditingDomain editingDomain = EditorUtils.getTransactionalEditingDomain();
         editingDomain.getCommandStack().execute(getCommand(editingDomain));
         return null;
@@ -139 +139 @@
     protected Command getCommand(TransactionalEditingDomain editingDomain) {
 
-        Element umlElement = null;
+        EObject umlElement = null;
         IStructuredSelection selection = (IStructuredSelection) PlatformUI.getWorkbench().getActiveWorkbenchWindow().getSelectionService().getSelection();      
         Object input = ((IStructuredSelection) selection).getFirstElement();
         
-        if (input instanceof ModelElementItem) {
-            EObject eObject = ((ModelElementItem)input).getEObject();
-            if ( eObject instanceof Element ) {
-                umlElement = (Element)eObject;
-            }
-        }
-        else if (input instanceof IUMLEditPart) {
-            umlElement = ((IUMLEditPart)input).getUMLElement();
-        }
-        
-        UmlModel umlModel = UmlUtils.getUmlModel();
-        modelFileURI = umlModel.getResourceURI().toString();
-
-        modelName = umlModel.getResourceURI().lastSegment();
-        project = umlModel.getResourceURI().path().replace(modelName, "")
-                .replace("/resource/", "");
-
-        
-//      String projectName = umlModel.getResourceURI().path().replace(modelName, "").replace("/resource/", "");
-        String projectName = umlModel.getResource().getURI().segment(1);
-        IWorkspace workspace = ResourcesPlugin.getWorkspace();
-        IWorkspaceRoot root = workspace.getRoot();
-        IProject iProject = root.getProject(projectName);
-        
-        String projectPath = iProject.getLocationURI().toString().replaceFirst("file:\\/", "");
-        
-        String modelElementQualifiedName = ((NamedElement)umlElement).getQualifiedName();
-        String temp_modelElementWoSpecChar = StringUtls.replaceSpecCharExceptThis(modelElementQualifiedName, "::");
-        final String modelElementDotPath = temp_modelElementWoSpecChar.replaceAll("::", ".");
-        final String packageMoFilePath = projectPath+"/code-gen/"+ StringUtls.replaceSpecChar(umlElement.getModel().getName()) + "/" +"package.mo";
-
-        URI chainURI = URI.createPlatformPluginURI("/org.openmodelica.modelicaml.gen.modelica/bin/code_generation.chain",true);
-        ResourceSet rs = new ResourceSetImpl();
-        Resource r = (Resource) rs.createResource(chainURI);
-        try {
-            r.load(null);
-        } catch (IOException e1) {
-            // TODO Auto-generated catch block
-            e1.printStackTrace();
-        }
-
-        myChain = (CChain) r.getContents().get(0);
-
-        String modelFilePath = modelFileURI.replace("platform:/resource/", "");
-        String outputFolderPath = project;
-        String logPath = project + "/errors.log";
-
-        // Don't create Parameter Files... simply set the correct path into them
-        EList<File> files = myChain.getParametersFiles();
-        for (File file : files) {
-            if (file instanceof CModel) {
-                file.setPath(modelFilePath);
-            } else if (file instanceof CFolder) {
-                file.setPath(outputFolderPath);
-            } else if (file instanceof CLog) {
-                file.setPath(logPath);
-            }
-        }
-
-        Job job = new Job("Modelica Simulation with OMCSim Center") {
-            protected IStatus run(IProgressMonitor monitor) {
-                Boolean RegenerateCodeBeforeEachSimulation = Platform.getPreferencesService().getBoolean("org.openmodelica.modelicaml.preferences", "RegenerateCodeBeforeEachSimulation", false, null);
-                if (RegenerateCodeBeforeEachSimulation) {
-                    runchain(monitor);
+//      if (input instanceof ModelElementItem) {
+//      EObject eObject = ((ModelElementItem)input).getEObject();
+//      if ( eObject instanceof Element ) {
+//          umlElement = (Element)eObject;
+//      }
+//  }
+//  else if (input instanceof IUMLEditPart) {
+//      umlElement = ((IUMLEditPart)input).getUMLElement();
+//  }
+        
+        umlElement = (EObject) adaptSelectedElement(getCurrentSelections().get(0));
+        
+        if (umlElement instanceof NamedElement) {
+        
+            UmlModel umlModel = UmlUtils.getUmlModel();
+            modelFileURI = umlModel.getResourceURI().toString();
+
+            modelName = umlModel.getResourceURI().lastSegment();
+            project = umlModel.getResourceURI().path().replace(modelName, "").replace("/resource/", "");
+            
+//          String projectName = umlModel.getResourceURI().path().replace(modelName, "").replace("/resource/", "");
+            String projectName = umlModel.getResource().getURI().segment(1);
+            IWorkspace workspace = ResourcesPlugin.getWorkspace();
+            IWorkspaceRoot root = workspace.getRoot();
+            IProject iProject = root.getProject(projectName);
+            
+            String projectPath = iProject.getLocationURI().toString().replaceFirst("file:\\/", "");
+            
+            String modelElementQualifiedName = ((NamedElement)umlElement).getQualifiedName();
+            String temp_modelElementWoSpecChar = StringUtls.replaceSpecCharExceptThis(modelElementQualifiedName, "::");
+            final String modelElementDotPath = temp_modelElementWoSpecChar.replaceAll("::", ".");
+            final String packageMoFilePath = projectPath+"/code-gen/"+ StringUtls.replaceSpecChar( ((NamedElement)umlElement).getModel().getName()) + "/" +"package.mo";
+
+            URI chainURI = URI.createPlatformPluginURI("/org.openmodelica.modelicaml.gen.modelica/bin/code_generation.chain",true);
+            ResourceSet rs = new ResourceSetImpl();
+            Resource r = (Resource) rs.createResource(chainURI);
+            try {
+                r.load(null);
+            } catch (IOException e1) {
+                // TODO Auto-generated catch block
+                e1.printStackTrace();
+            }
+
+            myChain = (CChain) r.getContents().get(0);
+
+            String modelFilePath = modelFileURI.replace("platform:/resource/", "");
+            String outputFolderPath = project;
+            String logPath = project + "/errors.log";
+
+            // Don't create Parameter Files... simply set the correct path into them
+            EList<File> files = myChain.getParametersFiles();
+            for (File file : files) {
+                if (file instanceof CModel) {
+                    file.setPath(modelFilePath);
+                } else if (file instanceof CFolder) {
+                    file.setPath(outputFolderPath);
+                } else if (file instanceof CLog) {
+                    file.setPath(logPath);
                 }
-                return Status.OK_STATUS;
-            }
-        };
-        job.setUser(true);
-        job.schedule();
-        
-        
-        NewSimulationProjectFromModellingEnvironmentWizard projectWizard = new NewSimulationProjectFromModellingEnvironmentWizard(packageMoFilePath, modelElementDotPath, job);
-        WizardDialog wizard = new WizardDialog(
-                this.shell, projectWizard
-                );
-        wizard.open();
-        
-        if (projectWizard.getCreateConfig().equals("interactive")) {
-            WizardDialog wizardinteractive = new WizardDialog(
-                    this.shell,
-                    new SessionConfiguration_InteractiveWizard());
-            wizardinteractive.open();
-        } else if (projectWizard.getCreateConfig().equals("noninteractive")) {
-            WizardDialog wizardnoninteractive = new WizardDialog(
-                    this.shell,
-                    new SessionConfiguration_NonInteractiveWizard());
-            wizardnoninteractive.open();
-        }
-        
-        CompoundCommand cc = new CompoundCommand("Modelica Code Generation");
-        return (cc.unwrap());
-        
-    }
-
+            }
+
+            Job job = new Job("Modelica Simulation with OMCSim Center") {
+                protected IStatus run(IProgressMonitor monitor) {
+                    Boolean RegenerateCodeBeforeEachSimulation = Platform.getPreferencesService().getBoolean("org.openmodelica.modelicaml.preferences", "RegenerateCodeBeforeEachSimulation", false, null);
+                    if (RegenerateCodeBeforeEachSimulation) {
+                        runchain(monitor);
+                    }
+                    return Status.OK_STATUS;
+                }
+            };
+            job.setUser(true);
+            job.schedule();
+            
+            
+            NewSimulationProjectFromModellingEnvironmentWizard projectWizard = new NewSimulationProjectFromModellingEnvironmentWizard(packageMoFilePath, modelElementDotPath, job);
+            WizardDialog wizard = new WizardDialog(
+                    this.shell, projectWizard
+                    );
+            wizard.open();
+            
+            if (projectWizard.getCreateConfig().equals("interactive")) {
+                WizardDialog wizardinteractive = new WizardDialog(
+                        this.shell,
+                        new SessionConfiguration_InteractiveWizard());
+                wizardinteractive.open();
+            } else if (projectWizard.getCreateConfig().equals("noninteractive")) {
+                WizardDialog wizardnoninteractive = new WizardDialog(
+                        this.shell,
+                        new SessionConfiguration_NonInteractiveWizard());
+                wizardnoninteractive.open();
+            }
+            
+            CompoundCommand cc = new CompoundCommand("Modelica Code Generation");
+            return (cc.unwrap());           
+        }
+        
+        return null;
+    }
+
+    
+    private List<Object> getCurrentSelections() {
+        ISelection selection = PlatformUI.getWorkbench().getActiveWorkbenchWindow().getSelectionService().getSelection();
+        if(selection instanceof IStructuredSelection) {
+            IStructuredSelection structuredSelection = (IStructuredSelection)selection;
+            return structuredSelection.toList();
+        }
+    
+        return null;
+    }
+    
+    protected EObject adaptSelectedElement( Object selection) {
+        EObject eObject = null;
+        if(selection != null) {
+            if(selection instanceof IAdaptable) {
+                selection = ((IAdaptable)selection).getAdapter(EObject.class);
+            }
+            Object businessObject = BusinessModelResolver.getInstance().getBusinessModel(selection);
+            if(businessObject instanceof EObject) {
+                eObject = (EObject)businessObject;
+            }
+        }
+        return eObject;
+    }
+    
     /**
      * Runchain.