private static Result OnFindCommand(UIDocument ui_doc,
FS_FEATURE_TYPE feature_type, string command_name, string shape_name,
Action <Document, FS_FEATURE_RESULT?> post_action)
{
uint index;
XYZ ray_origin = XYZ.Zero, ray_direction = XYZ.Zero;
try
{
GenerateRay(ui_doc, out ray_origin, out ray_direction);
}
catch (Autodesk.Revit.Exceptions.OperationCanceledException) { return(Result.Cancelled); }
// cast ray to point cloud and get an index of the point hit.
if (OutlierPointCloudEngine.Cast(ray_origin, ray_direction, 0.01f, out index) == false)
{
TaskDialog.Show(command_name, "There is no points picked.");
return(Result.Failed);
}
// run FindSurface to find shape
FS_FEATURE_RESULT?result = FindSurface.RunFindSurface(feature_type, index);
if (result == null)
{
TaskDialog.Show(command_name, $"FindSurface failed to find a {shape_name}.");
return(Result.Failed);
}
post_action(ui_doc.Document, result);
return(Result.Succeeded);
}
public static Result OnFindThinRingTorusCommand(UIDocument ui_doc)
{
uint index0, index1, index2;
XYZ ray0_origin = XYZ.Zero, ray0_direction = XYZ.Zero;
XYZ ray1_origin = XYZ.Zero, ray1_direction = XYZ.Zero;
XYZ ray2_origin = XYZ.Zero, ray2_direction = XYZ.Zero;
try
{
GenerateRay(ui_doc, out ray0_origin, out ray0_direction);
GenerateRay(ui_doc, out ray1_origin, out ray1_direction);
GenerateRay(ui_doc, out ray2_origin, out ray2_direction);
}
catch (Autodesk.Revit.Exceptions.OperationCanceledException) { return(Result.Cancelled); }
// cast ray to point cloud and get an index of the point hit.
if (OutlierPointCloudEngine.Cast(ray0_origin, ray0_direction, 0.01f, out index0) == false)
{
TaskDialog.Show("FindThinRingTorus", "There is no points picked." + Environment.NewLine + "Command cancelled.");
return(Result.Failed);
}
if (OutlierPointCloudEngine.Cast(ray1_origin, ray1_direction, 0.01f, out index1) == false)
{
TaskDialog.Show("FindThinRingTorus", "There is no points picked." + Environment.NewLine + "Command cancelled.");
return(Result.Failed);
}
if (OutlierPointCloudEngine.Cast(ray2_origin, ray2_direction, 0.01f, out index2) == false)
{
TaskDialog.Show("FindThinRingTorus", "There is no points picked." + Environment.NewLine + "Command cancelled.");
return(Result.Failed);
}
// run FindSurface to find shape
FS_FEATURE_RESULT?result = FindSurface.RunFindThinRingTorus(index0, index1, index2);
if (result == null)
{
TaskDialog.Show("FindThinRingTorus", $"FindSurface failed to find a thin ring torus.");
return(Result.Failed);
}
switch (result.Value.type)
{
case FS_FEATURE_TYPE.FS_TYPE_CYLINDER: DrawCylinderAndInliers(ui_doc.Document, result); break;
case FS_FEATURE_TYPE.FS_TYPE_SPHERE: DrawSphereAndInliers(ui_doc.Document, result); break;
case FS_FEATURE_TYPE.FS_TYPE_TORUS: DrawTorusAndInliers(ui_doc.Document, result); break;
}
return(Result.Succeeded);
}
/// <summary>
/// Cleans up FindSurface, custom point cloud engines, DirectShapeManager instances before Revit is terminated.
/// </summary>
public static void OnPluginShutdown()
{
// clean up FindSurface
FindSurface.CleanUp();
// clean up XYZPointCloudEngine
List <string> engine_names = PointCloudEngineRegistry.GetSupportedEngines() as List <string>;
foreach (string engine_name in engine_names)
{
if (engine_name == S_OUTLIER_PC_ENGINE_IDENTIFIER || engine_name == S_INLIER_PC_ENGINE_IDENTIFIER)
{
PointCloudEngineRegistry.UnregisterPointCloudEngine(engine_name);
}
}
}
/// <summary>
/// Resets the plugin to the state of when users imported a point cloud file.
/// </summary>
/// <returns>false if any point cloud file has been opened or true otherwise</returns>
public static bool OnResetCommand(Document document)
{
OnCleanUpCommand();
CloudPoint[] points;
if (s_xyz == null || s_xyz.Length == 0)
{
return(false);
}
FindSurfaceRevitPluginUtils.Pack(s_xyz, s_color, out points);
OutlierPointCloudEngine.CreatePointCloud(document, s_file_full_name, points, Transform.Identity, s_subdivision);
FindSurface.CleanUp();
FindSurface.SetPointCloud(s_xyz);
return(true);
}
/// <summary>
/// Creates a point cloud and send the data to FindSurface when users imported a point cloud file using Open button in FindSurface tab.
/// </summary>
/// <param name="doc">The active document that Revit uses</param>
/// <param name="file_full_name">The name of imported file</param>
/// <param name="xyz_array">Imported xyz data</param>
/// <param name="color_array">Imported color data</param>
/// <param name="subdivision">Subdivision number</param>
/// <param name="unit">The base unit of when the inspector displays information</param>
public static void OnOpenCommand(Document doc, string file_full_name, float[] xyz_array, int[] color_array, int subdivision, Units unit)
{
s_file_full_name = file_full_name;
s_xyz = xyz_array;
s_color = color_array;
s_subdivision = subdivision;
s_measurementUnit = unit;
CloudPoint[] points;
FindSurfaceRevitPluginUtils.Pack(xyz_array, color_array, out points);
s_temporary_outlier_deletion = true;
OutlierPointCloudEngine.CreatePointCloud(doc, file_full_name, points, Transform.Identity, subdivision);
FindSurface.CleanUp();
FindSurface.SetPointCloud(xyz_array);
}
public static Result OnFindDiskCylinderCommand(UIDocument ui_doc)
{
uint index0, index1, index2;
XYZ ray0_origin = XYZ.Zero, ray0_direction = XYZ.Zero;
XYZ ray1_origin = XYZ.Zero, ray1_direction = XYZ.Zero;
XYZ ray2_origin = XYZ.Zero, ray2_direction = XYZ.Zero;
try
{
GenerateRay(ui_doc, out ray0_origin, out ray0_direction);
GenerateRay(ui_doc, out ray1_origin, out ray1_direction);
GenerateRay(ui_doc, out ray2_origin, out ray2_direction);
}
catch (Autodesk.Revit.Exceptions.OperationCanceledException) { return(Result.Cancelled); }
// cast ray to point cloud and get an index of the point hit.
if (OutlierPointCloudEngine.Cast(ray0_origin, ray0_direction, 0.01f, out index0) == false)
{
TaskDialog.Show("FindDiskCylinder", "There is no points picked." + Environment.NewLine + "Command cancelled.");
return(Result.Failed);
}
if (OutlierPointCloudEngine.Cast(ray1_origin, ray1_direction, 0.01f, out index1) == false)
{
TaskDialog.Show("FindDiskCylinder", "There is no points picked." + Environment.NewLine + "Command cancelled.");
return(Result.Failed);
}
if (OutlierPointCloudEngine.Cast(ray2_origin, ray2_direction, 0.01f, out index2) == false)
{
TaskDialog.Show("FindDiskCylinder", "There is no points picked." + Environment.NewLine + "Command cancelled.");
return(Result.Failed);
}
// run FindSurface to find shape
FS_FEATURE_RESULT?result = FindSurface.RunFindDiskCylinder(index0, index1, index2);
if (result == null)
{
TaskDialog.Show("FindDiskCylinder", $"FindSurface failed to find a disk cylinder.");
return(Result.Failed);
}
DrawCylinderAndInliers(ui_doc.Document, result);
return(Result.Succeeded);
}
private static void ExtractInlierPointCloud(Document doc, string inlier_identifier, int highlight_color)
{
CloudPoint[] original_points = OutlierPointCloudEngine.GetCloudPoint();
List <CloudPoint> inlier_point_list = new List <CloudPoint>();
List <CloudPoint> outlier_point_list = new List <CloudPoint>();
List <float> outlier_xyz = new List <float>();
// this is a flag array of which each element is set when the point of its corresponding location is in outlier.
bool[] outlier_flags = FindSurface.GetInOutlierFlags();
for (int k = 0; k < outlier_flags.Length; k++)
{
float x = original_points[k].X;
float y = original_points[k].Y;
float z = original_points[k].Z;
int color = original_points[k].Color;
if (outlier_flags[k]) // outlier
{
outlier_point_list.Add(new CloudPoint(x, y, z, color));
outlier_xyz.AddRange(new float[] { x, y, z });
}
else // inlier
{
color = FindSurfaceRevitPluginUtils.MixARGB(color, highlight_color);
inlier_point_list.Add(new CloudPoint(x, y, z, color));
}
}
Transform outlier_transform = OutlierPointCloudEngine.GetPointCloudTransform();
// replace outlier (or original one) to new outlier points.
s_temporary_outlier_deletion = true;
OutlierPointCloudEngine.CreatePointCloud(doc, "Outlier", outlier_point_list.ToArray(), outlier_transform, s_subdivision);
// give new outlier points to FindSurface.
FindSurface.CleanUp();
FindSurface.SetPointCloud(outlier_xyz.ToArray());
// create inlier point cloud.
InlierPointCloudEngine.CreatePointCloud(doc, inlier_identifier, inlier_point_list.ToArray(), outlier_transform);
}
/// <summary>
/// Initializes FindSurface, custom point cloud engines, DirectShapeManager instances when Revit loaded this plugin.
/// </summary>
public static bool OnPluginStartUp()
{
FS_CONTEXT_CREATION_ERROR error = FS_CONTEXT_CREATION_ERROR.FS_NO_ERROR;
// initialize FindSurface
s_find_surface = FindSurface.GetInstance(ref error);
switch (error)
{
case FS_CONTEXT_CREATION_ERROR.FS_OUT_OF_MEMORY:
TaskDialog.Show("FindSurface", "Context creation failed: FS_OUT_OF_MEMORY");
return(false);
case FS_CONTEXT_CREATION_ERROR.FS_LICENSE_EXPIRED:
TaskDialog.Show("FindSurface", "Context creation failed: FS_LICENSE_EXPIRED");
return(false);
case FS_CONTEXT_CREATION_ERROR.FS_LICENSE_UNKNOWN:
TaskDialog.Show("FindSurface", "Context creation failed: FS_LICENSE_UNKNOWN");
return(false);
}
// initialize parameters temporarily
FindSurface.Accuracy = 0.003f; // Set Sensor Measurement Accuracy
FindSurface.MeanDistance = 0.01f; // Set Mean Distance of Neighboring Points
FindSurface.TouchRadius = 0.025f; // Touch Size of the Seed Region
// initialize Custom Point Cloud Engines
s_outlier_point_cloud_engine = new UniformGridPointCloudEngine(S_OUTLIER_PC_ENGINE_IDENTIFIER);
PointCloudEngineRegistry.RegisterPointCloudEngine(S_OUTLIER_PC_ENGINE_IDENTIFIER, s_outlier_point_cloud_engine, false);
s_inlier_point_cloud_engine = new OctreePointCloudEngine(S_INLIER_PC_ENGINE_IDENTIFIER);
PointCloudEngineRegistry.RegisterPointCloudEngine(S_INLIER_PC_ENGINE_IDENTIFIER, s_inlier_point_cloud_engine, false);
// initialize DirectShape Engines
s_direct_shape_manager = new DirectShapeEngine();
return(true);
}
