The only problem with your theory is that Zimmerman did not come after Martin with a gun. (Hence no arrest) Zimmerman was walking back to his vehicle, when Martin jumped him. The gun was exposed in the struggle and Martin tried to take it. (Probably not to admire the craftsmanship). At that point, Martin was the assailant, and Martin was the one going for a gun.
A jury will have to decide that.