Imagine

Does anyone know if Imagine was written about the breakup with Mac (before he passed) or is it about his death?