###########################################################################
# 01-07-09
# Melanie Martin
# counts.py 
#
# USAGE:  
#  python counts.py <input file> <team> <mission> <output file>
# 
# to count words, lines, and occurrences of & in a file
# for use on saso corpus
##########################################################################


import sys
import string
import re
import os


# Open input file
file = sys.argv[1] 
infile = open(file,'r')

#team = '01'
fteam = sys.argv[2]

#mission = '04'
fmission = sys.argv[3]

# Open output file
filename = sys.argv[4] 
outfile = open(filename,'w')


#---------------------------------------------------

line = infile.readline()
wordcount = 0
linecounter = 0
ambcounter = 0

while line != "":
	
	words = line.split()
	wordcount = wordcount + len(words)

	for word in words:
		if word == '&':
			ambcounter = ambcounter + 1

	linecounter = linecounter + 1
	line = infile.readline()

#output is team,mission,&,words,lines\n
outfile.write("%s, %s, %s, %s, %s\n" % (fteam, fmission, ambcounter, wordcount, linecounter) )
#outfile.write("lines = %s, words = %s, & = %s\n" % (linecounter,wordcount, ambcounter) )
infile.close()
outfile.close()