root/hodgestar/PythonCode/DmTools/kintani-skills.py

#!/usr/bin/env python

"""Script for calculating XP spent on skills for Adrianna's
   Kintani Valley campaign. Skills buying uses a two times
   triangular XP cost structure:
   
   Rank 1: 2 xp
   Rank 2: + 4 xp
   Rank 3: + 6 xp
   Rank 4: + 8 xp
   Rank 5: + 10 xp
   Rank 6: + 12 xp
   Rank 7: + 14 xp
   
   In addition, emphasis must be bought at least every odd rank.
   Emphasis are also bought using a two times triangular XP cost:
   
   First emphasis: 2 xp
   Second emphasis: +4 xp
   Third emphasis: +6 xp
   """

import re

def twice_triangular(n):
    """Calculate twice triangular XP cost for rank n."""
    return n*(n+1)

def skill_cost(ranks, emphasises):
    """Total XP cost for a given skill rank and number of emphasises."""
    return twice_triangular(ranks) + twice_triangular(emphasises)

def zim_skill_parser(fIn):
    """Parse a set of skills and emphasis from a ZimWiki file.

       Results are yielded as (skill name, rank, list of emphases) tuples.
       In the list of emphases names, free emphases are suffixed with an *.
       The characters school skill is also suffixed with an *.
       """
    heading_re = re.compile(r'\s*([=]+)(?P<sec>.*)\1\s*')
    skill_re = re.compile(r'(?P<linestart>[*/\s+]*)'
        r'(?P<name>[^*/.]+)[*/\s]+'
        r'(?P<rank>\d+)[*/\s]+'
        r'\((?P<emphs>.*)\)\s*')

    def process_emph(emph):
        """Convert free emph to emph* notation."""
        emph = emph.strip()
        if emph.startswith('//'):
            emph = emph.strip('/') + '*'
        return emph

    in_skills = False

    for line in fIn:
        match = heading_re.match(line)
        if match:
            sec = match.group('sec').strip()
            if sec.lower() in ['skills', 'skill']:
                in_skills = True
            else:
                in_skills = False
            continue

        if not in_skills:
            continue

        match = skill_re.match(line)
        if match:
            d = match.groupdict()
            school_skill = d['linestart'].endswith('//')
            name = d['name'].strip() + (school_skill and '*' or '')
            rank = int(d['rank'])
            emphs = [process_emph(x) for x in d['emphs'].split(',')]
            # remove skill rank description
            emphs = emphs[1:]
            yield name, rank, emphs

def main(fIn):
    total = 0

    print "[Skill]", "[Rank]", "[Emphases]", "[XP Cost]"
    print "============================================"

    for skill, rank, emphs in zim_skill_parser(fIn):
        non_free_emphs = [x for x in emphs if not x.endswith('*')]
        cost = skill_cost(rank, len(non_free_emphs))
        total += cost
        print skill, rank, "(" + ", ".join(emphs) + ")", cost

    print "\nTotal XP:", total

if __name__ == "__main__":
    import sys
    fIn = file(sys.argv[1], "rb")
    main(fIn)