torchref.restraints package

Restraints module for crystallographic refinement.

This module provides classes for building and managing geometry restraints (bonds, angles, torsions, planes, chirals, VDW contacts) from CIF dictionaries.

Restraint geometry parameters are sourced from the CCP4 Monomer Library, which derives ideal values from the Cambridge Structural Database.

References

Long, F., et al. (2017). AceDRG: a stereochemical description generator

for ligands. Acta Cryst. D73, 112-122.

Sherri, L.N., et al. (2018). Updated CCP4 Monomer Library.

Acta Cryst. D74, 641-655.

Classes

RestraintsNew

Refactored restraints handler using builder pattern (faster, more testable).

ResidueIterator

Efficient iterator over residues with pre-grouped data.

RestraintBuilder

Abstract base class for restraint builders.

BondRestraintBuilder

Builder for bond length restraints.

AngleRestraintBuilder

Builder for angle restraints.

TorsionRestraintBuilder

Builder for torsion angle restraints.

PlaneRestraintBuilder

Builder for planarity restraints.

ChiralRestraintBuilder

Builder for chiral volume restraints.

InterResidueBondBuilder

Builder for inter-residue bond restraints (peptide, disulfide).

InterResidueAngleBuilder

Builder for inter-residue angle restraints.

InterResidueTorsionBuilder

Builder for inter-residue torsion restraints (phi, psi, omega).

InterResiduePlaneBuilder

Builder for inter-residue plane restraints.

torchref.restraints.Restraints

alias of RestraintsNew

class torchref.restraints.ResidueIterator(pdb, filter_atom_type=None)

Bases: object

Efficient iterator over residues with pre-grouped data.

Pre-groups PDB data by (chainid, resseq) once at initialization, avoiding O(N) DataFrame filtering per residue. This reduces the complexity from O(N×R) to O(N log N) where N is the number of atoms and R is the number of residues.

Parameters:

• pdb (pd.DataFrame) – PDB DataFrame with columns ‘chainid’, ‘resseq’, ‘resname’, ‘name’, ‘index’.

• filter_atom_type (str, optional) – If provided, only include atoms with this ATOM type (e.g., ‘ATOM’ for protein).

pdb

Reference to the input DataFrame.

Type:

pd.DataFrame

groups

List of (chainid, resseq) tuples for iteration.

Type:

list of tuple

Examples

iterator = ResidueIterator(model.pdb)
for chain_id, resseq, residue_df in iterator:
    print(f"Processing {chain_id}:{resseq}")

__init__(pdb, filter_atom_type=None)

Initialize with pre-grouping of residues.

__iter__()

Iterate over residues.

Yields:

• chain_id (str) – Chain identifier.

• resseq (int) – Residue sequence number.

• residue (pd.DataFrame) – DataFrame containing all atoms in this residue.

__len__()

Return number of residues.

get_consecutive_pairs()

Iterate over consecutive residue pairs within each chain.

Useful for building inter-residue restraints (peptide bonds, etc.).

Yields:

• residue_i (pd.DataFrame) – First residue in the pair.

• residue_next (pd.DataFrame) – Next consecutive residue (resseq + 1).

class torchref.restraints.RestraintBuilder(verbose=0)

Bases: ABC

Abstract base class for restraint builders.

Provides common functionality for accumulating restraint data during residue iteration and finalizing to sorted tensors. Child classes implement the specific logic for each restraint type.

Parameters:

verbose (int, default 0) – Verbosity level for debug output.

_indices

Accumulated index arrays.

Type:

list of np.ndarray

_references

Accumulated reference value arrays.

Type:

list of np.ndarray

_sigmas

Accumulated sigma arrays.

Type:

list of np.ndarray

_count

Total number of restraints added.

Type:

int

restraint_type: str = 'base'

atom_columns: List[str] = []

n_atoms: int = 0

__init__(verbose=0)

Initialize empty accumulator lists.

reset()

Clear all accumulated data.

