Сorrect display of damage in the unit's weapon

Hello! I have 3 types of units: Organic, Mechanical, and Energy. There are also 4 types of armor: Light, Medium, Heavy, and Fortified. I used to do this through a set of effects where I put damage effects for each unit and armor association. That is, the damage for organic light and organic medium units was different. I had a problem that in the weapon field where I needed to display the damage effect, I couldn’t do anything, since there were several damage effects. Please tell me, with the release of patch 5.0, how can I make sure that different types of damage for different types of armor can be displayed in the icon of the unit’s weapons?