The node you are looking for is at
Manually Categorising Illumination Passes
.