RevitSdkSamples/SDK/Samples/ShaftHolePuncher/CS/ShaftHolePuncherForm.cs

//
// (C) Copyright 2003-2019 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.ComponentModel;
using System.Data;
using System.Drawing;
using System.Text;
using System.Windows.Forms;
using Autodesk.Revit.DB;
using Autodesk.Revit.UI;
using Point = System.Drawing.Point;

namespace Revit.SDK.Samples.ShaftHolePuncher.CS
{
    /// <summary>
    /// window form contains one picture box to show the 
    /// profile of a wall or floor, and three command buttons.
    /// User can draw curves of opening in picture box.
    /// </summary>
    public partial class ShaftHolePuncherForm : System.Windows.Forms.Form
    {
        #region class members
        private Profile m_profile;  //save the profile data
        private ITool m_tool = null; //current using tool
        Size m_sizePictureBox; //size of picture box
        #endregion

        /// <summary>
        /// constructor
        /// </summary>
        public ShaftHolePuncherForm()
        {
            InitializeComponent();
        }

        /// <summary>
        /// constructor
        /// </summary>
        /// <param name="profile">ProfileWall, ProfileFloor or ProfileNull</param>
        public ShaftHolePuncherForm(Profile profile)
            : this()
        {
            m_profile = profile;
            m_sizePictureBox = this.pictureBox.Size;

            if (profile is ProfileWall)
            {
                m_tool = new RectangleTool();
            }
            else
            {
                m_tool = new LineTool();
            }

            if (profile is ProfileNull)
            {
                this.ScaleComboBox.Visible = true;
                this.ScaleComboBox.SelectedIndex = 0;
                this.scaleLabel.Visible = true;
            }
            else if (profile is ProfileBeam)
            {
                this.DirectionPanel.Visible = true;
                this.DirectionComboBox.SelectedIndex = 0;
            }
        }

        /// <summary>
        /// store mouse location when mouse down
        /// </summary>
        /// <param name="sender">object who sent this event</param>
        /// <param name="e">event args</param>
        private void PictureBox_MouseDown(object sender, MouseEventArgs e)
        {
            Graphics graphics = this.pictureBox.CreateGraphics();
            graphics.SmoothingMode = System.Drawing.Drawing2D.SmoothingMode.HighQuality;
            m_tool.OnMouseDown(e);
            this.pictureBox.Refresh();
        }

        /// <summary>
        /// draw the line to where mouse moved
        /// </summary>
        /// <param name="sender">object who sent this event</param>
        /// <param name="e">event args</param>
        private void PictureBox_MouseMove(object sender, MouseEventArgs e)
        {
            this.pictureBox.Refresh();
            Graphics graphics = this.pictureBox.CreateGraphics();
            graphics.SmoothingMode = System.Drawing.Drawing2D.SmoothingMode.HighQuality;
            m_tool.OnMouseMove(graphics, e);
        }

        /// <summary>
        /// draw the curve of floor (or wall) and curves of Opening
        /// </summary>
        /// <param name="sender">object who sent this event</param>
        /// <param name="e">event args</param>
        private void PictureBox_Paint(object sender, PaintEventArgs e)
        {
            e.Graphics.SmoothingMode = System.Drawing.Drawing2D.SmoothingMode.HighQuality;

            //Draw the pictures in the m_tools
            m_tool.Draw(e.Graphics);

            //get transform matrix
            m_profile.ComputeScaleMatrix(m_sizePictureBox);
            Matrix4 trans = m_profile.Compute3DTo2DMatrix();

            //draw profile
            m_profile.Draw2D(e.Graphics, Pens.Blue, trans);
        }

        /// <summary>
        /// clear all the curves of the Opening
        /// </summary>
        /// <param name="sender">object who sent this event</param>
        /// <param name="e">event args</param>
        private void ButtonClean_Click(object sender, EventArgs e)
        {
            m_tool.Clear();
            m_tool.Finished = false;
            this.pictureBox.Refresh();
        }

        /// <summary>
        /// create Shaft Opening in Revit
        /// </summary>
        /// <param name="sender">object who sent this event</param>
        /// <param name="e">event args</param>
        private void CreateButton_Click(object sender, EventArgs e)
        {
            List<Point> points = m_tool.Points;

            if (!m_tool.Finished)
            {
                TaskDialog.Show("Revit", "Please finish the curve of Opening first!");
                return;
            }

            List<Vector4> ps3D = m_profile.Transform2DTo3D(points.ToArray());

            try
            {
                m_profile.CreateOpening(ps3D);
                this.Close();
            }
            catch (Exception ex)
            {
                TaskDialog.Show("Revit", ex.Message);
                ButtonClean_Click(null, null);
            }
        }

        /// <summary>
        /// close the form
        /// </summary>
        /// <param name="sender">object who sent this event</param>
        /// <param name="e">event args</param>
        private void CancelButton_Click(object sender, EventArgs e)
        {
            this.Close();
        }

        /// <summary>
        /// set the scale of profile when create Shaft Opening
        /// </summary>
        /// <param name="sender">object who sent this event</param>
        /// <param name="e">event args</param>
        private void ScaleComboBox_SelectedIndexChanged(object sender, EventArgs e)
        {
            ProfileNull profile = m_profile as ProfileNull;
            profile.Scale = (float)Convert.ToDouble(this.ScaleComboBox.Text);
            m_profile.ComputeScaleMatrix(m_sizePictureBox);
        }

        private void DirectionComboBox_SelectedIndexChanged(object sender, EventArgs e)
        {
            m_tool.Clear();
            m_tool.Finished = false;
            ProfileBeam profile = m_profile as ProfileBeam;
            if (0 == this.DirectionComboBox.SelectedIndex)
            {
                profile.ChangeTransformMatrix(true);
            }
            else if (1 == this.DirectionComboBox.SelectedIndex)
            {
                profile.ChangeTransformMatrix(false);
            }
            this.pictureBox.Refresh();
        }
    }
}