LogarithmPlotter/run.py

94 lines
3 KiB
Python

"""
* Logarithm Graph Creator - Create graphs with logarithm scales.
* Copyright (C) 2020 Ad5001
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
"""
from PySide2.QtWidgets import QApplication, QFileDialog
from PySide2.QtQml import QQmlApplicationEngine, qmlRegisterType
from PySide2.QtCore import Qt, QObject, Signal, Slot, Property
import os
import tempfile
from platform import release as os_release
from json import dumps
from sys import platform
tempfile = tempfile.mkstemp(suffix = '.png')[1]
def get_linux_theme():
des = {
"KDE": "org.kde.desktop", # org.kde.desktop resolves to universal in PySide2.
"gnome": "default",
"lxqt": "fusion",
"mate": "fusion",
}
if "XDG_SESSION_DESKTOP" in os.environ:
return des[os.environ["XDG_SESSION_DESKTOP"]] if os.environ["XDG_SESSION_DESKTOP"] in des else "fusion"
else:
# Android
return "Material"
os.environ["QT_QUICK_CONTROLS_STYLE"] = {
"linux": get_linux_theme(),
"freebsd": get_linux_theme(),
"win32": "universal" if os_release == "10" else "fusion",
"cygwin": "fusion",
"darwin": "imagine"
}[platform]
class Helper(QObject):
@Slot(str, str)
def write(self, filename, filedata):
if os.path.exists(os.path.dirname(os.path.realpath(filename))):
f = open(os.path.realpath(filename), 'w', -1, 'utf8')
f.write(filedata)
f.close()
@Slot(str, result=str)
def load(self, filename):
if os.path.exists(os.path.realpath(filename)):
f = open(os.path.realpath(filename), 'r', -1, 'utf8')
data = f.read()
f.close()
return data
return '{}'
@Slot(result=str)
def gettmpfile(self):
global tempfile
return tempfile
@Slot()
def copyImageToClipboard(self):
global tempfile
# TODO: Better copy system
os.system("xclip -selection clipboard -t image/png -i " + tempfile)
app = QApplication([])
app.setApplicationName("Logarithmic Plotter")
app.setOrganizationName("Ad5001")
engine = QQmlApplicationEngine()
helper = Helper()
engine.rootContext().setContextProperty("Helper", helper)
engine.addImportPath(os.path.realpath(os.path.join(os.getcwd(), "qml")))
engine.load(os.path.realpath(os.path.join(os.getcwd(), "qml", "LogGraph.qml")))
if not engine.rootObjects():
print("No root object")
exit(-1)
app.exec_()
os.remove(tempfile)