The Spheres of Power system might work for this, it separates the magic into different schools of magic and it should be able to replicate your different forms of magic through it's Spellcasting Traditions system.

The only downsides I see is that Spheres of Power is not free, and the elemental schools (such as Air) are separated into Destruction, Nature, and Weather.