UnrealEngine/Engine/Source/Programs/Enterprise/Datasmith/DatasmithRevitExporter/Private/DatasmithRevitSettingsManager.cs

// Copyright Epic Games, Inc. All Rights Reserved.

using System;
using System.Collections.Generic;
using Autodesk.Revit.DB;
using Autodesk.Revit.DB.ExtensibleStorage;

namespace DatasmithRevitExporter
{
	public class FSettings
	{
		public enum EInsertionPoint
		{
			Default,
			BasePoint,
			SurveyPoint
		};

		private IList<string> _MetadataParamNamesFilter = new List<string>();
		private IList<int> _MetadataParamGroupsFilter = new List<int>();

		// Derived data for fast lookup
		private HashSet<string> MetadataParamNamesSet;
		private HashSet<int> MetadataParamGroupsSet;

		public int LevelOfTesselation { get; set; } = 8;
		public EInsertionPoint InsertionPoint { get; set; } = EInsertionPoint.Default;
		public ElementId SyncViewId { get; set; } = ElementId.InvalidElementId;

		public IList<string> MetadataParamNamesFilter 
		{
			get => _MetadataParamNamesFilter;
			set
			{
				_MetadataParamNamesFilter = value;
				MetadataParamNamesSet = new HashSet<string>(_MetadataParamNamesFilter);
			}
		}

		public IList<int> MetadataParamGroupsFilter 
		{
			get => _MetadataParamGroupsFilter;
			set
			{
				_MetadataParamGroupsFilter = value;
				MetadataParamGroupsSet = new HashSet<int>(_MetadataParamGroupsFilter);
			}
		}

		// Returns true if the parameter passes the filter, false otherwise.
		public bool MatchParameterByMetadata(Parameter InParam)
		{
			if (MetadataParamNamesSet != null && MetadataParamNamesSet.Count > 0)
			{
				bool bNameMatch = false;
				foreach (string NameFilter in MetadataParamNamesSet)
				{
					// Check if NameFilter is contained in the current param name
					if (InParam.Definition.Name.IndexOf(NameFilter, StringComparison.OrdinalIgnoreCase) >= 0)
					{
						bNameMatch = true;
						break;
					}
				}

				if (!bNameMatch)
				{
					return false;
				}
			}

			// Name filter passes (or empty), so check group filter.
			bool bGroupMatch = true;
			if (MetadataParamGroupsSet != null && MetadataParamGroupsSet.Count > 0)
			{
				bGroupMatch = MetadataParamGroupsSet.Contains((int)InParam.Definition.ParameterGroup);
			}
		
			return bGroupMatch;
		}
	}

	public class FSettingsManager
	{
		static class FSettingsSchema
		{
			readonly static Guid SchemaGuid = new Guid("{67D3D706-4881-4BC3-9AA1-75768DE40415}");
			
			public readonly static string MetadataParamNamesField = "MetadataParamNames";
			public readonly static string MetadataParamGroupsField = "MetadataParamGroups";
			public readonly static string LevelOfTesselationField = "LevelOfTesselation";
			public readonly static string InsertionPointField = "InsertionPoint";
			public readonly static string SyncViewField = "SyncView";

			public static Schema GetSchema()
			{
				Schema Schema = Schema.Lookup(SchemaGuid);
				if (Schema != null)
				{
					return Schema;
				}

				SchemaBuilder SchemaBuilder = new SchemaBuilder(SchemaGuid);
				SchemaBuilder.SetSchemaName("DatasmithRevitExportSettings");

				SchemaBuilder.AddArrayField(MetadataParamNamesField, typeof(string));
				SchemaBuilder.AddArrayField(MetadataParamGroupsField, typeof(int));
				SchemaBuilder.AddSimpleField(LevelOfTesselationField, typeof(int));
				SchemaBuilder.AddSimpleField(InsertionPointField, typeof(int));
				SchemaBuilder.AddSimpleField(SyncViewField, typeof(int));

				return SchemaBuilder.Finish();
			}
		}

		static class DataStorageUniqueIdSchema
		{
			static readonly Guid SchemaGuid = new Guid("{B03CF2DB-3617-4C67-90D4-A34C734A54E4}");

			public static Schema GetSchema()
			{
				Schema Schema = Schema.Lookup(SchemaGuid);
				if (Schema != null)
				{
					return Schema;
				}

				SchemaBuilder SchemaBuilder = new SchemaBuilder(SchemaGuid);
				SchemaBuilder.SetSchemaName("DataStorageUniqueId");
				SchemaBuilder.AddSimpleField("Id", typeof(Guid));

				return SchemaBuilder.Finish();
			}
		}

		static readonly Guid SettingDataStorageId = new Guid("{5933427B-3E44-44AC-85CF-74C396662E05}");

		public static event EventHandler SettingsUpdated;