abstractmethod process_residue(residue, cif_restraints)

Process a single residue, matching atoms to CIF restraints.

Parameters:

• residue (pd.DataFrame) – Residue atoms with ‘name’ and ‘index’ columns.

• cif_restraints (pd.DataFrame) – Restraints from CIF dictionary for this residue type.

Returns:

Number of restraints added.

Return type:

int

finalize(device, sort_indices=True, min_sigma=0.0001)

Convert accumulated data to sorted tensors.

Parameters:

• device (torch.device) – Target device for tensors.

• sort_indices (bool, default True) – Whether to sort by first atom index for memory locality.

• min_sigma (float, default 1e-4) – Minimum sigma value to avoid division by zero.

Returns:

Dictionary with ‘indices’, ‘references’, ‘sigmas’ tensors, or None if no restraints were accumulated.

Return type:

dict or None

property count: int

Return total number of restraints accumulated.

class torchref.restraints.BondRestraintBuilder(verbose=0)

Bases: RestraintBuilder

Builder for bond length restraints.

Bond restraints define expected distances between pairs of bonded atoms. Each restraint specifies: atom1, atom2, target_distance, sigma.

Examples

builder = BondRestraintBuilder()
for residue in residues:
    builder.process_residue(residue, cif_bonds)
result = builder.finalize(device)
print(result['indices'].shape)  # (n_bonds, 2)

restraint_type: str = 'bond'

atom_columns: List[str] = ['atom1', 'atom2']

n_atoms: int = 2

process_residue(residue, cif_restraints)

Process bond restraints for a single residue.

Parameters:

• residue (pd.DataFrame) – Residue atoms with ‘name’ and ‘index’ columns.

• cif_restraints (pd.DataFrame) – Bond restraints with ‘atom1’, ‘atom2’, ‘value’, ‘sigma’ columns.

Returns:

Number of bond restraints added.

Return type:

int

class torchref.restraints.AngleRestraintBuilder(verbose=0)

Bases: RestraintBuilder

Builder for angle restraints.

Angle restraints define expected angles between three atoms (1-2-3). The angle is measured at the central atom (atom2).

Examples

builder = AngleRestraintBuilder()
for residue in residues:
    builder.process_residue(residue, cif_angles)
result = builder.finalize(device)
print(result['indices'].shape)  # (n_angles, 3)

restraint_type: str = 'angle'

atom_columns: List[str] = ['atom1', 'atom2', 'atom3']

n_atoms: int = 3

process_residue(residue, cif_restraints)

Process angle restraints for a single residue.

Parameters:

• residue (pd.DataFrame) – Residue atoms with ‘name’ and ‘index’ columns.

• cif_restraints (pd.DataFrame) – Angle restraints with ‘atom1’, ‘atom2’, ‘atom3’, ‘value’, ‘sigma’ columns.

Returns:

Number of angle restraints added.

Return type:

int

class torchref.restraints.TorsionRestraintBuilder(verbose=0)

Bases: RestraintBuilder

Builder for torsion angle restraints.

Torsion restraints define expected dihedral angles between four atoms. Includes periodicity information for symmetric torsions (e.g., period=3 for methyl groups).

_periods

Accumulated periodicity arrays.

Type:

list of np.ndarray

Examples

builder = TorsionRestraintBuilder()
for residue in residues:
    builder.process_residue(residue, cif_torsions)
result = builder.finalize(device)
print(result['indices'].shape)  # (n_torsions, 4)
print(result['periods'].shape)  # (n_torsions,)

restraint_type: str = 'torsion'

atom_columns: List[str] = ['atom1', 'atom2', 'atom3', 'atom4']

n_atoms: int = 4

__init__(verbose=0)

Initialize with additional periods accumulator.

reset()

Clear all accumulated data including periods.

process_residue(residue, cif_restraints)

Process torsion restraints for a single residue.

Filters out torsions with sigma=0 (undefined/flexible torsions).

