#############################################################
#Dice_roller.py
#
#Purpose:  A random number generation program that simulates
#  various dice rolls.
#Author:  Cody Jackson
#Date:  4/11/06
#
#Copyright 2006 Cody Jackson
#This program is free software; you can redistribute it and/or modify
#it under the terms of the GNU General Public License as published by
#the Free Software Foundation; either version 2 of the License, or
#(at your option) any later version.
#
#This program is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#GNU General Public License for more details.
#
#You should have received a copy of the GNU General Public License
#along with this program; if not, write to the Free Software
#Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
#############################################################

import random #randint

def randomNumGen(choice):
    """Get a random number to simulate a d6, d10, or d100 roll."""
    
    if choice == 1: #d6 roll
    	die = random.randint(1, 6)
    elif choice == 2: #d10 roll
        die = random.randint(1, 10)
    elif choice == 3: #d100 roll
        die = random.randint(1, 100)
    else:   #simple error message
        print "Shouldn't be here.  Invalid choice"
    return die

def multiDie(dice_number, die_type):
    """Add die rolls together, e.g. 2d6, 4d10, etc."""
    
#---Initialize variables    
    final_roll = 0
    val = 0
    
    while val < dice_number:
        final_roll += randomNumGen(die_type)
        val += 1
    print final_roll

def test():
    """Test criteria to show script works."""
    
    print "1d6 = ",    
    multiDie(1,1)   #1d6
    print "\n2d6 = ",
    multiDie(2,1)   #2d6
    print "\n3d6 = ",
    multiDie(3,1)   #3d6
    print "\n4d6 = ",
    multiDie(4,1)   #4d6
    print "\n1d10 = ",
    multiDie(1,2)   #1d10
    print "\n2d10 = ",
    multiDie(2,2)   #2d10
    print "\n3d10 = ",
    multiDie(2,2)   #3d10
    print "\n1d100 = ",
    multiDie(1,3)   #d100
    
if __name__ == "__main__":  #run test() if calling as a separate program
    test()