		public static FSettings ReadSettings(Document Doc)
		{
			FSettings Settings = new FSettings();

			void InitDefaultSettings()
			{
				// No settings in document, create defaults
				Settings.LevelOfTesselation = 8;
				Settings.InsertionPoint = 0;
				Settings.SyncViewId = ElementId.InvalidElementId;
				Settings.MetadataParamGroupsFilter.Add((int)Autodesk.Revit.DB.BuiltInParameterGroup.PG_GEOMETRY);
				Settings.MetadataParamGroupsFilter.Add((int)Autodesk.Revit.DB.BuiltInParameterGroup.PG_IDENTITY_DATA);
				Settings.MetadataParamGroupsFilter.Add((int)Autodesk.Revit.DB.BuiltInParameterGroup.PG_MATERIALS);
				Settings.MetadataParamGroupsFilter.Add((int)Autodesk.Revit.DB.BuiltInParameterGroup.PG_PHASING);
				Settings.MetadataParamGroupsFilter.Add((int)Autodesk.Revit.DB.BuiltInParameterGroup.PG_STRUCTURAL);

				WriteSettings(Doc, Settings);
			}

			Entity SettingsEntity = GetSettingsEntity(Doc);

			if (SettingsEntity == null || !SettingsEntity.IsValid())
			{
				InitDefaultSettings();
			}
			else
			{
				try
				{
					Settings.MetadataParamNamesFilter = SettingsEntity.Get<IList<string>>(FSettingsSchema.MetadataParamNamesField);
					Settings.MetadataParamGroupsFilter = SettingsEntity.Get<IList<int>>(FSettingsSchema.MetadataParamGroupsField);
					Settings.LevelOfTesselation = SettingsEntity.Get<int>(FSettingsSchema.LevelOfTesselationField);
					Settings.InsertionPoint = (FSettings.EInsertionPoint)SettingsEntity.Get<int>(FSettingsSchema.InsertionPointField);
					Settings.SyncViewId = new ElementId(SettingsEntity.Get<int>(FSettingsSchema.SyncViewField));
				}
				catch
				{
					InitDefaultSettings();
				}
			}

			return Settings;
		}

		public static void WriteSettings(Document Doc, FSettings Settings)
		{
			using (Transaction T = new Transaction(Doc, "Write settings"))
			{
				try
				{
					T.Start();

					DataStorage SettingDataStorage = GetSettingsDataStorage(Doc);
					if (SettingDataStorage == null)
					{
						SettingDataStorage = DataStorage.Create(Doc);
					}

					Entity SettingsEntity = new Entity(FSettingsSchema.GetSchema());
					SettingsEntity.Set(FSettingsSchema.MetadataParamNamesField, Settings.MetadataParamNamesFilter);
					SettingsEntity.Set(FSettingsSchema.MetadataParamGroupsField, Settings.MetadataParamGroupsFilter);
					SettingsEntity.Set(FSettingsSchema.LevelOfTesselationField, Settings.LevelOfTesselation);
					SettingsEntity.Set(FSettingsSchema.InsertionPointField, (int)Settings.InsertionPoint);
					SettingsEntity.Set(FSettingsSchema.SyncViewField, Settings.SyncViewId.IntegerValue);

					// Identify settings data storage
					Entity IdEntity = new Entity(DataStorageUniqueIdSchema.GetSchema());
					IdEntity.Set("Id", SettingDataStorageId);

					SettingDataStorage.SetEntity(IdEntity);
					SettingDataStorage.SetEntity(SettingsEntity);

					T.Commit();

					SettingsUpdated?.Invoke(Settings, null);
				}
				catch { }
			}
		}

		private static DataStorage GetSettingsDataStorage(Document Doc)
		{
			// Retrieve all data storages from project
			FilteredElementCollector Collector = new FilteredElementCollector(Doc);
			var DataStorages = Collector.OfClass(typeof(DataStorage));

			// Find setting data storage
			foreach (DataStorage DataStorage in DataStorages)
			{
				Entity SettingIdEntity = DataStorage.GetEntity(DataStorageUniqueIdSchema.GetSchema());

				if (!SettingIdEntity.IsValid())
				{
					continue;
				}

				Guid Id = SettingIdEntity.Get<Guid>("Id");

				if (!Id.Equals(SettingDataStorageId))
				{
					continue;
				}

				return DataStorage;
			}
			return null;
		}

		private static Entity GetSettingsEntity(Document Doc)
		{
			FilteredElementCollector Collector = new FilteredElementCollector(Doc);
			var DataStorages = Collector.OfClass(typeof(DataStorage));

			// Find setting data storage
			foreach (DataStorage DataStorage in DataStorages)
			{
				Entity SettingEntity = DataStorage.GetEntity(FSettingsSchema.GetSchema());

				// If a DataStorage contains setting entity, we found it
				if (!SettingEntity.IsValid())
				{
					continue;
				}

				return SettingEntity;
			}

			return null;
		}
	}
}