RevitSdkSamples/snapshot/2012/SDK/RevitLookup/CS/TestCmds.cs

//
// Copyright 2003-2010 by Autodesk, Inc. 
//
// Permission to use, copy, modify, and distribute this software in
// object code form for any purpose and without fee is hereby granted, 
// provided that the above copyright notice appears in all copies and 
// that both that copyright notice and the limited warranty and
// restricted rights notice below appear in all supporting 
// documentation.
//
// AUTODESK PROVIDES THIS PROGRAM "AS IS" AND WITH ALL FAULTS. 
// AUTODESK SPECIFICALLY DISCLAIMS ANY IMPLIED WARRANTY OF
// MERCHANTABILITY OR FITNESS FOR A PARTICULAR USE.  AUTODESK, INC. 
// DOES NOT WARRANT THAT THE OPERATION OF THE PROGRAM WILL BE
// UNINTERRUPTED OR ERROR FREE.
//
// Use, duplication, or disclosure by the U.S. Government is subject to 
// restrictions set forth in FAR 52.227-19 (Commercial Computer
// Software - Restricted Rights) and DFAR 252.227-7013(c)(1)(ii)
// (Rights in Technical Data and Computer Software), as applicable.
//

using System;
using System.Collections.Generic;
using System.Windows.Forms;
using System.Collections;
using System.Data;

using Autodesk.Revit;
using Autodesk.Revit.DB;
using Autodesk.Revit.UI;
using RevitLookup.Test;
using Autodesk.Revit.Attributes;

// Each command is implemented as a class that provides the IExternalCommand Interface
//

namespace RevitLookup
{
   /// <summary>
   /// The classic bare-bones test.  Just brings up an Alert box to show that the connection to the external module is working.
   /// </summary>
   [Autodesk.Revit.Attributes.Transaction(Autodesk.Revit.Attributes.TransactionMode.Manual)]
   [Autodesk.Revit.Attributes.Regeneration(Autodesk.Revit.Attributes.RegenerationOption.Manual)]
   public class HelloWorld : IExternalCommand
   {
      public Result Execute(Autodesk.Revit.UI.ExternalCommandData cmdData, ref string msg, ElementSet elems)
      {
         TaskDialog helloDlg = new TaskDialog("Autodesk Revit");
         helloDlg.MainContent = "Hello World.";
         helloDlg.Show();

         return Result.Cancelled;
      }
   }

   /// <summary>
   /// SnoopDB command:  Browse all Elements in the current Document
   /// </summary>

   [Autodesk.Revit.Attributes.Transaction(Autodesk.Revit.Attributes.TransactionMode.Manual)]
   [Autodesk.Revit.Attributes.Regeneration(Autodesk.Revit.Attributes.RegenerationOption.Manual)]
   public class CmdSnoopDb : IExternalCommand
   {
      public Autodesk.Revit.UI.Result Execute(Autodesk.Revit.UI.ExternalCommandData cmdData, ref string msg, ElementSet elems)
      {
         Autodesk.Revit.UI.Result result;

         try
         {
            Snoop.CollectorExts.CollectorExt.m_app = cmdData.Application;	// TBD: see note in CollectorExt.cs

            // iterate over the collection and put them in an ArrayList so we can pass on
            // to our Form
            Autodesk.Revit.DB.Document doc = cmdData.Application.ActiveUIDocument.Document;
            FilteredElementCollector elemTypeCtor = (new FilteredElementCollector(doc)).WhereElementIsElementType();
            FilteredElementCollector notElemTypeCtor = (new FilteredElementCollector(doc)).WhereElementIsNotElementType();
            FilteredElementCollector allElementCtor = elemTypeCtor.UnionWith(notElemTypeCtor);
            ICollection<Element> founds = allElementCtor.ToElements();

            ArrayList objs = new ArrayList();
            foreach (Element element in founds)
            {
               objs.Add(element);               
            }

            System.Diagnostics.Trace.WriteLine(founds.Count.ToString());
            Snoop.Forms.Objects form = new Snoop.Forms.Objects(objs);
            form.ShowDialog();

            result = Autodesk.Revit.UI.Result.Succeeded;
         }
         catch (System.Exception e)
         {
            msg = e.Message;
            result = Autodesk.Revit.UI.Result.Failed;
         }

         return result;
      }
   }

