I have the same thing, I can easily break out of code and get right back in to where I was, I can even pretty much do it with hours break, unless things are going very badly with the code in which case I cannot handle being interrupted.

I wonder if it's my ADHD that does it actually, I can handle quick changes of attention because after all that is what I do, but if things are a problem I need all effort to not switch attention and lose what I am doing.