Parameters:

• residue (pd.DataFrame) – Residue atoms with ‘name’ and ‘index’ columns.

• cif_restraints (pd.DataFrame) – Torsion restraints with ‘atom1’-‘atom4’, ‘value’, ‘sigma’, and optionally ‘periodicity’ columns.

Returns:

Number of torsion restraints added.

Return type:

int

finalize(device, sort_indices=True, min_sigma=0.0001)

Convert accumulated data to sorted tensors, including periods.

Parameters:

• device (torch.device) – Target device for tensors.

• sort_indices (bool, default True) – Whether to sort by first atom index for memory locality.

• min_sigma (float, default 1e-4) – Minimum sigma value.

Returns:

Dictionary with ‘indices’, ‘references’, ‘sigmas’, ‘periods’ tensors.

Return type:

dict or None

class torchref.restraints.PlaneRestraintBuilder(verbose=0)

Bases: RestraintBuilder

Builder for planarity restraints.

Plane restraints define groups of atoms that should be coplanar. Unlike other restraints, planes have variable atom counts (3-10 atoms). Results are grouped by atom count for efficient tensor operations.

_planes_by_count

Dictionary mapping atom_count -> {‘indices’: […], ‘sigmas’: […]}.

Type:

dict

Examples

builder = PlaneRestraintBuilder()
for residue in residues:
    builder.process_residue(residue, cif_planes)
result = builder.finalize(device)
# Returns dict like {'4_atoms': {...}, '6_atoms': {...}}

restraint_type: str = 'plane'

atom_columns: List[str] = ['atom']

n_atoms: int = None

__init__(verbose=0)

Initialize with planes-by-count dictionary.

reset()

Clear all accumulated data.

process_residue(residue, cif_restraints)

Process plane restraints for a single residue.

Groups plane atoms by plane_id and stores separately by atom count.

Parameters:

• residue (pd.DataFrame) – Residue atoms with ‘name’ and ‘index’ columns.

• cif_restraints (pd.DataFrame) – Plane restraints with ‘plane_id’, ‘atom’, ‘sigma’ columns.

Returns:

Number of plane restraints added.

Return type:

int

finalize(device, sort_indices=True, min_sigma=0.0001)

Convert accumulated data to tensors grouped by atom count.

Parameters:

• device (torch.device) – Target device for tensors.

• sort_indices (bool, default True) – Whether to sort by first atom index.

• min_sigma (float, default 1e-4) – Minimum sigma value.

Returns:

Dictionary with keys like ‘4_atoms’, ‘6_atoms’, each containing ‘indices’ and ‘sigmas’ tensors.

Return type:

dict or None

class torchref.restraints.ChiralRestraintBuilder(ideal_volume=2.5, sigma=0.2, verbose=0)

Bases: RestraintBuilder

Builder for chiral volume restraints.

Chiral restraints define the expected handedness (R/S configuration) of tetrahedral centers. Each restraint specifies a center atom and three neighbors, with a signed ideal volume.

_ideal_volumes

Accumulated ideal volume arrays (signed based on chirality).

Type:

list of np.ndarray

ideal_volume

Magnitude of ideal chiral volume in ų.

Type:

float

sigma

Standard deviation for restraint in ų.

Type:

float

Examples

builder = ChiralRestraintBuilder(ideal_volume=2.5, sigma=0.2)
for residue in residues:
    builder.process_residue(residue, cif_chirals)
result = builder.finalize(device)
print(result['indices'].shape)  # (n_chirals, 4)
print(result['ideal_volumes'].shape)  # (n_chirals,)

restraint_type: str = 'chiral'

atom_columns: List[str] = ['atom_centre', 'atom1', 'atom2', 'atom3']

n_atoms: int = 4

__init__(ideal_volume=2.5, sigma=0.2, verbose=0)

Initialize with chiral volume parameters.

Parameters:

• ideal_volume (float, default 2.5) – Magnitude of ideal chiral volume in ų.

