Add an axis mask to the blur shader