public Result Execute(
ExternalCommandData commandData,
ref string message,
ElementSet elements)
{
UIApplication uiapp = commandData.Application;
UIDocument uidoc = uiapp.ActiveUIDocument;
Application app = uiapp.Application;
Document doc = uidoc.Document;
FilteredElementCollector allSystems
= new FilteredElementCollector(doc)
.OfClass(typeof(MEPSystem));
int nAllSystems = allSystems.Count <Element>();
IEnumerable <MEPSystem> desirableSystems
= allSystems.Cast <MEPSystem>().Where <MEPSystem>(
s => IsDesirableSystemPredicate(s));
int nDesirableSystems = desirableSystems
.Count <Element>();
// Check for shared parameter
// to store graph information.
Definition def = SharedParameterMgr.GetDefinition(
desirableSystems.First <MEPSystem>());
if (null == def)
{
//message = "Please initialise the MEP graph "
// + "storage shared parameter before "
// + "launching this command.";
//return Result.Failed;
SharedParameterMgr.Create(doc);
def = SharedParameterMgr.GetDefinition(
desirableSystems.First <MEPSystem>());
if (null == def)
{
message = "Error creating the "
+ "storage shared parameter.";
return(Result.Failed);
}
}
string outputFolder = GetTemporaryDirectory();
string json;
int nXmlFiles = 0;
int nJsonGraphs = 0;
int nJsonBytes = 0;
// Collect one JSON string per system.
List <string> json_collector = new List <string>();
using (Transaction t = new Transaction(doc))
{
t.Start("Determine MEP Graph Structure and Store in JSON Shared Parameter");
System.Text.StringBuilder[] sbs = new System.Text.StringBuilder[3];
for (int i = 0; i < 3; ++i)
{
sbs[i] = new System.Text.StringBuilder();
sbs[i].Append("[");
}
foreach (MEPSystem system in desirableSystems)
{
Debug.Print(system.Name);
// Debug test -- limit to HWS systems.
//if( !system.Name.StartsWith( "HWS" ) ) { continue; }
FamilyInstance root = system.BaseEquipment;
// Traverse the system and dump the
// traversal graph into an XML file
TraversalTree tree = new TraversalTree(system);
if (tree.Traverse())
{
string filename = system.Id.IntegerValue.ToString();
filename = Path.ChangeExtension(
Path.Combine(outputFolder, filename), "xml");
tree.DumpIntoXML(filename);
// Uncomment to preview the
// resulting XML structure
//Process.Start( fileName );
json = Options.StoreJsonGraphBottomUp
? tree.DumpToJsonBottomUp()
: tree.DumpToJsonTopDown();
Debug.Assert(2 < json.Length,
"expected valid non-empty JSON graph data");
Debug.Print(json);
json_collector.Add(json);
Parameter p = system.get_Parameter(def);
p.Set(json);
nJsonBytes += json.Length;
++nJsonGraphs;
++nXmlFiles;
}
tree.CollectUniqueIds(sbs);
}
for (int i = 0; i < 3; ++i)
{
if (sbs[i][sbs[i].Length - 1] == ',')
{
sbs[i].Remove(sbs[i].Length - 1, 1);
}
sbs[i].Append("]");
}
System.Text.StringBuilder sb = new System.Text.StringBuilder();
sb.Append("{\"id\": 1 , \"name\" : \"MEP Systems\" , \"children\" : [{\"id\": 2 , \"name\": \"Mechanical System\",\"children\":");
sb.Append(sbs[0].ToString());
sb.Append("},{\"id\":3,\"name\":\"Electrical System\", \"children\":");
sb.Append(sbs[1].ToString());
sb.Append("},{\"id\":4,\"name\":\"Piping System\", \"children\":");
sb.Append(sbs[2].ToString());
sb.Append("}]}");
System.IO.StreamWriter file = new System.IO.StreamWriter(Path.ChangeExtension(Path.Combine(outputFolder, @"jsonData"), "json"));
file.WriteLine(sb.ToString());
file.Flush();
file.Close();
t.Commit();
}
string main = string.Format(
"{0} XML files and {1} JSON graphs ({2} bytes) "
+ "generated in {3} ({4} total systems, {5} desirable):",
nXmlFiles, nJsonGraphs, nJsonBytes,
outputFolder, nAllSystems, nDesirableSystems);
List <string> system_list = desirableSystems
.Select <Element, string>(e =>
string.Format("{0}({1})", e.Id, e.Name))
.ToList <string>();
system_list.Sort();
string detail = string.Join(", ",
system_list.ToArray <string>());
TaskDialog dlg = new TaskDialog(
nXmlFiles.ToString() + " Systems");
dlg.MainInstruction = main;
dlg.MainContent = detail;
dlg.Show();
string json_systems = string.Join(",", json_collector);
const string _json_format_to_store_systrms_in_root
= "{{"
+ "\"id\" : {0}, "
+ "\"text\" : \"{1}\", "
+ "\"children\" : [{2}]}}";
json = string.Format(
_json_format_to_store_systrms_in_root,
-1, doc.Title, json_systems);
Debug.Print(json);
return(Result.Succeeded);
}
public Result Execute(
ExternalCommandData commandData,
ref string message,
ElementSet elements)
{
UIApplication uiapp = commandData.Application;
UIDocument uidoc = uiapp.ActiveUIDocument;
Application app = uiapp.Application;
Document doc = uidoc.Document;
FilteredElementCollector allSystems
= new FilteredElementCollector(doc)
.OfClass(typeof(MEPSystem));
int nAllSystems = allSystems.Count <Element>();
IEnumerable <MEPSystem> desirableSystems
= allSystems.Cast <MEPSystem>().Where <MEPSystem>(
s => IsDesirableSystemPredicate(s));
int nDesirableSystems = desirableSystems
.Count <Element>();
// Check for shared parameter
// to store graph information.
// Determine element from which to retrieve
// shared parameter definition.
Element json_storage_element
= Options.StoreSeparateJsonGraphOnEachSystem
? desirableSystems.First <MEPSystem>()
: new FilteredElementCollector(doc)
.OfClass(typeof(ProjectInfo))
.FirstElement();
Definition def = SharedParameterMgr.GetDefinition(
json_storage_element);
if (null == def)
{
SharedParameterMgr.Create(doc);
def = SharedParameterMgr.GetDefinition(
json_storage_element);
if (null == def)
{
message = "Error creating the "
+ "storage shared parameter.";
return(Result.Failed);
}
}
string outputFolder = GetTemporaryDirectory();
int nXmlFiles = 0;
int nJsonGraphs = 0;
int nJsonBytes = 0;
// Collect one JSON string per system.
string json;
// Three separate collections for mechanical,
// electrical and piping systems:
List <string>[] json_collector
= new List <string>[(int)MepDomain.Count] {
new List <string>(),
new List <string>(),
new List <string>()
};
using (Transaction t = new Transaction(doc))
{
t.Start("Determine MEP Graph Structure and Store in JSON Shared Parameter");
StringBuilder[] sbs = new StringBuilder[3];
for (int i = 0; i < 3; ++i)
{
sbs[i] = new StringBuilder();
sbs[i].Append("[");
}
foreach (MEPSystem system in desirableSystems)
{
Debug.Print(system.Name);
FamilyInstance root = system.BaseEquipment;
// Traverse the system and dump the
// traversal graph into an XML file
TraversalTree tree = new TraversalTree(system);
if (tree.Traverse())
{
string filename = system.Id.IntegerValue.ToString();
filename = Path.ChangeExtension(
Path.Combine(outputFolder, filename), "xml");
tree.DumpIntoXML(filename);
// Uncomment to preview the
// resulting XML structure
//Process.Start( fileName );
json = Options.StoreJsonGraphBottomUp
? tree.DumpToJsonBottomUp()
: tree.DumpToJsonTopDown();
Debug.Assert(2 < json.Length,
"expected valid non-empty JSON graph data");
Debug.Print(json);
// Save this system hierarchy JSON in the
// appropriate domain specific collector.
json_collector[(int)GetMepDomain(system)].Add(json);
if (Options.StoreSeparateJsonGraphOnEachSystem)
{
Parameter p = system.get_Parameter(def);
p.Set(json);
}
nJsonBytes += json.Length;
++nJsonGraphs;
++nXmlFiles;
}
tree.CollectUniqueIds(sbs);
}
for (int i = 0; i < 3; ++i)
{
if (sbs[i][sbs[i].Length - 1] == ',')
{
sbs[i].Remove(sbs[i].Length - 1, 1);
}
sbs[i].Append("]");
}
StringBuilder sb = new StringBuilder();
sb.Append("{\"id\": 1 , \"name\" : \"MEP Systems\" , \"children\" : [{\"id\": 2 , \"name\": \"Mechanical System\",\"children\":");
sb.Append(sbs[0].ToString());
sb.Append("},{\"id\":3,\"name\":\"Electrical System\", \"children\":");
sb.Append(sbs[1].ToString());
sb.Append("},{\"id\":4,\"name\":\"Piping System\", \"children\":");
sb.Append(sbs[2].ToString());
sb.Append("}]}");
StreamWriter file = new StreamWriter(
Path.ChangeExtension(
Path.Combine(outputFolder, "jsonData"),
"json"));
file.WriteLine(sb.ToString());
file.Flush();
file.Close();
t.Commit();
}
string msg = string.Format(
"{0} XML files and {1} JSON graphs ({2} bytes) "
+ "generated in {3} ({4} total systems, {5} desirable):",
nXmlFiles, nJsonGraphs, nJsonBytes,
outputFolder, nAllSystems, nDesirableSystems);
List <string> system_list = desirableSystems
.Select <Element, string>(e =>
string.Format("{0}({1})", e.Id, e.Name))
.ToList <string>();
system_list.Sort();
string detail = string.Join(", ",
system_list.ToArray <string>());
TaskDialog dlg = new TaskDialog(
nXmlFiles.ToString() + " Systems");
dlg.MainInstruction = msg;
dlg.MainContent = detail;
dlg.Show();
string[] json_systems = new string[3];
int id = doc.Title.GetHashCode();
for (MepDomain d = MepDomain.Mechanical;
d < MepDomain.Count; ++d)
{
// Compare the systems using the label value,
// which comes after the first comma.
json_collector[(int)d].Sort((s, t)
=> string.Compare(
s.Substring(s.IndexOf(",")),
t.Substring(t.IndexOf(","))));
json_systems[(int)d]
= TreeNode.CreateJsonParentNode(
(++id).ToString(), d.ToString(),
json_collector[(int)d].ToArray <string>());
}
json = TreeNode.CreateJsonParentNode(
doc.Title.GetHashCode().ToString(),
doc.Title, json_systems);
Debug.Print(json);
if (Options.StoreEntireJsonGraphOnProjectInfo)
{
using (Transaction t = new Transaction(doc))
{
t.Start("Store MEP Graph Structure "
+ "in JSON Shared Parameter");
Parameter p = json_storage_element
.get_Parameter(def);
p.Set(json);
t.Commit();
}
}
return(Result.Succeeded);
}