def union(gra1, gra2):
""" a union of two graphs
"""
assert not atom_keys(gra1) & atom_keys(gra2)
atm_dct = {}
atm_dct.update(atoms(gra1))
atm_dct.update(atoms(gra2))
bnd_dct = {}
bnd_dct.update(bonds(gra1))
bnd_dct.update(bonds(gra2))
return _create.from_atoms_and_bonds(atm_dct, bnd_dct)
def add_atoms(gra, sym_dct, imp_hyd_vlc_dct=None, ste_par_dct=None):
""" add atoms to this molecular graph, setting their keys
"""
atm_keys = atom_keys(gra)
atm_sym_dct = atom_symbols(gra)
atm_imp_hyd_vlc_dct = atom_implicit_hydrogen_valences(gra)
atm_ste_par_dct = atom_stereo_parities(gra)
keys = set(sym_dct.keys())
imp_hyd_vlc_dct = {} if imp_hyd_vlc_dct is None else imp_hyd_vlc_dct
ste_par_dct = {} if ste_par_dct is None else ste_par_dct
assert not keys & atm_keys
assert set(imp_hyd_vlc_dct.keys()) <= keys
assert set(ste_par_dct.keys()) <= keys
atm_sym_dct.update(sym_dct)
atm_imp_hyd_vlc_dct.update(imp_hyd_vlc_dct)
atm_ste_par_dct.update(ste_par_dct)
atm_dct = _create.atoms_from_data(
atom_symbols=atm_sym_dct,
atom_implicit_hydrogen_valences=atm_imp_hyd_vlc_dct,
atom_stereo_parities=atm_ste_par_dct)
bnd_dct = bonds(gra)
gra = _create.from_atoms_and_bonds(atoms=atm_dct, bonds=bnd_dct)
return gra
def bond_induced_subgraph(gra, bnd_keys):
""" the subgraph induced by a subset of the bonds
"""
atm_keys = set(itertools.chain(*bnd_keys))
bnd_keys = set(bnd_keys)
assert atm_keys <= atom_keys(gra)
atm_dct = dict_.by_key(atoms(gra), atm_keys)
bnd_dct = dict_.by_key(bonds(gra), bnd_keys)
return _create.from_atoms_and_bonds(atm_dct, bnd_dct)
def subgraph(gra, atm_keys):
""" the subgraph induced by a subset of the atoms
"""
atm_keys = set(atm_keys)
assert atm_keys <= atom_keys(gra)
bnd_keys = set(filter(lambda x: x <= atm_keys, bond_keys(gra)))
atm_dct = dict_.by_key(atoms(gra), atm_keys)
bnd_dct = dict_.by_key(bonds(gra), bnd_keys)
return _create.from_atoms_and_bonds(atm_dct, bnd_dct)
def remove_bonds(gra, bnd_keys, check=True):
""" remove bonds from the molecular graph
"""
all_bnd_keys = bond_keys(gra)
bnd_keys = set(map(frozenset, bnd_keys))
if check:
assert bnd_keys <= all_bnd_keys
bnd_keys = all_bnd_keys - bnd_keys
atm_dct = atoms(gra)
bnd_dct = dict_.by_key(bonds(gra), bnd_keys)
return _create.from_atoms_and_bonds(atm_dct, bnd_dct)
def frozen(gra):
""" hashable, sortable, immutable container of graph data
"""
atm_keys = sorted(atom_keys(gra))
bnd_keys = sorted(bond_keys(gra), key=sorted)
# make it sortable by replacing Nones with -infinity
atm_vals = numpy.array(dict_.values_by_key(atoms(gra), atm_keys))
bnd_vals = numpy.array(dict_.values_by_key(bonds(gra), bnd_keys))
atm_vals[numpy.equal(atm_vals, None)] = -numpy.inf
bnd_vals[numpy.equal(bnd_vals, None)] = -numpy.inf
frz_atms = tuple(zip(atm_keys, map(tuple, atm_vals)))
frz_bnds = tuple(zip(bnd_keys, map(tuple, bnd_vals)))
return (frz_atms, frz_bnds)