create main window and trace headers tab

fixhdr/pyside.py

+import os
+import sys
+import time
+from itertools import repeat
+
+from getopt import getopt
+from glob import glob
+from operator import mod
+from PySide2 import QtWidgets, QtGui, QtCore
+
+#from fixhdr.LibTrace import *
+from LibTrace import *
+
+VERSION = "2022.1.0.0"
+
+SPACE = " "
+DATALOC = ""
+
+# vars used for batchmode
+BatchFile = ""
+Endianess = ""
+RunHeaderBatch = 0
+RunTimeBatch = 0
+RunEndianBatch = 0
+BATCHMODE = 0
+
+
+def Usage(err=''):
+    print(err)
+    print("Usage\nfixhdr \nfixhdr -#\nfixhdr -h\nfixhdr [-d DataDirs] "
+          "[-m batch_file] [-t batch_file] [-e endianess]")
+    print("\n       Invoked with no arguements, fixhdr will launch the GUI\n")
+    print("       -d DataDirs - colon separated list of data directories "
+          "[default: cwd]")
+    print("       -e endianess - convert trace headers to endianess ["
+          "big or little]")
+    print("       -h - this usage page")
+    print("       -m batch_file - correct mseed headers in batch mode using "
+          "batch_file")
+    print("       -t batch_file - correct trace timing in batch mode using "
+          "batch_file")
+    print("       -# - print version number")
+    print("      NOTE: -m, -t & -e are mutually exclusive. If you wish to do")
+    print("             both timing and headers, run time corrections first.")
+
+
+# get commandline args
+try:
+    opts, pargs = getopt(sys.argv[1:], 'hd:t:m:e:#')
+except Exception:
+    err = "\nERROR: Invalid command line usage\n"
+    Usage(err)
+    sys.exit(1)
+else:
+    for flag, arg in opts:
+        if flag == "-h":
+            Usage()
+            sys.exit(0)
+        if flag == "-d":
+            DATALOC = arg
+        if flag == "-e":
+            Endianess = arg
+            if Endianess != "big" and Endianess != "little":
+                err = "\nError: Endianess must be either 'big' or 'little'"
+                Usage(err)
+            RunEndianBatch = 1
+            BATCHMODE = 1
+        if flag == "-m":
+            BatchFile = arg
+            RunHeaderBatch = 1
+            BATCHMODE = 1
+        if flag == "-t":
+            BatchFile = arg
+            RunTimeBatch = 1
+            BATCHMODE = 1
+        if flag == "-#":
+            print(VERSION)
+            sys.exit(0)
+    if RunTimeBatch + RunHeaderBatch + RunEndianBatch > 1:
+        err = ("\nERROR: Cannot correct timing, headers and/or endianess "
+               "simultaniously\n")
+        Usage(err)
+        sys.exit(1)
+print("\n", os.path.basename(sys.argv[0]), VERSION)
+
+
+# main
+class MainWindow(QtWidgets.QWidget):
+    def __init__(self, BatchFile=''):
+        super().__init__()
+        self.resize(600,400)
+        self.setWindowTitle("fixhdr " + VERSION)
+
+        # set timezone to UTC
+        os.environ['TZ'] = 'GMT'
+        time.tzname = ('GMT', 'GMT')
+        time.timezone = 0
+
+        # set global vars
+        self.initVars()
+
+        # build main window and tabs
+        self.buildWindow()
+        self.buildTraceHeaders()
+        self.buildGlobalModify()
+        self.buildTimeShift()
+        self.buildEndianess()
+        self.buildLog()
+        self.buildHelp()
+        self.buildMSEEDformat()
+
+    def initVars(self):
+        """
+        Initialize variables, lists, and dictionaries 
+        associated with traces and trace headers
+        """
+
+        self.e_width = 24
+        self.OldSID = ""
+        self.DataDirs = ""
+        self.StatSel = ""
+        self.StatSelList = []
+        self.TraceDict = {}
+        self.UpdateHdrDict = {}
+        self.StatChanList = []
+        self.StationName = ""
+        self.Station = ""
+        self.NetCode = ""
+        self.SampleRate = 0
+        self.Channel = ""
+        self.LocCode = ""
+        self.StatChanListVars = [
+            ["Stat:Chan:Loc:Net:Sps", self.StationName],
+            ["Station_Name", self.Station],
+            ["Channel", self.Channel],
+            ["Location_Code", self.LocCode],
+            ["Network_Code", self.NetCode],
+            ["Sample_Rate", self.SampleRate]
+        ]
+
+        # New Values on Trace Headers notebook
+        self.NStation = ""
+        self.NChannel = ""
+        self.NLocCode = ""
+        self.NNetCode = ""
+        self.NewVars = [
+            ["Station_Name", self.NStation],
+            ["Channel", self.NChannel],
+            ["Location_Code", self.NLocCode],
+            ["Network_Code", self.NNetCode]
+        ]
+
+        # Modified button on Trace Headers notebook
+        # denote whether applied updates or not
+        self.SetStat = 0
+        self.SetChan = 0
+        self.SetLoc = 0
+        self.SetNet = 0
+        self.SetHdrVars = [
+            ["Station_Name", self.SetStat],
+            ["Channel", self.SetChan],
+            ["Location_Code", self.SetLoc],
+            ["Network_Code", self.SetNet]
+        ]
+        self.SetHdrDict = {}
+
+        # vars used in global nb
+        self.GlobalStation = ""
+        self.GlobalChannel = ""
+        self.GlobalLocCode = ""
+        self.GlobalSps = ""
+        self.GlobalNetCode = ""
+
+        # global update variables
+        self.GUStation = ""
+        self.GUChannel = ""
+        self.GULocCode = ""
+        self.GUSps = ""
+        self.GUNetCode = ""
+
+        # lists for dropdown menus on Global Modify notebook
+        self.StationList = []
+        self.ChannelList = []
+        self.LocList = []
+        self.NetList = []
+        self.SpsList = []
+
+        self.GlobalListVars = [
+            ["Station_Name", self.GlobalStation, self.StationList,
+             self.GUStation],
+            ["Channel", self.GlobalChannel, self.ChannelList, self.GUChannel],
+            ["Location_Code", self.GlobalLocCode, self.LocList,
+             self.GULocCode],
+            ["Network_Code", self.GlobalNetCode, self.NetList,
+             self.GUNetCode],
+            ["Sample_Rate", self.GlobalSps, self.SpsList, self.GUSps]
+        ]
+
+        # vars used in TimeShift nb
+        self.TimeShiftName = ""
+        self.OldShiftName = ""
+        self.TSStartTime = ""
+        self.TSEndTime = ""
+        self.TSShift = 0
+        self.TypeTimeCor = 0
+        self.TimeTagQuest = ""
+        self.WildLoc = 0
+        self.PopUpHelp = 0
+        self.UpdateTimeDict = {}
+        self.TraceTimeDict = {}
+        self.SetTimeDict = {}
+        self.TimeCorrectedFiles = {}
+
+        self.ShiftVars = [
+            ["Start_Time", self.TSStartTime],
+            ["End_Time", self.TSEndTime],
+            ["Time_Shift_sec", self.TSShift],
+            ["Time_Tag_Quest", self.TimeTagQuest],
+            ["Corr_Type", self.TypeTimeCor]
+        ]
+
+        self.SetTime = 0
+        self.SetTQuest = 0
+        self.SetTimeVars = [
+            ["Time_Shift_sec", self.SetTime],
+            ["Time_Tag_Quest", self.SetTQuest]
+        ]
+
+        # vars used in Endianess nb
+        self.Savefile = ""
+        self.DataDirList = []
+        self.BigEndianList = []
+        self.LittleEndianList = []
+        self.NumLittleFiles = 0
+        self.NumBigFiles = 0
+        self.ToEndian = ""
+
+        # log vars and lists
+        self.LogList = [
+            ["All", 0],
+            ["Header", 1],
+            ["Timing", 2],
+            ["Endian", 3]
+        ]
+
+        self.LogVar = 0
+        self.LogAll = []
+        self.LogHeader = []
+        self.LogTime = []
+        self.LogEndian = []
+        self.LogSummary = []
+
+        # activity and warning vars
+        self.ActiveText = ""
+        self.AlteredText = ""
+
+        # set of vars watched for active threading event
+        self.RunBuilddb = 0
+        self.RunModHdrs = 0
+        self.RunApplyTimeCor = 0
+        self.RunChangeEndian = 0
+        self.FoundFiles = 0
+        self.RWErrors = 0
+        self.CancelWarn = 0
+        self.IgnoreWarn = 0
+        self.UseNewOnlyTime = 0
+
+        # if DATALOC not set on commandline set here
+        if DATALOC:
+            self.DataDirs = DATALOC
+        else:
+            self.DataDirs = os.getcwd()
+
+        # if a batch file exists on the command line,
+        # we just run necessary processes for modifying headers
+        self.BatchFile = ""
+        self.BatchFile = BatchFile
+
+        # INSERT BATCHMODE HERE
+        """
+        if BATCHMODE:
+            # address endian change first since unique i.e. no batchfile
+            print("Running in Batch Mode")
+            if RunEndianBatch:
+                print("Changing Endianess to: ", Endianess)
+                self.ToEndian.set(Endianess)
+                self.RunChangeEndian.set(1)
+                print("Finding Files beneath directory: ",
+                      self.DataDirs.get())
+                self.BuildTrcList()
+                self.ChangeEndian()
+            else:
+                if RunTimeBatch:
+                    print("Correcting Timing")
+                if RunHeaderBatch:
+                    print("Correcting Headers")
+
+                print("Using Template: ", self.BatchFile.get())
+                self.LoadTemplate()
+                if RunTimeBatch:
+                    if not self.UpdateTimeDict:
+                        print("No Timing Corrections in Template.\nDone")
+                        sys.exit(1)
+                    print("Finding Files beneath directory: ",
+                          self.DataDirs.get())
+                    self.BuildTrcList()
+                    self.RunApplyTimeCor.set(1)
+                    self.ApplyTimeCor()
+                if RunHeaderBatch:
+                    if not self.UpdateHdrDict:
+                        print("No Header Corrections in Template.\nDone")
+                        sys.exit(1)
+                    print("Finding Files beneath directory: ",
+                          self.DataDirs.get())
+                    self.BuildTrcList()
+                    self.RunModHdrs.set(1)
+                    self.ModHdrs()
+            print("Done")
+            sys.exit(0)
+            """
+
+    def buildWindow(self):
+        """
+        Build widgets for root window
+        """
+
+        # tab widget
+        self.tabWidget = QtWidgets.QTabWidget()
+        self.tabWidget.addTab(QtWidgets.QWidget(), "Trace Headers")
+        self.tabWidget.addTab(QtWidgets.QWidget(), "Global Modify")
+        self.tabWidget.addTab(QtWidgets.QWidget(), "Time Shift")
+        self.tabWidget.addTab(QtWidgets.QWidget(), "Endianess")
+        self.tabWidget.addTab(QtWidgets.QWidget(), "Log")
+        self.tabWidget.addTab(QtWidgets.QWidget(), "Help")
+        self.tabWidget.addTab(QtWidgets.QWidget(), "MSEED format")
+
+        # save
+        self.save_b = QtWidgets.QPushButton("Save Template")
+        self.save_b.clicked.connect(self.clickedSaveTemplate)
+        self.save_b.setToolTip("Export saved header & \ntiming corrections to file")
+        self.save_b.setStyleSheet("QPushButton::hover{background-color:green;}")
+
+        # load
+        self.load_b = QtWidgets.QPushButton("Load Template")
+        self.load_b.clicked.connect(self.clickedLoadTemplate)
+        self.load_b.setToolTip("Import saved header &\ntiming corrections file")
+        self.load_b.setStyleSheet("QPushButton::hover{background-color:green;}")
+
+        # exit
+        exit_b = QtWidgets.QPushButton("Exit")
+        exit_b.clicked.connect(lambda: quit())
+        exit_b.setToolTip("Exit program")
+        exit_b.setStyleSheet("QPushButton::hover{background-color:red;}")
+
+        # groupbox for buttons/widgets under tab widget
+        box = QtWidgets.QGroupBox()
+        
+        self.PopUpHelp_cb = QtWidgets.QCheckBox("PopUp Help")
+        self.PopUpHelp_cb.setChecked(True)
+        self.PopUpHelp_cb.clicked.connect(self.clickedPopUpHelp)
+        self.PopUpHelp_cb.setToolTip("Toggles 'PopUp Help' on and off")
+
+
+        # groupbox layout
+        boxLayout = QtWidgets.QHBoxLayout()
+        boxLayout.addWidget(self.PopUpHelp_cb)
+        boxLayout.addStretch()
+        boxLayout.addWidget(self.load_b)
+        boxLayout.addWidget(self.save_b)
+        boxLayout.addWidget(exit_b)
+        boxLayout.setMargin(0)
+        box.setLayout(boxLayout)
+
+        # info bar
+        self.InfoString_l = QtWidgets.QLineEdit()
+        self.InfoString_l.setReadOnly(True)
+        self.InfoString_l.setStyleSheet("background-color:yellow")
+
+        # main window layout
+        mainLayout = QtWidgets.QVBoxLayout()
+        mainLayout.addWidget(self.tabWidget)
+        mainLayout.addWidget(box)
+        mainLayout.addWidget(self.InfoString_l)
+        mainLayout.setMargin(0)
+        self.setLayout(mainLayout)
+
+
+    def buildTraceHeaders(self):
+        """
+        Build widgets for Trace Headers tab for selecting traces, building trace database, and correcting
+        headers one key at a time
+        """
+
+        # data directory find and clear buttons, and entry field
+        # button to build trace db
+        box1 = QtWidgets.QGroupBox()
+
+        DataDirs_l = QtWidgets.QLabel("Data Directories:")
+        DataDirs_l.setToolTip("Search path(s) for finding mseed files.\n"
+                              "Colon separate multiple entries")
+        
+        self.DataDir_le = QtWidgets.QLineEdit()
+        self.DataDir_le.setText(self.DataDirs)
+        
+        self.BuildTrcList_b = QtWidgets.QPushButton("Build Trace db")
+        self.BuildTrcList_b.setStyleSheet("""
+                                          QPushButton{background-color:lightblue;}
+                                          QPushButton::hover{background-color:green;}
+                                          """)
+        self.BuildTrcList_b.setToolTip("Build a Trace db using\n"
+                                       "'Data Directories'\n"
+                                       "as top level directories")
+
+        self.FindDataDir_b = QtWidgets.QPushButton("Find")
+        self.FindDataDir_b.setStyleSheet("QPushButton::hover{background-color:green;}")
+        self.FindDataDir_b.setToolTip("Dialog window to\n"
+                                      "select Data Directories")
+
+        self.ClearDataDir_b = QtWidgets.QPushButton("Clear")
+        self.ClearDataDir_b.setStyleSheet("QPushButton::hover{background-color:orange;}")
+        self.ClearDataDir_b.setToolTip("Clears 'Data Directories' entry")
+
+        layout1 = QtWidgets.QHBoxLayout(box1)
+        layout1.addWidget(DataDirs_l)
+        layout1.addWidget(self.DataDir_le)
+        layout1.addWidget(self.BuildTrcList_b)
+        layout1.addWidget(self.FindDataDir_b)
+        layout1.addWidget(self.ClearDataDir_b)
+        layout1.setMargin(0)
+
+        # station selection frame and entry box
+        box2 = QtWidgets.QGroupBox()
+
+        StatSel_l = QtWidgets.QLabel("Find only stations (colon separated list):")
+        
+        self.StatSel_le = QtWidgets.QLineEdit()
+        self.StatSel_le.setToolTip("Filter trace search\n"
+                                   "for these stations")
+        
+        self.ClearStatSel_b = QtWidgets.QPushButton("Clear")
+        self.ClearStatSel_b.setStyleSheet("QPushButton::hover{background-color:orange;}")
+        self.ClearStatSel_b.setToolTip("Clear station filter")
+
+        layout2 = QtWidgets.QHBoxLayout(box2)
+        layout2.addWidget(StatSel_l)
+        layout2.addWidget(self.StatSel_le)
+        layout2.addWidget(self.ClearStatSel_b)
+        layout2.setMargin(0)
+
+        # global substitutions
+        box3 = QtWidgets.QGroupBox()
+
+        layout3 = QtWidgets.QHBoxLayout(box3)
+        layout31 = QtWidgets.QFormLayout()
+        layout32 = QtWidgets.QFormLayout()
+        layout3.addLayout(layout31)
+        layout3.addLayout(layout32)
+
+        for widget in self.StatChanListVars:
+            if widget[0] == "Stat:Chan:Loc:Net:Sps":
+                layout31.addRow(QtWidgets.QLabel(widget[0]), QtWidgets.QComboBox())
+            else:
+                layout31.addRow(QtWidgets.QLabel(widget[0]), QtWidgets.QLineEdit())
+        
+        layout32.addRow(QtWidgets.QLabel("New Values"), QtWidgets.QLabel("Applied"))
+
+        for _ in repeat(None, 4):
+            layout32.addRow(QtWidgets.QLineEdit(), QtWidgets.QCheckBox())
+
+
+        # 4th groupbox
+        box4 = QtWidgets.QGroupBox()
+        box4.setStyleSheet("QGroupBox{border:0px;}")
+
+        self.ListTraces_b = QtWidgets.QPushButton("List Traces")
+        self.ListTraces_b.setStyleSheet("QPushButton::hover{background-color:green;}")
+        self.ListTraces_b.setToolTip("Lists traces for selected\n"
+                                     "Stat:Chan:Loc:Net:Sps")
+
+        self.ModHdr_b = QtWidgets.QPushButton("Modify Headers")
+        self.ModHdr_b.setStyleSheet("""
+                                    QPushButton{background-color:lightblue;}
+                                    QPushButton::hover{background-color:green;}
+                                    """)
+        self.ModHdr_b.setToolTip("Modifies headers using current entries\n"
+                                 "and all entries in 'Update Dictionary")
+        
+        self.ClearCurrent_b = QtWidgets.QPushButton("Clear Current")
+        self.ClearCurrent_b.setStyleSheet("QPushButton::hover{background-color:orange;}")
+        self.ClearCurrent_b.setToolTip("Clears entries in 'Update Dictionary'\n"
+                                       "for selected Stat:Chan:Loc:Net:Sps")
+
+        self.ClearUpdateDict_b = QtWidgets.QPushButton("Clear Update Dictionary")
+        self.ClearUpdateDict_b.setStyleSheet("QPushButton::hover{background-color:orange;}")
+        self.ClearUpdateDict_b.setToolTip("Clears all entries\n"
+                                             "in 'Update Dictionary'")
+
+        layout4 = QtWidgets.QHBoxLayout(box4)
+        layout4.addWidget(self.ListTraces_b)
+        layout4.addStretch()
+        layout4.addWidget(self.ModHdr_b)
+        layout4.addWidget(self.ClearCurrent_b)
+        layout4.addWidget(self.ClearUpdateDict_b)
+        layout4.setMargin(0)
+
+        tabLayout = QtWidgets.QVBoxLayout(self.tabWidget.widget(0))
+        tabLayout.addWidget(box1)
+        tabLayout.addWidget(box2)
+        tabLayout.addWidget(box3)
+        tabLayout.addStretch()
+        tabLayout.addWidget(box4)
+        tabLayout.setMargin(0)
+
+
+    def buildGlobalModify(self):
+        """
+        Build widgets for Global Modify tab for making global changes to headers
+        """
+
+
+    def buildTimeShift(self):
+        """
+        Build widgets for Time Shift tab for applying time corrections to headers
+        """
+
+
+    def buildEndianess(self):
+        """
+        Build widgets for Endianness tab for swapping endianess of headers
+        """
+
+
+    def buildLog(self):
+        """
+        Build widgets for Log tab for keeping track of changes
+        """
+
+
+    def buildHelp(self):
+        """
+        Build widgets for Help tab
+        """
+
+
+    def buildMSEEDformat(self):
+        """
+        Build widgets for MSEED format tab
+        """
+
+
+    def clickedPopUpHelp(self):
+        """
+        """
+        
+
+    def clickedLoadTemplate(self):
+        """
+        """
+
+
+    def clickedSaveTemplate(self):
+        """
+        """
+    
+    def clickedBuildTrace(self):
+        """
+        """
+    
+    def clickedFind(self):
+        """
+        """
+
+
+
+if __name__ == "__main__":
+    app = QtWidgets.QApplication()
+    window = MainWindow()
+    window.show()
+    sys.exit(app.exec_())
+