Class: Kumi::IR::DF::ImportInliner

Inherits:
Object
  • Object
show all
Defined in:
lib/kumi/ir/df/import_inliner.rb

Instance Method Summary collapse

Constructor Details

#initialize(axis_map:, extra_axes: []) ⇒ ImportInliner

Returns a new instance of ImportInliner.



9
10
11
12
 
# File 'lib/kumi/ir/df/import_inliner.rb', line 9

def initialize(axis_map:, extra_axes: [])
  @axis_map = normalize_axis_map(axis_map)
  @extra_axes = Array(extra_axes).map { |axis| axis&.to_sym }.compact
end

Instance Method Details

#remap_function(function) ⇒ Object 



14
15
16
17
18
19
20
21
22
 
# File 'lib/kumi/ir/df/import_inliner.rb', line 14

def remap_function(function)
  new_blocks = function.blocks.map { remap_block(_1) }
  Kumi::IR::DF::Function.new(
    name: function.name,
    parameters: function.parameters,
    blocks: new_blocks,
    return_stamp: function.return_stamp
  )
end