You could do what I did, go back to work, but do the job slower, take more brain breaks, only focus on one thing at a time. A good boss will understand and know you're trying to get back in the groove. It will take a while, and they are right about months before the majority of the headaches are gone, but they still can come back, so it's a matter of learning to cope, as well.

Can you ask your doctor if asprin which is a blood thinner, might help the headaches? I took some heavy duty pain meds when they found out about the tooth and it made working easier. (Though with an addictive personality I don't know if that's a good thing.)