def update_connect_status(self, item):
try:
img = QImage()
if item[0]:
logger.info('connect to {} success, report: {}'.format(
self.handler.addr, self.handler.report_type))
if img.load(connect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
self.btn_connect.setText(i18n[self.lang]['Disconnect'])
self.btn_connect.setStyleSheet('''color: red;font:bold;''')
self.set_disable(False)
else:
logger.info('disconnect from or failed connect {}'.format(
self.handler.addr))
self.handler.cmd_que.queue.clear()
if img.load(disconnect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
self.btn_connect.setText(i18n[self.lang]['Connect'])
self.btn_connect.setStyleSheet(
'''color: green;font:bold;''')
self.set_disable(True)
if item[1]:
if img.load(connect_icon_path):
self.label_reported.setPixmap(QPixmap.fromImage(img))
else:
if img.load(disconnect_icon_path):
self.label_reported.setPixmap(QPixmap.fromImage(img))
except Exception as e:
print(e)
def _set_common_top_ui(self):
top_frame = QFrame()
top_frame.setMaximumHeight(60)
top_layout = QVBoxLayout(top_frame)
self.layout.addWidget(top_frame)
common_top_frame = QFrame()
common_top_frame.setMinimumHeight(50)
common_top_frame.setMaximumHeight(50)
common_top_layout = QHBoxLayout(common_top_frame)
top_layout.addWidget(common_top_frame)
common_top_layout.addStretch(0)
label = QLabel(i18n[self.lang]['Type'] + ':')
self.label_type = QLabel('')
self.label_type.setStyleSheet('''color: gray;font:bold;''')
common_top_layout.addWidget(label)
common_top_layout.addWidget(self.label_type)
label = QLabel(i18n[self.lang]['Mode'] + ':')
self.label_mode = QLabel('')
self.label_mode.setStyleSheet('''color: gray;font:bold;''')
common_top_layout.addWidget(label)
common_top_layout.addWidget(self.label_mode)
label = QLabel(i18n[self.lang]['HardwareVersion'] + ':')
self.label_hard_version = QLabel('')
self.label_hard_version.setStyleSheet('''color: gray;font:bold;''')
common_top_layout.addWidget(label)
common_top_layout.addWidget(self.label_hard_version)
label = QLabel(i18n[self.lang]['FirmwareVersion'] + ':')
self.label_firm_version = QLabel('')
self.label_firm_version.setStyleSheet('''color: gray;font:bold;''')
common_top_layout.addWidget(label)
common_top_layout.addWidget(self.label_firm_version)
label_1 = QLabel(i18n[self.lang]['Connected'] + ':')
self.label_connected = QLabel()
img = QImage()
self.label_connected.setMaximumHeight(20)
self.label_connected.setMaximumWidth(20)
self.label_connected.setScaledContents(True)
if img.load(disconnect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
self.lnt_addr = QLineEdit('COM12')
self.lnt_addr.setMaximumWidth(50)
self.lnt_addr.setMinimumWidth(30)
self.btn_connect = QPushButton(i18n[self.lang]['Connect'])
# self.btn_connect.setMaximumWidth(50)
# common_top_layout.addStretch(0)
common_top_layout.setSpacing(10)
common_top_layout.addWidget(label_1)
common_top_layout.addWidget(self.label_connected)
common_top_layout.addWidget(self.lnt_addr)
common_top_layout.addWidget(self.btn_connect)
def _load_texture(self, file, id):
path = pt.join(self.IMAGE_PATH, file)
img = QImage()
if img.load(path):
self.textures[id] = QGLWidget.convertToGLFormat(img)
print(f'Texture {path} loaded...[OK]')
else:
print(f'Texture {path} loaded...[FAIL]')
def load_image(self):
fname = QFileDialog.getOpenFileName(self.main_ui.window, 'Open file',
'', '*.svg')
if fname and fname[0]:
img = QImage()
if img.load(fname[0]):
self.label_img.setPixmap(QPixmap.fromImage(img))
with open(fname[0], 'rb') as f:
self.handler.source = f.read()
self.handler.template = None
self.gcode = None
self.up_frame.show()
def update_connect_status(self, item):
try:
img = QImage()
if item and self.status != 1:
self.status = 1
logger.info('connect to {} success'.format(self.handler.port))
if img.load(connect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
self.btn_connect.setText(i18n[self.lang]['Disconnect'])
self.btn_connect.setStyleSheet('''color: red;font:bold;''')
self.set_disable(False)
elif not item and self.status != 0:
self.status = 0
logger.info('disconnect from {0} or failed connect {0}'.format(self.handler.port))
self.handler.cmd_que.queue.clear()
if img.load(disconnect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
self.btn_connect.setText(i18n[self.lang]['Connect'])
self.btn_connect.setStyleSheet('''color: green;font:bold;''')
self.set_disable(True)
except Exception as e:
print(e)
def load_jxr_data(data):
with TemporaryDirectory() as tdir:
if iswindows and isinstance(tdir, type('')):
tdir = tdir.encode('mbcs')
with lopen(os.path.join(tdir, 'input.jxr'), 'wb') as f:
f.write(data)
cmd = [get_exe_path('JxrDecApp'), '-i', 'input.jxr', '-o', 'output.tif', '-c', '0']
creationflags = 0x08 if iswindows else 0
subprocess.Popen(cmd, cwd=tdir, stdout=lopen(os.devnull, 'wb'), stderr=subprocess.STDOUT, creationflags=creationflags).wait()
i = QImage()
if not i.load(os.path.join(tdir, 'output.tif')):
raise NotImage('Failed to convert JPEG-XR image')
return i
def encode_jpeg(file_path, quality=80):
from calibre.utils.speedups import ReadOnlyFileBuffer
quality = max(0, min(100, int(quality)))
exe = get_exe_path('cjpeg')
cmd = [exe] + '-optimize -progressive -maxmemory 100M -quality'.split() + [str(quality)]
img = QImage()
if not img.load(file_path):
raise ValueError('%s is not a valid image file' % file_path)
ba = QByteArray()
buf = QBuffer(ba)
buf.open(QBuffer.WriteOnly)
if not img.save(buf, 'PPM'):
raise ValueError('Failed to export image to PPM')
return run_optimizer(file_path, cmd, as_filter=True, input_data=ReadOnlyFileBuffer(ba.data()))
def encode_jpeg(file_path, quality=80):
from calibre.utils.speedups import ReadOnlyFileBuffer
quality = max(0, min(100, int(quality)))
exe = get_exe_path('cjpeg')
cmd = [exe] + '-optimize -progressive -maxmemory 100M -quality'.split() + [str(quality)]
img = QImage()
if not img.load(file_path):
raise ValueError('%s is not a valid image file' % file_path)
ba = QByteArray()
buf = QBuffer(ba)
buf.open(QBuffer.WriteOnly)
if not img.save(buf, 'PPM'):
raise ValueError('Failed to export image to PPM')
return run_optimizer(file_path, cmd, as_filter=True, input_data=ReadOnlyFileBuffer(ba.data()))
def _set_up_frame_ui(self):
self.up_frame = QFrame()
self.up_frame.setMinimumHeight(300)
self.up_frame.setMaximumHeight(500)
self.up_layout = QHBoxLayout(self.up_frame)
up_left_frame = QFrame()
# up_left_frame.setMinimumWidth(self.main_ui.window.geometry().width() / 2)
# up_left_frame.setMaximumWidth(self.geometry().width() / 2)
up_right_frame = QFrame()
# up_right_frame.setMinimumWidth(self.main_ui.window.geometry().width() / 2)
# up_right_frame.setMaximumWidth(self.geometry().width() / 2)
self.up_left_layout = QHBoxLayout(up_left_frame)
self.up_right_layout = QHBoxLayout(up_right_frame)
self.up_layout.addWidget(up_left_frame)
# self.up_layout.addWidget(up_right_frame)
self.layout.addWidget(self.up_frame)
self.label_img = QLabel()
# self.label_img.setMaximumHeight(320)
# self.label_img.setMaximumWidth(480)
# self.label_img.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
# self.label_img.setScaledContents(True)
img = QImage()
if img.load(os.path.join(icon_path, 'tmp.svg')):
self.label_img.setPixmap(QPixmap.fromImage(img))
with open(os.path.join(icon_path, 'tmp.svg'), 'rb') as f:
self.handler.source = f.read()
self.up_left_layout.addWidget(self.label_img)
self.label_img_preview = QLabel()
# self.label_img_preview.setMaximumHeight(320)
# self.label_img_preview.setMaximumWidth(480)
self.label_img_preview.setDisabled(True)
# # self.label_img_preview.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
# # self.label_img_preview.setScaledContents(True)
# data = np.zeros(320 * 240)
# img = QImage(data, 320, 240, QImage.Format_RGB888)
# pixmap = QPixmap.fromImage(img)
# self.label_img_preview.setPixmap(pixmap)
self.up_right_layout.addWidget(self.label_img_preview)
def _populate_covers(self):
'''
Display calibre cover for both unless mismatch
'''
def _fetch_marvin_cover(border_width=0):
'''
Retrieve LargeCoverJpg from cache
'''
#self._log_location('border_width: {0}'.format(border_width))
con = sqlite3.connect(self.marvin_db_path)
with con:
con.row_factory = sqlite3.Row
# Fetch Hash from mainDb
cover_cur = con.cursor()
cover_cur.execute('''SELECT
Hash
FROM Books
WHERE ID = '{0}'
'''.format(self.book_id))
row = cover_cur.fetchone()
book_hash = row[b'Hash']
large_covers_subpath = self.connected_device._cover_subpath(size="large")
cover_path = '/'.join([large_covers_subpath, '%s.jpg' % book_hash])
stats = self.parent.ios.exists(cover_path)
if stats:
self._log("fetching large cover from cache")
#self._log("cover size: {:,} bytes".format(int(stats['st_size'])))
cover_bytes = self.parent.ios.read(cover_path, mode='rb')
m_image = QImage()
m_image.loadFromData(cover_bytes)
if border_width:
# Construct a QPixmap with oversized yellow background
m_image = m_image.scaledToHeight(
self.COVER_ICON_SIZE - border_width * 2,
Qt.SmoothTransformation)
self.m_pixmap = QPixmap(
QSize(m_image.width() + border_width * 2,
m_image.height() + border_width * 2))
m_painter = QPainter(self.m_pixmap)
m_painter.setRenderHints(m_painter.Antialiasing)
m_painter.fillRect(self.m_pixmap.rect(), self.MISMATCH_COLOR)
m_painter.drawImage(border_width,
border_width,
m_image)
else:
m_image = m_image.scaledToHeight(
self.COVER_ICON_SIZE,
Qt.SmoothTransformation)
self.m_pixmap = QPixmap(
QSize(m_image.width(),
m_image.height()))
m_painter = QPainter(self.m_pixmap)
m_painter.setRenderHints(m_painter.Antialiasing)
m_painter.drawImage(0, 0, m_image)
self.marvin_cover.setPixmap(self.m_pixmap)
else:
# No cover available, use generic
self._log("No cached cover, using generic")
pixmap = QPixmap()
pixmap.load(I('book.png'))
pixmap = pixmap.scaled(self.COVER_ICON_SIZE,
self.COVER_ICON_SIZE,
aspectRatioMode=Qt.KeepAspectRatio,
transformMode=Qt.SmoothTransformation)
self.marvin_cover.setPixmap(pixmap)
self.calibre_cover.setMaximumSize(QSize(self.COVER_ICON_SIZE, self.COVER_ICON_SIZE))
self.calibre_cover.setText('')
self.calibre_cover.setScaledContents(False)
self.marvin_cover.setMaximumSize(QSize(self.COVER_ICON_SIZE, self.COVER_ICON_SIZE))
self.marvin_cover.setText('')
self.marvin_cover.setScaledContents(False)
if self.cid:
db = self.opts.gui.current_db
if 'cover_hash' not in self.mismatches:
mi = db.get_metadata(self.cid, index_is_id=True, get_cover=True, cover_as_data=True)
c_image = QImage()
if mi.has_cover:
c_image.loadFromData(mi.cover_data[1])
c_image = c_image.scaledToHeight(self.COVER_ICON_SIZE,
Qt.SmoothTransformation)
self.c_pixmap = QPixmap(QSize(c_image.width(),
c_image.height()))
c_painter = QPainter(self.c_pixmap)
c_painter.setRenderHints(c_painter.Antialiasing)
c_painter.drawImage(0, 0, c_image)
else:
c_image.load(I('book.png'))
c_image = c_image.scaledToWidth(135,
Qt.SmoothTransformation)
# Construct a QPixmap with dialog background
self.c_pixmap = QPixmap(
QSize(c_image.width(),
c_image.height()))
c_painter = QPainter(self.c_pixmap)
c_painter.setRenderHints(c_painter.Antialiasing)
bgcolor = self.palette().color(QPalette.Background)
c_painter.fillRect(self.c_pixmap.rect(), bgcolor)
c_painter.drawImage(0, 0, c_image)
# Set calibre cover
self.calibre_cover.setPixmap(self.c_pixmap)
if self.opts.prefs.get('development_mode', False):
# Show individual covers
_fetch_marvin_cover()
else:
# Show calibre cover on both sides
self.marvin_cover.setPixmap(self.c_pixmap)
else:
# Covers don't match - render with border
# Construct a QImage with the cover sized to fit inside border
c_image = QImage()
cdata = db.cover(self.cid, index_is_id=True)
if cdata is None:
c_image.load(I('book.png'))
self.calibre_cover.setScaledContents(True)
else:
c_image.loadFromData(cdata)
c_image = c_image.scaledToHeight(
self.COVER_ICON_SIZE - self.BORDER_WIDTH * 2,
Qt.SmoothTransformation)
# Construct a QPixmap with yellow background
self.c_pixmap = QPixmap(
QSize(c_image.width() + self.BORDER_WIDTH * 2,
c_image.height() + self.BORDER_WIDTH * 2))
c_painter = QPainter(self.c_pixmap)
c_painter.setRenderHints(c_painter.Antialiasing)
c_painter.fillRect(self.c_pixmap.rect(),self.MISMATCH_COLOR)
c_painter.drawImage(self.BORDER_WIDTH, self.BORDER_WIDTH, c_image)
self.calibre_cover.setPixmap(self.c_pixmap)
# Render Marvin cover with small border if different covers,
# large cover if no cover hash (loaded via OPDS)
border_width = self.BORDER_WIDTH
if self.mismatches['cover_hash']['Marvin'] is None:
border_width = self.BORDER_WIDTH * 3
_fetch_marvin_cover(border_width=border_width)
else:
_fetch_marvin_cover()
def fillwithpic(self, filename):
img = QImage()
img.load(filename)
self.img = img
self.__board = QPixmap(img)
def _set_common_top_ui(self):
top_frame = QFrame()
top_frame.setMaximumHeight(60)
top_layout = QVBoxLayout(top_frame)
self.layout.addWidget(top_frame)
common_top_frame = QFrame()
common_top_frame.setMinimumHeight(50)
common_top_layout = QHBoxLayout(common_top_frame)
top_layout.addWidget(common_top_frame)
label_1 = QLabel(i18n[self.lang]['Connected'] + ':')
self.label_connected = QLabel()
img = QImage()
self.label_connected.setMaximumHeight(20)
self.label_connected.setMaximumWidth(20)
self.label_connected.setScaledContents(True)
if img.load(disconnect_icon_path):
self.label_connected.setPixmap(QPixmap.fromImage(img))
label_2 = QLabel(i18n[self.lang]['Reported'] + ':')
self.label_reported = QLabel()
img = QImage()
self.label_reported.setMaximumHeight(20)
self.label_reported.setMaximumWidth(20)
self.label_reported.setScaledContents(True)
if img.load(disconnect_icon_path):
self.label_reported.setPixmap(QPixmap.fromImage(img))
self.lnt_addr = QLineEdit('192.168.1.182')
self.lnt_addr.setMaximumWidth(100)
self.lnt_addr.setMinimumWidth(60)
self.btn_connect = QPushButton(i18n[self.lang]['Connect'])
# self.btn_connect.setMaximumWidth(50)
# common_top_layout.addStretch(0)
common_top_layout.setSpacing(10)
common_top_layout.addWidget(label_1)
common_top_layout.addWidget(self.label_connected)
common_top_layout.addWidget(label_2)
common_top_layout.addWidget(self.label_reported)
common_top_layout.addWidget(self.lnt_addr)
common_top_layout.addWidget(self.btn_connect)
# common_down_frame = QFrame()
# common_down_layout = QHBoxLayout(common_down_frame)
# common_down_layout.setSpacing(0)
# top_layout.addWidget(common_down_frame)
common_down_layout = common_top_layout
label = QLabel(i18n[self.lang]['WarnCode'] + ':')
self.label_warn_code = QLabel('0')
self.label_warn_code.setStyleSheet('''color: gray;font:bold;''')
common_down_layout.addWidget(label)
common_down_layout.addWidget(self.label_warn_code)
label = QLabel(i18n[self.lang]['ErrorCode'] + ':')
self.label_error_code = QLabel('0')
self.label_error_code.setStyleSheet('''color: gray;font:bold;''')
common_down_layout.addWidget(label)
common_down_layout.addWidget(self.label_error_code)
label = QLabel(i18n[self.lang]['CmdCount'] + ':')
self.label_cmd_count = QLabel('0')
self.label_cmd_count.setStyleSheet('''color: gray;font:bold;''')
common_down_layout.addWidget(label)
common_down_layout.addWidget(self.label_cmd_count)
label = QLabel(i18n[self.lang]['State'] + ':')
self.label_state = QLabel('4')
self.label_state.setStyleSheet('''color: gray;font:bold;''')
common_down_layout.addWidget(label)
common_down_layout.addWidget(self.label_state)
label = QLabel(i18n[self.lang]['Maable'] + ':')
self.label_maable = QLabel('128')
self.label_maable.setStyleSheet('''color: gray;font:bold;''')
common_down_layout.addWidget(label)
common_down_layout.addWidget(self.label_maable)
label = QLabel(i18n[self.lang]['Mtbrake'] + ':')
self.label_mtbrake = QLabel('128')
self.label_mtbrake.setStyleSheet('''color: gray;font:bold;''')
common_down_layout.addWidget(label)
common_down_layout.addWidget(self.label_mtbrake)
def _populate_covers(self):
'''
Display calibre cover for both unless mismatch
'''
def _fetch_marvin_cover(border_width=0):
'''
Retrieve LargeCoverJpg from cache
'''
#self._log_location('border_width: {0}'.format(border_width))
con = sqlite3.connect(self.marvin_db_path)
with con:
con.row_factory = sqlite3.Row
# Fetch Hash from mainDb
cover_cur = con.cursor()
cover_cur.execute('''SELECT
Hash
FROM Books
WHERE ID = '{0}'
'''.format(self.book_id))
row = cover_cur.fetchone()
book_hash = row[b'Hash']
large_covers_subpath = self.connected_device._cover_subpath(
size="large")
cover_path = '/'.join([large_covers_subpath, '%s.jpg' % book_hash])
stats = self.parent.ios.exists(cover_path)
if stats:
self._log("fetching large cover from cache")
#self._log("cover size: {:,} bytes".format(int(stats['st_size'])))
cover_bytes = self.parent.ios.read(cover_path, mode='rb')
m_image = QImage()
m_image.loadFromData(cover_bytes)
if border_width:
# Construct a QPixmap with oversized yellow background
m_image = m_image.scaledToHeight(
self.COVER_ICON_SIZE - border_width * 2,
Qt.SmoothTransformation)
self.m_pixmap = QPixmap(
QSize(m_image.width() + border_width * 2,
m_image.height() + border_width * 2))
m_painter = QPainter(self.m_pixmap)
m_painter.setRenderHints(m_painter.Antialiasing)
m_painter.fillRect(self.m_pixmap.rect(),
self.MISMATCH_COLOR)
m_painter.drawImage(border_width, border_width, m_image)
else:
m_image = m_image.scaledToHeight(self.COVER_ICON_SIZE,
Qt.SmoothTransformation)
self.m_pixmap = QPixmap(
QSize(m_image.width(), m_image.height()))
m_painter = QPainter(self.m_pixmap)
m_painter.setRenderHints(m_painter.Antialiasing)
m_painter.drawImage(0, 0, m_image)
self.marvin_cover.setPixmap(self.m_pixmap)
else:
# No cover available, use generic
self._log("No cached cover, using generic")
pixmap = QPixmap()
pixmap.load(I('book.png'))
pixmap = pixmap.scaled(self.COVER_ICON_SIZE,
self.COVER_ICON_SIZE,
aspectRatioMode=Qt.KeepAspectRatio,
transformMode=Qt.SmoothTransformation)
self.marvin_cover.setPixmap(pixmap)
self.calibre_cover.setMaximumSize(
QSize(self.COVER_ICON_SIZE, self.COVER_ICON_SIZE))
self.calibre_cover.setText('')
self.calibre_cover.setScaledContents(False)
self.marvin_cover.setMaximumSize(
QSize(self.COVER_ICON_SIZE, self.COVER_ICON_SIZE))
self.marvin_cover.setText('')
self.marvin_cover.setScaledContents(False)
if self.cid:
db = self.opts.gui.current_db
if 'cover_hash' not in self.mismatches:
mi = db.get_metadata(self.cid,
index_is_id=True,
get_cover=True,
cover_as_data=True)
c_image = QImage()
if mi.has_cover:
c_image.loadFromData(mi.cover_data[1])
c_image = c_image.scaledToHeight(self.COVER_ICON_SIZE,
Qt.SmoothTransformation)
self.c_pixmap = QPixmap(
QSize(c_image.width(), c_image.height()))
c_painter = QPainter(self.c_pixmap)
c_painter.setRenderHints(c_painter.Antialiasing)
c_painter.drawImage(0, 0, c_image)
else:
c_image.load(I('book.png'))
c_image = c_image.scaledToWidth(135,
Qt.SmoothTransformation)
# Construct a QPixmap with dialog background
self.c_pixmap = QPixmap(
QSize(c_image.width(), c_image.height()))
c_painter = QPainter(self.c_pixmap)
c_painter.setRenderHints(c_painter.Antialiasing)
bgcolor = self.palette().color(QPalette.Background)
c_painter.fillRect(self.c_pixmap.rect(), bgcolor)
c_painter.drawImage(0, 0, c_image)
# Set calibre cover
self.calibre_cover.setPixmap(self.c_pixmap)
if self.opts.prefs.get('development_mode', False):
# Show individual covers
_fetch_marvin_cover()
else:
# Show calibre cover on both sides
self.marvin_cover.setPixmap(self.c_pixmap)
else:
# Covers don't match - render with border
# Construct a QImage with the cover sized to fit inside border
c_image = QImage()
cdata = db.cover(self.cid, index_is_id=True)
if cdata is None:
c_image.load(I('book.png'))
self.calibre_cover.setScaledContents(True)
else:
c_image.loadFromData(cdata)
c_image = c_image.scaledToHeight(
self.COVER_ICON_SIZE - self.BORDER_WIDTH * 2,
Qt.SmoothTransformation)
# Construct a QPixmap with yellow background
self.c_pixmap = QPixmap(
QSize(c_image.width() + self.BORDER_WIDTH * 2,
c_image.height() + self.BORDER_WIDTH * 2))
c_painter = QPainter(self.c_pixmap)
c_painter.setRenderHints(c_painter.Antialiasing)
c_painter.fillRect(self.c_pixmap.rect(), self.MISMATCH_COLOR)
c_painter.drawImage(self.BORDER_WIDTH, self.BORDER_WIDTH,
c_image)
self.calibre_cover.setPixmap(self.c_pixmap)
# Render Marvin cover with small border if different covers,
# large cover if no cover hash (loaded via OPDS)
border_width = self.BORDER_WIDTH
if self.mismatches['cover_hash']['Marvin'] is None:
border_width = self.BORDER_WIDTH * 3
_fetch_marvin_cover(border_width=border_width)
else:
_fetch_marvin_cover()