any_write works with DeviceReferences. Switch for writing ports: lyga…

…dgets.cell_translation.do_write_ports

lygadgets/cell_translation.py

@@ -7,6 +7,7 @@
 import os
 
 default_phidl_portlayer = 41
+do_write_ports = True
 
 def celltype_to_write_function(celltype):
     ''' Takes a class that represents a layout Cell and gives a function that writes its geometry to disc
@@ -35,18 +36,31 @@ def not_supported_error(language_name):
     try: import phidl
     except ImportError: pass
     else:
-        if issubclass(celltype, phidl.Device):
-            def write_with_ports(device, filename, *args, port_layer=None, **kwargs):
-                if port_layer is None:
-                    port_layer = default_phidl_portlayer
-                try:
-                    wgp = phidl.geometry.with_geometric_ports
-                except AttributeError:  # it is an older version of phidl
-                    pass
-                else:
-                    device = wgp(device, layer=port_layer)
-                device.write_gds(filename, *args, **kwargs)
-            return write_with_ports
+        if issubclass(celltype, (phidl.Device, phidl.device_layout.DeviceReference)):
+            if do_write_ports:
+                def write_with_ports(device, filename, *args, port_layer=None, **kwargs):
+                    if port_layer is None:
+                        port_layer = default_phidl_portlayer
+                    # If its a reference, use its parent
+                    if issubclass(celltype, phidl.device_layout.DeviceReference):
+                        device = device.parent
+                    # Try to convert to geometric ports
+                    try:
+                        port2geom = phidl.geometry.ports_to_geometry
+                    except AttributeError:  # it is an older version of phidl
+                        pass
+                    else:
+                        device = port2geom(device, layer=port_layer)
+                    # The actual write
+                    device.write_gds(filename, *args, **kwargs)
+                return write_with_ports
+            else:
+                def write_parent(device, filename, *args, **kwargs):
+                    # If its a reference, use its parent
+                    if issubclass(celltype, phidl.device_layout.DeviceReference):
+                        device = device.parent
+                    device.write_gds(filename, *args, **kwargs)
+                return write_parent
 
     # try: import gdspy
     # except ImportError: pass
@@ -175,10 +189,13 @@ def anyCell_to_anyCell(initial_cell, final_cell):
 
         The supported types and their mapping to write methods are contained in celltype_to_write_function and celltype_to_read_function.
     '''
+    do_write_ports_orig = do_write_ports
+    do_write_ports = True
     tempfile = os.path.realpath('temp_cellTranslation.gds')
     any_write(initial_cell, tempfile)
     new_cell = any_read(final_cell, tempfile)
     os.remove(tempfile)
+    do_write_ports = do_write_ports_orig
 
     # Transfer other data (ports, metadata, CML files, etc.)
     pass  # TODO