diff --git a/pylot/core/util/widgets.py b/pylot/core/util/widgets.py index 761dbdfc..35ba0949 100644 --- a/pylot/core/util/widgets.py +++ b/pylot/core/util/widgets.py @@ -1726,9 +1726,9 @@ class PickDlg(QDialog): spe = picks['spe'] if picktype == 'manual': + color = pick_color_plt(picktype, phaseID, quality) if not textOnly: linestyle_mpp, width_mpp = pick_linestyle_plt(picktype, 'mpp') - color = pick_color_plt(picktype, phaseID, quality) if picks['epp'] and picks['lpp']: ax.fill_between([epp, lpp], ylims[0], ylims[1], alpha=.25, color=color, label='EPP, LPP') @@ -1744,7 +1744,7 @@ class PickDlg(QDialog): ax.plot([mpp, mpp], ylims, color=color, linestyle=linestyle_mpp, linewidth=width_mpp, label='{}-Pick (NO PICKERROR)'.format(phase), picker=5) # append phase text (if textOnly: draw with current ylims) - self.phaseText.append(ax.text(mpp, ylims[1], phase)) + self.phaseText.append(ax.text(mpp, ylims[1], phase, color=color)) elif picktype == 'auto': color = pick_color_plt(picktype, phaseID, quality) linestyle_mpp, width_mpp = pick_linestyle_plt(picktype, 'mpp') @@ -1754,7 +1754,7 @@ class PickDlg(QDialog): ax.vlines(mpp, ylims[0], ylims[1], color=color, linestyle=linestyle_mpp, linewidth=width_mpp, picker=5, label='{}-Autopick (quality: {})'.format(phase, quality)) # append phase text (if textOnly: draw with current ylims) - self.phaseText.append(ax.text(mpp, ylims[1], phase)) + self.phaseText.append(ax.text(mpp, ylims[1], phase, color=color)) else: raise TypeError('Unknown picktype {0}'.format(picktype))