SketchUpCMS/readDDL_unique.rb at master · adavidzh/SketchUpCMS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#!/usr/bin/env ruby
# Tai Sakuma <sakuma@fnal.gov>

$LOAD_PATH.unshift(File.expand_path(File.dirname(__FILE__)))
$LOAD_PATH.unshift(File.expand_path(File.dirname(__FILE__)) + "/lib")
$LOAD_PATH.unshift(File.expand_path(File.dirname(__FILE__)) + "/gratr/lib")

require 'buildGeometryManager'
require 'buildDDLCallBacks'
require 'readXMLFiles'

require 'graph_functions'
require 'graph_functions_DD'

require 'gratr'
require 'gratr/dot'

require "benchmark"

require 'LogicalPartInstance'

##__________________________________________________________________||
def cmsmain

  lines = [ ]
  lines << Benchmark::CAPTION
  lines << Benchmark.measure { read_xmlfiles() }
  lines << Benchmark.measure { draw_geom() }
  puts lines

end

##__________________________________________________________________||
def draw_geom

  # all PosParts in the XML file
  graphAll = GRATR::DirectedPseudoGraph.new
  $posPartsManager.parts.each { |pp| graphAll.add_edge!(pp.parentName, pp.childName, pp) }

  vertices = Hash[$logicalPartsManager.parts.map { |p| [p.name, LogicalPartInstance.new($geometryManager, p)] } ]

  topName = :"cms:CMSE"

  subNames = [:"muonBase:MBWheel_2N", :"muonBase:MBWheel_1N"]

  nameDepthList = [
    {:name => :"mb1:MB1ChimHoneycombBox", :depth => 0},
    {:name => :"mb1:MB1ChimSuperLayerZ", :depth => 0},
    {:name => :"mb1:MB1ChimSuperLayerPhi", :depth => 0},
    {:name => :"mb1:MB1HoneycombBox", :depth => 0},

    # :"mb1:MB1SuperLayerZ"
    {:name => :"mb1:MB1SLZLayer_58Cells", :depth => 0},
    {:name => :"mb1:MB1SLZLayer_56Cells", :depth => 0},
    {:name => :"mb1:MB1SLZLayer_57Cells", :depth => 0},
    {:name => :"mb1:MB1SLZAlPlateInner", :depth => 0},
    {:name => :"mb1:MB1SLZAlPlateOuter", :depth => 0},

    # :"mb1:MB1SuperLayerPhi"
    {:name => :"mb1:MB1SLPhiLayer_48Cells", :depth => 0},
    {:name => :"mb1:MB1SLPhiLayer_50Cells", :depth => 0},
    {:name => :"mb1:MB1SLPhiLayer_49Cells", :depth => 0},
    {:name => :"mb1:MB1SLPhiAlPlateInner", :depth => 0},
    {:name => :"mb1:MB1SLPhiAlPlateOuter", :depth => 0},
  ]

  graphTopToSub = subgraph_from_to(graphAll, topName, subNames)

  names = nameDepthList.collect { |e| e[:name] }
  graphSubToNames = subgraph_from_to(graphAll, subNames, names)

  graphNamesToDepths = graphAll.class.new
  nameDepthList.each do |e|
    graphNamesToDepths = graphNamesToDepths + subgraph_from_depth(graphAll, e[:name], e[:depth])
  end

  graph = graphTopToSub + graphSubToNames + graphNamesToDepths

  edge = graph.adjacent(:"mb1:MB1N", {:direction => :in, :type => :edges})[0, 5]
  make_logicalPart_unique(graph, edge, vertices, true)

  edgesToRemove = graph.adjacent(:"mb1:MB1N#1", {:direction => :out, :type => :edges})
  edgesToRemove.each { |e| graph.remove_edge! e }

  graph = subgraph_from(graph, topName)

  graph.write_to_graphic_file('pdf','graph')

  n_instances = n_paths(graph, topName)

  graph.topsort.each do |v|
    puts " %-30s %10d   %s" % [v, n_instances[v], vertices[v].inspect]
  end

end

##__________________________________________________________________||
def read_xmlfiles
  topDir = File.expand_path(File.dirname(__FILE__))
  xmlfileList = ['Geometry_YB1N_sample.xml']
  # xmlfileList = ['GeometryExtended.xml']
  xmlfileList.map! {|f| f = File.join(topDir, f) }
  p xmlfileList
  geometryManager = buildGeometryManager()
  callBacks = buildDDLCallBacks(geometryManager)
  readXMLFiles(xmlfileList, callBacks)
end

##__________________________________________________________________||

cmsmain