-
Notifications
You must be signed in to change notification settings - Fork 235
Add Figure.timestamp to plot the GMT timestamp logo #2208
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from all commits
c8206ec
2096760
e0bb6bb
4d6d777
fffe3cd
54f4307
fe8eeac
97574ac
1e8de17
79460fb
7031448
7ad112c
b057ad8
f92d272
cecf6de
7dd74d7
06049a6
876d963
752273e
f6795a9
9eba2a9
92c285a
776981b
c771183
386882e
e3580e7
41f5419
4def360
bc64e8c
78c2307
e6248c2
d989b22
960508f
16c0368
98b6762
b8a7fbb
688f6b0
6e168e1
5857866
01a3cbb
4fa2e72
0c07a48
e4bd94c
0079643
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -518,6 +518,7 @@ def _repr_html_(self): | |
| subplot, | ||
| ternary, | ||
| text, | ||
| timestamp, | ||
| velo, | ||
| wiggle, | ||
| ) | ||
|
|
||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,102 @@ | ||
| """ | ||
| timestamp - Plot the GMT timestamp logo. | ||
| """ | ||
| from packaging.version import Version | ||
| from pygmt import __gmt_version__ | ||
| from pygmt.clib import Session | ||
| from pygmt.exceptions import GMTInvalidInput | ||
| from pygmt.helpers import build_arg_string, is_nonstr_iter | ||
|
|
||
| __doctest_skip__ = ["timestamp"] | ||
|
|
||
|
|
||
| def timestamp( | ||
| self, | ||
| text=None, | ||
| label=None, | ||
| justification="BL", | ||
| offset=("-54p", "-54p"), | ||
| font="Helvetica,black", | ||
| timefmt="%Y %b %d %H:%M:%S", | ||
| ): | ||
| r""" | ||
| Plot the GMT timestamp logo. | ||
|
|
||
| Parameters | ||
| ---------- | ||
| text : None or str | ||
| If ``None``, the current UNIX timestamp is shown in the GMT timestamp | ||
| logo. Set this parameter to replace the UNIX timestamp with a | ||
| custom text string instead. The text must be less than 64 characters. | ||
| *Requires GMT>=6.5.0*. | ||
|
Member
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I'm still unsure what's the best way to document features that require new GMT versions.
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Hm. Is there a similar situation for the Julia wrapper, which can serve as an orientation? |
||
| label : None or str | ||
| The text string shown after the GMT timestamp logo. | ||
yvonnefroehlich marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| justification : str | ||
| Justification of the timestamp. The *justification* is a two-character | ||
| code that is a combination of a horizontal (**L**\ (eft), | ||
| **C**\ (enter), or **R**\ (ight)) and a vertical (**T**\ (op), | ||
| **M**\ (iddle), or **B**\ (ottom)) code. | ||
| offset : str or tuple | ||
| *offset* or (*offset_x*, *offset_y*). | ||
| Offset the anchor point of the timestamp by *offset_x* and *offset_y*. | ||
yvonnefroehlich marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| If a single value *offset* is given, *offset_y* = *offset_x* = | ||
| *offset*. | ||
| font : str | ||
| Font of the timestamp and the optional label. The parameter can't | ||
| change the font color for GMT<=6.4.0, only the font ID. | ||
| timefmt : str | ||
| Format string for the UNIX timestamp. The format string is parsed by | ||
| the C function ``strftime``, so that virtually any text can be used | ||
| (even not containing any time information). | ||
|
|
||
| Examples | ||
| -------- | ||
| >>> # Plot the GMT timestamp logo. | ||
yvonnefroehlich marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| >>> import pygmt | ||
| >>> fig = pygmt.Figure() | ||
| >>> fig.timestamp() | ||
| >>> fig.show() | ||
| <IPython.core.display.Image object> | ||
|
|
||
| >>> # Plot the GMT timestamp logo with a custom label. | ||
yvonnefroehlich marked this conversation as resolved.
Show resolved
Hide resolved
|
||
| >>> fig = pygmt.Figure() | ||
| >>> fig.timestamp(label="Powered by PyGMT") | ||
| >>> fig.show() | ||
| <IPython.core.display.Image object> | ||
| """ | ||
| self._preprocess() # pylint: disable=protected-access | ||
|
|
||
| # Build the options passed to the "plot" module | ||
| kwdict = {"T": True, "U": ""} | ||
| if label is not None: | ||
| kwdict["U"] += f"{label}" | ||
| kwdict["U"] += f"+j{justification}" | ||
|
|
||
| if is_nonstr_iter(offset): # given a tuple | ||
| kwdict["U"] += "+o" + "/".join(f"{item}" for item in offset) | ||
| else: # given a single value | ||
| if "/" not in offset and Version(__gmt_version__) <= Version("6.4.0"): | ||
| # Giving a single offset doesn't work in GMT <= 6.4.0. | ||
| # See https://github.com/GenericMappingTools/gmt/issues/7107. | ||
| kwdict["U"] += f"+o{offset}/{offset}" | ||
| else: | ||
| kwdict["U"] += f"+o{offset}" | ||
|
|
||
| # The +t modifier was added in GMT 6.5.0. | ||
| # See https://github.com/GenericMappingTools/gmt/pull/7127. | ||
| if text is not None: | ||
| if Version(__gmt_version__) < Version("6.5.0"): | ||
| raise GMTInvalidInput("The parameter 'text' requires GMT>=6.5.0.") | ||
| if len(str(text)) > 64: | ||
| raise GMTInvalidInput( | ||
| "The parameter 'text' must be less than 64 characters." | ||
| ) | ||
| kwdict["U"] += f"+t{text}" | ||
|
|
||
| with Session() as lib: | ||
| lib.call_module( | ||
| module="plot", | ||
| args=build_arg_string( | ||
| kwdict, confdict={"FONT_LOGO": font, "FORMAT_TIME_STAMP": timefmt} | ||
| ), | ||
| ) | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| outs: | ||
| - md5: b4ca6cb54d80463606bb3d28eb077227 | ||
| size: 1668 | ||
| path: test_timestamp.png |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| outs: | ||
| - md5: e1c3c80a85ebaadc0a820bcc7954a539 | ||
| size: 2840 | ||
| path: test_timestamp_font.png |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| outs: | ||
| - md5: c2bfca8624e4bf966eeb57bbc4aa0726 | ||
| size: 8843 | ||
| path: test_timestamp_justification.png |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| outs: | ||
| - md5: 868d6a5912b586bc7e095b9923ce8d83 | ||
| size: 3036 | ||
| path: test_timestamp_label.png |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,4 @@ | ||
| outs: | ||
| - md5: be0021731b881e32e3d8a46d6fa8abc3 | ||
| size: 9450 | ||
| path: test_timestamp_offset.png |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,88 @@ | ||
| """ | ||
| Tests for Figure.timestamp. | ||
| """ | ||
| import pytest | ||
| from packaging.version import Version | ||
| from pygmt import Figure, __gmt_version__ | ||
|
|
||
|
|
||
| @pytest.fixture(scope="module", name="faketime") | ||
| def fixture_faketime(): | ||
| """ | ||
| Fake datetime that will be passed to the "timefmt" parameter, so that the | ||
| timestamp string always has a fixed value. | ||
| """ | ||
| return "9999-99-99T99:99:99" | ||
|
|
||
|
|
||
| @pytest.mark.mpl_image_compare | ||
| def test_timestamp(faketime): | ||
| """ | ||
| Test that the simplest timestamp() call works. | ||
| """ | ||
| fig = Figure() | ||
| fig.timestamp(timefmt=faketime) | ||
| return fig | ||
|
|
||
|
|
||
| @pytest.mark.mpl_image_compare | ||
| def test_timestamp_label(faketime): | ||
| """ | ||
| Check if the "label" parameter works. | ||
| """ | ||
| fig = Figure() | ||
| fig.timestamp(label="Powered by PyGMT", timefmt=faketime) | ||
| return fig | ||
|
|
||
|
|
||
| @pytest.mark.mpl_image_compare | ||
| def test_timestamp_justification(): | ||
| """ | ||
| Check if the "justification" parameter works. | ||
|
|
||
| Only a subset of justification codes are tested to avoid overlapping | ||
| timestamps. | ||
| """ | ||
| fig = Figure() | ||
| fig.basemap(projection="X10c/5c", region=[0, 10, 0, 5], frame=0) | ||
| for just in ["BL", "BR", "TL", "TR"]: | ||
| fig.timestamp(justification=just, timefmt=just) | ||
| return fig | ||
|
Comment on lines
+39
to
+50
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more.
Member
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Unlike other embellishments (e.g., inset, legend or direction rose), the timestamp logo is always located at the lower left corner of the map. I believe this is GMT's design. See https://docs.generic-mapping-tools.org/dev/gmt.conf.html#term-MAP_LOGO_POS
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Interesting design choice 😅 Actually, I just realized I was thinking of
Member
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. The default values are clear from the function definition:
I'm hesitate to document the defaults in the docstring.
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Ah ok, was thinking of adding |
||
|
|
||
|
|
||
| @pytest.mark.mpl_image_compare | ||
| def test_timestamp_offset(): | ||
| """ | ||
| Check if the "offset" parameter works. | ||
| """ | ||
| fig = Figure() | ||
| fig.basemap(projection="X10c/5c", region=[0, 10, 0, 5], frame="g1") | ||
| for offset in ["1c", "1c/2c", ("1c", "3c")]: | ||
| fig.timestamp(offset=offset, timefmt=f"offset={offset}") | ||
| return fig | ||
|
|
||
|
|
||
| @pytest.mark.mpl_image_compare | ||
| def test_timestamp_font(faketime): | ||
| """ | ||
| Test if the "font" parameter works. | ||
| """ | ||
| fig = Figure() | ||
| fig.timestamp(font="Times-Roman", label="Powered by GMT", timefmt=faketime) | ||
| return fig | ||
|
|
||
|
|
||
| @pytest.mark.skipif( | ||
| Version(__gmt_version__) < Version("6.5.0"), | ||
| reason="The 'text' parameter requires GMT>=6.5.0", | ||
| ) | ||
| @pytest.mark.mpl_image_compare(filename="test_timestamp.png") | ||
| def test_timestamp_text(faketime): | ||
| """ | ||
| Test if the "text" parameter works. | ||
|
|
||
| Requires GMT>=6.5.0. | ||
| """ | ||
| fig = Figure() | ||
| fig.timestamp(text=faketime) | ||
| return fig | ||
Uh oh!
There was an error while loading. Please reload this page.