mirror of
https://github.com/ARM-software/workload-automation.git
synced 2024-10-06 19:01:15 +01:00
8878cc20d4
- Add copyright headers to files that did not already have them - Update the year to the last year the file was modified in files with existing headers.
109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
# Copyright 2018 ARM Limited
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
# 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)
|
|
|