//
// (C) Copyright 2003-2014 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.Text;
using System.Linq;
using Autodesk.Revit;
using Autodesk.Revit.Creation;
using Autodesk.Revit.DB;
using Autodesk.Revit.UI;
using System.Windows.Forms;
using System.Diagnostics;

namespace Revit.SDK.Samples.ElementsBatchCreation.CS
{
   using View = Autodesk.Revit.DB.View;

   /// <summary>
   /// This class will demonstrate how to create many elements via batch creation methods
   /// </summary>
   public class ElementsBatchCreation
   {
      /// <summary>
      /// A reference to the external application
      /// </summary>
      private ExternalCommandData m_cmdData;

      /// <summary>
      /// A reference to active document
      /// </summary>
      private Autodesk.Revit.DB.Document m_doc;

      /// <summary>
      /// A reference to Level 1
      /// </summary>
      private Level m_level;

      /// <summary>
      /// A reference to ViewPlan named "Level 1;
      /// </summary>
      private ViewPlan m_viewPlan;

      /// <summary>
      /// Constructor of ElementsBatchCreation
      /// </summary>
      /// <param name="cmdData">A reference to the external application</param>
      public ElementsBatchCreation(ExternalCommandData cmdData)
      {
         m_cmdData = cmdData;
         m_doc = cmdData.Application.ActiveUIDocument.Document;
      }

      /// <summary>
      /// Batch Creations of several elements, it will call separate methods for each element
      /// </summary>
      /// <returns>If all batch creations succeed, return true; otherwise, return false</returns>
      public bool CreateElements()
      {
         //Get common informations for batch creation
         Transaction tran = new Transaction(m_doc, "ElementBatchCreation");
         tran.Start();
         PreCreate();
         m_doc.AutoJoinElements();
         m_doc.Regenerate();
         tran.Commit();

         //Prepare messages to notify user of succeed and failed operations      
         String failedMethods = "";
         String succeedMethods = "";
         bool success = false;
         tran.Start();
         success = CreateAreas();
         m_doc.AutoJoinElements();
         m_doc.Regenerate();
         tran.Commit();
         if (success)
         {
            succeedMethods += " Area";
         }
         else
         {
            failedMethods += " Area";
         }

         tran.Start();
         success = CreateColumns();
         m_doc.AutoJoinElements();
         m_doc.Regenerate();
         tran.Commit();
         if (success)
         {
            succeedMethods += " Column";
         }
         else
         {
            failedMethods += " Column";
         }

         tran.Start();
         success = CreateRooms();
         m_doc.AutoJoinElements();
         m_doc.Regenerate();
         tran.Commit();
         if (success)
         {
            succeedMethods += " Room";
         }
         else
         {
            failedMethods += " Room";
         }

         tran.Start();
         success = CreateTextNotes();
         m_doc.AutoJoinElements();
         m_doc.Regenerate();
         tran.Commit();
         if (success)
         {
            succeedMethods += " TextNote";
         }
         else
         {
            failedMethods += " TextNote";
         }

         tran.Start();
         success = CreateWalls();
         m_doc.AutoJoinElements();
         m_doc.Regenerate();
         tran.Commit();
         if (success)
         {
            succeedMethods += " Wall";
         }
         else
         {
            failedMethods += " Wall";
         }

         bool result = true;
         if (String.IsNullOrEmpty(succeedMethods))
         {
            TaskDialog.Show("ElementsBatchCreation", "Batch creations of" + failedMethods + " failed", TaskDialogCommonButtons.Close);
            result = false;
         }
         else if (String.IsNullOrEmpty(failedMethods))
         {
            TaskDialog.Show("ElementsBatchCreation", "Batch creations of" + succeedMethods + " succeed", TaskDialogCommonButtons.Close);
         }
         else
         {
            TaskDialog.Show("ElementsBatchCreation", "Batch creations of" + succeedMethods + " succeed," + " Batch creations of" + failedMethods + " failed", TaskDialogCommonButtons.Close);
            result = false;
         }
         return result;
      }