• sigma (float, default 0.2) – Standard deviation for restraint in ų.

• verbose (int, default 0) – Verbosity level.

reset()

Clear all accumulated data.

process_residue(residue, cif_restraints)

Process chiral restraints for a single residue.

Parameters:

• residue (pd.DataFrame) – Residue atoms with ‘name’ and ‘index’ columns.

• cif_restraints (pd.DataFrame) – Chiral restraints with ‘atom_centre’, ‘atom1’, ‘atom2’, ‘atom3’, ‘volume_sign’ columns.

Returns:

Number of chiral restraints added.

Return type:

int

finalize(device, sort_indices=True, min_sigma=0.0001)

Convert accumulated data to sorted tensors.

Parameters:

• device (torch.device) – Target device for tensors.

• sort_indices (bool, default True) – Whether to sort by first atom index.

• min_sigma (float, default 1e-4) – Minimum sigma value.

Returns:

Dictionary with ‘indices’, ‘ideal_volumes’, ‘sigmas’ tensors.

Return type:

dict or None

class torchref.restraints.InterResidueBondBuilder(verbose=0)

Bases: object

Builder for inter-residue bond restraints (peptide bonds, disulfide bonds).

Unlike intra-residue builders, this handles bonds between atoms in different residues. It processes residue pairs and link definitions.

Parameters:

verbose (int, default 0) – Verbosity level for debug output.

_indices

Accumulated bond index arrays.

Type:

list of np.ndarray

_references

Accumulated reference distance arrays.

Type:

list of np.ndarray

_sigmas

Accumulated sigma arrays.

Type:

list of np.ndarray

Examples

builder = InterResidueBondBuilder()
for res_i, res_next in iterator.get_consecutive_pairs():
    builder.process_peptide_bond(res_i, res_next, trans_link)
result = builder.finalize(device)

__init__(verbose=0)

Initialize empty accumulator lists.

reset()

Clear all accumulated data.

process_peptide_bond(residue_i, residue_next, link_bonds)

Process peptide bond restraints between consecutive residues.

Parameters:

• residue_i (pd.DataFrame) – First residue (C-terminal of bond).

• residue_next (pd.DataFrame) – Second residue (N-terminal of bond).

• link_bonds (pd.DataFrame) – Bond definitions from link dictionary with columns: ‘atom_1_comp_id’, ‘atom1’, ‘atom_2_comp_id’, ‘atom2’, ‘value’, ‘sigma’.

Returns:

Number of bond restraints added.

Return type:

int

process_disulfide_bond(sg1_idx, sg2_idx, bond_length, bond_sigma)

Process a single disulfide bond restraint.

Parameters:

• sg1_idx (int) – Index of first SG atom.

• sg2_idx (int) – Index of second SG atom.

• bond_length (float) – Target bond length in Å.

• bond_sigma (float) – Sigma for restraint in Å.

Returns:

Always returns 1.

Return type:

int

finalize(device, sort_indices=True, min_sigma=0.0001)

Convert accumulated data to sorted tensors.

Parameters:

• device (torch.device) – Target device for tensors.

• sort_indices (bool, default True) – Whether to sort by first atom index.

• min_sigma (float, default 1e-4) – Minimum sigma value.

Returns:

Dictionary with ‘indices’, ‘references’, ‘sigmas’ tensors.

Return type:

dict or None

property count: int

Return total number of restraints accumulated.

class torchref.restraints.InterResidueAngleBuilder(verbose=0)

Bases: object

Builder for inter-residue angle restraints (peptide angles).

Handles angles that span two consecutive residues.

Parameters:

verbose (int, default 0) – Verbosity level for debug output.

__init__(verbose=0)

Initialize empty accumulator lists.

reset()

Clear all accumulated data.

process_peptide_angles(residue_i, residue_next, link_angles)

Process peptide angle restraints between consecutive residues.

Parameters:

• residue_i (pd.DataFrame) – First residue.

