WebSVN – DevTools – Diff – /EA_ReqPro/trunk/EA_ReqPro/ReqProMain.cs

       private Thread ourThread = null;
       public readonly static String GUI_OUTPUT_TAB_NAME = "EA_ReqPro";
       public static EA.Repository EA_Repository = null;
+      public static int abortRequests = 0;
       public static bool mustAbort = false;
       /// <summary>
       /// Called when EA initialised. Creates an output tab for the add in.
       /// </summary>
                return "-&EA_ReqPro";
             case "-&EA_ReqPro":
                if (ourThread != null && ourThread.IsAlive == true)
+               {
-                  ar = new string[] {"&Abort Import"};
+                  ar = new string[] {"&Abort Import/Export"};
                   return ar;
+               }
                else
+               {
                   ar = new string[]{
                                    "&Import for Traceability Use",
                                    "&Display Change Log",
                                    "&Import for Document Model Use",
+                                   "&Export to ReqPro",
                                    "&Export to ReqPro Compatible CSV File",
                                    "&Create Requirement Diagram",
+                                   "&Create Requirement Diagram Using Model Search"
                   };
                   return ar;
+               }
+         }
          if (IsProjectOpen(repository))
+         {
             switch (itemName)
+            {
-               case "&Abort Import":
+               case "&Abort Import/Export":
                case "&Import for Document Model Use":
+               case "&Export to ReqPro":
                case "&Export to ReqPro Compatible CSV File":
                case "&Create Requirement Diagram":
+               case "&Create Requirement Diagram Using Model Search":
                case "&Import for Traceability Use":
                   isEnabled = true;
                   break;
                case "&Display Change Log":
                   else
+                  {
                      isEnabled = false;
+                  }
                   break;
+            }
+         }
          else
