home *** CD-ROM | disk | FTP | other *** search
/ GameStar 2006 January / Gamestar_80_2006-01_dvd.iso / Dema / Civilization4 / data1.cab / Civ4DemoComponent / Assets / Python / CvDiplomacy.py < prev    next >
Encoding:
Python Source  |  2005-11-09  |  30.3 KB  |  697 lines
  1. ## Sid Meier's Civilization 4
  2. ## Copyright Firaxis Games 2005
  3. from CvPythonExtensions import *
  4. import CvUtil
  5. import PyHelpers
  6.  
  7. PyPlayer = PyHelpers.PyPlayer
  8.  
  9. DebugLogging = False
  10.  
  11. gc = CyGlobalContext()
  12.  
  13. class CvDiplomacy:
  14.     "Code used by Civ Diplomacy interface"
  15.     
  16.     def __init__(self):
  17.         "constructor - set up class vars, AI and User strings"
  18.         if DebugLogging:
  19.             print "Launching Diplomacy"
  20.         
  21.         self.iLastResponseID = -1
  22.  
  23.         self.diploScreen = CyDiplomacy()
  24.             
  25.     def setDebugLogging(self, bDebugLogging):
  26.         global DebugLogging
  27.         DebugLogging = bDebugLogging
  28.     
  29.     def determineResponses (self, eComment):
  30.         "Will determine the user responses given an AI comment"
  31.         if DebugLogging:
  32.             print "CvDiplomacy.determineResponses: %s" %(eComment,)
  33.         
  34.         # Eliminate previous comments
  35.         self.diploScreen.clearUserComments()
  36.         
  37.         # If the AI is declaring war
  38.         if (self.isComment(eComment, "AI_DIPLOCOMMENT_DECLARE_WAR") ):
  39.  
  40.             # We respond to their declaration
  41.             self.addUserComment("USER_DIPLOCOMMENT_WAR_RESPONSE", -1, -1)
  42.             self.diploScreen.endTrade()
  43.  
  44.         # If this is the first time we are being contacted by the AI
  45.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_FIRST_CONTACT") ):
  46.  
  47.             # if you are on different teams and NOT at war, give the user the option to declare war
  48.             if (gc.getTeam(gc.getGame().getActiveTeam()).canDeclareWar(gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam())):
  49.                 self.addUserComment("USER_DIPLOCOMMENT_WAR", -1, -1)
  50.  
  51.             # We say hi and begin our peace
  52.             self.addUserComment("USER_DIPLOCOMMENT_PEACE", -1, -1)
  53.  
  54.             self.diploScreen.endTrade()
  55.  
  56.         # The AI refuses to talk
  57.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_REFUSE_TO_TALK") ):
  58.     
  59.             # Give the option to exit
  60.             self.addUserComment("USER_DIPLOCOMMENT_EXIT", -1, -1)
  61.             self.diploScreen.endTrade();
  62.  
  63.         # If the AI is offering a city oo
  64.         # If the AI is giving help
  65.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_OFFER_CITY") or
  66.                     self.isComment(eComment, "AI_DIPLOCOMMENT_GIVE_HELP")):
  67.  
  68.             # We can accept their offer
  69.             self.addUserComment("USER_DIPLOCOMMENT_ACCEPT_OFFER", -1, -1)
  70.             # Or reject it...
  71.             self.addUserComment("USER_DIPLOCOMMENT_REJECT_OFFER", -1, -1)
  72.  
  73.         # If the AI is offering a deal
  74.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_OFFER_PEACE") or
  75.                     self.isComment(eComment, "AI_DIPLOCOMMENT_OFFER_DEAL")):
  76.  
  77.             # We can accept their offer
  78.             self.addUserComment("USER_DIPLOCOMMENT_ACCEPT_OFFER", -1, -1)
  79.             # Or we can try to negotiate
  80.             self.addUserComment("USER_DIPLOCOMMENT_RENEGOTIATE", -1, -1)
  81.             # Or reject it...
  82.             self.addUserComment("USER_DIPLOCOMMENT_REJECT_OFFER", -1, -1)
  83.  
  84.         # If the AI is cancelling a deal
  85.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_CANCEL_DEAL")):
  86.  
  87.             # We can try to renegotiate
  88.             self.addUserComment("USER_DIPLOCOMMENT_RENEGOTIATE", -1, -1)
  89.             # Or just exit...
  90.             self.addUserComment("USER_DIPLOCOMMENT_NO_RENEGOTIATE", -1, -1)
  91.  
  92.         # If the AI is demanding tribute
  93.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_ASK_FOR_HELP")):
  94.  
  95.             # We can give them help
  96.             self.addUserComment("USER_DIPLOCOMMENT_GIVE_HELP", -1, -1)
  97.             # Or refuse...
  98.             self.addUserComment("USER_DIPLOCOMMENT_REFUSE_HELP", -1, -1)
  99.  
  100.         # If the AI is demanding tribute
  101.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_DEMAND_TRIBUTE")):
  102.     
  103.             # We can accept their demands
  104.             self.addUserComment("USER_DIPLOCOMMENT_ACCEPT_DEMAND", -1, -1)
  105.             # Or reject them...
  106.             self.addUserComment("USER_DIPLOCOMMENT_REJECT_DEMAND", -1, -1)
  107.  
  108.         # If the AI is pressuring us to convert to their religion
  109.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_RELIGION_PRESSURE")):
  110.     
  111.             # We can accept their demands
  112.             self.addUserComment("USER_DIPLOCOMMENT_CONVERT", -1, -1)
  113.             # Or reject them...
  114.             self.addUserComment("USER_DIPLOCOMMENT_NO_CONVERT", -1, -1)
  115.     
  116.         # If the AI is pressuring us to switch to their favorite civic
  117.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_CIVIC_PRESSURE")):
  118.     
  119.             # We can accept their demands
  120.             self.addUserComment("USER_DIPLOCOMMENT_REVOLUTION", -1, -1)
  121.             # Or reject them...
  122.             self.addUserComment("USER_DIPLOCOMMENT_NO_REVOLUTION", -1, -1)
  123.     
  124.         # If the AI is pressuring us to join their war
  125.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_JOIN_WAR")):
  126.     
  127.             # We can accept their demands
  128.             self.addUserComment("USER_DIPLOCOMMENT_JOIN_WAR", -1, -1)
  129.             # Or reject them...
  130.             self.addUserComment("USER_DIPLOCOMMENT_NO_JOIN_WAR", -1, -1)
  131.     
  132.         # If the AI is pressuring us to stop trading with their enemy
  133.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_STOP_TRADING")):
  134.     
  135.             # We can accept their demands
  136.             self.addUserComment("USER_DIPLOCOMMENT_STOP_TRADING", -1, -1)
  137.             # Or reject them...
  138.             self.addUserComment("USER_DIPLOCOMMENT_NO_STOP_TRADING", -1, -1)
  139.  
  140.         # If we are viewing our current deals or
  141.         # If we are trading or
  142.         # If we are trying another proposal or
  143.         # If they reject our offer or
  144.         # If they reject our demand
  145.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_CURRENT_DEALS") or
  146.                     self.isComment(eComment, "AI_DIPLOCOMMENT_TRADING") or 
  147.                     self.isComment(eComment, "AI_DIPLOCOMMENT_REJECT") or
  148.                     self.isComment(eComment, "AI_DIPLOCOMMENT_SORRY") or
  149.                     self.isComment(eComment, "AI_DIPLOCOMMENT_TRY_THIS_DEAL") or 
  150.                     self.isComment(eComment, "AI_DIPLOCOMMENT_NO_DEAL") or
  151.                     self.isComment(eComment, "AI_DIPLOCOMMENT_REJECT_ASK") or
  152.                     self.isComment(eComment, "AI_DIPLOCOMMENT_REJECT_DEMAND")):
  153.     
  154.             # If no one is currently offering anything
  155.             if (self.diploScreen.ourOfferEmpty() == 1 and self.diploScreen.theirOfferEmpty() == 1):
  156.     
  157.                 # If we are at war, allow us to suggest a peace treaty
  158.                 if (self.diploScreen.atWar()):
  159.                     self.addUserComment("USER_DIPLOCOMMENT_PROPOSE", -1, -1)
  160.                     self.addUserComment("USER_DIPLOCOMMENT_OFFER_PEACE", -1, -1)
  161.     
  162.             # If one of us has something on the table
  163.             if (self.diploScreen.ourOfferEmpty() == 0 or self.diploScreen.theirOfferEmpty() == 0):
  164.     
  165.                 # If the offer is from the AI
  166.                 if (self.diploScreen.isAIOffer()):
  167.     
  168.                     # We can accept or reject the offer
  169.                     self.addUserComment("USER_DIPLOCOMMENT_ACCEPT", -1, -1)
  170.                     self.addUserComment("USER_DIPLOCOMMENT_REJECT", -1, -1)
  171.  
  172.                 # Otherwise this is a player offer to the AI
  173.                 else:
  174.  
  175.                     # This is a two way deal
  176.                     if (self.diploScreen.ourOfferEmpty() == 0 and self.diploScreen.theirOfferEmpty() == 0):
  177.                         # Insert the propose trade button
  178.                         self.addUserComment("USER_DIPLOCOMMENT_PROPOSE", -1, -1)
  179.  
  180.                         # If we are at war, use these items to fish for peace
  181.                         if (self.diploScreen.atWar()):
  182.                             self.addUserComment("USER_DIPLOCOMMENT_OFFER_PEACE", -1, -1)
  183.  
  184.                         # During peace, see what we can get for these items
  185.                         else:
  186.                             if (gc.getGame().getActiveTeam() != gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam()):
  187.                                 self.addUserComment("USER_DIPLOCOMMENT_COMPLETE_DEAL", -1, -1)
  188.  
  189.                     # Otherwise they have something on the table and we dont
  190.                     elif (self.diploScreen.theirOfferEmpty() == 0):
  191.  
  192.                         # If we are at war, demand the items for peace or ask what they want
  193.                         if (self.diploScreen.atWar()):
  194.                             self.addUserComment("USER_DIPLOCOMMENT_PROPOSE", -1, -1)
  195.                             self.addUserComment("USER_DIPLOCOMMENT_OFFER_PEACE", -1, -1)
  196.  
  197.                         # Otherwise (during peacetime) ask what they want for our item or demand they give it to us
  198.                         else:
  199.                             if (gc.getGame().getActiveTeam() == gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam()):
  200.                                 self.addUserComment("USER_DIPLOCOMMENT_DEMAND_TEAM", -1, -1)
  201.  
  202.                             else:
  203.                                 self.addUserComment("USER_DIPLOCOMMENT_OFFER", -1, -1)
  204.  
  205.                                 if (gc.getPlayer(self.diploScreen.getWhoTradingWith()).AI_getAttitude(gc.getGame().getActivePlayer()) >= AttitudeTypes.ATTITUDE_PLEASED):
  206.                                     self.addUserComment("USER_DIPLOCOMMENT_ASK", -1, -1)
  207.                                 elif (gc.getTeam(gc.getGame().getActiveTeam()).canDeclareWar(gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam())):
  208.                                     self.addUserComment("USER_DIPLOCOMMENT_DEMAND", -1, -1)
  209.     
  210.                     # Otherwise we have something on the table and they dont
  211.                     else:
  212.  
  213.                         # If we are at war, use this item to fish for peace or propose peace with the items
  214.                         if (self.diploScreen.atWar()):
  215.                             self.addUserComment("USER_DIPLOCOMMENT_PROPOSE", -1, -1)
  216.                             self.addUserComment("USER_DIPLOCOMMENT_OFFER_PEACE", -1, -1)
  217.  
  218.                         # During peace, see what we can get for these items or simply gift them to the AI
  219.                         else:
  220.                             if (gc.getGame().getActiveTeam() != gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam()):
  221.                                 self.addUserComment("USER_DIPLOCOMMENT_FISH_FOR_DEAL", -1, -1)
  222.  
  223.                             self.addUserComment("USER_DIPLOCOMMENT_GIFT", -1, -1)
  224.  
  225.             # Exit option
  226.             self.addUserComment("USER_DIPLOCOMMENT_NEVERMIND", -1, -1)
  227.             self.addUserComment("USER_DIPLOCOMMENT_EXIT", -1, -1)
  228.             self.diploScreen.startTrade( eComment, self.isComment(eComment, "AI_DIPLOCOMMENT_CURRENT_DEALS") )
  229.  
  230.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_SOMETHING_ELSE")):
  231.             if (gc.getTeam(gc.getGame().getActiveTeam()).canDeclareWar(gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam())):
  232.                 self.addUserComment("USER_DIPLOCOMMENT_WAR", -1, -1)
  233.  
  234.             self.addUserComment("USER_DIPLOCOMMENT_ATTITUDE", -1, -1)
  235.  
  236.             if (gc.getGame().getActiveTeam() == gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam()):
  237.                 self.addUserComment("USER_DIPLOCOMMENT_RESEARCH", -1, -1)
  238.  
  239.             if (gc.getTeam(gc.getGame().getActiveTeam()).AI_shareWar(gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam())):
  240.                 self.addUserComment("USER_DIPLOCOMMENT_TARGET", -1, -1)
  241.  
  242.             self.addUserComment("USER_DIPLOCOMMENT_NEVERMIND", -1, -1)
  243.             self.addUserComment("USER_DIPLOCOMMENT_EXIT", -1, -1)
  244.  
  245.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_RESEARCH")):
  246.             for i in range(gc.getNumTechInfos()):
  247.                 if (gc.getPlayer(self.diploScreen.getWhoTradingWith()).canResearch(i, False)):
  248.                     self.addUserComment("USER_DIPLOCOMMENT_RESEARCH_TECH", i, -1, gc.getTechInfo(i).getTextKey())
  249.  
  250.             self.addUserComment("USER_DIPLOCOMMENT_SOMETHING_ELSE", -1, -1)
  251.             self.addUserComment("USER_DIPLOCOMMENT_EXIT", -1, -1)
  252.  
  253.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_ATTITUDE") or
  254.                     self.isComment(eComment, "AI_DIPLOCOMMENT_ATTITUDE_PLAYER_FURIOUS") or 
  255.                     self.isComment(eComment, "AI_DIPLOCOMMENT_ATTITUDE_PLAYER_ANNOYED") or
  256.                     self.isComment(eComment, "AI_DIPLOCOMMENT_ATTITUDE_PLAYER_CAUTIOUS") or
  257.                     self.isComment(eComment, "AI_DIPLOCOMMENT_ATTITUDE_PLAYER_PLEASED") or 
  258.                     self.isComment(eComment, "AI_DIPLOCOMMENT_ATTITUDE_PLAYER_FRIENDLY")):
  259.             for i in range(gc.getMAX_CIV_PLAYERS()):
  260.                 if (gc.getPlayer(i).isAlive()):
  261.                     if ((i != gc.getGame().getActivePlayer()) and (i != self.diploScreen.getWhoTradingWith())):
  262.                         if (gc.getTeam(gc.getGame().getActiveTeam()).isHasMet(gc.getPlayer(i).getTeam()) and gc.getTeam(gc.getPlayer(self.diploScreen.getWhoTradingWith()).getTeam()).isHasMet(gc.getPlayer(i).getTeam())):
  263.                             self.addUserComment("USER_DIPLOCOMMENT_ATTITUDE_PLAYER", i, -1, gc.getPlayer(i).getNameKey())
  264.  
  265.             self.addUserComment("USER_DIPLOCOMMENT_SOMETHING_ELSE", -1, -1)
  266.             self.addUserComment("USER_DIPLOCOMMENT_EXIT", -1, -1)
  267.  
  268.         elif (self.isComment(eComment, "AI_DIPLOCOMMENT_TARGET")):
  269.             for i in range(gc.getMAX_CIV_PLAYERS()):
  270.                 if (gc.getPlayer(i).isAlive()):
  271.                     if (gc.getTeam(gc.getGame().getActiveTeam()).isAtWar(gc.getPlayer(i).getTeam())):
  272.                         player = PyPlayer(i)
  273.                         cityList = player.getCityList()
  274.                         for city in cityList:
  275.                             if (city.isRevealed(gc.getGame().getActiveTeam())):
  276.                                 self.addUserComment("USER_DIPLOCOMMENT_TARGET_CITY", i, city.getID(), city.getNameKey())
  277.  
  278.             self.addUserComment("USER_DIPLOCOMMENT_SOMETHING_ELSE", -1, -1)
  279.             self.addUserComment("USER_DIPLOCOMMENT_EXIT", -1, -1)
  280.  
  281.         # The default...
  282.         else:
  283.  
  284.             if (gc.getPlayer(gc.getGame().getActivePlayer()).canTradeWith(self.diploScreen.getWhoTradingWith())):
  285.                 # Allow us to begin another proposal
  286.                 self.addUserComment("USER_DIPLOCOMMENT_PROPOSAL", -1, -1)
  287.  
  288.             if (not self.isComment(eComment, "AI_DIPLOCOMMENT_PEACE") and not self.isComment(eComment, "AI_DIPLOCOMMENT_NO_PEACE")):
  289.                 # If we are at war, allow to suggest peace
  290.                 if (self.diploScreen.atWar()):
  291.                     self.addUserComment("USER_DIPLOCOMMENT_SUGGEST_PEACE", -1, -1)
  292.  
  293.             # If we have a current deal, allow us to see the deals
  294.             if (self.diploScreen.hasAnnualDeal()):
  295.                 self.addUserComment("USER_DIPLOCOMMENT_CURRENT_DEALS", -1, -1)
  296.  
  297.             self.addUserComment("USER_DIPLOCOMMENT_SOMETHING_ELSE", -1, -1)
  298.  
  299.             # Exit potential
  300.             self.addUserComment("USER_DIPLOCOMMENT_EXIT", -1, -1)
  301.             self.diploScreen.endTrade();
  302.  
  303.     def addUserComment(self, eComment, iData1, iData2, *args):
  304.         " Helper for adding User Comments "
  305.         iComment = self.getCommentID( eComment )
  306.         self.diploScreen.addUserComment( iComment, iData1, iData2, self.getDiplomacyComment(iComment), args)
  307.         
  308.     def setAIComment (self, eComment, *args):
  309.         " Handles the determining the AI comments"
  310.         AIString = self.getDiplomacyComment(eComment)
  311.  
  312.         if DebugLogging:
  313.             print "CvDiplomacy.setAIComment: %s" %(eComment,)
  314.             if (len(args)):
  315.                 print "args", args
  316.             AIString = "(%d) - %s" %(self.getLastResponseID(), AIString)
  317.         
  318.         self.diploScreen.setAIString(AIString, args)
  319.         self.diploScreen.setAIComment(eComment)
  320.         self.determineResponses(eComment)
  321.         self.performHeadAction(eComment)
  322.  
  323.     def performHeadAction( self, eComment ):
  324.     
  325.         if ( eComment == self.getCommentID("AI_DIPLOCOMMENT_NO_PEACE") or
  326.              eComment == self.getCommentID("AI_DIPLOCOMMENT_REJECT") or
  327.              eComment == self.getCommentID("AI_DIPLOCOMMENT_NO_DEAL") or
  328.              eComment == self.getCommentID("AI_DIPLOCOMMENT_CANCEL_DEAL") or
  329.              eComment == self.getCommentID("AI_DIPLOCOMMENT_REJECT_ASK") or
  330.              eComment == self.getCommentID("AI_DIPLOCOMMENT_REJECT_DEMAND") ):
  331.             self.diploScreen.performHeadAction( LeaderheadAction.LEADERANIM_DISAGREE ) 
  332.         elif ( eComment == self.getCommentID("AI_DIPLOCOMMENT_ACCEPT") or
  333.                eComment == self.getCommentID("AI_DIPLOCOMMENT_TRY_THIS_DEAL") ):
  334.             self.diploScreen.performHeadAction( LeaderheadAction.LEADERANIM_AGREE ) 
  335.         elif ( eComment == self.getCommentID("AI_DIPLOCOMMENT_FIRST_CONTACT") or
  336.                eComment == self.getCommentID("AI_DIPLOCOMMENT_GREETINGS") or
  337.                eComment == self.getCommentID("AI_DIPLOCOMMENT_WORST_ENEMY") or
  338.                eComment == self.getCommentID("AI_DIPLOCOMMENT_UNIT_BRAG") or
  339.                eComment == self.getCommentID("AI_DIPLOCOMMENT_NUKES") or
  340.                eComment == self.getCommentID("AI_DIPLOCOMMENT_OFFER_PEACE") or
  341.                eComment == self.getCommentID("AI_DIPLOCOMMENT_OFFER_CITY") or
  342.                eComment == self.getCommentID("AI_DIPLOCOMMENT_OFFER_DEAL") or
  343.                eComment == self.getCommentID("AI_DIPLOCOMMENT_GIVE_HELP") or
  344.                eComment == self.getCommentID("AI_DIPLOCOMMENT_RELIGION_PRESSURE") or
  345.                eComment == self.getCommentID("AI_DIPLOCOMMENT_CIVIC_PRESSURE") or
  346.                eComment == self.getCommentID("AI_DIPLOCOMMENT_JOIN_WAR") or
  347.                eComment == self.getCommentID("AI_DIPLOCOMMENT_STOP_TRADING") or
  348.                eComment == self.getCommentID("AI_DIPLOCOMMENT_ASK_FOR_HELP") or
  349.                eComment == self.getCommentID("AI_DIPLOCOMMENT_DEMAND_TRIBUTE") ):
  350.             self.diploScreen.performHeadAction( LeaderheadAction.LEADERANIM_GREETING ) 
  351.         elif ( eComment == self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_FURIOUS") or
  352.                eComment == self.getCommentID("AI_DIPLOCOMMENT_WORST_ENEMY_TRADING") or
  353.                eComment == self.getCommentID("AI_DIPLOCOMMENT_HELP_REFUSED") or
  354.                eComment == self.getCommentID("AI_DIPLOCOMMENT_DEMAND_REJECTED") or
  355.                eComment == self.getCommentID("AI_DIPLOCOMMENT_RELIGION_DENIED") or
  356.                eComment == self.getCommentID("AI_DIPLOCOMMENT_CIVIC_DENIED") or
  357.                eComment == self.getCommentID("AI_DIPLOCOMMENT_JOIN_DENIED") or
  358.                eComment == self.getCommentID("AI_DIPLOCOMMENT_STOP_DENIED") ):
  359.             self.diploScreen.performHeadAction( LeaderheadAction.LEADERANIM_FURIOUS ) 
  360.         elif ( eComment == self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_ANNOYED") ):
  361.             self.diploScreen.performHeadAction( LeaderheadAction.LEADERANIM_ANNOYED ) 
  362.         elif ( eComment == self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_CAUTIOUS") ):
  363.             self.diploScreen.performHeadAction( LeaderheadAction.LEADERANIM_CAUTIOUS ) 
  364.         elif ( eComment == self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_PLEASED") or
  365.                eComment == self.getCommentID("AI_DIPLOCOMMENT_SORRY") ):
  366.             self.diploScreen.performHeadAction( LeaderheadAction.LEADERANIM_PLEASED ) 
  367.         elif ( eComment == self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_FRIENDLY") or
  368.                eComment == self.getCommentID("AI_DIPLOCOMMENT_GLAD") or
  369.                eComment == self.getCommentID("AI_DIPLOCOMMENT_THANKS") ):
  370.             self.diploScreen.performHeadAction( LeaderheadAction.LEADERANIM_FRIENDLY ) 
  371.     
  372.         return
  373.  
  374.     def getDiplomacyComment (self, eComment):
  375.         "Function to get the user String"
  376.         debugString = "CvDiplomacy.getDiplomacyComment: %s" %(eComment,)
  377.         eComment = int(eComment)
  378.         if DebugLogging:
  379.             print debugString, eComment
  380.         
  381.         szString = ""
  382.         szFailString = "Error***: No string found for eComment: %s"
  383.         
  384.         if ( gc.getDiplomacyInfo(eComment) ):
  385.             DiplomacyTextInfo = gc.getDiplomacyInfo(eComment)
  386.             if ( not DiplomacyTextInfo ):
  387.                 print "%s IS AN INVALID DIPLOCOMMENT" %(eComment,)
  388.                 CvUtil.pyAssert(True, "CvDiplomacy.getDiplomacyComment: %s does not have a DiplomacyTextInfo" %(eComment,))
  389.                 return szFailString %(eComment,)
  390.             
  391.             szString = self.filterUserResponse(DiplomacyTextInfo)
  392.             
  393.         else:
  394.             szString = szFailString %(eComment,)
  395.         
  396.         return szString
  397.     
  398.     def setLastResponseID(self, iResponse):
  399.         self.iLastResponseID = iResponse
  400.     
  401.     def getLastResponseID(self):
  402.         return self.iLastResponseID
  403.  
  404.     def isUsed(self, var, i, num):
  405.         "returns true if any element in the var list is true"
  406.         for j in range(num):
  407.             if (var(i, j)):
  408.                 return true
  409.         return false
  410.         
  411.     def filterUserResponse(self, diploInfo):
  412.         "pick the user's response from a CvDiplomacyTextInfo, based on response conditions"
  413.         if (self.diploScreen.getWhoTradingWith() == -1):
  414.             return ""
  415.             
  416.         theirPlayer = gc.getPlayer(self.diploScreen.getWhoTradingWith())
  417.         ourPlayer = gc.getActivePlayer()
  418.         responses = []
  419.         
  420.         for i in range(diploInfo.getNumResponses()):    
  421.             
  422.             # check attitude of other player towards me
  423.             if (self.isUsed(diploInfo.getAttitudeTypes, i, AttitudeTypes.NUM_ATTITUDE_TYPES)):
  424.                 att = theirPlayer.AI_getAttitude(CyGame().getActivePlayer())
  425.                 if (not diploInfo.getAttitudeTypes(i, att)):
  426.                     continue
  427.             
  428.             # check civ type
  429.             if (self.isUsed(diploInfo.getCivilizationTypes, i, gc.getNumCivilizationInfos()) and
  430.                 not diploInfo.getCivilizationTypes(i, theirPlayer.getCivilizationType())):
  431.                 continue
  432.                 
  433.             # check leader type
  434.             if (self.isUsed(diploInfo.getLeaderHeadTypes, i, gc.getNumLeaderHeadInfos()) and
  435.                 not diploInfo.getLeaderHeadTypes(i, theirPlayer.getLeaderType())):
  436.                 continue
  437.  
  438.             # check power type
  439.             if (self.isUsed(diploInfo.getDiplomacyPowerTypes, i, DiplomacyPowerTypes.NUM_DIPLOMACYPOWER_TYPES)):
  440.                 theirPower = theirPlayer.getPower()
  441.                 ourPower = ourPlayer.getPower()
  442.                 
  443.                 if (ourPower < (theirPower / 2)):
  444.                     if not diploInfo.getDiplomacyPowerTypes(i, DiplomacyPowerTypes.DIPLOMACYPOWER_STRONGER):
  445.                         continue
  446.                         
  447.                 elif (ourPower > (theirPower * 2)):
  448.                     if not diploInfo.getDiplomacyPowerTypes(i, DiplomacyPowerTypes.DIPLOMACYPOWER_WEAKER):
  449.                         continue
  450.                         
  451.                 else:
  452.                     if not diploInfo.getDiplomacyPowerTypes(i, DiplomacyPowerTypes.DIPLOMACYPOWER_EQUAL):
  453.                         continue
  454.                 
  455.             # passed all tests, so add to response list
  456.             for j in range(diploInfo.getNumDiplomacyText(i)):
  457.                 responses.append(diploInfo.getDiplomacyText(i, j))
  458.                     
  459.         # pick a random response
  460.         numResponses = len(responses)
  461.         if (numResponses>0):
  462.             iResponse = gc.getASyncRand().get(numResponses, "Python Diplomacy ASYNC")
  463.             self.setLastResponseID(iResponse)
  464.             return responses[iResponse]
  465.         
  466.         return ""    # no responses matched
  467.             
  468.     def handleUserResponse(self, eComment, iData1, iData2):
  469.         if DebugLogging:
  470.             print "CvDiplomacy.handleUserResponse: %s" %(eComment,)
  471.             
  472.         diploScreen = CyDiplomacy()
  473.  
  474.         # If we accept peace
  475.         if (self.isComment(eComment, "USER_DIPLOCOMMENT_PEACE")):
  476.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_PEACE"))
  477.     
  478.         # If we choose war
  479.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_WAR")):
  480.             diploScreen.declareWar()
  481.             #diploScreen.closeScreen()
  482.  
  483.         # If we wish to make a trade proposal or try to renegotiate
  484.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_PROPOSAL") or
  485.                     self.isComment(eComment, "USER_DIPLOCOMMENT_RENEGOTIATE")):
  486.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_TRADING"))
  487.             diploScreen.showAllTrade(True)
  488.  
  489.         # If we want to propose a trade
  490.         elif(self.isComment(eComment, "USER_DIPLOCOMMENT_PROPOSE")):
  491.             if (diploScreen.offerDeal() == 1):
  492.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ACCEPT"))
  493.             else:
  494.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_REJECT"))
  495.     
  496.         # If we ask for peace
  497.         elif(self.isComment(eComment, "USER_DIPLOCOMMENT_SUGGEST_PEACE")):
  498.             if (diploScreen.offerDeal() == 1):
  499.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_PEACE"))
  500.     
  501.             else:
  502.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_NO_PEACE"))
  503.     
  504.         # If we accept a trade
  505.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_ACCEPT")):
  506.             diploScreen.implementDeal()
  507.             diploScreen.setAIOffer(0)
  508.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_GLAD"))
  509.     
  510.         # If we reject a trade
  511.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_REJECT")):
  512.             diploScreen.setAIOffer(0)
  513.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_SORRY"))
  514.  
  515.         # If we offer a deal, or is we are fishing for a deal, or if we are offering peace or fishing for peace
  516.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_OFFER") or
  517.                     self.isComment(eComment, "USER_DIPLOCOMMENT_COMPLETE_DEAL") or
  518.                     self.isComment(eComment, "USER_DIPLOCOMMENT_FISH_FOR_DEAL") or
  519.                     self.isComment(eComment, "USER_DIPLOCOMMENT_OFFER_PEACE")):
  520.             if (diploScreen.counterPropose() == 1):
  521.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_TRY_THIS_DEAL"))
  522.             else:
  523.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_NO_DEAL"))
  524.  
  525.         # if we are asking for something
  526.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_ASK")):
  527.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_ASK_HELP, -1, -1)
  528.             if (diploScreen.offerDeal()):
  529.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ACCEPT_ASK"))
  530.             else:
  531.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_REJECT_ASK"))
  532.  
  533.         # if we are demanding something
  534.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_DEMAND")):
  535.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_MADE_DEMAND, -1, -1)
  536.             if (diploScreen.offerDeal()):
  537.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ACCEPT_DEMAND"))
  538.             else:
  539.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_REJECT_DEMAND"))
  540.  
  541.         # if we are demanding something from our teammate
  542.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_DEMAND_TEAM")):
  543.             diploScreen.offerDeal()
  544.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ACCEPT_DEMAND_TEAM"))
  545.  
  546.         # If we are giving a gift
  547.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_GIFT")):
  548.             diploScreen.offerDeal()
  549.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_THANKS"))
  550.  
  551.         # If we decide to view current deals
  552.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_CURRENT_DEALS")):
  553.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_CURRENT_DEALS"))
  554.             diploScreen.showAllTrade(False)
  555.  
  556.         # If we give help
  557.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_GIVE_HELP")):
  558.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_GIVE_HELP, -1, -1)
  559.             diploScreen.implementDeal()
  560.             diploScreen.setAIOffer(0)
  561.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_THANKS"))
  562.  
  563.         # If we accept their demand
  564.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_ACCEPT_DEMAND")):
  565.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_ACCEPT_DEMAND, -1, -1)
  566.             diploScreen.implementDeal()
  567.             diploScreen.setAIOffer(0)
  568.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_THANKS"))
  569.  
  570.         # If we accept the offer
  571.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_ACCEPT_OFFER")):
  572.             diploScreen.implementDeal()
  573.             diploScreen.setAIOffer(0)
  574.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_THANKS"))
  575.  
  576.         # If we refuse to help
  577.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_REFUSE_HELP")):
  578.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_REFUSED_HELP, -1, -1)
  579.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_HELP_REFUSED"))
  580.  
  581.         # If we reject their demand
  582.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_REJECT_DEMAND")):
  583.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_REJECTED_DEMAND, -1, -1)
  584.             
  585.             if (gc.getPlayer(self.diploScreen.getWhoTradingWith()).AI_demandRebukedWar(gc.getGame().getActivePlayer())):
  586.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_DECLARE_WAR"))
  587.                 diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_DEMAND_WAR, -1, -1)
  588.             else:
  589.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_DEMAND_REJECTED"))
  590.  
  591.         # If we convert to their state religion
  592.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_CONVERT")):
  593.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_CONVERT, -1, -1)
  594.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_THANKS"))
  595.  
  596.         # If we refuse to convert to their state religion
  597.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_NO_CONVERT")):
  598.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_NO_CONVERT, -1, -1)
  599.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_RELIGION_DENIED"))
  600.  
  601.         # If we adopt their favorite civic
  602.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_REVOLUTION")):
  603.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_REVOLUTION, -1, -1)
  604.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_THANKS"))
  605.  
  606.         # If we refuse to adopt their favorite civic
  607.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_NO_REVOLUTION")):
  608.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_NO_REVOLUTION, -1, -1)
  609.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_CIVIC_DENIED"))
  610.  
  611.         # If we join their war
  612.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_JOIN_WAR")):
  613.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_JOIN_WAR, diploScreen.getData(), -1)
  614.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_THANKS"))
  615.  
  616.         # If we refuse to join their war
  617.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_NO_JOIN_WAR")):
  618.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_NO_JOIN_WAR, -1, -1)
  619.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_JOIN_DENIED"))
  620.  
  621.         # If we stop the trading
  622.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_STOP_TRADING")):
  623.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_STOP_TRADING, diploScreen.getData(), -1)
  624.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_THANKS"))
  625.  
  626.         # If we refuse to stop the trading
  627.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_NO_STOP_TRADING")):
  628.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_NO_STOP_TRADING, -1, -1)
  629.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_STOP_DENIED"))
  630.  
  631.         # If we want to go back to first screen
  632.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_NEVERMIND")):
  633.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_WELL"))
  634.  
  635.         # If we want to discuss something else
  636.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_SOMETHING_ELSE")):
  637.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_SOMETHING_ELSE"))
  638.  
  639.         # If we want to ask them to change their research
  640.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_RESEARCH")):
  641.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_RESEARCH"))
  642.  
  643.         # If we want to ask them to change their research to a specific tech
  644.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_RESEARCH_TECH")):
  645.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_RESEARCH_TECH, iData1, -1)
  646.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_RESEARCH_TECH"))
  647.  
  648.         # If we want to ask them to what their attitude is
  649.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_ATTITUDE")):
  650.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE"))
  651.  
  652.         # If we want to ask them to what their attitude is on a specific player
  653.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_ATTITUDE_PLAYER")):
  654.             eAttitude = gc.getPlayer(self.diploScreen.getWhoTradingWith()).AI_getAttitude(iData1)
  655.             
  656.             if (eAttitude == AttitudeTypes.ATTITUDE_FURIOUS):
  657.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_FURIOUS"), gc.getPlayer(iData1).getNameKey())
  658.             elif (eAttitude == AttitudeTypes.ATTITUDE_ANNOYED):
  659.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_ANNOYED"), gc.getPlayer(iData1).getNameKey())
  660.             elif (eAttitude == AttitudeTypes.ATTITUDE_CAUTIOUS):
  661.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_CAUTIOUS"), gc.getPlayer(iData1).getNameKey())
  662.             elif (eAttitude == AttitudeTypes.ATTITUDE_PLEASED):
  663.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_PLEASED"), gc.getPlayer(iData1).getNameKey())
  664.             else:
  665.                 self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_ATTITUDE_PLAYER_FRIENDLY"), gc.getPlayer(iData1).getNameKey())
  666.  
  667.         # If we want to ask them to change their target
  668.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_TARGET")):
  669.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_TARGET"))
  670.  
  671.         # If we want to ask them to change their target to a specific city
  672.         elif (self.isComment(eComment, "USER_DIPLOCOMMENT_TARGET_CITY")):
  673.             diploScreen.diploEvent(DiploEventTypes.DIPLOEVENT_TARGET_CITY, iData1, iData2)
  674.             self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_TARGET_CITY"))
  675.  
  676.         else:
  677.             diploScreen.closeScreen()
  678.  
  679.     def dealCanceled( self ):
  680.     
  681.         self.setAIComment(self.getCommentID("AI_DIPLOCOMMENT_TRADING"))
  682.         
  683.         return
  684.  
  685.     def isComment(self, eComment, strComment):
  686.         'bool - comment matching'
  687.         if ( gc.getDiplomacyInfo(eComment).getType() == strComment ):
  688.             return True
  689.         return False
  690.     
  691.     def getCommentID(self, strComment):
  692.         'int - ID for DiploCommentType'
  693.         for i in range(gc.getNumDiplomacyInfos()):
  694.             if ( gc.getDiplomacyInfo(i).getType() == strComment ):
  695.                 return i
  696.         return -1
  697.