      /// <summary>
      /// Get common informations for batch creation
      /// </summary>
      private void PreCreate()
      {
         try
         {
            Autodesk.Revit.Creation.Application appCreation = m_cmdData.Application.Application.Create;

            //Try to get Level named "Level 1" which will be used in most creations
            m_level = (from elem in
                          new FilteredElementCollector(m_doc).OfClass(typeof(Level)).ToElements()
                       let level = elem as Level
                       where level != null && "Level 1" == level.Name
                       select level).First();

            //If ViewPlan "Level 1" does not exist, try to create one.
            if (null != m_level)
            {
               ElementId AreaSchemeId = new FilteredElementCollector(m_doc).OfClass(typeof(AreaScheme)).FirstOrDefault(a => a.Name == "Rentable").Id;
               try
               {
                  m_viewPlan = ViewPlan.CreateAreaPlan(m_doc, AreaSchemeId, m_level.Id);
                  m_viewPlan.Name = "Level 1";
               }
               catch
               {
               }
            }

            if (null == m_level && null == m_viewPlan)
            {
               return;
            }



            //List of Curve is used to store Area's boundary lines
            List<Curve> curves = new List<Curve>();

            Autodesk.Revit.DB.XYZ pt1 = new Autodesk.Revit.DB.XYZ(-4, 95, 0);
            Autodesk.Revit.DB.XYZ pt2 = new Autodesk.Revit.DB.XYZ(-106, 95, 0);
            Line line = Line.CreateBound(pt1, pt2);
            curves.Add(line);
            pt1 = new Autodesk.Revit.DB.XYZ(-4, 105, 0);
            pt2 = new Autodesk.Revit.DB.XYZ(-106, 105, 0);
            line = Line.CreateBound(pt1, pt2);
            curves.Add(line);


            for (int i = 0; i < 11; i++)
            {
               pt1 = new Autodesk.Revit.DB.XYZ(-5 - i * 10, 94, 0);
               pt2 = new Autodesk.Revit.DB.XYZ(-5 - i * 10, 106, 0);
               line = Line.CreateBound(pt1, pt2);
               if (null != line)
               {
                  curves.Add(line);
               }

            }

            // Create Area Boundary Line for Area
            // It is necessary if need to create closed region for Area
            // But for Room, it is not necessary.
            Autodesk.Revit.DB.XYZ origin = new Autodesk.Revit.DB.XYZ(0, 0, 0);
            Autodesk.Revit.DB.XYZ norm = new Autodesk.Revit.DB.XYZ(0, 0, 1);
            Plane plane = appCreation.NewPlane(norm, origin);
            if (null != plane)
            {
               SketchPlane sketchPlane = SketchPlane.Create(m_doc, plane);
               if (null != sketchPlane)
               {
                  foreach (Curve curve in curves)
                  {
                     m_doc.Create.NewAreaBoundaryLine(sketchPlane, curve, m_viewPlan);
                  }
               }
            }


            //Create enclosed region using Walls for Room                
            pt1 = new Autodesk.Revit.DB.XYZ(5, -5, 0);
            pt2 = new Autodesk.Revit.DB.XYZ(55, -5, 0);
            line = Line.CreateBound(pt1, pt2);
            Wall.Create(m_doc, line, m_level.Id, true);

            pt1 = new Autodesk.Revit.DB.XYZ(5, 5, 0);
            pt2 = new Autodesk.Revit.DB.XYZ(55, 5, 0);
            line = Line.CreateBound(pt1, pt2);
            Wall.Create(m_doc, line, m_level.Id, true);

            for (int i = 0; i < 6; i++)
            {
               pt1 = new Autodesk.Revit.DB.XYZ(5 + i * 10, -5, 0);
               pt2 = new Autodesk.Revit.DB.XYZ(5 + i * 10, 5, 0);
               line = Line.CreateBound(pt1, pt2);
               Wall.Create(m_doc, line, m_level.Id, true);
            }

         }
         catch (Exception)
         {
         }
      }

      /// <summary>
      /// Batch creation of Areas
      /// </summary>
      /// <returns>If batch creation succeeds, return true; otherwise, return false</returns>
      private bool CreateAreas()
      {
         try
         {
            if (null == m_viewPlan)
            {
               return false;
            }

            List<AreaCreationData> areaCreationDatas = new List<AreaCreationData>();
            //Create AreaCreateDatas for Areas' batch creation 
            for (int i = 1; i < 11; i++)
            {
               Autodesk.Revit.DB.UV point = new Autodesk.Revit.DB.UV(i * -10, 100);
               Autodesk.Revit.DB.UV tagPoint = new Autodesk.Revit.DB.UV(i * -10, 100);
               AreaCreationData areaCreationData = new AreaCreationData(m_viewPlan, point);
               if (null != areaCreationData)
               {
                  areaCreationData.TagPoint = tagPoint;
                  areaCreationDatas.Add(areaCreationData);
               }
            }

            // Create Areas
            if (0 == areaCreationDatas.Count)
            {
               return false;
            }
            m_doc.Create.NewAreas(areaCreationDatas);
         }
         catch (Exception)
         {
            return false;
         }

         return true;
      }

      /// <summary>
      /// Batch creation of Columns
      /// </summary>
      /// <returns>If batch creation succeeds, return true; otherwise, return false</returns>
      private bool CreateColumns()
      {
         try
         {
            List<FamilyInstanceCreationData> fiCreationDatas = new List<FamilyInstanceCreationData>();

            if (null == m_level)
            {
               return false;
            }

            //Try to get a FamilySymbol
            FamilySymbol familySymbol = (from elem in
                                            new FilteredElementCollector(m_doc).OfCategory(BuiltInCategory.OST_StructuralColumns).OfType<FamilySymbol>()
                                         let var = elem as FamilySymbol
                                         where var != null && var.Category != null && "Structural Columns" == var.Category.Name
                                         select var).First();

            if (null == familySymbol)
            {
               return false;
            }

            //Create FamilyInstanceCreationData for FamilyInstances' batch creation 
            for (int i = 1; i < 11; i++)
            {
               Autodesk.Revit.DB.XYZ location = new Autodesk.Revit.DB.XYZ(i * 10, 100, 0);
               m_doc.Create.NewFamilyInstance(location, familySymbol, m_level, Autodesk.Revit.DB.Structure.StructuralType.Column);

            }

         }
         catch (Exception)
         {
            return false;
         }

         return true;
      }

