1 # -*- coding=utf-8 -*-
  2 from PIL import Image
  3 import os
  4 import shutil
  5 
  6 SUFIX_PNG = "png"
  7 SUFIX_JPEG = "jpeg"
  8 SUFIX_LIST = [SUFIX_PNG, SUFIX_JPEG]
  9 NEED_RESIZE_TAG = True
 10 SMALL_SIZE = (126,126)
 11 
 12 
 13 def isWhite(color):
 14     r, g, b = color[0], color[1], color[2]
 15     if r == 255 and g == 255 and  b==255:
 16         return True
 17     return False
 18 
 19 def getImgFileList(folderPath):
 20     modelFileNames = []
 21     for dirpath, dirnames, filenames in os.walk(folderPath):
 22         for filename in filenames:
 23             sufix = os.path.splitext(filename)[1][1:]
 24             if sufix in SUFIX_LIST:
 25                 modelFileNames.append(filename)
 26     modelFileNames.sort()
 27     return modelFileNames
 28 
 29 def getFileName(fileFullName):
 30     return fileFullName[:fileFullName.rindex(".")]
 31 
 32 def checkImgSize(sizeDict, widthSum, heightSum, imgCnt):
 33     # imgCnt = len(sizeDict.keys())
 34     averWidth, averHeight = widthSum/imgCnt, heightSum/imgCnt
 35     for sizeKey, sizeValue in sizeDict.items():
 36         if sizeValue[‘width‘] != averWidth:
 37             print "[ERROE],{IMG} width is wrong! please check it again!".format(IMG=sizeKey)
 38             return False
 39         if sizeValue[‘height‘] != averHeight:
 40             print "[ERROE, ]{IMG} height is wrong!please check it again!".format(IMG=sizeKey)
 41             return False
 42     return True
 43 
 44 
 45 def test():
 46     folderPath = ‘./model/‘
 47     srcPath = ‘./src/‘
 48     srcBackupPath = ‘./srcBackup/‘
 49     targetPath = ‘./done/‘
 50     currPath = ‘.‘
 51     # check model &src folder 
 52     if not os.path.exists(folderPath) or not os.path.exists(srcPath):
 53         print ‘[ERROR], dir [model] or [src] does not exist‘
 54         return
 55 
 56     # creat the target folder
 57     if not os.path.exists(targetPath):
 58         os.mkdir(targetPath)
 59 
 60     partImgNames = getImgFileList(folderPath)
 61     srcImgNames = getImgFileList(srcPath)
 62 
 63     partDict = {}
 64     srcDict = {}
 65     sizeDict = {}
 66     imgCnt = 0
 67     widthSum, heightSum = 0, 0
 68 
 69     # change src  img file sufix 把之前的png原图改成jpeg格式
 70     # 需不需要改格式
 71     # for _, srcImgName in enumerate(srcImgNames):
 72     #     if srcImgName.endswith(SUFIX_PNG):
 73     #         im = Image.open(srcPath+srcImgName)
 74     #         bg = Image.new("RGB", im.size, (255,255,255))
 75     #         newSrcImgName = srcImgName.replace(SUFIX_PNG, SUFIX_JPEG)
 76     #         bg.paste(im, im)
 77     #         bg.save(srcPath+newSrcImgName)
 78     #         if os.path.exists(srcPath+srcImgName):
 79     #             if not os.path.exists(srcBackupPath):
 80     #                 os.mkdir(srcBackupPath)
 81     #             shutil.move(srcPath+srcImgName, srcBackupPath)
 82 
 83     srcImgNames = getImgFileList(srcPath)
 84     for index, partImgName in enumerate(partImgNames):
 85         partDict[index] = Image.open(folderPath+partImgName)
 86         width, height = partDict[index].size
 87         sizeDict[partImgName] = {}
 88         sizeDict[partImgName][‘width‘] = width
 89         sizeDict[partImgName][‘height‘] = height
 90         widthSum += width
 91         heightSum += height
 92         imgCnt += 1
 93 
 94     for index, srcImgName in enumerate(srcImgNames):
 95         srcDict[index] = Image.open(srcPath+srcImgName)
 96         width, height = srcDict[index].size
 97         sizeDict[srcImgName] = {}
 98         sizeDict[srcImgName][‘width‘] = width
 99         sizeDict[srcImgName][‘height‘] = height
100         widthSum += width
101         heightSum += height
102         imgCnt += 1
103 
104     # img size check, 检查图片尺寸
105     if not checkImgSize(sizeDict, widthSum, heightSum, imgCnt):
106         return 
107 
108     for srcIndex, srcImgName in enumerate(srcImgNames):
109         tempSrcImg = srcDict[srcIndex].copy()
110         colorDataSrc1 = tempSrcImg.getdata()
111 
112         for modelIndex, partImgName in enumerate(partImgNames):
113             tempImg = partDict[modelIndex].copy() # copy img , 复制图片
114             colorData2 = tempImg.getdata()
115             colorStore = []  # to save colors
116             for index, color2 in enumerate (colorData2):
117                 if not isWhite(color2):
118                     colorStore.append(colorDataSrc1[index])
119                 else:
120                     colorStore.append((0,0,0,0))
121 
122             tempImg.putdata(colorStore)
123             # make target files names
124             newFileName = "{TARGET}{NAME}-{INDEX}.{SUFIX}".format(
125                 TARGET=targetPath, 
126                 NAME=getFileName(srcImgName),
127                 INDEX=modelIndex+1, 
128                 SUFIX=SUFIX_PNG)
129             tempImg.save(newFileName, SUFIX_PNG)
130 
131             if NEED_RESIZE_TAG:
132                 smallTargetPath = ‘./smalldone/‘
133                 if not os.path.exists(smallTargetPath):
134                     os.mkdir(smallTargetPath)
135                 smallImg = tempImg.resize(SMALL_SIZE)
136                 newFileName = "{TARGET}{NAME}-{INDEX}.{SUFIX}".format(
137                 TARGET=smallTargetPath, 
138                 NAME=getFileName(srcImgName),
139                 INDEX=modelIndex+1, 
140                 SUFIX=SUFIX_PNG)
141                 smallImg.save(newFileName, SUFIX_PNG)
142             print ‘------create {NAME} done-----‘.format(NAME=newFileName)
143     print ‘>>>>>>>>>>>>>>>>>Job Done<<<<<<<<<<<<<<<<‘
144 
145 
146 if __name__ == "__main__":
147     test()