def setRandomCode(self, codeSize=-1):
if codeSize == -1:
codeSize = self.__defaultCodeSize
random.seed()
self.__pegList = []
for i in range(codeSize):
x = peg.Peg(random.randint(0,colour.Colours.numberOfColours-1))
self.__pegList.append(x)
def compare(self,code):
resultPegs = []
secretUsed = []
guessUsed = []
count = 0
codeLength = len(self.__pegList)
for i in range(codeLength):
secretUsed.append(False)
guessUsed.append(False)
"""
Black pegs first: correct colour in correct position
"""
for i in range(codeLength):
if (self.__pegList[i].equals(code.getPegs()[i])):
secretUsed[i] = True
guessUsed[i] = True
resultPegs.append(peg.Peg(colour.Colours.black))
count += 1
"""
White pegs: trickier
White pegs are for pegs of the correct colour, but in the wrong
place. Each peg should only be evaluated once, hence the "used"
lists.
Condition below is a shortcut- if there were 3 blacks pegs
then the remaining peg can't be a correct colour (think about it)
"""
if (count < codeLength-1):
for i in range(codeLength):
if (guessUsed[i]):
continue
for j in range(codeLength):
if (i != j and not secretUsed[j] \
and not guessUsed[i] \
and self.__pegList[j].equals(code.getPegs()[i])):
resultPegs.append(peg.Peg(colour.Colours.white))
secretUsed[j] = True
guessUsed[i] = True
return Code(resultPegs)
def setRandomCode(self, codeSize=-1):
if codeSize == -1:
codeSize = self.__defaultCodeSize
random.seed()
self.__pegList = []
for i in range(codeSize):
#Avoid to guess the fake code (change lower limit from 0 to 1)
x = peg.Peg(random.randint(1, colour.Colours.numberOfColours - 1))
self.__pegList.append(x)
def __parseColour(self, s):
if (re.search("^r", s) is not None):
return peg.Peg(colour.Colours.red)
elif (re.search("^p", s) is not None):
return peg.Peg(colour.Colours.purple)
elif (re.search("^g", s) is not None):
return peg.Peg(colour.Colours.green)
elif (re.search("^y", s) is not None):
return peg.Peg(colour.Colours.yellow)
elif (re.search("^w", s) is not None):
return peg.Peg(colour.Colours.white)
elif (re.search("^b", s) is not None):
return peg.Peg(colour.Colours.black)
else:
return None