| |
|
| | from ..common.Common import * |
| | from ..common.Execute import * |
| | from ..common.Log import * |
| | from ..common.FileTypes import * |
| |
|
| | class PhotoInfo: |
| | |
| | |
| | |
| | |
| | |
| | |
| |
|
| | def __init__(self, file_path): |
| | |
| | command = "ffprobe -v error -select_streams v:0 -show_entries stream=codec_name,height,width,color_space -of csv=s=x:p=0 " + file_path |
| | |
| | stdout, _, errorCode = ExecuteCommand(command) |
| |
|
| | stdout = stdout.split("\n")[0] |
| | stdout = stdout.strip() |
| |
|
| | info = stdout.split("x") |
| |
|
| | self.file_path = file_path |
| | self.file_size = GetFileSize(file_path) |
| |
|
| | if len(info) < 4: |
| | log.error("File " + file_path + " has size " + str(self.Size)) |
| | log.error("Error running command (return code: " + str(errorCode) + "): " + command + " | Returned string is " + stdout + " | Array is " + str(info)) |
| |
|
| | index = 0 |
| |
|
| | self.image_codec = info[index] |
| | index = index + 1 |
| |
|
| | self.width = int(info[index]) |
| | index = index + 1 |
| |
|
| | self.height = int(info[index]) |
| | index = index + 1 |
| |
|
| | self.color_space = info[index] |
| | index = index + 1 |
| |
|
| | if IsPNGFile(file_path): |
| | self.bit_depth = 10 |
| | elif IsJPGFile(file_path): |
| | self.bit_depth = 8 |
| | else: |
| | log.fatal("Unsupported photo file " + file_path) |
| |
|
| | def String(self): |
| | return json.dumps(self, default=lambda o: o.__dict__, |
| | sort_keys=True, indent=4) |
| |
|
| | def Print(self, stream): |
| | stream.write(self.String()) |
