Full conversion

bin/full_conversion.py

@@ -0,0 +1,49 @@
+
+import argparse
+import ROOT
+
+def main():
+    parser = argparse.ArgumentParser(description='Convert detector')
+    parser.add_argument('-c', '--compact', help='Compact file location(s)',
+                        required=True, type=str, nargs='+')
+    parser.add_argument('-o', '--out', help='Converted file path',
+                        default='nothing', type=str)
+    parser.add_argument('-v', '--visibility', help='Level of layers in detector',
+                        default=9, type=int)
+
+    args = parser.parse_args()
+
+    convert(args.compact, args.out, args.visibility)
+
+
+def convert(compact_files, out_path, visibility):
+    print('INFO: Converting following compact file(s):')
+    for cfile in compact_files:
+        print('      ' + cfile)
+
+    ROOT.gSystem.Load('libDDCore')
+    description = ROOT.dd4hep.Detector.getInstance()
+    for cfile in compact_files:
+        description.fromXML(cfile)
+
+    ROOT.gGeoManager.SetVisLevel(visibility)
+    ROOT.gGeoManager.SetVisOption(0)
+
+    if out_path == 'nothing':
+
+        counter = 0
+        slash_list = []
+        for cfile in compact_files:
+            for i in cfile:
+                counter += 1
+                if i == '/':
+                    slash_list.append(counter)
+
+            last_slash = max(slash_list)
+            ROOT.gGeoManager.Export(f'{cfile[last_slash:len(cfile)-4]}.root')
+
+    else:
+        ROOT.gGeoManager.Export(out_path)
+
+if __name__ == '__main__':
+    main()

bin/index.js

@@ -57,6 +57,14 @@ if (options.outputFile) {
   outFileName = options.outputFile;
 }
 
