Yes I can relate. I've heard that psychotic episodes can be like having a concussion. It took me a few months after my break for me to feel like my brain was back at a normal functioning level. Medication can cause cognitive issues too.
Take it easy, continue journalling, and maybe take some omega 3 fatty acids. You will get it back. I remember my pdoc telling me that a break/episode a major medical emergency so treat is as such, and remember it will take a while to recover from. I felt like I was hit by a truck so that made me feel a little better.