AITA for leaving my friend's house cause he was being cause snappy at me cause his grandma just died

So last night, my friends grandma died. He didn't know her very well, but he's having a hard time with it because his dad's been gone for a couple weeks being with her in her last days. My friend doesn't have a good relationship with his mom and he's been super cranky and irritated with his mom and been complaining about her for the last 2 weeks cause he's been stuck at home with her without his dad. So he's just irritable all around.

So anyway, I went over there to hangout with him. He was obviously not in a good mood, understandably, I didn't expect him to be. But, I felt like he was being rude and aggressive in a way.

-Telling me not to turn up the TV volume when I grabbed the remote even though I could BARELY HEAR IT. he said it's too loud.

-Saying me 'its people like you' bla bla bla (something about me and star wars???) because he was pissed about the Star Wars movie we were watching not being very good. So I said, I'm not into it either, should we change it? And he said no and looked so irritated at me for asking. Even though he was complaining about it for 10 minutes straight.

-i asked if he was okay if I moved a chair to kick my feet up on since his recliners broken and he was like "the coffee table is the exact same height as the chair". Even though it wasn't, and the coffee table was all the way over where he was with his legs kicked up on it. So it made more sense to me to just move a chair instead of moving over the coffee table.

Anyway, I ended up just saying "hey, I know you're feeling irritable for obvious reasons, and I kind of am too today, so maybe today's not the best day to hangout and we can get together again in a couple days". And he just said FINE and pointed and the door. So I just grabbed my bag and started heading to the door. He said one more thing and I can't even remember what it was cause I was just seeing red and I just said "you're being an asshole to me, so I'm leaving and I'll talk to you later" and I kinda slammed the door and left.

Do I need to apologize? I feel like an asshole. But I feel like there's no excuse to take it out on me.

So I went over there