There are two pre-cooked macros:

omac/mask_atom.omac to add an area around an atom to the mask omac/unmask_atom.omac to remove such an area from the mask