def pack(directory, quiet, algorithm, size):
cprint('working directory {}'.format(directory), color='green')
cprint('quiet {}'.format(quiet), color='green')
cprint('algorithm {}'.format(algorithm), color='green')
cprint('size {}'.format(size))
folder = QDir(path=directory)
folder.setNameFilters(['*.png'])
folder.setFilter(QDir.Files or QDir.NoDotAndDotDot)
dit = QDirIterator(folder, flags=QDirIterator.Subdirectories, filters=QDir.Files)
filenames = []
while dit.hasNext():
im = folder.relativeFilePath(dit.next())
if not re.search('atlas', im):
filenames.append(im)
images = [Image(directory, filename) for filename in filenames]
BinPacking.bin_packing_progress = MyBinPackingProgress()
bin_packer = FirstFitShelfBinPacking(Size(2048, 2048), images, bin_parameters={})
bin_packer.saveAtlases(directory)
print ''
def harvestDirectory(self, directory):
folder = QDir(path=directory)
folder.setNameFilters(["*.png"])
folder.setFilter(QDir.Files or QDir.NoDotAndDotDot)
dit = QDirIterator(folder, flags=QDirIterator.Subdirectories, filters=QDir.Files)
while dit.hasNext():
im = folder.relativeFilePath(dit.next())
if not re.search("atlas", im):
self.images.append(im)
self.bin_packing_thread.setDirectory(self.directory)
self.bin_packing_thread.setImages(self.images)
self.updateImages(self.images)
def open_folder(self):
self.imagesList.clear()
if not self.current_directory:
self.current_directory = QFileDialog.getExistingDirectory(parent=self,
caption=u'Add images from directory')
folder = QDir(self.current_directory)
folder.setNameFilters(['*.png'])
folder.setFilter(QDir.Files or QDir.NoDotAndDotDot)
dir_iterator = QDirIterator(folder, flags=QDirIterator.Subdirectories, filters=QDir.Files)
images = []
while dir_iterator.hasNext():
images.append(folder.relativeFilePath(dir_iterator.next()))
self.white_albatross.addImages(folder, images)
self.imagesList.addItems(images)
def open_folder(self):
self.imagesList.clear()
if not self.current_directory:
self.current_directory = QFileDialog.getExistingDirectory(
parent=self, caption=u'Add images from directory')
folder = QDir(self.current_directory)
folder.setNameFilters(['*.png'])
folder.setFilter(QDir.Files or QDir.NoDotAndDotDot)
dir_iterator = QDirIterator(folder,
flags=QDirIterator.Subdirectories,
filters=QDir.Files)
images = []
while dir_iterator.hasNext():
images.append(folder.relativeFilePath(dir_iterator.next()))
self.white_albatross.addImages(folder, images)
self.imagesList.addItems(images)
def browse_folder(self):
# fname = QtWidgets.QFileDialog.getOpenFileName()
fname = QtGui.QFileDialog.getOpenFileName()
currentDir = QDir(QDir.currentPath())
self.leInputFile.setText(currentDir.relativeFilePath(fname[0]))
