def show_heatmap(self):
# print("a")
directory = QDir(os.getcwd())
directory.setFilter(QDir.Files | QDir.NoDotDot | QDir.NoDotAndDotDot)
directory.setSorting(QDir.Time)
set_filter = ["*.jpg"]
directory.setNameFilters(set_filter)
# print("b")
for show in directory.entryInfoList():
print("%s %s" % (show.size(), show.fileName()))
if "heatmap" in show.fileName():
out = show.fileName()
break
# print("c")
w = self.heatmapLabel.width()
h = self.heatmapLabel.height()
# print("d")
# print("%r", self.zone_cutted)
if self.zone_cutted:
x1, y1, x2, y2 = get_zone_position()
imageObject = Image.open(out)
cropped = imageObject.crop((x1, y1, x2, y2))
name_of_file = QFileInfo(out).baseName() + "_zone.jpg"
cropped.save(name_of_file)
pixmap = QPixmap(name_of_file)
else:
pixmap = QPixmap(out)
# print("e")
pixmap = pixmap.scaled(w, h, QtCore.Qt.KeepAspectRatio)
self.heatmapLabel.setPixmap(pixmap)
def show_video(self):
directory = QDir(os.getcwd())
directory.setFilter(QDir.Files | QDir.NoDotDot | QDir.NoDotAndDotDot)
directory.setSorting(QDir.Time)
set_filter = ["*.avi"]
directory.setNameFilters(set_filter)
for show in directory.entryInfoList():
print("%s %s" % (show.size(), show.fileName()))
out = show.fileName()
break
# print(1)
if self.zone_cutted:
# print(2)
x1, y1, x2, y2 = get_zone_position()
clip = VideoFileClip(out)
cut_video = crop(clip, x1, y1, x2, y2)
name_of_file = QFileInfo(out).baseName() + "_zone.avi"
cut_video.write_videofile(name_of_file, codec='mpeg4', audio=False)
else:
# print(3)
name_of_file = out
# print(4)
self.mediaPlayer.setMedia(
QMediaContent(QUrl.fromLocalFile(os.getcwd() + "\\" + name_of_file))
)
# print(5)
# self.mediaPlayer.setPlaybackRate()
self.play()
def related_file(FBTS, filename, encoding=None):
qd = QDir( FBTS.folderpath() )
qd.setFilter(QDir.Files | QDir.Readable)
qd.setSorting(QDir.Type | QDir.Reversed)
qd.setNameFilters( [filename] ) # literal name or 'foo*.*'
names = qd.entryList()
if names : # list is not empty, open the first
a_file = QFile( qd.absoluteFilePath(names[0]) )
return _qfile_to_stream(a_file, QIODevice.ReadOnly, encoding)
return None
def related_file(FBTS, filename, encoding=None):
qd = QDir(FBTS.folderpath())
qd.setFilter(QDir.Files | QDir.Readable)
qd.setSorting(QDir.Type | QDir.Reversed)
qd.setNameFilters([filename]) # literal name or 'foo*.*'
names = qd.entryList()
if names: # list is not empty, open the first
a_file = QFile(qd.absoluteFilePath(names[0]))
return _qfile_to_stream(a_file, QIODevice.ReadOnly, encoding)
return None
def recursive(self, path, search_string, paths):
directory = QDir(path)
directory.setFilter(QDir.Files | QDir.Dirs | QDir.NoDotAndDotDot)
directory.setSorting(QDir.DirsLast)
for entry in directory.entryInfoList():
entry_path = entry.absoluteFilePath()
ignored = self.is_ignored(entry_path)
if not ignored and entry.isFile() and entry_path.endswith('php'):
current_path = entry_path[len(self.root) + 1:]
if self.is_fuzzy(current_path, search_string):
paths.append(current_path)
elif not ignored and entry.isDir():
paths = self.recursive(entry_path, search_string, paths)
return paths
def show_counting(self):
directory = QDir(os.getcwd())
directory.setFilter(QDir.Files | QDir.NoDotDot | QDir.NoDotAndDotDot)
directory.setSorting(QDir.Time)
set_filter = ["*.jpg"]
directory.setNameFilters(set_filter)
for show in directory.entryInfoList():
print("%s %s" % (show.size(), show.fileName()))
if "graph" in show.fileName():
out = show.fileName()
break
w = self.countingLabel.width()
h = self.countingLabel.height()
pixmap = QPixmap(out)
pixmap = pixmap.scaled(w, h, QtCore.Qt.KeepAspectRatio)
self.countingLabel.setPixmap(pixmap)