+         {
             isEnabled = false;
+      {
          EA_Repository = repository;
          switch( itemName )
+         {
+            // This menu option appears only when a thread is active. It allows the user to initiate
+            // an abort of the thread.
-            case "&Abort Import":
+            case "&Abort Import/Export":
                if (ourThread != null && ourThread.IsAlive == true)
+               {
+                  if (abortRequests > 0)
+                  {
+                     // if user has made a second abort request, it is probably because the thread does not
+                     // appear to be aborting. This can happen sometimes. I have noticed occasional lockups,
+                     // and every time this happens, the thread seems to be locked up in a call to an EA
+                     // automation interface function, so perhaps there is some wierd interaction between
+                     // the thread and the use of the menu in the main EA process? Anyway, lets try to abort
+                     // the thread now because it does not appear as if it is going to quit gracefully.
+                     ourThread.Abort();
+                     ReqProDatabase.close();
+                     break;
+                  }
+                  else
+                  {
+                     // try to quit gracefully.
-                  mustAbort = true;
+                     mustAbort = true;
+                     abortRequests++;
+                  }
+               }
                break;
             case "&Import for Traceability Use":
                if (ourThread == null || ourThread.IsAlive == false)
                   ourThread.ApartmentState = ApartmentState.STA;
                   ourThread.Start();
+               }
                break;
+            case "&Export to ReqPro":
+               if (ourThread == null || ourThread.IsAlive == false)
+               {
+                  ourThread = new Thread(new ThreadStart(ExportToReqPro_Thread));
+                  ourThread.ApartmentState = ApartmentState.STA;
+                  ourThread.Start();
+               }
+               break;
             case "&Export to ReqPro Compatible CSV File":
                repository.EnsureOutputVisible(GUI_OUTPUT_TAB_NAME);
                repository.WriteOutput(GUI_OUTPUT_TAB_NAME,"",-1);
                ExportToReqProCSVForm export_dlg = new ExportToReqProCSVForm();
                DialogResult dlgRes = export_dlg.ShowDialog();
                allowedElementTypes.Add("Requirement");
                //allowedElementTypes.Add("UseCase");
                ElementAccumulator reqLister = new ElementAccumulator(allowedElementTypes);
                EA.Package thePackage = (EA.Package)repository.GetTreeSelectedObject();
                EA_Utilities.findAndProcessPackageElements( thePackage, reqLister, true );
-               EA.Diagram newDiagram = (EA.Diagram)thePackage.Diagrams.AddNew("Requirements","Logical");
+               createRequirementDiagram(thePackage, reqLister.Elements);
-               newDiagram.Update();
+               break;
-               thePackage.Update();
+            }
-               foreach (EA.Element element in reqLister.Elements)
+            case "&Create Requirement Diagram Using Model Search":
+            {
+               QueryForm qf = new QueryForm();
+               DialogResult qf_dlgRes = qf.ShowDialog();
+               if (qf_dlgRes == DialogResult.OK)
+               {
-                  EA.DiagramObject newDiagramObject = (EA.DiagramObject)newDiagram.DiagramObjects.AddNew("l=1;r=10;t=1;b=1", "");
+                  EA.Package thePackage = (EA.Package)repository.GetTreeSelectedObject();
-                  newDiagramObject.ElementID = element.ElementID;
+                  EA.Collection c = EA_Repository.GetElementsByQuery(qf.textBox_query_name.Text, qf.textBox_search_term.Text);
-                  newDiagramObject.Update();
+                  createRequirementDiagram(thePackage, c);
+               }
-               newDiagram.DiagramObjects.Refresh();
-               thePackage.Update();
-               repository.GetProjectInterface().LayoutDiagramEx( newDiagram.DiagramGUID,
-                  EA.ConstLayoutStyles.lsDiagramDefault, 4, 20, 20, false);
                break;
+            }
+         }
+      }
+      private void createRequirementDiagram(EA.Package containerPackage, EA.Collection c)
+      {
+         EA.Diagram newDiagram = (EA.Diagram)containerPackage.Diagrams.AddNew("Requirements","Logical");
+         newDiagram.Update();
+         containerPackage.Update();
+         foreach (EA.Element element in c)
+         {
+            EA.DiagramObject newDiagramObject = (EA.DiagramObject)newDiagram.DiagramObjects.AddNew("l=1;r=10;t=1;b=1", "");
+            newDiagramObject.ElementID = element.ElementID;
+            newDiagramObject.Update();
+         }
+         newDiagram.DiagramObjects.Refresh();
+         containerPackage.Update();
+         EA_Repository.GetProjectInterface().LayoutDiagramEx( newDiagram.DiagramGUID,
+            EA.ConstLayoutStyles.lsDiagramDefault, 4, 20, 20, false);
+      }
+      private void createRequirementDiagram(EA.Package containerPackage, ArrayList elements)
+      {
+         EA.Diagram newDiagram = (EA.Diagram)containerPackage.Diagrams.AddNew("Requirements","Logical");
+         newDiagram.Update();
+         containerPackage.Update();
+         foreach (EA.Element element in elements)
+         {
+            EA.DiagramObject newDiagramObject = (EA.DiagramObject)newDiagram.DiagramObjects.AddNew("l=1;r=10;t=1;b=1", "");
+            newDiagramObject.ElementID = element.ElementID;
+            newDiagramObject.Update();
+         }
+         newDiagram.DiagramObjects.Refresh();
+         containerPackage.Update();
+         EA_Repository.GetProjectInterface().LayoutDiagramEx( newDiagram.DiagramGUID,
+            EA.ConstLayoutStyles.lsDiagramDefault, 4, 20, 20, false);
+      }
+      /// <summary>
+      /// Thread function for import for traceability use operation
+      /// </summary>
       private static void ImportReqProDatabase_Thread()