• residue_next (pd.DataFrame) – Second residue.

• link_angles (pd.DataFrame) – Angle definitions from link dictionary.

Returns:

Number of angle restraints added.

Return type:

int

process_disulfide_angles(res1_atoms, res2_atoms, link_angles)

Process disulfide angle restraints.

Parameters:

• res1_atoms (pd.DataFrame) – First cysteine residue atoms.

• res2_atoms (pd.DataFrame) – Second cysteine residue atoms.

• link_angles (pd.DataFrame) – Angle definitions from disulfide link.

Returns:

Number of angle restraints added.

Return type:

int

finalize(device, sort_indices=True, min_sigma=0.0001)

Convert accumulated data to sorted tensors.

property count: int

Return total number of restraints accumulated.

class torchref.restraints.InterResidueTorsionBuilder(verbose=0)

Bases: object

Builder for inter-residue torsion restraints (phi, psi, omega).

Handles backbone torsion angles that span consecutive residues. Separates phi, psi, and omega angles for different treatment.

Parameters:

verbose (int, default 0) – Verbosity level for debug output.

__init__(verbose=0)

Initialize empty accumulator lists for each torsion type.

reset()

Clear all accumulated data.

process_peptide_torsions(residue_i, residue_next, link_torsions)

Process backbone torsion restraints between consecutive residues.

Parameters:

• residue_i (pd.DataFrame) – First residue.

• residue_next (pd.DataFrame) – Second residue.

• link_torsions (pd.DataFrame) – Torsion definitions from link dictionary with ‘id’ column indicating ‘phi’, ‘psi’, or ‘omega’.

Returns:

Number of (phi, psi, omega) torsions added.

Return type:

tuple of (int, int, int)

process_disulfide_torsions(res1_atoms, res2_atoms, link_torsions)

Process disulfide torsion restraints.

Parameters:

• res1_atoms (pd.DataFrame) – First cysteine residue atoms.

• res2_atoms (pd.DataFrame) – Second cysteine residue atoms.

• link_torsions (pd.DataFrame) – Torsion definitions from disulfide link.

Returns:

Number of torsion restraints added.

Return type:

int

finalize_phi(device, sort_indices=True)

Finalize phi angle indices and periods.

finalize_psi(device, sort_indices=True)

Finalize psi angle indices and periods.

finalize_omega(device, sort_indices=True)

Finalize omega angle restraints.

finalize_disulfide(device, sort_indices=True)

Finalize disulfide torsion restraints.

class torchref.restraints.InterResiduePlaneBuilder(verbose=0)

Bases: object

Builder for inter-residue plane restraints (peptide planes).

Handles planes that span consecutive residues (e.g., the planar peptide bond group).

Parameters:

verbose (int, default 0) – Verbosity level for debug output.

__init__(verbose=0)

Initialize with planes-by-count dictionary.

reset()

Clear all accumulated data.

process_peptide_planes(residue_i, residue_next, link_planes)

Process peptide plane restraints between consecutive residues.

Parameters:

• residue_i (pd.DataFrame) – First residue.

• residue_next (pd.DataFrame) – Second residue.

• link_planes (pd.DataFrame) – Plane definitions from link dictionary with ‘plane_id’, ‘atom_comp_id’, ‘atom’, ‘sigma’ columns.

Returns:

Number of plane restraints added.

Return type:

int

finalize(device, sort_indices=True, min_sigma=0.0001)

Convert accumulated data to tensors grouped by atom count.

property count: int

Return total number of restraints accumulated.

torchref.restraints.get_library_manager(verbose=1)

Get the global MonomerLibraryManager singleton.

Parameters:

verbose (int, optional) – Verbosity level (only used on first call).

Return type:

MonomerLibraryManager

Submodules

