private void button_Go_Click(object sender, RoutedEventArgs e) { if (listBox_schemaList.SelectedItems.Count == 0) { return; } DirectoryInfo dInfo = new DirectoryInfo(textBox_folderLocation.Text); if (dInfo == null) { return; } if (aggregateEntities == null) { aggregateEntities = new SortedSet <string>(); } aggregateEntities.Clear(); if (!Directory.Exists(textBox_outputFolder.Text)) { textBox_outputFolder.Text = ""; return; } logF = new StreamWriter(System.IO.Path.Combine(outputFolder, "entityList.log")); IList <IFCEntityAndPsetList> fxEntityNPsetList = new List <IFCEntityAndPsetList>(); string jsonFile = outputFolder + @"\IFCEntityAndPsetDefs.json"; if (File.Exists(jsonFile)) { File.Delete(jsonFile); } FileStream fs = File.Create(jsonFile); DataContractJsonSerializer ser = new DataContractJsonSerializer(typeof(List <IFCEntityAndPsetList>)); foreach (string fileName in listBox_schemaList.SelectedItems) { FileInfo f = dInfo.GetFiles(fileName).First(); processSchema(f); ProcessPsetDefinition procPdef = new ProcessPsetDefinition(logF); string schemaName = f.Name.Replace(".xsd", ""); IDictionary <string, IfcSchemaEntityNode> entDict = IfcSchemaEntityTree.GetEntityDictFor(f.Name); IFCEntityAndPsetList schemaEntities = new IFCEntityAndPsetList(); schemaEntities.Version = schemaName; schemaEntities.EntityList = new HashSet <IFCEntityInfo>(); schemaEntities.PsetDefList = new HashSet <IFCPropertySetDef>(); Dictionary <ItemsInPsetQtoDefs, string> keywordsToProcess = PsetOrQto.PsetOrQtoDefItems[PsetOrQtoSetEnum.PROPERTYSET]; DirectoryInfo[] psdFolders = new DirectoryInfo(System.IO.Path.Combine(textBox_folderLocation.Text, schemaName)).GetDirectories("psd", SearchOption.AllDirectories); DirectoryInfo[] underpsdFolders = psdFolders[0].GetDirectories(); if (underpsdFolders.Count() > 0) { foreach (DirectoryInfo subDir in psdFolders[0].GetDirectories()) { procPdef.ProcessSchemaPsetDef(schemaName, subDir, keywordsToProcess); } } else { procPdef.ProcessSchemaPsetDef(schemaName, psdFolders[0], keywordsToProcess); } keywordsToProcess = PsetOrQto.PsetOrQtoDefItems[PsetOrQtoSetEnum.QTOSET]; DirectoryInfo[] qtoFolders = new DirectoryInfo(System.IO.Path.Combine(textBox_folderLocation.Text, schemaName)).GetDirectories("qto", SearchOption.AllDirectories); if (qtoFolders.Count() > 0) { DirectoryInfo[] underqtoFolders = qtoFolders[0].GetDirectories(); if (underqtoFolders.Count() > 0) { foreach (DirectoryInfo subDir in qtoFolders[0].GetDirectories()) { procPdef.ProcessSchemaPsetDef(schemaName, subDir, keywordsToProcess); } } else { procPdef.ProcessSchemaPsetDef(schemaName, qtoFolders[0], keywordsToProcess); } } procPdef.ProcessPredefinedPsets(schemaName); //Collect information on applicable Psets for Entity IDictionary <string, HashSet <string> > entPsetDict = new Dictionary <string, HashSet <string> >(); foreach (KeyValuePair <string, IList <VersionSpecificPropertyDef> > pdefEntry in procPdef.allPDefDict) { foreach (VersionSpecificPropertyDef vPdef in pdefEntry.Value) { //if (vPdef.IfcVersion.Equals(schemaName, StringComparison.InvariantCultureIgnoreCase)) { IFCPropertySetDef psetDef = new IFCPropertySetDef(); psetDef.PsetName = vPdef.PropertySetDef.Name; IList <string> props = new List <string>(); foreach (PropertySet.PsetProperty property in vPdef.PropertySetDef.properties) { props.Add(property.Name); } psetDef.Properties = props; schemaEntities.PsetDefList.Add(psetDef); // TODO: to check the appl classes either a type or not and check whether the pair (type or without) exists in entDict, if there is add foreach (string applEntity in vPdef.PropertySetDef.ApplicableClasses) { if (entPsetDict.ContainsKey(applEntity)) { entPsetDict[applEntity].Add(vPdef.PropertySetDef.Name); } else { entPsetDict.Add(applEntity, new HashSet <string>() { vPdef.PropertySetDef.Name }); } // The Pset will be valid for both the Instance and the Type. Check for that here and add if found string entOrTypePair; if (applEntity.Length > 4 && applEntity.EndsWith("Type")) { entOrTypePair = applEntity.Substring(0, applEntity.Length - 4); } else { entOrTypePair = applEntity + "Type"; } if (aggregateEntities.Contains(entOrTypePair)) { if (entPsetDict.ContainsKey(entOrTypePair)) { entPsetDict[entOrTypePair].Add(vPdef.PropertySetDef.Name); } else { entPsetDict.Add(entOrTypePair, new HashSet <string>() { vPdef.PropertySetDef.Name }); } } } } } } // For every entity of the schema, collect the list of PredefinedType (obtained from the xsd), and collect all applicable // Pset Definitions collected above foreach (KeyValuePair <string, IfcSchemaEntityNode> ent in entDict) { IFCEntityInfo entInfo = new IFCEntityInfo(); // The abstract entity type is not going to be listed here as they can never be created if (ent.Value.isAbstract) { continue; } // Collect only the IfcProducts or IfcGroup if (!ent.Value.IsSubTypeOf("IfcProduct") && !ent.Value.IsSubTypeOf("IfcGroup", strict:false) && !ent.Value.IsSubTypeOf("IfcTypeProduct")) { continue; } entInfo.Entity = ent.Key; if (!string.IsNullOrEmpty(ent.Value.PredefinedType)) { if (IfcSchemaEntityTree.PredefinedTypeEnumDict.ContainsKey(ent.Value.PredefinedType)) { entInfo.PredefinedType = IfcSchemaEntityTree.PredefinedTypeEnumDict[ent.Value.PredefinedType]; } } // Get Pset list that is applicable to this entity type if (entPsetDict.ContainsKey(entInfo.Entity)) { entInfo.PropertySets = entPsetDict[entInfo.Entity].ToList(); } // Collect Pset that is applicable to the supertype of this entity IList <IfcSchemaEntityNode> supertypeList = IfcSchemaEntityTree.FindAllSuperTypes(entInfo.Entity, "IfcProduct", "IfcTypeProduct", "IfcGroup"); if (supertypeList != null && supertypeList.Count > 0) { foreach (IfcSchemaEntityNode superType in supertypeList) { if (entPsetDict.ContainsKey(superType.Name)) { if (entInfo.PropertySets == null) { entInfo.PropertySets = new List <string>(); } foreach (string pset in entPsetDict[superType.Name]) { entInfo.PropertySets.Add(pset); } } } } schemaEntities.EntityList.Add(entInfo); } fxEntityNPsetList.Add(schemaEntities); } ser.WriteObject(fs, fxEntityNPsetList); fs.Close(); if (aggregateEntities.Count > 0) { string entityList; entityList = "using System;" + "\r\nusing System.Collections.Generic;" + "\r\nusing System.Linq;" + "\r\nusing System.Text;" + "\r\n" + "\r\nnamespace Revit.IFC.Common.Enums" + "\r\n{" + "\r\n /// <summary>" + "\r\n /// IFC entity types. Combining IFC2x3 and IFC4 (Add2) entities." + "\r\n /// List of Entities for IFC2x is found in IFC2xEntityType.cs" + "\r\n /// List of Entities for IFC4 is found in IFC4EntityType.cs" + "\r\n /// </summary>" + "\r\n public enum IFCEntityType" + "\r\n {"; foreach (string ent in aggregateEntities) { entityList += "\r\n /// <summary>" + "\r\n /// IFC Entity " + ent + " enumeration" + "\r\n /// </summary>" + "\r\n " + ent + ",\n"; } entityList += "\r\n Unknown," + "\r\n DontExport" + "\r\n }" + "\r\n}"; System.IO.File.WriteAllText(outputFolder + @"\IFCEntityType.cs", entityList); } foreach (IFCEntityAndPsetList fxEntityNPset in fxEntityNPsetList) { string entityList; entityList = "using System;" + "\r\nusing System.Collections.Generic;" + "\r\nusing System.Linq;" + "\r\nusing System.Text;" + "\r\n" + "\r\nnamespace Revit.IFC.Common.Enums." + fxEntityNPset.Version + "\r\n{" + "\r\n /// <summary>" + "\r\n /// List of Entities for " + fxEntityNPset.Version + "\r\n /// </summary>" + "\r\n public enum EntityType" + "\r\n {"; foreach (IFCEntityInfo entInfo in fxEntityNPset.EntityList) { entityList += "\r\n /// <summary>" + "\r\n /// IFC Entity " + entInfo.Entity + " enumeration" + "\r\n /// </summary>" + "\r\n " + entInfo.Entity + ",\r\n"; } entityList += "\r\n Unknown," + "\r\n DontExport" + "\r\n }" + "\r\n}"; System.IO.File.WriteAllText(outputFolder + @"\" + fxEntityNPset.Version + "EntityType.cs", entityList); } // Only allows test when only one schema is selected if (listBox_schemaList.SelectedItems.Count == 1) { button_subtypeTest.IsEnabled = true; button_supertypeTest.IsEnabled = true; button_ExportInfoPair.IsEnabled = true; } else { button_subtypeTest.IsEnabled = false; button_supertypeTest.IsEnabled = false; button_ExportInfoPair.IsEnabled = false; } if (logF != null) { logF.Close(); } }
private void button_Go_Click(object sender, RoutedEventArgs e) { string tempFolder = System.IO.Path.GetTempPath(); logF = new StreamWriter(Path.Combine(tempFolder, "GeneratePsetDefWin.log")); textBox_OutputMsg.Clear(); string parFileNameOut = Path.Combine(Path.GetDirectoryName(SharedParFileName), Path.GetFileNameWithoutExtension(SharedParFileName) + "_out.txt"); stSharedPar = File.CreateText(parFileNameOut); IDictionary <string, SharedParameterDef> existingParDict = ProcessPsetDefinition.processExistingParFile(SharedParFileName); IDictionary <string, SharedParameterDef> existingTypeParDict = new Dictionary <string, SharedParameterDef>(); if (File.Exists(SharedParFileNameType)) { string typeParFileNameOut = Path.Combine(Path.GetDirectoryName(SharedParFileNameType), Path.GetFileNameWithoutExtension(SharedParFileNameType) + "_out.txt"); stSharedParType = File.CreateText(typeParFileNameOut); existingTypeParDict = ProcessPsetDefinition.processExistingParFile(SharedParFileNameType); } else { stSharedParType = File.CreateText(SharedParFileNameType); } ProcessPsetDefinition procPsetDef = new ProcessPsetDefinition(logF); if (string.IsNullOrEmpty(textBox_PSDSourceDir.Text) || string.IsNullOrEmpty(textBox_OutputFile.Text)) { return; } var psdFolders = new DirectoryInfo(textBox_PSDSourceDir.Text).GetDirectories("psd", SearchOption.AllDirectories); var qtoFolders = new DirectoryInfo(textBox_PSDSourceDir.Text).GetDirectories("qto", SearchOption.AllDirectories); string dirName = Path.GetDirectoryName(textBox_OutputFile.Text); string outFileName = Path.GetFileNameWithoutExtension(textBox_OutputFile.Text); string penumFileName = Path.Combine(dirName, outFileName); // Collect all Pset definition for psd folders Dictionary <ItemsInPsetQtoDefs, string> keywordsToProcess = PsetOrQto.PsetOrQtoDefItems[PsetOrQtoSetEnum.PROPERTYSET]; HashSet <string> IfcSchemaProcessed = new HashSet <string>(); foreach (DirectoryInfo psd in psdFolders) { string schemaFolder = psd.FullName.Remove(0, textBox_PSDSourceDir.Text.Length + 1).Split('\\')[0]; logF.WriteLine("\r\n*** Processing " + schemaFolder); foreach (DirectoryInfo subDir in psd.GetDirectories()) { procPsetDef.ProcessSchemaPsetDef(schemaFolder, subDir, keywordsToProcess); } procPsetDef.ProcessSchemaPsetDef(schemaFolder, psd, keywordsToProcess); IfcSchemaProcessed.Add(schemaFolder); } // Collect all QtoSet definition for qto folders keywordsToProcess = PsetOrQto.PsetOrQtoDefItems[PsetOrQtoSetEnum.QTOSET]; foreach (DirectoryInfo qto in qtoFolders) { string schemaFolder = qto.FullName.Remove(0, textBox_PSDSourceDir.Text.Length + 1).Split('\\')[0]; logF.WriteLine("\r\n*** Processing " + schemaFolder); foreach (DirectoryInfo subDir in qto.GetDirectories()) { procPsetDef.ProcessSchemaPsetDef(schemaFolder, subDir, keywordsToProcess); } procPsetDef.ProcessSchemaPsetDef(schemaFolder, qto, keywordsToProcess); } // Process predefined properties foreach (string schemaName in IfcSchemaProcessed) { logF.WriteLine("\r\n*** Processing " + schemaName); procPsetDef.ProcessPredefinedPsets(schemaName); } // For testing purpose: Dump all the propertyset definition in a text file if (checkBox_Dump.IsChecked.HasValue && checkBox_Dump.IsChecked.Value) { string pSetDump = ""; foreach (KeyValuePair <string, IList <VersionSpecificPropertyDef> > psetDefEntry in procPsetDef.allPDefDict) { pSetDump += "**** Property Set Name: " + psetDefEntry.Key; foreach (VersionSpecificPropertyDef vPdef in psetDefEntry.Value) { pSetDump += "\r\n ===> IfcVersion: " + vPdef.IfcVersion; pSetDump += "\r\n" + vPdef.PropertySetDef.ToString() + "\r\n"; } pSetDump += "\r\n\n"; } string dumpDir = Path.GetDirectoryName(textBox_OutputFile.Text); string dumpFile = Path.GetFileNameWithoutExtension(textBox_OutputFile.Text) + ".txt"; string dumpFilePath = Path.Combine(dumpDir, dumpFile); if (File.Exists(dumpFilePath)) { File.Delete(dumpFilePath); } StreamWriter tx = new StreamWriter(dumpFilePath); tx.Write(pSetDump); tx.Close(); } IDictionary <string, int> groupParamDict = new Dictionary <string, int>(); string[] outFNameParts = outFileName.Split('_'); // Do it for the predefined propserty sets string fNameToProcess = Path.Combine(dirName, outFNameParts[0] + "_PredefPset.cs"); if (File.Exists(fNameToProcess)) { File.Delete(fNameToProcess); } StreamWriter outF = new StreamWriter(fNameToProcess); // Group ID 1 and 2 are reserved int offset = 3; offset = WriteGeneratedCode(outF, procPsetDef, penumFileName, "Ifc", groupParamDict, offset); // Do it for the predefined propserty sets fNameToProcess = Path.Combine(dirName, outFNameParts[0] + "_PsetDef.cs"); if (File.Exists(fNameToProcess)) { File.Delete(fNameToProcess); } outF = new StreamWriter(fNameToProcess); offset = WriteGeneratedCode(outF, procPsetDef, penumFileName, "Pset", groupParamDict, offset); // Do it for the predefined propserty sets fNameToProcess = Path.Combine(dirName, outFNameParts[0] + "_QsetDef.cs"); if (File.Exists(fNameToProcess)) { File.Delete(fNameToProcess); } outF = new StreamWriter(fNameToProcess); offset = WriteGeneratedCode(outF, procPsetDef, penumFileName, "Qto", groupParamDict, offset); // Close the Enum files procPsetDef.endWriteEnumFile(); WriteRevitSharedParam(stSharedPar, existingParDict, groupParamDict, false, out IList <string> deferredParList); AppendDeferredParamList(stSharedPar, deferredParList); stSharedPar.Close(); WriteRevitSharedParam(stSharedParType, existingTypeParDict, groupParamDict, true, out IList <string> deferredParTypeList); AppendDeferredParamList(stSharedParType, deferredParTypeList); stSharedParType.Close(); logF.Close(); }