Blog

#!/usr/bin/python
import re
import sys
import matplotlib.pyplot as plt

noPVP = 1
if len(sys.argv) > 1:
	if sys.argv[1] == "-pvp":
		noPVP = 0


def main():
#VARIABLES
	inputFile = "combat.log"
	outputFile = "output.tab"
	playerName = "Xiluthus"
	playerServer = "Antonidas"
	logEvents = ["SPELL_DAMAGE", "SPELL_PERIODIC_DAMAGE"]
	targetList = []
	targetDictionary = {}
	spellList = {}
	for l in logEvents:
		spellList[l] = []
	logData = {}
	
	playerID = playerName+"-"+playerServer
	
	
	#Open input
	try: 
		inFH = open(inputFile, "rt")
	except IOError:
		print ("\n\n\tERROR (input): Could not open/find",inputFile,"\n")
	
	print("\n\nParsing file ",inputFile,"...",sep="",end="")
	
	#loop through file and do the parsing
	for line in inFH:
		if re.match("^\d+",line) and (re.search(logEvents[0],line) or re.search(logEvents[1],line)): 
	
			splitLine = line.split(",")
			# splitLine now contains:
			# 0: timestamp + event name
			# 2: character name 
			# 5: target id 
			# 6: target name 
			# 10: spell name
			# 29: damage
	
			if splitLine[2] == "\""+playerID+"\"":
				# dirty hack: split at double whitespace
				timeStamp, eventName = splitLine[0].split("  ")
				
				# not really necessary, but just for easy reading
				targetID = splitLine[5]
				targetName = splitLine[6]
				spellName = splitLine[10]
				damageValue = splitLine[29]

				# skip player targets
				if noPVP == 1 and (re.match("Player",targetID) or re.match("Pet",targetID)):
					continue

				# add to dictionary
				if eventName not in logData.keys():
					logData[eventName] = {}
				
				if targetID not in logData[eventName].keys():
					logData[eventName][targetID] = {}
				
				if spellName not in logData[eventName][targetID].keys():
					logData[eventName][targetID][spellName] = []
				
				logData[eventName][targetID][spellName].append([timeStamp,damageValue])
		
				# filter some info right here
				# get unique spells and targets
				if spellName not in spellList[eventName]:
					spellList[eventName].append(spellName)
				if targetID not in targetList:
					targetList.append(targetID)	
				# create target ID<->name dictionary
				if targetID not in targetDictionary.keys():
					targetDictionary[targetID] = targetName
	
	
	inFH.close()
	
	print("done\n")

	# 1. get average dmg for all spells
	for e in logEvents:
		print ("Evaluation for event "+e+":\n")
		
		targetLine = ""
		targetNameLength = getTargetNameLength(targetList,targetDictionary)

		for t in targetList:
			targetLine = targetLine+" "+targetDictionary[t].center(targetNameLength)+" |"

		spellNameLength = getSpellnameLength(spellList[e])
		space = " " * spellNameLength		

		print(space+" |"+targetLine)

		for s in spellList[e]:
			outLine = s.ljust(spellNameLength)+" |"
			for t in targetList:
				if t in logData[e].keys():
					if s in logData[e][t].keys():
						outLine = outLine+" "+str(getAverageDmg(logData[e][t][s])).center(targetNameLength)+" |"
					else:
						outLine = outLine+" "+str(0).center(targetNameLength)+" |"
				else:
					outLine = outLine+" "+str(0).center(targetNameLength)+" |"
			print(outLine)

		print()

	# 2. plot damage over time for one target
	print("Select target for total damage over time plot:")
	i = 1
	for t in targetList:
		print("\t"+str(i)+".",targetDictionary[t])
		i += 1

	check = 0
	while check == 0:
		selection = input("Which target number?: ")
		if re.match("^\d+$",selection) and int(selection) < len(targetList)+1 and int(selection) > 0:
			check = 1
		

	selectedTarget = targetList[int(selection)-1]

	damageTuple = []
	for e in logEvents:
		for s in spellList[e]:
			if t in logData[e].keys():
				if s in logData[e][selectedTarget].keys():
					for i in logData[e][selectedTarget][s]:
						# create list of tuples
						damageTuple.append((i[0],i[1]))
	# sort list by first value of tuple
	damageTuple.sort(key=lambda pair:pair[0])

	totalDamage = 0
	totalTime = 0
	prevTime = 0
	i = 0
	for entry in damageTuple:
		# unpack tuple
		(currentTime, currentDamage) = entry
		totalDamage += int(currentDamage)
		if prevTime != 0:
			totalTime += convertToSeconds(prevTime,currentTime)
		prevTime = currentTime
		# update with new value
		damageTuple[i] = (totalTime,totalDamage)
		i += 1
		
	plt.title(targetDictionary[selectedTarget])
	plt.xlabel("Time / Seconds")
	plt.ylabel("Total Damage")
	plt.plot(*zip(*damageTuple))
	plt.show()		

def convertToSeconds(prev,curr):

	(date,time) = re.split("\s+",prev)		
	(h,m,sp) = time.split(":")
	sp = float(sp) + int(h)*3600
	sp = sp + int(m)*60
	
	(date,time) = re.split("\s+",curr)		
	(h,m,sc) = time.split(":")
	sc = float(sc) + int(h)*3600
	sc = sc + int(m)*60
	
	return(sc-sp)

def getTargetNameLength(data,dict):

	length = 0

	for d in range(0,len(data)):
		if len(dict[data[d]]) > length:
			length = len(dict[data[d]])

	return length

def getSpellnameLength(data):

	length = 0

	for d in range(0,len(data)):
		if len(data[d]) > length:
			length = len(data[d])

	return length


def getAverageDmg(data):

	totalDamage = 0
	
	for i in range(0,len(data)):
		timeStamp,damageValue = data[i]
		totalDamage += int(damageValue)

	return(round(totalDamage/len(data),2))

if __name__ == "__main__":
	main()