chroma calculator

#!/usr/bin/python

def calc(pstr=1, hpmod=0, armod=0, qt=False, nrgmod=0):

    hp = float(300 + (100 * hpmod))
    hp_final = hp
    if qt:
        nrg = 225 + (150 * nrgmod)
        qtehp = nrg * 2.4
        hp_final += qtehp

    armor_before = 350 * (1 + armod)
    armor_ele = 350 * (1 + armod + 1.5 * pstr)
    armor_both = armor_ele * 3.5 * pstr

    ehp_multi = 1 + (armor_both / 300)
    ehp_final = hp_final * ehp_multi
    print 'HP: %s' % hp
    print 'Armor: %s' % armor_both
    print 'EHP Mult: %s' % ehp_multi
    print 'Final EHP: %s' % ehp_final
    return ehp_final