• torchref.restraints.builders module
  • Classes
  • ResidueIterator
    • ResidueIterator.pdb
    • ResidueIterator.groups
    • ResidueIterator.__init__()
    • ResidueIterator.__iter__()
    • ResidueIterator.__len__()
    • ResidueIterator.get_consecutive_pairs()
  • RestraintBuilder
    • RestraintBuilder._indices
    • RestraintBuilder._references
    • RestraintBuilder._sigmas
    • RestraintBuilder._count
    • RestraintBuilder.restraint_type
    • RestraintBuilder.atom_columns
    • RestraintBuilder.n_atoms
    • RestraintBuilder.__init__()
    • RestraintBuilder.reset()
    • RestraintBuilder.process_residue()
    • RestraintBuilder.finalize()
    • RestraintBuilder.count
  • BondRestraintBuilder
    • BondRestraintBuilder.restraint_type
    • BondRestraintBuilder.atom_columns
    • BondRestraintBuilder.n_atoms
    • BondRestraintBuilder.process_residue()
  • AngleRestraintBuilder
    • AngleRestraintBuilder.restraint_type
    • AngleRestraintBuilder.atom_columns
    • AngleRestraintBuilder.n_atoms
    • AngleRestraintBuilder.process_residue()
  • TorsionRestraintBuilder
    • TorsionRestraintBuilder._periods
    • TorsionRestraintBuilder.restraint_type
    • TorsionRestraintBuilder.atom_columns
    • TorsionRestraintBuilder.n_atoms
    • TorsionRestraintBuilder.__init__()
    • TorsionRestraintBuilder.reset()
    • TorsionRestraintBuilder.process_residue()
    • TorsionRestraintBuilder.finalize()
  • PlaneRestraintBuilder
    • PlaneRestraintBuilder._planes_by_count
    • PlaneRestraintBuilder.restraint_type
    • PlaneRestraintBuilder.atom_columns
    • PlaneRestraintBuilder.n_atoms
    • PlaneRestraintBuilder.__init__()
    • PlaneRestraintBuilder.reset()
    • PlaneRestraintBuilder.process_residue()
    • PlaneRestraintBuilder.finalize()
  • ChiralRestraintBuilder
    • ChiralRestraintBuilder._ideal_volumes
    • ChiralRestraintBuilder.ideal_volume
    • ChiralRestraintBuilder.sigma
    • ChiralRestraintBuilder.restraint_type
    • ChiralRestraintBuilder.atom_columns
    • ChiralRestraintBuilder.n_atoms
    • ChiralRestraintBuilder.__init__()
    • ChiralRestraintBuilder.reset()
    • ChiralRestraintBuilder.process_residue()
    • ChiralRestraintBuilder.finalize()
  • InterResidueBondBuilder
    • InterResidueBondBuilder._indices
    • InterResidueBondBuilder._references
    • InterResidueBondBuilder._sigmas
    • InterResidueBondBuilder.__init__()
    • InterResidueBondBuilder.reset()
    • InterResidueBondBuilder.process_peptide_bond()
    • InterResidueBondBuilder.process_disulfide_bond()
    • InterResidueBondBuilder.finalize()
    • InterResidueBondBuilder.count
  • InterResidueAngleBuilder
    • InterResidueAngleBuilder.__init__()
    • InterResidueAngleBuilder.reset()
    • InterResidueAngleBuilder.process_peptide_angles()
    • InterResidueAngleBuilder.process_disulfide_angles()
    • InterResidueAngleBuilder.finalize()
    • InterResidueAngleBuilder.count
  • InterResidueTorsionBuilder
    • InterResidueTorsionBuilder.__init__()
    • InterResidueTorsionBuilder.reset()
    • InterResidueTorsionBuilder.process_peptide_torsions()
    • InterResidueTorsionBuilder.process_disulfide_torsions()
    • InterResidueTorsionBuilder.finalize_phi()
    • InterResidueTorsionBuilder.finalize_psi()
    • InterResidueTorsionBuilder.finalize_omega()
    • InterResidueTorsionBuilder.finalize_disulfide()
  • InterResiduePlaneBuilder
    • InterResiduePlaneBuilder.__init__()
    • InterResiduePlaneBuilder.reset()
    • InterResiduePlaneBuilder.process_peptide_planes()
    • InterResiduePlaneBuilder.finalize()
    • InterResiduePlaneBuilder.count
