def test_equ_netlist_complete_test_1(self):
netlist = Netlist()
str_netlist = "M0001 N0002 IN001 N0001 GND NMOS\n"
netlist.set_netlist(str_netlist)
equ_netlists = []
script_dir = os.path.dirname(os.path.realpath(__file__)) + '/'
with open(script_dir + 'complete_test_results_1.txt') as results:
temp_netlist = ''
for line in results:
if line == "\n":
equ_netlists.append(temp_netlist)
temp_netlist = ''
continue
temp_netlist += line
self.assertEqual(len(equ_netlists),
len(list(netlist.get_equ_netlists())))
self.assertCountEqual(equ_netlists, netlist.get_equ_netlists())
def test_equ_netlist_swap_diff(self):
netlist = Netlist()
str_netlist = "M0001 VDD IN001 IN002 GND NMOS\n"
netlist.set_netlist(str_netlist)
equ_netlists = [
"M0001 VDD IN001 IN002 GND NMOS\n",
"M0001 IN002 IN001 VDD GND NMOS\n",
"M0001 VDD IN002 IN001 GND NMOS\n",
"M0001 IN001 IN002 VDD GND NMOS\n"
]
self.assertCountEqual(equ_netlists, netlist.get_equ_netlists())
class Cell:
def __init__(self, db):
self.netlist_ = Netlist()
self.upstream_db_ = db
self.cell_id_ = None
self.bsf_ = None
self.bsf_weak_ = None
self.bsf_unified_ = None
self.bsf_weak_unified_ = None
def init_based_on_id(self, id_cell):
self.cell_id_ = id_cell
self.bsf_ = None
self.bsf_weak_ = None
self.bsf_unified_ = None
self.bsf_weak_unified_ = None
row = self.upstream_db_.get_query_row(
'SELECT CELL_BSF, CELL_BSF_UNIFIED, CELL_BSF_weak, '
'CELL_BSF_weak_UNIFIED, CELL_NETLIST '
f'FROM {self.upstream_db_.table_} WHERE idCELL=%s', [id_cell])
if row is None:
raise ValueError(
f'Cell #{id_cell} not found in {self.upstream_db_.table_}')
self.netlist_.set_netlist(row['CELL_NETLIST'])
self.bsf_ = row['CELL_BSF'].decode("utf-8")
self.bsf_weak_ = row['CELL_BSF_weak'].decode("utf-8")
self.bsf_unified_ = row['CELL_BSF_UNIFIED'].decode("utf-8")
self.bsf_weak_unified_ = row['CELL_BSF_weak_UNIFIED'].decode("utf-8")
def init_based_on_netlist(self, netlist):
self.cell_id_ = None
self.bsf_ = None
self.bsf_weak_ = None
self.bsf_unified_ = None
self.bsf_weak_unified_ = None
self.netlist_.set_netlist(netlist)
def cal_bsf(self):
self.bsf_, self.bsf_weak_ = csim(self.netlist_.get_netlist_string())
self.bsf_unified_, bsf_list = gen_equ_bsf(self.bsf_)
self.bsf_weak_unified_, bsf_list = gen_equ_bsf(self.bsf_weak_)
def get_upstream_db(self):
return self.upstream_db_
def set_upstream_db(self, db_):
self.upstream_db_ = db_
def get_bsf(self):
if not self.bsf_:
self.cal_bsf()
return self.bsf_
def get_bsf_unified(self):
if not self.bsf_unified_:
self.cal_bsf()
return self.bsf_unified_
def get_bsf_weak(self):
if not self.bsf_weak_:
self.cal_bsf()
return self.bsf_weak_
def fetch_ids(self):
query = f'SELECT Q.ID FROM ' \
f'(SELECT idCELL AS ID, CELL_NETLIST AS Netlist ' \
f'FROM {self.upstream_db_.get_table()} WHERE CELL_BSF_UNIFIED=%s)' \
f' AS Q WHERE Q.Netlist IN ('
temp = "', '".join(self.netlist_.get_equ_netlists())
temp = "'" + temp + "'"
query += temp
query += ')'
id_list = list()
for row in self.upstream_db_.run_query(query,
[self.get_bsf_unified()]):
id_list.append(int(row['ID']))
return id_list
def get_id(self):
if self.cell_id_ is None:
id_list = self.fetch_ids()
if id_list:
self.cell_id_ = id_list[0]
else:
self.cell_id_ = 0
return self.cell_id_
def get_family(self):
id_cell = self.get_id()
cell_family = self.upstream_db_.get_query_value(
'CELL_FAMILY',
f'SELECT CELL_FAMILY FROM {self.upstream_db_.get_table()} '
'WHERE idCELL=%s', [id_cell])
return cell_family
def clear_family(self):
id_cell = self.get_id()
self.upstream_db_.run_sql(
f'UPDATE {self.upstream_db_.get_table()} SET CELL_FAMILY=NULL WHERE '
f'idCELL={id_cell}')
def add_to_family(self, family_name):
id_cell = self.get_id()
if id_cell == 0:
return
cell_family = self.get_family()
if not cell_family:
cell_family = family_name
elif family_name not in cell_family:
cell_family += ',' + family_name
else:
# cell already belongs to this family
return
self.upstream_db_.update(id_cell, 'idCELL',
{'CELL_FAMILY': cell_family})
