Hi Kedar,
You can try the CIF user exit for stock, based on the criteria you can bring these stocks with different ATP categories. Or else if you maintain the restricted batch at different storage location then based on storage location you can bring the stock to APO with different ATP category and you can decide whether the batch can be used for planning or not.
BR
Thamizh