• torchref.restraints.builders_fast module
  • PreprocessedPDB
    • PreprocessedPDB.__init__()
    • PreprocessedPDB.get_residue_data()
    • PreprocessedPDB.has_duplicate_atoms()
    • PreprocessedPDB.has_altlocs()
    • PreprocessedPDB.get_altloc_conformations()
  • PreprocessedCIF
    • PreprocessedCIF.__init__()
  • RestraintBuilder
    • RestraintBuilder.__init__()
    • RestraintBuilder.build()
  • BondRestraintBuilder
    • BondRestraintBuilder.build()
  • AngleRestraintBuilder
    • AngleRestraintBuilder.build()
  • TorsionRestraintBuilder
    • TorsionRestraintBuilder.build()
  • PlaneRestraintBuilder
    • PlaneRestraintBuilder.build()
  • ChiralRestraintBuilder
    • ChiralRestraintBuilder.build()
  • build_all_restraints()
  • PreprocessedLinkData
    • PreprocessedLinkData.__init__()
  • InterResidueBondBuilder
    • InterResidueBondBuilder.__init__()
    • InterResidueBondBuilder.reset()
    • InterResidueBondBuilder.process_disulfide_bond()
    • InterResidueBondBuilder.finalize()
    • InterResidueBondBuilder.count
    • InterResidueBondBuilder.build()
  • InterResidueAngleBuilder
    • InterResidueAngleBuilder.__init__()
    • InterResidueAngleBuilder.reset()
    • InterResidueAngleBuilder.process_disulfide_angles()
    • InterResidueAngleBuilder.finalize()
    • InterResidueAngleBuilder.count
    • InterResidueAngleBuilder.build()
  • InterResidueTorsionBuilder
    • InterResidueTorsionBuilder.__init__()
    • InterResidueTorsionBuilder.reset()
    • InterResidueTorsionBuilder.process_disulfide_torsions()
    • InterResidueTorsionBuilder.finalize_disulfide()
    • InterResidueTorsionBuilder.disulfide_count
    • InterResidueTorsionBuilder.build()
  • InterResiduePlaneBuilder
    • InterResiduePlaneBuilder.__init__()
    • InterResiduePlaneBuilder.build()
  • ResidueIterator
    • ResidueIterator.__init__()
    • ResidueIterator.__iter__()
    • ResidueIterator.__len__()
    • ResidueIterator.get_consecutive_pairs()
• torchref.restraints.builders_numba module
  • find_atom_index()
  • match_bonds_numba()
  • match_angles_numba()
  • match_torsions_numba()
  • match_chirals_numba()
  • PreprocessedPDB
    • PreprocessedPDB.__init__()
    • PreprocessedPDB.get_residue_atoms()
  • PreprocessedCIF
    • PreprocessedCIF.__init__()
  • FastResidueIterator
    • FastResidueIterator.__init__()
    • FastResidueIterator.__iter__()
  • FastBondBuilder
    • FastBondBuilder.__init__()
    • FastBondBuilder.process_residue()
    • FastBondBuilder.finalize()
    • FastBondBuilder.reset()
  • FastAngleBuilder
    • FastAngleBuilder.__init__()
    • FastAngleBuilder.process_residue()
    • FastAngleBuilder.finalize()
    • FastAngleBuilder.reset()
  • FastTorsionBuilder
    • FastTorsionBuilder.__init__()
    • FastTorsionBuilder.process_residue()
    • FastTorsionBuilder.finalize()
    • FastTorsionBuilder.reset()
  • FastPlaneBuilder
    • FastPlaneBuilder.__init__()
    • FastPlaneBuilder.process_residue()
    • FastPlaneBuilder.finalize()
    • FastPlaneBuilder.reset()
  • FastChiralBuilder
    • FastChiralBuilder.__init__()
    • FastChiralBuilder.process_residue()
    • FastChiralBuilder.finalize()
    • FastChiralBuilder.reset()
  • build_all_restraints_fast()
