def removeRedundantNode(self, listXmlNode, cvNode):
xmlX, xmlY, cvX, cvT = 0, 0, 0, 0
cvX, cvY = calc.nodeCenter(cvNode)
listFlag = []
for xmlNode in listXmlNode:
xmlX, xmlY = calc.nodeCenter(xmlNode)
if calc.euclideanDistance(xmlX, xmlY, cvX, cvY) > 10:
listFlag.append(True)
else:
listFlag.append(False)
return all(listFlag)
def compareState(self):
pathToStateData = './state/'
#initial first png and xml
#shutil.copy('0.png', pathToStateData + str(0) + '.png')
#shutil.copy('0.xml', pathToStateData + str(0) + '.xml')
imgFiles = [name for name in os.listdir(pathToStateData) if name.endswith('.png')]
xmlFiles = [name for name in os.listdir(pathToStateData) if name.endswith('.xml')]
txtFiles = [name for name in os.listdir('./state/') if name.endswith('.txt')]
listCheckState = []
for i in xrange(len(imgFiles)):
listCheckState.append(calc.pixelCompare('0.png', pathToStateData + imgFiles[i], 0.1))
print "check state: " + str(listCheckState)
#all -> and list, any or list
if all(listCheckState) == True:
for i in xrange(len(imgFiles) +1 ):
if str(i) + '.png' not in imgFiles:
shutil.copy('0.png', pathToStateData + str(i) + '.png')
shutil.copy('0.xml', pathToStateData + str(i) + '.xml')
shutil.copy('xmlcv.txt', './state/' + str(i) + '.txt')
print 'state: ' + str(i)
else:
for i,j in enumerate(listCheckState):
if j == False:
print 'state: ' + str(i)
break
'''
