mirror of
https://github.com/ARM-software/workload-automation.git
synced 2025-01-19 12:24:32 +00:00
42539bbe0d
Changing the way target descriptions work from a static mapping to something that is dynamically generated and is extensible via plugins. Also moving core target implementation stuff under "framework".
94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
# Adapted from
|
|
# https://gist.github.com/jtriley/1108174
|
|
# pylint: disable=bare-except,unpacking-non-sequence
|
|
import os
|
|
import shlex
|
|
import struct
|
|
import platform
|
|
import subprocess
|
|
|
|
|
|
def get_terminal_size():
|
|
""" getTerminalSize()
|
|
- get width and height of console
|
|
- works on linux,os x,windows,cygwin(windows)
|
|
originally retrieved from:
|
|
http://stackoverflow.com/questions/566746/how-to-get-console-window-width-in-python
|
|
"""
|
|
current_os = platform.system()
|
|
tuple_xy = None
|
|
if current_os == 'Windows':
|
|
tuple_xy = _get_terminal_size_windows()
|
|
if tuple_xy is None:
|
|
# needed for window's python in cygwin's xterm
|
|
tuple_xy = _get_terminal_size_tput()
|
|
if current_os in ['Linux', 'Darwin'] or current_os.startswith('CYGWIN'):
|
|
tuple_xy = _get_terminal_size_linux()
|
|
if tuple_xy is None or tuple_xy == (0, 0):
|
|
tuple_xy = (80, 25) # assume "standard" terminal
|
|
return tuple_xy
|
|
|
|
|
|
def _get_terminal_size_windows():
|
|
# pylint: disable=unused-variable,redefined-outer-name,too-many-locals
|
|
try:
|
|
from ctypes import windll, create_string_buffer
|
|
# stdin handle is -10
|
|
# stdout handle is -11
|
|
# stderr handle is -12
|
|
h = windll.kernel32.GetStdHandle(-12)
|
|
csbi = create_string_buffer(22)
|
|
res = windll.kernel32.GetConsoleScreenBufferInfo(h, csbi)
|
|
if res:
|
|
(bufx, bufy, curx, cury, wattr,
|
|
left, top, right, bottom,
|
|
maxx, maxy) = struct.unpack("hhhhHhhhhhh", csbi.raw)
|
|
sizex = right - left + 1
|
|
sizey = bottom - top + 1
|
|
return sizex, sizey
|
|
except:
|
|
pass
|
|
|
|
|
|
def _get_terminal_size_tput():
|
|
# get terminal width
|
|
# src: http://stackoverflow.com/questions/263890/how-do-i-find-the-width-height-of-a-terminal-window
|
|
try:
|
|
cols = int(subprocess.check_call(shlex.split('tput cols')))
|
|
rows = int(subprocess.check_call(shlex.split('tput lines')))
|
|
return (cols, rows)
|
|
except:
|
|
pass
|
|
|
|
|
|
def _get_terminal_size_linux():
|
|
def ioctl_GWINSZ(fd):
|
|
try:
|
|
import fcntl
|
|
import termios
|
|
cr = struct.unpack('hh',
|
|
fcntl.ioctl(fd, termios.TIOCGWINSZ, '1234'))
|
|
return cr
|
|
except:
|
|
pass
|
|
cr = ioctl_GWINSZ(0) or ioctl_GWINSZ(1) or ioctl_GWINSZ(2)
|
|
if not cr:
|
|
try:
|
|
fd = os.open(os.ctermid(), os.O_RDONLY)
|
|
cr = ioctl_GWINSZ(fd)
|
|
os.close(fd)
|
|
except:
|
|
pass
|
|
if not cr:
|
|
try:
|
|
cr = (os.environ['LINES'], os.environ['COLUMNS'])
|
|
except:
|
|
return None
|
|
return int(cr[1]), int(cr[0])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sizex, sizey = get_terminal_size()
|
|
print 'width =', sizex, 'height =', sizey
|
|
|