this is indexed with the wrong semantics. `dis_indices` are indices into the array _after_ slicing out the atom_indices.