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 '''