def_parser/colorize_partition.py at master · parastuffs/def_parser · 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
import def_parser
# import Image
from PIL import Image
import random
from natsort import natsorted # https://pypi.python.org/pypi/natsort
import os

"""
Partition 0: (255, 0, 0)
Partition 1: (0, 0, 255)
"""


if __name__ == "__main__":

    def_parser.MEMORY_MACROS = False # spc: True, others: False
    # def_parser.MEMORY_MACROS = True # spc: True, others: False
    def_parser.UNITS_DISTANCE_MICRONS = 10000 #flipr, boomcore, ldpc
    # def_parser.UNITS_DISTANCE_MICRONS = 1000  #ccx, spc

    # ldpc, boomcore, flipr: "7nm"
    # ccx, spc: "45nm"
    def_parser.extractStdCells("7nm")
    # def_parser.extractStdCells("45nm")
    if def_parser.MEMORY_MACROS:
        def_parser.extractMemoryMacros(14,4)

    # If you change the deffile, also change the leffile, MEMORY_MACROS and UNITS_DISTANCE_MICRONS
    def_parser.deffile = "7nm_Jul2017/ldpc.def"
    # def_parser.deffile = "ldpc_4x4/ldpc-4x4.def"
    # def_parser.deffile = "7nm_Jul2017/BoomCore.def"
    # def_parser.deffile = "7nm_Jul2017/flipr.def"
    # def_parser.deffile = "7nm_Jul2017/ccx.def"
    # def_parser.deffile = "7nm_Jul2017/SPC/spc.def" # Don't forget to turn the MEMORY_MACROS on.
    # def_parser.deffile = "ldpc_4x4_serial.def/ldpc-4x4-serial.def"
    # def_parser.deffile = "ldpc_4x4/ldpc-4x4.def"


    """
    The folder tree should be as follows:
    Main folder
    |_ Cluster level x
        |_ partition
    """
    # partitionFile = "/home/para/dev/metis_unicorn/temp_design/metis_02_1-NoWires_area.hgr.part"
    # partitionFile = "/home/para/dev/def_parser/2018-01-24_19-37-08/BoomCore_random_9/partitions_2018-02-20_13-03-40/metis_02_1-NoWires_area.hgr.part"
    # mainDir = "/home/para/dev/def_parser/2018-03-14_19-51-44"
    mainDir = "/home/para/dev/def_parser/2019-05-20_13-13-11_ldpc_hierarchical-geometric/"

    def_parser.design = def_parser.Design()
    def_parser.design.ReadArea()
    def_parser.design.ExtractCells()

    imgW = int(def_parser.design.width*10) # Width of the design in 10^-1 um
    imgH = int(def_parser.design.height*10)
    imgSize = (imgW) * (imgH)
    data = [0] * imgSize

    for clusterDir in natsorted(os.listdir(mainDir)):
        clusterDir = os.path.join(mainDir, clusterDir)
        if os.path.isdir(clusterDir):
            for partitionDir in natsorted(os.listdir(clusterDir)):
                partitionDir = os.path.join(clusterDir, partitionDir)
                if os.path.isdir(partitionDir):
                    for partitionFile in natsorted(os.listdir(partitionDir)):
                        # I just want to generate the image for the 'metis_01' file.
                        # CF 2017-02-21, MPD: "Comme disait mon pere, c'est des moules dans du sirop.", en parlant du projet batiment.
                        if "01" in partitionFile and partitionFile.split('.')[-1] == "part":
                        # if "metis_01" in partitionFile and partitionFile.split('.')[-1] == "part":

                            clusters = dict()
                            partColors = [(255, 0, 0), (0, 0, 255)]

                            with open(os.path.join(partitionDir, partitionFile), 'r') as f:
                                for line in f.readlines():
                                    gate = line.split()[0]
                                    x = def_parser.design.gates.get(gate).x
                                    y = def_parser.design.gates.get(gate).y
                                    index = (int(y*10) * (imgW) ) + int(x*10)
                                    data[index] = partColors[int(line.split()[1])]

                            print("Create the image (" + str(imgW) + ", " + str(imgH) + ")")
                            img = Image.new('RGB', (imgW, imgH))
                            print("Put data")
                            img.putdata(data)
                            outfile = os.path.join(partitionDir, clusterDir.split('/')[-1] + "_" + partitionFile.split('.')[0] + ".png")
                            print("save image " + outfile)
                            img.save(outfile)
                        elif "03" in partitionFile and partitionFile.split('.')[-1] == "part":

                            clusters = dict()
                            partColors = [(255, 0, 0), (0, 0, 255)]

                            with open(os.path.join(partitionDir, partitionFile), 'r') as f:
                                for line in f.readlines():
                                    gate = line.split()[0]
                                    x = def_parser.design.gates.get(gate).x
                                    y = def_parser.design.gates.get(gate).y
                                    index = (int(y*10) * (imgW) ) + int(x*10)
                                    data[index] = partColors[int(line.split()[1])]

                            print("Create the image (" + str(imgW) + ", " + str(imgH) + ")")
                            img = Image.new('RGB', (imgW, imgH))
                            print("Put data")
                            img.putdata(data)
                            outfile = os.path.join(partitionDir, clusterDir.split('/')[-1] + "_" + partitionFile.split('.')[0] + ".png")
                            print("save image " + outfile)
                            img.save(outfile)