+      {
+         bool cancelled = false;
          mustAbort = false;
-         ImportReqProDatabase import_parser = new ImportReqProDatabase();
+         abortRequests = 0;
          EA_Repository.EnsureOutputVisible(GUI_OUTPUT_TAB_NAME);
          EA_Repository.WriteOutput(GUI_OUTPUT_TAB_NAME,"",-1);
+         ImportReqProDatabase import_parser = new ImportReqProDatabase();
-         import_parser.prompt_and_parse(ReqProDB_Artifact.MODE.TRACEABILITY);
+         import_parser.prompt_and_parse(ReqProDB_Artifact.MODE.TRACEABILITY, out cancelled);
          if (mustAbort)
+         {
+            EA_Repository.WriteOutput(GUI_OUTPUT_TAB_NAME, "Import Aborted", -1);
             MessageBoxEx.Show("Import Aborted", "Progress");
+         }
+         else if (cancelled)
+         {
+            Main.WriteOutput("Import Cancelled",-1);
+         }
+         abortRequests = 0;
+         ReqProDatabase.close();
+      }
+      /// <summary>
+      /// Thread function for import for document model use operation
+      /// </summary>
       private static void CopyReqProDatabase_Thread()
+      {
+         bool cancelled = false;
          mustAbort = false;
+         abortRequests = 0;
          EA_Repository.EnsureOutputVisible(GUI_OUTPUT_TAB_NAME);
          EA_Repository.WriteOutput(GUI_OUTPUT_TAB_NAME,"",-1);
          CopyReqProDatabase copy_parser = new CopyReqProDatabase();
-         copy_parser.prompt_and_parse(ReqProDB_Artifact.MODE.DOC_MODEL);
+         copy_parser.prompt_and_parse(ReqProDB_Artifact.MODE.DOC_MODEL, out cancelled);
          if (mustAbort)
+         {
+            EA_Repository.WriteOutput(GUI_OUTPUT_TAB_NAME, "Import Aborted", -1);
             MessageBoxEx.Show("Import Aborted", "Progress");
+         }
+         else if (cancelled)
+         {
+            Main.WriteOutput("Import Cancelled",-1);
+         }
+         abortRequests = 0;
+         ReqProDatabase.close();
+      }
+      /// <summary>
+      /// Thread function for export to reqpro database operation
+      /// </summary>
+      private static void ExportToReqPro_Thread()
+      {
+         bool cancelled = false;
+         mustAbort = false;
+         abortRequests = 0;
+         EA_Repository.EnsureOutputVisible(GUI_OUTPUT_TAB_NAME);
+         EA_Repository.WriteOutput(GUI_OUTPUT_TAB_NAME,"",-1);
+         ExportToReqProDatabase export_parser = new ExportToReqProDatabase();
+         export_parser.prompt_and_parse(ReqProDB_Artifact.MODE.EXPORT, out cancelled);
+         if (mustAbort)
+         {
+            EA_Repository.WriteOutput(GUI_OUTPUT_TAB_NAME, "Export Aborted", -1);
+            MessageBoxEx.Show("Export Aborted", "Progress");
+         }
+         else if (cancelled)
+         {
+            Main.WriteOutput("Export Cancelled",-1);
+         }
+         abortRequests = 0;
+         ReqProDatabase.close();
+      }
       public static void MessageBoxException(Exception exc)
+      {
          MessageBoxException(exc, "Exception");
+      }
       public static void MessageBoxException(Exception exc, string caption)
+      {
-         MessageBoxEx.Show(exc.ToString(), caption);
+         MessageBoxEx.Show(exc.ToString(), caption, MessageBoxIcon.Error);
+      }
+      public static void WriteOutput(string s, int id)
+      {
+         EA_Repository.WriteOutput( GUI_OUTPUT_TAB_NAME, s, id);
+      }
+      public static void ClearOutput()
+      {
+         EA_Repository.ClearOutput(GUI_OUTPUT_TAB_NAME);
+      }
+      public static void WriteSeperator()
+      {
+         EA_Repository.WriteOutput( GUI_OUTPUT_TAB_NAME, "----------------------------------------------------------------------------------------------------------------------", -1);
+      }
+   }
+}

Subversion Repositories DevTools

(root)/EA_ReqPro/trunk/EA_ReqPro/ReqProMain.cs – Rev 2153 → 2155