Source code ev3dev2/console.py

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
# -----------------------------------------------------------------------------
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
# -----------------------------------------------------------------------------

import os


class Console():
    """
    A class that represents the EV3 LCD console, which implements ANSI codes
    for cursor positioning, text color, and resetting the screen. Supports changing
    the console font using standard system fonts.
    """
    def __init__(self, font="Lat15-TerminusBold24x12"):
        """
        Construct the Console instance, optionally with a font name specified.

        Parameter:

        - ``font`` (string): Font name, as found in ``/usr/share/consolefonts/``

        """
        self._font = None
        self._columns = 0
        self._rows = 0
        self._echo = False
        self._cursor = False
        self.set_font(font, reset_console=False)  # don't reset the screen during construction
        self.cursor = False
        self.echo = False

    @property
    def columns(self):
        """
        Return (int) number of columns on the EV3 LCD console supported by the current font.
        """
        return self._columns

    @property
    def rows(self):
        """
        Return (int) number of rows on the EV3 LCD console supported by the current font.
        """
        return self._rows

    @property
    def echo(self):
        """
        Return (bool) whether the console echo mode is enabled.
        """
        return self._echo

    @echo.setter
    def echo(self, value):
        """
        Enable/disable console echo (so that EV3 button presses do not show the escape characters on
        the LCD console). Set to True to show the button codes, or False to hide them.
        """
        self._echo = value
        os.system("stty {}".format("echo" if value else "-echo"))

    @property
    def cursor(self):
        """
        Return (bool) whether the console cursor is visible.
        """
        return self._cursor

    @cursor.setter
    def cursor(self, value):
        """
        Enable/disable console cursor (to hide the cursor on the LCD).
        Set to True to show the cursor, or False to hide it.
        """
        self._cursor = value
        print("\x1b[?25{}".format('h' if value else 'l'), end='')

    def text_at(self, text, column=1, row=1, reset_console=False, inverse=False, alignment="L"):
        """
        Display ``text`` (string) at grid position (``column``, ``row``).
        Note that the grid locations are 1-based (not 0-based).

        Depending on the font, the number of columns and rows supported by the EV3 LCD console
        can vary. Large fonts support as few as 11 columns and 4 rows, while small fonts support
        44 columns and 21 rows. The default font for the Console() class results in a grid that
        is 14 columns and 5 rows.

        Using the ``inverse=True`` parameter will display the ``text`` with more emphasis and contrast,
        as the background of the text will be black, and the foreground is white. Using inverse
        can help in certain situations, such as to indicate when a color sensor senses
        black, or the gyro sensor is pointing to zero.

        Use the ``alignment`` parameter to enable the function to align the ``text`` differently to the
        column/row values passed-in. Use ``L`` for left-alignment (default), where the first character
        in the ``text`` will show at the column/row position. Use ``R`` for right-alignment, where the
        last character will show at the column/row position. Use ``C`` for center-alignment, where the
        text string will centered at the column/row position (as close as possible using integer
        division--odd-length text string will center better than even-length).

        Parameters:

        - ``text`` (string): Text to display
        - ``column`` (int): LCD column position to start the text (1 = left column);
          text will wrap when it reaches the right edge
        - ``row`` (int): LCD row position to start the text (1 = top row)
        - ``reset_console`` (bool): ``True`` to reset the EV3 LCD console before showing
          the text; default is ``False``
        - ``inverse`` (bool): ``True`` for white on black, otherwise black on white;
          default is ``False``
        - ``alignment`` (string): Align the ``text`` horizontally. Use ``L`` for left-alignment (default),
          ``R`` for right-alignment, or ``C`` for center-alignment

        """

        if reset_console:
            self.reset_console()

        if alignment == "R":
            column = column - len(text) + 1
        elif alignment == "C":
            column -= len(text) // 2

        if inverse:
            text = "\x1b[7m{}\x1b[m".format(text)

        print("\x1b[{};{}H{}".format(row, column, text), end='')

    def set_font(self, font="Lat15-TerminusBold24x12", reset_console=True):
        """
        Set the EV3 LCD console font and optionally reset the EV3 LCD console
        to clear it and turn off the cursor.

        Parameters:

        - ``font`` (string): Font name, as found in ``/usr/share/consolefonts/``
        - ``reset_console`` (bool): ``True`` to reset the EV3 LCD console
          after the font change; default is ``True``

        """
        if font is not None and font != self._font:
            self._font = font
            os.system("setfont {}".format(font))
            rows, columns = os.popen('stty size').read().strip().split(" ")
            self._rows = int(rows)
            self._columns = int(columns)

        if reset_console:
            self.reset_console()

    def clear_to_eol(self, column=None, row=None):
        """
        Clear to the end of line from the ``column`` and ``row`` position
        on the EV3 LCD console. Default to current cursor position.

        Parameters:

        - ``column`` (int): LCD column position to move to before clearing
        - ``row`` (int): LCD row position to move to before clearing

        """
        if column is not None and row is not None:
            print("\x1b[{};{}H".format(row, column), end='')
        print("\x1b[K", end='')

    def reset_console(self):
        """
        Clear the EV3 LCD console using ANSI codes, and move the cursor to 1,1
        """
        print("\x1b[2J\x1b[H", end='')