# Count the number of PDB entries found for each pair of proteins.

import json
with open('pdb_dimers.json', 'r') as f:
    pdb_dimers = json.load(f)

print('contact pdbs', sum(len(pdb_chains) for seq1, seq2, pdb_chains in pdb_dimers))

pdb_ids = set()
for seq1, seq2, pdb_chains in pdb_dimers:
    for pdb_id, chains1, chains2 in pdb_chains:
        pdb_ids.add(pdb_id)
print('unique contact pdbs', len(pdb_ids)) 

for seq1, seq2, pdb_chains in pdb_dimers:
    print(seq1, seq2, len(pdb_chains))