• torchref.restraints.hydrogen_topology module
  • HydrogenTopology
    • HydrogenTopology.h_parent_idx
    • HydrogenTopology.h_bond_length
    • HydrogenTopology.h_vdw_radius
    • HydrogenTopology.h_placement_type
    • HydrogenTopology.h_slot_in_parent
    • HydrogenTopology.parent_neighbor_idx
    • HydrogenTopology.parent_neighbor_count
    • HydrogenTopology.h_chainid_enc
    • HydrogenTopology.h_resseq
    • HydrogenTopology.__init__()
    • HydrogenTopology.n_hydrogens
    • HydrogenTopology.has_candidates
  • build_hydrogen_topology()
  • place_riding_hydrogens()
  • build_h_candidate_pairs()
• torchref.restraints.library module
  • MonomerLibraryManager
    • MonomerLibraryManager.__init__()
    • MonomerLibraryManager.get_cif_file()
    • MonomerLibraryManager.get_link_definitions_path()
    • MonomerLibraryManager.monomer_dir
  • get_library_manager()
• torchref.restraints.neighbor_search module
  • prefilter_symop_offsets()
  • assign_to_grid()
  • build_cell_list()
  • find_pairs_periodic_grid()
  • find_pairs_periodic_grid_v2()
  • exclusion_set_to_hash()
  • filter_pairs()
  • build_vdw_restraints_gpu()
  • find_h_vdw_pairs_gpu()
• torchref.restraints.ramachandran module
  • Surface types
  • load_nll_surfaces()
  • classify_residue()
• torchref.restraints.restraints module
  • RestraintsNew
    • RestraintsNew.pdb
    • RestraintsNew.xyz_fn
    • RestraintsNew.adp_fn
    • RestraintsNew.vdw_radii_fn
    • RestraintsNew.cif_dict
    • RestraintsNew.restraints
    • RestraintsNew.__init__()
    • RestraintsNew.xyz()
    • RestraintsNew.adp()
    • RestraintsNew.get_vdw_radii()
    • RestraintsNew.restraints
    • RestraintsNew.expand_altloc()
    • RestraintsNew.build_restraints()
    • RestraintsNew.h_topo
    • RestraintsNew.h_excl_hash
    • RestraintsNew.rebuild_vdw_restraints()
    • RestraintsNew.summary()
    • RestraintsNew.__repr__()
    • RestraintsNew.bond_lengths()
    • RestraintsNew.copy()
    • RestraintsNew.bond_deviations()
    • RestraintsNew.nll_bonds()
    • RestraintsNew.angles()
    • RestraintsNew.angle_deviations()
    • RestraintsNew.nll_angles()
    • RestraintsNew.cat_dict()
    • RestraintsNew.torsions()
    • RestraintsNew.torsion_deviations()
    • RestraintsNew.torsion_deviations_with_sigmas()
    • RestraintsNew.nll_torsions()
    • RestraintsNew.nll_planes()
    • RestraintsNew.nll_vdw()
    • RestraintsNew.adp_b_differences()
    • RestraintsNew.adp_similarity_loss()
• torchref.restraints.restraints_helper module
  • validate_restraint_data()
  • read_cif()
  • split_respecting_quotes()
  • find_cif_file_in_library()
  • read_link_definitions()
  • build_restraints_bondlength()
  • build_restraints_angles()
  • build_restraints_torsion()
  • build_restraints_planes()
  • build_restraints()
  • calculate_restraints_bondlength()
  • calculate_restraints_angles()
  • calculate_restraints_torsion()
  • calculate_restraints_all()
  • read_for_component()
  • read_comp_list()