   /// <summary>
   /// Snoop ModScope command:  Browse all Elements in the current selection set
   /// </summary>

   [Autodesk.Revit.Attributes.Transaction(Autodesk.Revit.Attributes.TransactionMode.Manual)]
   [Autodesk.Revit.Attributes.Regeneration(Autodesk.Revit.Attributes.RegenerationOption.Manual)]
   public class CmdSnoopModScope : IExternalCommand
   {
      public Autodesk.Revit.UI.Result Execute(Autodesk.Revit.UI.ExternalCommandData cmdData, ref string msg, ElementSet elems)
      {
         Autodesk.Revit.UI.Result result;

         try
         {
            Snoop.CollectorExts.CollectorExt.m_app = cmdData.Application;	// TBD: see note in CollectorExt.cs

            Snoop.Forms.Objects form = new Snoop.Forms.Objects(cmdData.Application.ActiveUIDocument.Selection.Elements);
            form.ShowDialog();

            result = Autodesk.Revit.UI.Result.Succeeded;
         }
         catch (System.Exception e)
         {
            msg = e.Message;
            result = Autodesk.Revit.UI.Result.Failed;
         }

         return result;
      }
   }

   /// <summary>
   /// Snoop App command:  Browse all objects that are part of the Application object
   /// </summary>

   [Autodesk.Revit.Attributes.Transaction(Autodesk.Revit.Attributes.TransactionMode.Manual)]
   [Autodesk.Revit.Attributes.Regeneration(Autodesk.Revit.Attributes.RegenerationOption.Manual)]
   public class CmdSnoopApp : IExternalCommand
   {
      public Autodesk.Revit.UI.Result Execute(Autodesk.Revit.UI.ExternalCommandData cmdData, ref string msg, ElementSet elems)
      {
         Autodesk.Revit.UI.Result result;

         try
         {
            Snoop.CollectorExts.CollectorExt.m_app = cmdData.Application;	// TBD: see note in CollectorExt.cs

            Snoop.Forms.Objects form = new Snoop.Forms.Objects(cmdData.Application.Application);
            form.ShowDialog();

            result = Autodesk.Revit.UI.Result.Succeeded;
         }
         catch (System.Exception e)
         {
            msg = e.Message;
            result = Autodesk.Revit.UI.Result.Failed;
         }

         return result;
      }
   }

   /// <summary>
   /// TestShell command:  The TestShell is a framework for adding small tests.  Each test is 
   /// plugged into the TestShell UI so that you don't have to write a new external command for
   /// each test and occupy additional menu items.  Create a Test by adding new RevitLookupTestFuncs objects.
   /// </summary>

   [Autodesk.Revit.Attributes.Transaction(Autodesk.Revit.Attributes.TransactionMode.Manual)]
   [Autodesk.Revit.Attributes.Regeneration(Autodesk.Revit.Attributes.RegenerationOption.Manual)]
   public class CmdTestShell : IExternalCommand
   {
      ArrayList m_tests = new ArrayList();
      Autodesk.Revit.UI.UIApplication m_app = null;

