Skip to content
Snippets Groups Projects
Commit 6763e1ce authored by Tammo Jan Dijkema's avatar Tammo Jan Dijkema
Browse files

Offset add argument

parent 3f3324d6
No related branches found
No related tags found
No related merge requests found
......@@ -35,7 +35,7 @@ from argparse import ArgumentParser
class PowerPlot:
"""A power plot with two time scales"""
def __init__(self, fig, yname):
def __init__(self, fig, yname, offset=None):
self.fig = fig
self.ax = fig.add_subplot()
self.ax.set_xlabel("Time (UTC)")
......@@ -46,6 +46,8 @@ class PowerPlot:
formatter.scaled[1 / (24 * 60)] = "%H:%M"
self.ax.xaxis.set_major_formatter(formatter)
if offset is not None:
self.ax.yaxis.get_major_formatter().set_useOffset(offset)
self.min, self.max = None, None
fig.suptitle(yname, fontsize=16)
......@@ -94,7 +96,7 @@ class MyToolbar(NavigationToolbar):
class PowerPlotMainWindow(QMainWindow):
"""QT Application that reads data from stdin, plots in a PowerPlot"""
def __init__(self, col_num=1):
def __init__(self, offset=None, col_num=1):
super().__init__()
self.setStyleSheet("background-color: white;")
self.autoscale_x = True
......@@ -151,7 +153,7 @@ class PowerPlotMainWindow(QMainWindow):
# fig.canvas.manager.set_window_title("Dwingeloo Radio Telescope")
self.powerplot = PowerPlot(fig, yname)
self.powerplot = PowerPlot(fig, yname, offset=offset)
self.read_stdin()
self.stdin_notifier = QSocketNotifier(
......@@ -202,6 +204,11 @@ def handle_interrupt_signal(signum, frame):
if __name__ == "__main__":
parser = ArgumentParser(description="Read files and numbers from stdin and plot them")
parser.add_argument("-o", "--offset", help="Offset along y-axis", type=float, default=None)
args = parser.parse_args()
app = QApplication(sys.argv)
icon_path = os.path.join(
os.path.dirname(os.path.realpath(__file__)), "sag_a_icon.png"
......@@ -210,7 +217,7 @@ if __name__ == "__main__":
app.setWindowIcon(QtGui.QIcon(icon_path))
except:
pass
main_window = PowerPlotMainWindow()
main_window = PowerPlotMainWindow(offset=args.offset)
main_window.show()
signal.signal(signal.SIGINT, handle_interrupt_signal)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment