X-Git-Url: https://harishankar.org/repos/?p=butaba-adventures.git;a=blobdiff_plain;f=utility.py;h=93dac190dba2d7a9d99fa1a10f1e392eadcd17c7;hp=05faab5b40a533e8b80855353f5364a0fd82f5b0;hb=8d8c32fdbd61956ccc4792524b621524d83b0467;hpb=4d50728c1454554fef2ad4841345d0c28101702d

diff --git a/utility.py b/utility.py
index 05faab5..93dac19 100644
--- a/utility.py
+++ b/utility.py
@@ -2,6 +2,67 @@
 import pygame
 import sys
 import os.path
+import xml.etree.cElementTree as ET
+import textwrap
+
+# function to run through an interactive dialogue and return the response ID
+def dialogue_play (screen, bgscreen, npc, responder,
+					marginleft=0, margintop=0, marginright=0, marginbottom=0):
+
+	# first check if NPC has a dialogue at all
+	if len (npc.dialogues) == 0 or npc.currentdialog >= len (npc.dialogues):
+		return None
+
+	# get the conversation as a dictionary
+	convtree = xml_to_dict (npc.dialogues[npc.currentdialog])
+
+	scrwidth = screen.get_width ()
+	scrheight = screen.get_height ()
+	bgwidth = bgscreen.get_width ()
+	bgheight = bgscreen.get_height ()
+	leftedge = scrwidth/2 - bgwidth/2
+	topedge = scrheight/2 - bgheight/2
+
+	# now initiate the conversation
+	dlg_id = "1"
+	while 1:
+		screen.blit (bgscreen, (leftedge, topedge))
+		if npc.portrait is not None:
+			screen.blit (npc.portrait, (leftedge, topedge))
+
+		# get the lines to display wrapped using textwrap module
+		lines = []
+		textnpc = textwrap.wrap (convtree[dlg_id][0], 40)
+		for text in textnpc:
+			lines.append ((10, 0, 0, 0, text))
+
+		put_lines (screen, lines,
+						pygame.Rect(leftedge+marginleft, topedge+margintop,
+							bgwidth-marginleft-marginright, bgheight-margintop-marginbottom))
+		pygame.display.update ()
+
+		for event in pygame.event.get ():
+			if event.type == pygame.QUIT:
+				sys.exit (0)
+
+# function to parse a conversation XML into a python dictionary
+def xml_to_dict (file):
+	# parse the dialogue XML file
+	dlgtree = ET.parse (file)
+	# get the root element
+	conversation = dlgtree.getroot ()
+
+	# build the conversation tree as a dictionary
+	convtree = dict ()
+	for dlg in conversation.findall ("dialogue"):
+		id = dlg.get ("id")
+		convtree[id] = []
+		convtree[id].append (dlg.find ("speech").text)
+		for resp in dlg.findall ("response"):
+			convtree[id].append ((resp.text, resp.get ("id"), resp.get ("nextdialogue")))
+
+	return convtree
+
 
 # function to draw text on surface
 def put_text (surface,  x, y, size, (r,g,b), text):
@@ -9,8 +70,9 @@ def put_text (surface,  x, y, size, (r,g,b), text):
 	textsurf = pygame.font.Font (harisfont, size).render (text, True, pygame.Color (r,g,b))
 	surface.blit (textsurf, (x, y))
 
-# function to draw several lines of text, centered horizontally and vertically on surface
-def put_lines (surface, text_lines):
+# function to draw several lines of text, centered horizontally and vertically on screen
+# or drawn centered on a rectangle
+def put_lines (surface, text_lines, rect=None):
 	textsurfs = []
 	height = 0
 	harisfont = os.path.join ("font", "harisgamefont.ttf")
@@ -21,12 +83,23 @@ def put_lines (surface, text_lines):
 		height = height + s.get_height()*1.5
 		textsurfs.append (s)
 
-	scrwidth = surface.get_width ()
-	scrheight = surface.get_height ()
 	i = 0
-	for s in textsurfs:
-		surface.blit (s, (scrwidth/2 - s.get_width()/2, scrheight/2 - height/2+ i* (s.get_height()*1.5)))
-		i += 1
+	# if no rectangle specified, center in screen
+	if rect is None:
+		scrwidth = surface.get_width ()
+		scrheight = surface.get_height ()
+
+		for s in textsurfs:
+			surface.blit (s, (scrwidth/2 - s.get_width()/2, scrheight/2 - height/2+ i* (s.get_height()*1.5)))
+			i += 1
+	# center on specified rectangular region
+	else:
+		midx = (rect.left + rect.right)/2
+		midy = (rect.top + rect.bottom)/2
+		num = len (textsurfs) / 2
+		for s in textsurfs:
+			surface.blit (s, (midx - s.get_width()/2, midy - (num-i) * s.get_height() * 3/2))
+			i += 1
 
 # function to ask a question and return answer
 def ask_question (surface, question, answers, bgscreen):
@@ -48,7 +121,9 @@ def ask_question (surface, question, answers, bgscreen):
 
 		surface.blit (bgscreen, (surface.get_width()/2 - bgscreen.get_width()/2,
 									surface.get_height()/2 - bgscreen.get_height()/2))
-		put_lines (surface, textarray)
+		put_lines (surface, textarray, pygame.Rect (surface.get_width()/2 - bgscreen.get_width()/2,
+												surface.get_height()/2 - bgscreen.get_height()/2,
+												bgscreen.get_width(), bgscreen.get_height()))
 
 		pygame.display.update ()