      /// <summary>
      /// Batch creation of Rooms
      /// </summary>
      /// <returns>If batch creation succeeds, return true; otherwise, return false</returns>
      private bool CreateRooms()
      {
         try
         {
            if (null == m_level)
            {
               return false;
            }

            //Try to get Phase used to create Rooms
            Phase phase = (from elem in
                              new FilteredElementCollector(m_doc).OfClass(typeof(Phase)).ToElements()
                           select elem).First() as Phase;

            if (null == phase)
            {
               return false;
            }

            for (int i = 1; i < 6; i++)
            {
               Autodesk.Revit.DB.UV point = new Autodesk.Revit.DB.UV(i * 10, 0);
               Autodesk.Revit.DB.UV tagPoint = new Autodesk.Revit.DB.UV(i * 10, 0);
               Autodesk.Revit.DB.Architecture.Room room = m_doc.Create.NewRoom(m_level, point);
            }

         }
         catch (Exception)
         {
            return false;
         }

         return true;
      }

      /// <summary>
      /// Batch creation of TextNotes
      /// </summary>
      /// <returns>If batch creation succeeds, return true; otherwise, return false</returns>
      private bool CreateTextNotes()
      {
         try
         {


            //Try to get View named "Level 1" where the TextNotes are
            View view = (from elem in
                            new FilteredElementCollector(m_doc).OfClass(typeof(ViewPlan)).ToElements()
                         let var = elem as View
                         where var != null && !var.IsTemplate && null != var.Name && "Level 1" == var.Name && ViewType.FloorPlan == var.ViewType
                         select var).First();
            if (null == view)
            {
               return false;
            }

            //Create TextNoteCreationData for TextNotes' batch creation 
            for (int i = 1; i < 6; i++)
            {
               Autodesk.Revit.DB.XYZ origin = new Autodesk.Revit.DB.XYZ(i * -20, -100, 0);
               Autodesk.Revit.DB.XYZ baseVec = new Autodesk.Revit.DB.XYZ(1, 0, 0);
               Autodesk.Revit.DB.XYZ upVec = new Autodesk.Revit.DB.XYZ(0, 0, 1);
               m_doc.Create.NewTextNote(view, origin, baseVec, upVec, 10, TextAlignFlags.TEF_ALIGN_CENTER, "TextNote");
            }

            // Create TextNotes

         }
         catch (Exception)
         {
            return false;
         }

         return true;
      }

      /// <summary>
      /// Batch creation of Walls
      /// </summary>
      /// <returns>If batch creation succeeds, return true; otherwise, return false</returns>
      private bool CreateWalls()
      {
         try
         {
            if (null == m_level)
            {
               return false;
            }

            Autodesk.Revit.Creation.Application appCreation = m_cmdData.Application.Application.Create;
            for (int i = 1; i < 11; i++)
            {
               // Create wall's profile which is a combination of rectangle and arc

               IList<Curve> curveArray = new List<Curve>();
               // Create three lines for rectangle part of profile.
               Autodesk.Revit.DB.XYZ pt1 = new Autodesk.Revit.DB.XYZ(i * 10, -80, 15);
               Autodesk.Revit.DB.XYZ pt2 = new Autodesk.Revit.DB.XYZ(i * 10, -80, 0);
               Autodesk.Revit.DB.XYZ pt3 = new Autodesk.Revit.DB.XYZ(i * 10, -90, 0);
               Autodesk.Revit.DB.XYZ pt4 = new Autodesk.Revit.DB.XYZ(i * 10, -90, 15);
               Line line1 = Line.CreateBound(pt1, pt2);
               Line line2 = Line.CreateBound(pt2, pt3);
               Line line3 = Line.CreateBound(pt3, pt4);

               // Create arc part of profile.
               Autodesk.Revit.DB.XYZ pointInCurve = new Autodesk.Revit.DB.XYZ(i * 10, -85, 20);
               Arc arc = Arc.Create(pt4, pt1, pointInCurve);

               if (null == line1 || null == line2 || null == line3 || null == arc)
               {
                  continue;
               }
               curveArray.Add(line1);
               curveArray.Add(line2);
               curveArray.Add(line3);
               curveArray.Add(arc);
               Wall.Create(m_doc, curveArray, true);
            }

            for (int i = 1; i < 11; i++)
            {
               Autodesk.Revit.DB.XYZ pt1 = new Autodesk.Revit.DB.XYZ(i * 10, -110, 0);
               Autodesk.Revit.DB.XYZ pt2 = new Autodesk.Revit.DB.XYZ(i * 10, -120, 0);
               Line line = Line.CreateBound(pt1, pt2);
               Wall.Create(m_doc, line, m_level.Id, true);
            }


         }
         catch (Exception)
         {
            return false;
         }
         return true;
      }
   }
}