+for (var [name, entry] of Object.entries(config.subParts)) {
+  let temp = [];
+  for (let path of entry[0]) {
+    temp.push(new RegExp(path));
+  }
+  entry[0] = temp;
+}
+
 convertGeometry(inFilePath,
                 outFileName,
                 config.maxLevel,

calling_root2gltf.py

@@ -0,0 +1,14 @@
+import argparse
+import subprocess
+
+parser = argparse.ArgumentParser(description='Convert detector (to .gltf)')
+parser.add_argument('root_file', help='Root file',
+                      type=str)
+parser.add_argument('-c', '--config_file', help='Json file of detector structure',
+                        default='a', type=str)
+parser.add_argument('-g', '--gltf_file', help='Gltf file',
+                        default='detector.gltf', type=str)
+
+args = parser.parse_args()
+
+subprocess.run(["node", ".", "-c", f'{args.config_file}', "-o", f'{args.gltf_file}', f'{args.root_file}'])

config_automatic.json

@@ -0,0 +1 @@
+{"childrenToHide": [], "subParts": {"BeamCal": [["BeamCal_(?!envelope)\\w+"], 0.8], "Coil": [["Coil_(?!envelope)\\w+"], 0.8], "EcalBarrel": [["EcalBarrel_(?!envelope)\\w+"], 0.8], "EcalEndcap": [["EcalEndcap_(?!envelope)\\w+"], 0.8], "EcalPlug": [["EcalPlug_(?!envelope)\\w+"], 0.8], "FTD": [["FTD_(?!envelope)\\w+", "FTDDisk_.*_negZ", "ftd_petal_negZ_.*_.*", "ftd_sensor_negZ_.*_.*_.*", "FTDDisk_.*_posZ", "ftd_petal_posZ_.*_.*", "ftd_sensor_posZ_.*_.*_.*", "FTD_support"], 0.8], "HcalBarrel": [["HcalBarrel_(?!envelope)\\w+"], 0.8], "HcalEndcap": [["HcalEndcap_(?!envelope)\\w+"], 0.8], "HcalRing": [["HcalRing_(?!envelope)\\w+"], 0.8], "LHCal": [["LHCal_(?!envelope)\\w+"], 0.8], "Lcal": [["Lcal_(?!envelope)\\w+"], 0.8], "QD0_cryostat": [["QD0_cryostat_(?!envelope)\\w+"], 0.8], "QD0_support": [["QD0_support_(?!envelope)\\w+"], 0.8], "SET": [["SET_(?!envelope)\\w+", "set_ladder_.*_.*", "set_ladder_.*_.*_.*"], 0.8], "SIT": [["SIT_(?!envelope)\\w+", "sit_ladder_.*_.*", "sit_ladder_.*_.*_.*"], 0.8], "SServices00": [["SServices00_(?!envelope)\\w+"], 0.8], "TPC": [["TPC_(?!envelope)\\w+", "tpc_endcap_bwd", "tpc_endcap_fwd", "tpc_sensGas_bwd", "tpc_row_bwd_.*", "tpc_sensGas_fwd", "tpc_row_fwd_.*"], 0.8], "Tube": [["Tube_(?!envelope)\\w+"], 0.8], "VXD": [["VXD_(?!envelope)\\w+", "BerylliumAnnulusBlock_.*_negZ_.*", "BerylliumAnnulusBlock_.*_posZ_.*", "ElectronicsEnd_.*_negZ_.*", "ElectronicsEnd_.*_posZ_.*", "SiActiveLayer_.*_negZ_.*", "SiActiveLayer_.*_posZ_.*", "VXD_support", "KaptonLines_.*_negZ", "KaptonLines_.*_posZ", "MetalLines_.*_negZ", "MetalLines_.*_posZ", "VXD_ExtKaptonCab_neg", "VXD_ExtKaptonCab_pos", "VXD_ExtMetalCab_neg", "VXD_ExtMetalCab_pos", "VXD_endplate_bwd", "VXD_endplate_fwd", "VXD_support_bwd_inner", "VXD_support_bwd_outer", "VXD_support_fwd_inner", "VXD_support_fwd_outer", "negShellCone", "posShellCone"], 0.8], "YokeBarrel": [["YokeBarrel_(?!envelope)\\w+"], 0.8], "YokeEndcap": [["YokeEndcap_(?!envelope)\\w+"], 0.8]}, "maxLevel": 3}

configfile_generator.py

@@ -0,0 +1,73 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+from dd4hep import Detector
+import re
+from argparse import ArgumentParser
+import pprint
+import json
+
+parser = ArgumentParser()
+
+parser.add_argument(
+    "--compactFile", help="DD4hep compact description xml", required=True
+)
+parser.add_argument(
+    "--maxDepth",
+    help="Maximum traversal depth of the detector tree",
+    default=10,
+    type=int,
+)
+
+args = parser.parse_args()
+
+theDetector = Detector.getInstance()
+theDetector.fromXML(args.compactFile)
+start = theDetector.world()
+
+def process_name(raw_name):
+    name = re.sub(r"\d+", ".*", raw_name)
+    return name
+
+def tree(detElement, depth, maxDepth):
+    nd = {}
+    depth += 1
+    children = detElement.children()
+    for raw_name, child in children:
+        if depth > maxDepth:
+            tree(child, depth, maxDepth)
+        else:
+            dictionary = tree(child, depth, maxDepth)
+            nd.update({raw_name: dictionary})
+    return nd
+
+detector_dict = tree(start, 0, args.maxDepth)
+#pprint.pprint(detector_dict)
+
+def post_processing(obj, main_parts, subParts={}, sublist= []):
+    for k, v in obj.items():
+        if k in main_parts:
+            sublist = [f'{k}_(?!envelope)\\w+']
+            outer_list = []
+            outer_list.append(sublist)
+            outer_list.append(0.8)
+            subParts.update({k: outer_list})
+            post_processing(v, main_parts, subParts, sublist)
+
+        else:
+            k_new = process_name(k)
+            x = re.search("module|stave|layer|Calorimeter", k_new)
+            if k_new not in sublist and x == None:
+                sublist.append(k_new)
+            post_processing(v, main_parts, subParts, sublist)
+    return subParts
+
+subPart_processed = post_processing(detector_dict, list(detector_dict.keys()))
+
+final_dict = {"childrenToHide": [],
+              "subParts": subPart_processed,
+              "maxLevel": 3}
+
+pprint.pprint(final_dict)
+with open("config_automatic.json", "w") as outfile: 
+    json.dump(final_dict, outfile)

configs/config_ild_trial2.json

@@ -0,0 +1,28 @@
+{
+  "childrenToHide": [
+    "SServices",
+    "QDX_cryo",
+    "QDX_support"
+  ],
+  "subParts": {
+    "EcalBarrel" : [["EcalBarrel_module\\w+"], 0.8],
+    "Coil" : [["coil_vol_0"], 0.8],
+    "Lcal" : [["Lcal_module_0", "Lcal_module_1"], 0.8],
+    "EcalEndcap" : [["EcalEndcapQuadrant_."], 0.8],
+    "EcalPlug" : [["ECRing_0", "ECRing_1"], 0.8],
+    "BeamCal" : [["Beam_Cal_module_0", "Beam_Cal_module_1"], 0.8],
+    "FTD" : [["FTDAirDiskLogicalPZ_._.", "FTDAirDiskLogicalNZ_._."], 0.8],
+    "HcalEndcap" : [["HcalEndcap_EndcapModule._.", "Hcal_endcap_FEE_."], 0.8],
+    "HcalRing" : [["HcalEndCapRingLogical_0", "HcalEndCapRingLogical_1"], 0.8],
+    "LHCal" : [["LHCal_module_0", "LHCal_module_1"], 0.8],
+    "HcalBarrel": [["HcalBarrel_module_.", "HcalBarrel_Module_lateral_plate_."], 0.8],
+    "SIT": [["SIT_LadderLogical\\w+"], 0.8],
+    "VXD": [["SiActiveLayer\\w+"], 0.8],
+    "SET": [["SET_LadderLogical\\w+"], 0.8],
+    "TPC": [["TPC\\w+"], 0.8],
+    "Tube": [["tube\\w+"], 0.8],
+    "YokeBarrel": [["YokeBarrel\\w+"], 0.8],
+    "YokeEndcap": [["YokeEndcap\\w+"], 0.8]
+  },
+  "maxLevel": 3
+}

conversion_xml2gltf_autoConfig.py

@@ -0,0 +1,81 @@
+
+import argparse
+import ROOT
+import subprocess
+
+def main():
+    parser = argparse.ArgumentParser(description='Convert detector')
+    parser.add_argument('-cm', '--compact_file', help='Compact file location(s)',
+                        required=True, type=str, nargs='+')
+    parser.add_argument('-cn', '--config_file', help='Json file of detector structure',
+                        default='nothing', type=str)
+    parser.add_argument('-ro', '--out_root', help='Converted root file path',
+                        default='nothing', type=str)
+    parser.add_argument('-ri', '--in_root', help='Input root file path (if a root file has already been obtained)',
+                        default='nothing', type=str)
+    parser.add_argument('-g', '--out_gltf', help='Converted gltf file path',
+                        default='nothing', type=str)
+    parser.add_argument('-d', '--depth', help='Level of layers in detector',
+                        default=9, type=int)
+
+    args = parser.parse_args()
+
+    #if args.in_root != 'nothing' and args.config_file != 'nothing':
+     #   gltf_convert(args.config_file, args.out_gltf, args.in_root)
+    #else:
+    print(args.compact_file)
+    root_path = root_convert(args.compact_file, args.out_root, args.depth)
+
+    if args.config_file == 'nothing':
+        subprocess.run(["python", "configfile_generator.py", "--compactFile", f'{args.compact_file}'])
+    else:
+        config_file = args.config_file
+
+    gltf_convert(config_file, args.out_gltf, root_path)
+
+
+def root_convert(compact_files, out_path, visibility):
+    print('INFO: Converting following compact file(s):')
+    for cfile in compact_files:
+        print('      ' + cfile)
+
+    ROOT.gSystem.Load('libDDCore')
+    description = ROOT.dd4hep.Detector.getInstance()
+    for cfile in compact_files:
+        description.fromXML(cfile)
+
+    ROOT.gGeoManager.SetVisLevel(visibility)
+    ROOT.gGeoManager.SetVisOption(0)
+
+    if out_path == 'nothing':
+        counter = 0
+        slash_list = []
+        for cfile in compact_files:
+            for i in cfile:
+                counter += 1
+                if i == '/':
+                    slash_list.append(counter)
+
+            last_slash = max(slash_list)
+            root_path = f'{cfile[last_slash:len(cfile)-4]}.root'
+
+    else:
+        root_path = out_path
+
+    ROOT.gGeoManager.Export(root_path)
+
+    return root_path
+
+def gltf_convert(config, gltf, root):
+
+    if gltf == 'nothing':
+        gltf_path = f'{root[:len(root)-5]}.gltf'
+
+    else:
+        gltf_path = gltf
+
+    subprocess.run(["node", ".", "-c", f'{config}', "-o", f'{gltf_path}', f'{root}'])
+
+
+if __name__ == '__main__':
+    main()

conversion_xml2root.py

@@ -0,0 +1,67 @@
+
+import argparse
+import ROOT
+import subprocess
+
+def main():
+    parser = argparse.ArgumentParser(description='Convert detector')
+    parser.add_argument('-cm', '--compact', help='Compact file location(s)',
+                        required=True, type=str, nargs='+')
+    parser.add_argument('-r', '--out_root', help='Converted root file path',
+                        default='nothing', type=str)
+    parser.add_argument('-v', '--visibility', help='Level of layers in detector',
+                        default=9, type=int)
+    parser.add_argument('-g', '--out_gltf', help='Converted gltf file path',
+                        default='nothing', type=str)
+    parser.add_argument('-cn', '--config_file', help='Json file of detector structure',
+                        default='a', type=str)
+    args = parser.parse_args()
+
+    root_path = root_convert(args.compact, args.out_root, args.visibility)
+    gltf_convert(args.config_file, args.out_gltf, root_path)
+
+def root_convert(compact_files, out_path, visibility):
+    print('INFO: Converting following compact file(s):')
+    for cfile in compact_files:
+        print('      ' + cfile)
+
+    ROOT.gSystem.Load('libDDCore')
+    description = ROOT.dd4hep.Detector.getInstance()
+    for cfile in compact_files:
+        description.fromXML(cfile)
+
+    ROOT.gGeoManager.SetVisLevel(visibility)
+    ROOT.gGeoManager.SetVisOption(0)
+
+    if out_path == 'nothing':
+        counter = 0
+        slash_list = []
+        for cfile in compact_files:
+            for i in cfile:
+                counter += 1
+                if i == '/':
+                    slash_list.append(counter)
+
+            last_slash = max(slash_list)
+            root_path = f'{cfile[last_slash:len(cfile)-4]}.root'
+
+    else:
+        root_path = out_path
+
+    ROOT.gGeoManager.Export(root_path)
+
+    return root_path
+
+
+def gltf_convert(config, gltf, root):
+
+    if gltf == 'nothing':
+        gltf_path = f'{root[:len(root)-5]}.gltf'
+
+    else:
+        gltf_path = gltf
+
+    subprocess.run(["node", ".", "-c", f'{config}', "-o", f'{gltf_path}', f'{root}'])
+
+if __name__ == '__main__':
+    main()