OpenRAM/compiler/pgates/ptx.py

import design
import debug
from tech import drc, spice
from vector import vector
from contact import contact
from globals import OPTS
import path

class ptx(design.design):
    """
    This module generates gds and spice of a parametrically NMOS or
    PMOS sized transistor.  Pins are accessed as D, G, S, B.  Width is
    the transistor width. Mults is the number of transistors of the
    given width. Total width is therefore mults*width.  Options allow
    you to connect the fingered gates and active for parallel devices.

    """
    def __init__(self, width=drc("minwidth_tx"), mults=1, tx_type="nmos", connect_active=False, connect_poly=False, num_contacts=None):
        # We need to keep unique names because outputting to GDSII
        # will use the last record with a given name. I.e., you will
        # over-write a design in GDS if one has and the other doesn't
        # have poly connected, for example.
        name = "{0}_m{1}_w{2}".format(tx_type, mults, width)
        if connect_active:
            name += "_a"
        if connect_poly:
            name += "_p"
        if num_contacts:
            name += "_c{}".format(num_contacts)
        # replace periods with underscore for newer spice compatibility
        name=name.replace('.','_')

        design.design.__init__(self, name)
        debug.info(3, "create ptx2 structure {0}".format(name))

        self.tx_type = tx_type
        self.mults = mults
        self.tx_width = width
        self.connect_active = connect_active
        self.connect_poly = connect_poly
        self.num_contacts = num_contacts

        self.create_netlist()
        # We must always create ptx layout for pbitcell
        # some transistor sizes in other netlist depend on pbitcell
        self.create_layout()


    
    def create_layout(self):
        """Calls all functions related to the generation of the layout"""
        self.setup_layout_constants()
        self.add_active()
        self.add_well_implant()  
        self.add_poly()
        self.add_active_contacts()
        self.translate_all(self.active_offset)

        # for run-time, we won't check every transitor DRC independently
        # but this may be uncommented for debug purposes
        #self.DRC()

    def create_netlist(self):
        self.add_pin_list(["D", "G", "S", "B"])
        
        # self.spice.append("\n.SUBCKT {0} {1}".format(self.name,
        #                                              " ".join(self.pins)))
        # Just make a guess since these will actually be decided in the layout later.
        area_sd = 2.5*drc("minwidth_poly")*self.tx_width
        perimeter_sd = 2*drc("minwidth_poly") + 2*self.tx_width
        self.spice_device="M{{0}} {{1}} {0} m={1} w={2}u l={3}u pd={4:.2f}u ps={4:.2f}u as={5:.2f}p ad={5:.2f}p".format(spice[self.tx_type],
                                                                                                                        self.mults,
                                                                                                                        self.tx_width,
                                                                                                                        drc("minwidth_poly"),
                                                                                                                        perimeter_sd,
                                                                                                                        area_sd)
        self.spice.append("\n* ptx " + self.spice_device)
        # self.spice.append(".ENDS {0}".format(self.name))

    def setup_layout_constants(self):
        """
        Pre-compute some handy layout parameters.
        """

        if self.num_contacts==None:
            self.num_contacts=self.calculate_num_contacts()

        # Determine layer types needed
        if self.tx_type == "nmos":
            self.implant_type = "n"
            self.well_type = "p"
        elif self.tx_type == "pmos":
            self.implant_type = "p"
            self.well_type = "n"
        else:
            self.error("Invalid transitor type.",-1)
            
            
        # This is not actually instantiated but used for calculations
        self.active_contact = contact(layer_stack=("active", "contact", "metal1"),
                                      dimensions=(1, self.num_contacts))

        
        # The contacted poly pitch (or uncontacted in an odd technology)
        self.poly_pitch = max(2*self.contact_to_gate + self.contact_width + self.poly_width,
                              self.poly_space)

        # The contacted poly pitch (or uncontacted in an odd technology)
        self.contact_pitch = 2*self.contact_to_gate + self.contact_width + self.poly_width
        
        # The enclosure of an active contact. Not sure about second term.
        active_enclose_contact = max(drc("active_enclosure_contact"),
                                     (self.active_width - self.contact_width)/2)
        # This is the distance from the edge of poly to the contacted end of active
        self.end_to_poly = active_enclose_contact + self.contact_width + self.contact_to_gate
        

        # Active width is determined by enclosure on both ends and contacted pitch,
        # at least one poly and n-1 poly pitches
        self.active_width = 2*self.end_to_poly + self.poly_width + (self.mults - 1)*self.poly_pitch

        # Active height is just the transistor width
        self.active_height = self.tx_width

        # Poly height must include poly extension over active
        self.poly_height = self.tx_width + 2*self.poly_extend_active

        # The active offset is due to the well extension
        self.active_offset = vector([self.well_enclose_active]*2)

        # Well enclosure of active, ensure minwidth as well
        if drc("has_{}well".format(self.well_type)):
            self.cell_well_width = max(self.active_width + 2*self.well_enclose_active,
                                  self.well_width)
            self.cell_well_height = max(self.tx_width + 2*self.well_enclose_active,
                                   self.well_width)
            # We are going to shift the 0,0, so include that in the width and height
            self.height = self.cell_well_height - self.active_offset.y
            self.width = self.cell_well_width - self.active_offset.x
        else:
            # If no well, use the boundary of the active and poly
            self.height = self.poly_height
            self.width = self.active_width
        
        # The active offset is due to the well extension
        self.active_offset = vector([self.well_enclose_active]*2)

        # This is the center of the first active contact offset (centered vertically)
        self.contact_offset = self.active_offset + vector(active_enclose_contact + 0.5*self.contact_width,
                                                          0.5*self.active_height)
                                     
        
        # Min area results are just flagged for now.
        debug.check(self.active_width*self.active_height>=drc("minarea_active"),"Minimum active area violated.")
        # We do not want to increase the poly dimensions to fix an area problem as it would cause an LVS issue.
        debug.check(self.poly_width*self.poly_height>=drc("minarea_poly"),"Minimum poly area violated.")

    def connect_fingered_poly(self, poly_positions):
        """
        Connect together the poly gates and create the single gate pin.
        The poly positions are the center of the poly gates
        and we will add a single horizontal connection.
        """
        # Nothing to do if there's one poly gate
        if len(poly_positions)<2:
            return

        # The width of the poly is from the left-most to right-most poly gate
        poly_width = poly_positions[-1].x - poly_positions[0].x + self.poly_width
        if self.tx_type == "pmos":
            # This can be limited by poly to active spacing or the poly extension
            distance_below_active = self.poly_width + max(self.poly_to_active,0.5*self.poly_height)
            poly_offset = poly_positions[0] - vector(0.5*self.poly_width, distance_below_active)
        else:
            # This can be limited by poly to active spacing or the poly extension
            distance_above_active = max(self.poly_to_active,0.5*self.poly_height)            
            poly_offset = poly_positions[0] + vector(-0.5*self.poly_width, distance_above_active)
        # Remove the old pin and add the new one
        self.remove_layout_pin("G") # only keep the main pin
        self.add_layout_pin(text="G",
                            layer="poly",
                            offset=poly_offset,
                            width=poly_width,
                            height=drc("minwidth_poly"))


    def connect_fingered_active(self, drain_positions, source_positions):
        """
        Connect each contact  up/down to a source or drain pin
        """
        
        # This is the distance that we must route up or down from the center
        # of the contacts to avoid DRC violations to the other contacts
        pin_offset = vector(0, 0.5*self.active_contact.second_layer_height \
                            + self.m1_space + 0.5*self.m1_width)
        # This is the width of a m1 extend the ends of the pin
        end_offset = vector(self.m1_width/2,0)

        # drains always go to the MIDDLE of the cell, so top of NMOS, bottom of PMOS
        # so reverse the directions for NMOS compared to PMOS.
        if self.tx_type == "pmos":
            drain_dir = -1
            source_dir = 1
        else:
            drain_dir = 1
            source_dir = -1
            
        if len(source_positions)>1: 
            source_offset = pin_offset.scale(source_dir,source_dir)
            self.remove_layout_pin("S") # remove the individual connections
            # Add each vertical segment
            for a in source_positions:
                self.add_path(("metal1"), [a,a+pin_offset.scale(source_dir,source_dir)])
            # Add a single horizontal pin
            self.add_layout_pin_segment_center(text="S",
                                               layer="metal1",
                                               start=source_positions[0]+source_offset-end_offset,
                                               end=source_positions[-1]+source_offset+end_offset)

        if len(drain_positions)>1:
            drain_offset = pin_offset.scale(drain_dir,drain_dir)
            self.remove_layout_pin("D") # remove the individual connections
            # Add each vertical segment
            for a in drain_positions:
                self.add_path(("metal1"), [a,a+drain_offset])
            # Add a single horizontal pin
            self.add_layout_pin_segment_center(text="D",
                                               layer="metal1",
                                               start=drain_positions[0]+drain_offset-end_offset,
                                               end=drain_positions[-1]+drain_offset+end_offset)
            
    def add_poly(self):
        """
        Add the poly gates(s) and (optionally) connect them.
        """
        # poly is one contacted spacing from the end and down an extension
        poly_offset = self.active_offset + vector(self.poly_width,self.poly_height).scale(0.5,0.5) \
                      + vector(self.end_to_poly, -self.poly_extend_active)

        # poly_positions are the bottom center of the poly gates
        poly_positions = []

        # It is important that these are from left to right, so that the pins are in the right
        # order for the accessors
        for i in range(0, self.mults):
            # Add this duplicate rectangle in case we remove the pin when joining fingers
            self.add_rect_center(layer="poly",
                                 offset=poly_offset,
                                 height=self.poly_height,
                                 width=self.poly_width)
            self.add_layout_pin_rect_center(text="G",
                                            layer="poly",
                                            offset=poly_offset,
                                            height=self.poly_height,
                                            width=self.poly_width)
            poly_positions.append(poly_offset)
            poly_offset = poly_offset + vector(self.poly_pitch,0)

        if self.connect_poly:
            self.connect_fingered_poly(poly_positions)
            
    def add_active(self):
        """ 
        Adding the diffusion (active region = diffusion region) 
        """
        self.add_rect(layer="active",
                      offset=self.active_offset,
                      width=self.active_width,
                      height=self.active_height)
        # If the implant must enclose the active, shift offset
        # and increase width/height
        enclose_width = drc("implant_enclosure_active")
        enclose_offset = [enclose_width]*2
        self.add_rect(layer="{}implant".format(self.implant_type),
                      offset=self.active_offset - enclose_offset,
                      width=self.active_width + 2*enclose_width,
                      height=self.active_height + 2*enclose_width)

    def add_well_implant(self):
        """
        Add an (optional) well and implant for the type of transistor.
        """
        if drc("has_{}well".format(self.well_type)):
            self.add_rect(layer="{}well".format(self.well_type),
                          offset=(0,0),
                          width=self.cell_well_width,
                          height=self.cell_well_height)
            self.add_rect(layer="vtg",
                          offset=(0,0),
                          width=self.cell_well_width,
                          height=self.cell_well_height)


    def calculate_num_contacts(self):
        """ 
        Calculates the possible number of source/drain contacts in a finger.
        For now, it is hard set as 1.
        """
        return 1


    def get_contact_positions(self):
        """
        Create a list of the centers of drain and source contact positions.
        """
        # The first one will always be a source
        source_positions = [self.contact_offset]
        drain_positions = []
        # It is important that these are from left to right, so that the pins are in the right
        # order for the accessors.
        for i in range(self.mults):
            if i%2:
                # It's a source... so offset from previous drain.
                source_positions.append(drain_positions[-1] + vector(self.contact_pitch,0))
            else:
                # It's a drain... so offset from previous source.
                drain_positions.append(source_positions[-1] + vector(self.contact_pitch,0))

        return [source_positions,drain_positions]
        
    def add_active_contacts(self):
        """
        Add the active contacts to the transistor.
        """

        [source_positions,drain_positions] = self.get_contact_positions()

        for pos in source_positions:
            contact=self.add_contact_center(layers=("active", "contact", "metal1"),
                                            offset=pos,
                                            size=(1, self.num_contacts),
                                            implant_type=self.implant_type,
                                            well_type=self.well_type)
            self.add_layout_pin_rect_center(text="S",
                                            layer="metal1",
                                            offset=pos,
                                            width=contact.mod.second_layer_width,
                                            height=contact.mod.second_layer_height)

                
        for pos in drain_positions:
            contact=self.add_contact_center(layers=("active", "contact", "metal1"),
                                            offset=pos,
                                            size=(1, self.num_contacts),
                                            implant_type=self.implant_type,
                                            well_type=self.well_type)
            self.add_layout_pin_rect_center(text="D",
                                            layer="metal1",
                                            offset=pos,
                                            width=contact.mod.second_layer_width,
                                            height=contact.mod.second_layer_height)
                
        if self.connect_active:
            self.connect_fingered_active(drain_positions, source_positions)

    def get_cin(self):
        """Returns the relative gate cin of the tx"""
        return self.tx_width/drc("minwidth_tx")