      public Autodesk.Revit.UI.Result Execute(Autodesk.Revit.UI.ExternalCommandData cmdData, ref string msg, ElementSet elems)
      {
         m_app = cmdData.Application;
         Snoop.CollectorExts.CollectorExt.m_app = cmdData.Application;	// TBD: see note in CollectorExt.cs   
         Autodesk.Revit.UI.Result result;

         try
         {
            // Since we dont have an "App" as such, there is no 
            // app-wide data - so just create tests for the duration
            // of the cmd and then destroy them
            CreateAndAddTests();

            Test.TestForm form = new Test.TestForm(RevitLookupTestFuncs.RegisteredTestFuncs());
            if (form.ShowDialog() == DialogResult.OK)
               form.DoTest();

            result = Autodesk.Revit.UI.Result.Succeeded;
         }
         catch (System.Exception e)
         {	// we want to catch it so we can see the problem, otherwise it just silently bails out

            /*System.Exception innerException = e.InnerException;

            while (innerException != null) {
                innerException = innerException.InnerException;
            }

            msg = innerException.Message;*/
            msg = e.Message;
            MessageBox.Show(msg);


            result = Autodesk.Revit.UI.Result.Failed;
         }

         finally
         {
            // if an exception happens anywhere, clean up before quitting
            RemoveAndFreeTestFuncs();
         }

         return result;
      }

      /// <summary>
      /// 
      /// </summary>
      private void CreateAndAddTests()
      {
         m_tests.Add(new Test.TestElements(m_app));
         m_tests.Add(new Test.TestDocument(m_app));
         m_tests.Add(new Test.TestGeometry(m_app));
         m_tests.Add(new Test.TestGraphicsStream(m_app));
         m_tests.Add(new Test.TestImportExport(m_app));
         m_tests.Add(new Test.TestUi(m_app));
         m_tests.Add(new Test.SDKSamples.SDKTestFuncs(m_app));
         m_tests.Add(new Test.TestApplication(m_app));

         foreach (RevitLookupTestFuncs testFunc in m_tests)
         {
            RevitLookupTestFuncs.AddTestFuncsToFramework(testFunc);
         }
      }

      /// <summary>
      /// 
      /// </summary>
      private void
      RemoveAndFreeTestFuncs()
      {
         foreach (RevitLookupTestFuncs testFunc in m_tests)
         {
            RevitLookupTestFuncs.RemoveTestFuncsFromFramework(testFunc);
         }
      }
   }

   /// <summary>
   /// Snoop ModScope command:  Browse all Elements in the current selection set
   /// </summary>

   [Autodesk.Revit.Attributes.Transaction(Autodesk.Revit.Attributes.TransactionMode.Manual)]
   [Autodesk.Revit.Attributes.Regeneration(Autodesk.Revit.Attributes.RegenerationOption.Manual)]
   public class CmdSampleMenuItemCallback : IExternalCommand
   {
      public Autodesk.Revit.UI.Result Execute(Autodesk.Revit.UI.ExternalCommandData cmdData, ref string msg, ElementSet elems)
      {
         Autodesk.Revit.UI.Result result;

         try
         {
            MessageBox.Show("Called back into RevitLookup by picking toolbar or menu item");
            result = Autodesk.Revit.UI.Result.Succeeded;
         }
         catch (System.Exception e)
         {
            msg = e.Message;
            result = Autodesk.Revit.UI.Result.Failed;
         }

         return result;
      }
   }

   [Autodesk.Revit.Attributes.Transaction(Autodesk.Revit.Attributes.TransactionMode.Manual)]
   [Autodesk.Revit.Attributes.Regeneration(Autodesk.Revit.Attributes.RegenerationOption.Manual)]
   public class CmdEvents : IExternalCommand
   {
      public Autodesk.Revit.UI.Result Execute(Autodesk.Revit.UI.ExternalCommandData cmdData, ref string msg, ElementSet elems)
      {
         EventTrack.Events.ApplicationEvents.m_app = cmdData.Application.Application;
         EventTrack.Events.DocEvents.m_docSet = cmdData.Application.Application.Documents;

         Autodesk.Revit.UI.Result result;

         try
         {
            RevitLookup.EventTrack.Forms.EventsForm dbox = new RevitLookup.EventTrack.Forms.EventsForm();
            dbox.ShowDialog();

            result = Autodesk.Revit.UI.Result.Succeeded;
         }
         catch (System.Exception e)
         {
            msg = e.Message;
            result = Autodesk.Revit.UI.Result.Failed;
         }

         return result;
      }
   }

}