It’s not just down to the company when you go back to work after such an awful incident. It’s down to you and your Doctor too I would have thought.
Obviously, the size and quality of the company would come into consideration when making any decision about returning to work, and how much time you had off. If you had more than two years service I would have thought they would have no option but to support you in the short to medium term, and I mean weeks or months there, not a few nights.
As an aside, I hope you feel better in time - but I’m guessing you realise now it will always be with you.
MickM