emcek / dcspy / 462

pending completion

Build # 462

Build Type

Pull #17

travis-ci

Committed by

web-flow

Commit Message

<a href="https://github.com/emcek/dcspy/commit/<a class=hub.com/emcek/dcspy/commit/6bfdb8ba611d3ca8dff71a6852a04116f72583ef">6bfdb8ba6<a href="https://github.com/emcek/dcspy/commit/6bfdb8ba611d3ca8dff71a6852a04116f72583ef">">Merge </a><a class="double-link" href="https://github.com/emcek/dcspy/commit/<a class="double-link" href="https://github.com/emcek/dcspy/commit/f8a3cead14a1a43c96ed1a696e77262c26255f4c">f8a3cead1</a>">f8a3cead1</a><a href="https://github.com/emcek/dcspy/commit/6bfdb8ba611d3ca8dff71a6852a04116f72583ef"> into 657c9ab7d">657c9ab7d</a>

Pull Request Pull Request #17: add G19 handling

Run Details

103 of 103 new or added lines in 6 files covered. (100.0%)

420 of 604 relevant lines covered (69.54%)

0.7 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

96.72

/dcspy/aircrafts.py

from functools import partial
from logging import getLogger
from string import whitespace
from typing import Dict, Union

from PIL import Image, ImageDraw

from dcspy import FONT_11, FONT_16, lcd_sdk, LcdSize

try:
    from typing_extensions import TypedDict
except ImportError:
    from typing import TypedDict

BIOS_VALUE = TypedDict('BIOS_VALUE', {'class': str, 'args': Dict[str, int], 'value': Union[int, str]})
LOG = getLogger(__name__)


class Aircraft:
    def __init__(self, lcd_type: LcdSize) -> None:
        """
        Basic constructor.

        :param lcd_type: LCD type
        """
        self.lcd = lcd_type
        self.bios_data: Dict[str, BIOS_VALUE] = {}

    def button_request(self, button: int, request: str = '\n') -> str:
        """
        Prepare aircraft specific DCS-BIOS request for button pressed.

        If button is out of scope new line is return.

        :param button: possible values 1-4
        :type: int
        :param request: valid DCS-BIOS command as string
        :type request: str
        :return: ready to send DCS-BIOS request
        :rtype: str
        """
        LOG.debug(f'{self.__class__.__name__} Button: {button}')
        LOG.debug(f'Request: {request.replace(whitespace[2], " ")}')
        return request

    @staticmethod
    def update_display(image: Image.Image) -> None:
        """Update display."""
        lcd_sdk.update_display(image)

    def prepare_image(self) -> Image.Image:
        """
        Prepare image to be send to LCD.

        :return: image instance ready display on LCD
        :rtype: Image.Image
        """
        img_for_lcd = {1: partial(Image.new, mode='1', size=(self.lcd.width, self.lcd.height), color=0),
                       2: partial(Image.new, mode='RGBA', size=(self.lcd.width, self.lcd.height), color=(0, 0, 0, 0))}
        if self.lcd.type in (1, 2):
            return img_for_lcd[self.lcd.type]()
        LOG.debug(f'Wrong LCD type: {self.lcd}')

    def set_bios(self, selector: str, value: str) -> None:
        """
        Set value for DCS-BIOS selector.

        :param selector:
        :param value:
        """
        self.bios_data[selector]['value'] = value
        LOG.debug(f'{self.__class__.__name__} {selector} value: "{value}"')
        lcd_image = self.prepare_image()
        self.update_display(lcd_image)

    def get_bios(self, selector: str) -> Union[str, int]:
        """
        Get value for DCS-BIOS selector.

        :param selector:
        """
        try:
            return self.bios_data[selector]['value']
        except KeyError:
            return ''


class FA18Chornet(Aircraft):
    def __init__(self, lcd_type: LcdSize) -> None:
        """
        Basic constructor.

        :param lcd_type: LCD type
        """
        super().__init__(lcd_type)
        self.bios_data: Dict[str, BIOS_VALUE] = {
            'ScratchpadStr1': {'class': 'StringBuffer', 'args': {'address': 0x744e, 'length': 2}, 'value': str()},
            'ScratchpadStr2': {'class': 'StringBuffer', 'args': {'address': 0x7450, 'length': 2}, 'value': str()},
            'ScratchpadNum': {'class': 'StringBuffer', 'args': {'address': 0x7446, 'length': 8}, 'value': str()},
            'OptionDisplay1': {'class': 'StringBuffer', 'args': {'address': 0x7432, 'length': 4}, 'value': str()},
            'OptionDisplay2': {'class': 'StringBuffer', 'args': {'address': 0x7436, 'length': 4}, 'value': str()},
            'OptionDisplay3': {'class': 'StringBuffer', 'args': {'address': 0x743a, 'length': 4}, 'value': str()},
            'OptionDisplay4': {'class': 'StringBuffer', 'args': {'address': 0x743e, 'length': 4}, 'value': str()},
            'OptionDisplay5': {'class': 'StringBuffer', 'args': {'address': 0x7442, 'length': 4}, 'value': str()},
            'COMM1': {'class': 'StringBuffer', 'args': {'address': 0x7424, 'length': 2}, 'value': str()},
            'COMM2': {'class': 'StringBuffer', 'args': {'address': 0x7426, 'length': 2}, 'value': str()},
            'OptionCueing1': {'class': 'StringBuffer', 'args': {'address': 0x7428, 'length': 1}, 'value': str()},
            'OptionCueing2': {'class': 'StringBuffer', 'args': {'address': 0x742a, 'length': 1}, 'value': str()},
            'OptionCueing3': {'class': 'StringBuffer', 'args': {'address': 0x742c, 'length': 1}, 'value': str()},
            'OptionCueing4': {'class': 'StringBuffer', 'args': {'address': 0x742e, 'length': 1}, 'value': str()},
            'OptionCueing5': {'class': 'StringBuffer', 'args': {'address': 0x7430, 'length': 1}, 'value': str()},
            'FuelTotal': {'class': 'StringBuffer', 'args': {'address': 0x748a, 'length': 6}, 'value': str()}}

    def prepare_image(self) -> Image.Image:
        """
        Prepare image for F/A-18C Hornet to be send to LCD.

        :return: image instance ready display on LCD
        """
        img = super().prepare_image()
        draw = ImageDraw.Draw(img)
        # Scrachpad
        draw.text(xy=(0, 0), text=f'{self.get_bios("ScratchpadStr1")}{self.get_bios("ScratchpadStr2")}{self.get_bios("ScratchpadNum")}', fill=255, font=FONT_16)
        draw.line(xy=(0, 20, 115, 20), fill=255, width=1)

        # comm1
        draw.rectangle(xy=(0, 29, 20, 42), fill=0, outline=255)
        draw.text(xy=(2, 29), text=self.get_bios('COMM1'), fill=255, font=FONT_16)

        # comm2
        offset_comm2 = 44
        draw.rectangle(xy=(139 - offset_comm2, 29, 159 - offset_comm2, 42), fill=0, outline=255)
        draw.text(xy=(140 - offset_comm2, 29), text=self.get_bios('COMM2'), fill=255, font=FONT_16)

        # option display 1..5 with cueing
        for i in range(1, 6):
            offset = (i - 1) * 8
            draw.text(xy=(120, offset), text=f'{i}{self.get_bios(f"OptionCueing{i}")}{self.get_bios(f"OptionDisplay{i}")}', fill=255, font=FONT_11)

        # Fuel Totaliser
        draw.text(xy=(36, 29), text=self.get_bios('FuelTotal'), fill=255, font=FONT_16)
        return img

    def set_bios(self, selector: str, value: str) -> None:
        """
        Set new data.

        :param selector:
        :param value:
        """
        if selector in ('ScratchpadStr1', 'ScratchpadStr2', 'COMM1', 'COMM2'):
            value = value.replace('`', '1').replace('~', '2')
        super().set_bios(selector, value)

    def button_request(self, button: int, request: str = '\n') -> str:
        """
        Prepare F/A-18 Hornet specific DCS-BIOS request for button pressed.

        If button is out of scope new line is return.

        :param button: possible values 1-4
        :type: int
        :param request: valid DCS-BIOS command as string
        :type request: str
        :return: ready to send DCS-BIOS request
        :rtype: str
        """
        action = {1: 'UFC_COMM1_CHANNEL_SELECT DEC\n',
                  2: 'UFC_COMM1_CHANNEL_SELECT INC\n',
                  3: 'UFC_COMM2_CHANNEL_SELECT DEC\n',
                  4: 'UFC_COMM2_CHANNEL_SELECT INC\n',
                  9: 'UFC_COMM1_CHANNEL_SELECT DEC\n',
                  10: 'UFC_COMM1_CHANNEL_SELECT INC\n',
                  14: 'UFC_COMM2_CHANNEL_SELECT DEC\n',
                  13: 'UFC_COMM2_CHANNEL_SELECT INC\n'}
        return super().button_request(button, action.get(button, '\n'))


class F16C50(Aircraft):
    def __init__(self, lcd_type: LcdSize) -> None:
        """
        Basic constructor.

        :param lcd_type: LCD type
        """
        super().__init__(lcd_type)
        self.bios_data: Dict[str, BIOS_VALUE] = {
            'DED_LINE_1': {'class': 'StringBuffer', 'args': {'address': 0x4502, 'length': 25}, 'value': str()},
            'DED_LINE_2': {'class': 'StringBuffer', 'args': {'address': 0x451c, 'length': 25}, 'value': str()},
            'DED_LINE_3': {'class': 'StringBuffer', 'args': {'address': 0x4536, 'length': 25}, 'value': str()},
            'DED_LINE_4': {'class': 'StringBuffer', 'args': {'address': 0x4550, 'length': 25}, 'value': str()},
            'DED_LINE_5': {'class': 'StringBuffer', 'args': {'address': 0x456a, 'length': 25}, 'value': str()}}

    def prepare_image(self) -> Image.Image:
        """
        Prepare image for F-16C Viper to be send to LCD.

        :return: image instance ready display on LCD
        """
        img = super().prepare_image()
        draw = ImageDraw.Draw(img)
        for i in range(1, 6):
            offset = (i - 1) * 8
            # replace 'o' to degree sign and 'a' with up-down arrow
            text = str(self.get_bios(f'DED_LINE_{i}')).replace('o', '\u00b0').replace('a', '\u2195')
            draw.text(xy=(0, offset), text=text, fill=255, font=FONT_11)
        return img


class Ka50(Aircraft):
    def __init__(self, lcd_type: LcdSize) -> None:
        """
        Basic constructor.

        :param lcd_type: LCD type
        """
        super().__init__(lcd_type)
        self.bios_data: Dict[str, BIOS_VALUE] = {
            'l1_apostr1': {'class': 'StringBuffer', 'args': {'address': 0x1934, 'length': 1}, 'value': str()},
            'l1_apostr2': {'class': 'StringBuffer', 'args': {'address': 0x1936, 'length': 1}, 'value': str()},
            'l1_point': {'class': 'StringBuffer', 'args': {'address': 0x1930, 'length': 1}, 'value': str()},
            'l1_sign': {'class': 'StringBuffer', 'args': {'address': 0x1920, 'length': 1}, 'value': str()},
            'l1_text': {'class': 'StringBuffer', 'args': {'address': 0x1924, 'length': 6}, 'value': str()},
            'l2_apostr1': {'class': 'StringBuffer', 'args': {'address': 0x1938, 'length': 1}, 'value': str()},
            'l2_apostr2': {'class': 'StringBuffer', 'args': {'address': 0x193a, 'length': 1}, 'value': str()},
            'l2_point': {'class': 'StringBuffer', 'args': {'address': 0x1932, 'length': 1}, 'value': str()},
            'l2_sign': {'class': 'StringBuffer', 'args': {'address': 0x1922, 'length': 1}, 'value': str()},
            'l2_text': {'class': 'StringBuffer', 'args': {'address': 0x192a, 'length': 6}, 'value': str()},
            'AP_ALT_HOLD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1936, 'mask': 0x8000, 'shift_by': 0xf}, 'value': int()},
            'AP_BANK_HOLD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1936, 'mask': 0x200, 'shift_by': 0x9}, 'value': int()},
            'AP_FD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1938, 'mask': 0x200, 'shift_by': 0x9}, 'value': int()},
            'AP_HDG_HOLD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1936, 'mask': 0x800, 'shift_by': 0xb}, 'value': int()},
            'AP_PITCH_HOLD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1936, 'mask': 0x2000, 'shift_by': 0xd}, 'value': int()}}

    def button_request(self, button: int, request: str = '\n') -> str:
        """
        Prepare Ka-50 Black Shark specific DCS-BIOS request for button pressed.

        If button is out of scope new line is return.

        :param button: possible values 1-4
        :type: int
        :param request: valid DCS-BIOS command as string
        :type request: str
        :return: ready to send DCS-BIOS request
        :rtype: str
        """
        action = {1: 'PVI_WAYPOINTS_BTN 1\nPVI_WAYPOINTS_BTN 0\n',
                  2: 'PVI_FIXPOINTS_BTN 1\nPVI_FIXPOINTS_BTN 0\n',
                  3: 'PVI_AIRFIELDS_BTN 1\nPVI_AIRFIELDS_BTN 0\n',
                  4: 'PVI_TARGETS_BTN 1\nPVI_TARGETS_BTN 0\n',
                  9: 'PVI_WAYPOINTS_BTN 1\nPVI_WAYPOINTS_BTN 0\n',
                  10: 'PVI_FIXPOINTS_BTN 1\nPVI_FIXPOINTS_BTN 0\n',
                  14: 'PVI_AIRFIELDS_BTN 1\nPVI_AIRFIELDS_BTN 0\n',
                  13: 'PVI_TARGETS_BTN 1\nPVI_TARGETS_BTN 0\n'}
        return super().button_request(button, action.get(button, '\n'))

    def prepare_image(self) -> Image.Image:
        """
        Prepare image for Ka-50 Black Shark to be send to LCD.

        :return: image instance ready display on LCD
        """
        img = super().prepare_image()
        draw = ImageDraw.Draw(img)
        text1, text2 = '', ''
        draw.rectangle(xy=(0, 1, 85, 18), fill=0, outline=255)
        draw.rectangle(xy=(0, 22, 85, 39), fill=0, outline=255)
        draw.rectangle(xy=(88, 1, 103, 18), fill=0, outline=255)
        draw.rectangle(xy=(88, 22, 103, 39), fill=0, outline=255)
        l1_text = str(self.get_bios('l1_text'))
        l2_text = str(self.get_bios('l2_text'))
        if l1_text:
            text1 = f'{l1_text[-6:-3]}{self.get_bios("l1_apostr1")}{l1_text[-3:-1]}{self.get_bios("l1_apostr2")}{l1_text[-1]}'
        if l2_text:
            text2 = f'{l2_text[-6:-3]}{self.get_bios("l2_apostr1")}{l2_text[-3:-1]}{self.get_bios("l2_apostr2")}{l2_text[-1]}'
        line1 = f'{self.get_bios("l1_sign")}{text1} {self.get_bios("l1_point")}'
        line2 = f'{self.get_bios("l2_sign")}{text2} {self.get_bios("l2_point")}'
        draw.text(xy=(2, 3), text=line1, fill=255, font=FONT_16)
        draw.text(xy=(2, 24), text=line2, fill=255, font=FONT_16)
        self._auto_pilot_switch(draw)
        return img

    def _auto_pilot_switch(self, draw_obj: ImageDraw) -> None:
        """
        Draw rectangle and add text for autopilot channels in correct coordinates.

        :param draw_obj: ImageDraw object form PIL
        """
        for c_rect, c_text, ap_channel, turn_on in (((111, 1, 124, 18), (114, 3), 'B', self.get_bios('AP_BANK_HOLD_LED')),
                                                    ((128, 1, 141, 18), (130, 3), 'P', self.get_bios('AP_PITCH_HOLD_LED')),
                                                    ((145, 1, 158, 18), (147, 3), 'F', self.get_bios('AP_FD_LED')),
                                                    ((111, 22, 124, 39), (114, 24), 'H', self.get_bios('AP_HDG_HOLD_LED')),
                                                    ((128, 22, 141, 39), (130, 24), 'A', self.get_bios('AP_ALT_HOLD_LED'))):
            if turn_on:
                draw_obj.rectangle(c_rect, fill=255, outline=255)
                draw_obj.text(xy=c_text, text=ap_channel, fill=0, font=FONT_16)
            else:
                draw_obj.rectangle(xy=c_rect, fill=0, outline=255)
                draw_obj.text(xy=c_text, text=ap_channel, fill=255, font=FONT_16)


class F14B(Aircraft):
    def __init__(self, lcd_type: LcdSize) -> None:
        """
        Basic constructor.

        :param lcd_type: LCD type
        """
        super().__init__(lcd_type)
        self.bios_data: Dict[str, BIOS_VALUE] = {
            'RIO_CAP_CLEAR': {'class': 'IntegerBuffer', 'args': {'address': 0x12d4, 'mask': 0x1, 'shift_by': 0x0}, 'value': int()},
            'RIO_CAP_SW': {'class': 'IntegerBuffer', 'args': {'address': 0x12d2, 'mask': 0x8000, 'shift_by': 0xf}, 'value': int()},
            'RIO_CAP_NE': {'class': 'IntegerBuffer', 'args': {'address': 0x12d2, 'mask': 0x4000, 'shift_by': 0xe}, 'value': int()},
            'RIO_CAP_ENTER': {'class': 'IntegerBuffer', 'args': {'address': 0x12d4, 'mask': 0x2, 'shift_by': 0x1}, 'value': int()}}

    def button_request(self, button: int, request: str = '\n') -> str:
        """
        Prepare F-14 Tomcat specific DCS-BIOS request for button pressed.

        If button is out of scope new line is return.

        :param button: possible values 1-4
        :type: int
        :param request: valid DCS-BIOS command as string
        :type request: str
        :return: ready to send DCS-BIOS request
        :rtype: str
        """
        action = {1: 'RIO_CAP_CLEAR 1\nRIO_CAP_CLEAR 0\n',
                  2: 'RIO_CAP_SW 1\nRIO_CAP_SW 0\n',
                  3: 'RIO_CAP_NE 1\nRIO_CAP_NE 0\n',
                  4: 'RIO_CAP_ENTER 1\nRIO_CAP_ENTER 0\n',
                  9: 'RIO_CAP_CLEAR 1\nRIO_CAP_CLEAR 0\n',
                  10: 'RIO_CAP_SW 1\nRIO_CAP_SW 0\n',
                  14: 'RIO_CAP_NE 1\nRIO_CAP_NE 0\n',
                  13: 'RIO_CAP_ENTER 1\nRIO_CAP_ENTER 0\n'}
        return super().button_request(button, action.get(button, '\n'))

    def prepare_image(self) -> Image.Image:
        """
        Prepare image for F-14B Tomcat to be send to LCD.

        :return: image instance ready display on LCD
        """
        img = super().prepare_image()
        draw = ImageDraw.Draw(img)
        draw.text(xy=(2, 3), text='F-14B Tomcat', fill=255, font=FONT_16)
        return img

1	from functools import partial	1✔
2	from logging import getLogger	1✔
3	from string import whitespace	1✔
4	from typing import Dict, Union	1✔
5
6	from PIL import Image, ImageDraw	1✔
7
8	from dcspy import FONT_11, FONT_16, lcd_sdk, LcdSize	1✔
9
10	try:	1✔
11	from typing_extensions import TypedDict	1✔
12	except ImportError:	×
13	from typing import TypedDict	×
14
15	BIOS_VALUE = TypedDict('BIOS_VALUE', {'class': str, 'args': Dict[str, int], 'value': Union[int, str]})	1✔
16	LOG = getLogger(__name__)	1✔
17
18
19	class Aircraft:	1✔
20	def __init__(self, lcd_type: LcdSize) -> None:	1✔
21	"""
22	Basic constructor.
23
24	:param lcd_type: LCD type
25	"""
26	self.lcd = lcd_type	1✔
27	self.bios_data: Dict[str, BIOS_VALUE] = {}	1✔
28
29	def button_request(self, button: int, request: str = '\n') -> str:	1✔
30	"""
31	Prepare aircraft specific DCS-BIOS request for button pressed.
32
33	If button is out of scope new line is return.
34
35	:param button: possible values 1-4
36	:type: int
37	:param request: valid DCS-BIOS command as string
38	:type request: str
39	:return: ready to send DCS-BIOS request
40	:rtype: str
41	"""
42	LOG.debug(f'{self.__class__.__name__} Button: {button}')	1✔
43	LOG.debug(f'Request: {request.replace(whitespace[2], " ")}')	1✔
44	return request	1✔
45
46	@staticmethod	1✔
47	def update_display(image: Image.Image) -> None:	1✔
48	"""Update display."""
49	lcd_sdk.update_display(image)	1✔
50
51	def prepare_image(self) -> Image.Image:	1✔
52	"""
53	Prepare image to be send to LCD.
54
55	:return: image instance ready display on LCD
56	:rtype: Image.Image
57	"""
58	img_for_lcd = {1: partial(Image.new, mode='1', size=(self.lcd.width, self.lcd.height), color=0),	1✔
59	2: partial(Image.new, mode='RGBA', size=(self.lcd.width, self.lcd.height), color=(0, 0, 0, 0))}
60	if self.lcd.type in (1, 2):	1✔
61	return img_for_lcd[self.lcd.type]()	1✔
62	LOG.debug(f'Wrong LCD type: {self.lcd}')	1✔
63
64	def set_bios(self, selector: str, value: str) -> None:	1✔
65	"""
66	Set value for DCS-BIOS selector.
67
68	:param selector:
69	:param value:
70	"""
71	self.bios_data[selector]['value'] = value	1✔
72	LOG.debug(f'{self.__class__.__name__} {selector} value: "{value}"')	1✔
73	lcd_image = self.prepare_image()	1✔
74	self.update_display(lcd_image)	1✔
75
76	def get_bios(self, selector: str) -> Union[str, int]:	1✔
77	"""
78	Get value for DCS-BIOS selector.
79
80	:param selector:
81	"""
82	try:	1✔
83	return self.bios_data[selector]['value']	1✔
84	except KeyError:	1✔
85	return ''	1✔
86
87
88	class FA18Chornet(Aircraft):	1✔
89	def __init__(self, lcd_type: LcdSize) -> None:	1✔
90	"""
91	Basic constructor.
92
93	:param lcd_type: LCD type
94	"""
95	super().__init__(lcd_type)	1✔
96	self.bios_data: Dict[str, BIOS_VALUE] = {	1✔
97	'ScratchpadStr1': {'class': 'StringBuffer', 'args': {'address': 0x744e, 'length': 2}, 'value': str()},
98	'ScratchpadStr2': {'class': 'StringBuffer', 'args': {'address': 0x7450, 'length': 2}, 'value': str()},
99	'ScratchpadNum': {'class': 'StringBuffer', 'args': {'address': 0x7446, 'length': 8}, 'value': str()},
100	'OptionDisplay1': {'class': 'StringBuffer', 'args': {'address': 0x7432, 'length': 4}, 'value': str()},
101	'OptionDisplay2': {'class': 'StringBuffer', 'args': {'address': 0x7436, 'length': 4}, 'value': str()},
102	'OptionDisplay3': {'class': 'StringBuffer', 'args': {'address': 0x743a, 'length': 4}, 'value': str()},
103	'OptionDisplay4': {'class': 'StringBuffer', 'args': {'address': 0x743e, 'length': 4}, 'value': str()},
104	'OptionDisplay5': {'class': 'StringBuffer', 'args': {'address': 0x7442, 'length': 4}, 'value': str()},
105	'COMM1': {'class': 'StringBuffer', 'args': {'address': 0x7424, 'length': 2}, 'value': str()},
106	'COMM2': {'class': 'StringBuffer', 'args': {'address': 0x7426, 'length': 2}, 'value': str()},
107	'OptionCueing1': {'class': 'StringBuffer', 'args': {'address': 0x7428, 'length': 1}, 'value': str()},
108	'OptionCueing2': {'class': 'StringBuffer', 'args': {'address': 0x742a, 'length': 1}, 'value': str()},
109	'OptionCueing3': {'class': 'StringBuffer', 'args': {'address': 0x742c, 'length': 1}, 'value': str()},
110	'OptionCueing4': {'class': 'StringBuffer', 'args': {'address': 0x742e, 'length': 1}, 'value': str()},
111	'OptionCueing5': {'class': 'StringBuffer', 'args': {'address': 0x7430, 'length': 1}, 'value': str()},
112	'FuelTotal': {'class': 'StringBuffer', 'args': {'address': 0x748a, 'length': 6}, 'value': str()}}
113
114	def prepare_image(self) -> Image.Image:	1✔
115	"""
116	Prepare image for F/A-18C Hornet to be send to LCD.
117
118	:return: image instance ready display on LCD
119	"""
120	img = super().prepare_image()	1✔
121	draw = ImageDraw.Draw(img)	1✔
122	# Scrachpad
123	draw.text(xy=(0, 0), text=f'{self.get_bios("ScratchpadStr1")}{self.get_bios("ScratchpadStr2")}{self.get_bios("ScratchpadNum")}', fill=255, font=FONT_16)	1✔
124	draw.line(xy=(0, 20, 115, 20), fill=255, width=1)	1✔
125
126	# comm1
127	draw.rectangle(xy=(0, 29, 20, 42), fill=0, outline=255)	1✔
128	draw.text(xy=(2, 29), text=self.get_bios('COMM1'), fill=255, font=FONT_16)	1✔
129
130	# comm2
131	offset_comm2 = 44	1✔
132	draw.rectangle(xy=(139 - offset_comm2, 29, 159 - offset_comm2, 42), fill=0, outline=255)	1✔
133	draw.text(xy=(140 - offset_comm2, 29), text=self.get_bios('COMM2'), fill=255, font=FONT_16)	1✔
134
135	# option display 1..5 with cueing
136	for i in range(1, 6):	1✔
137	offset = (i - 1) * 8	1✔
138	draw.text(xy=(120, offset), text=f'{i}{self.get_bios(f"OptionCueing{i}")}{self.get_bios(f"OptionDisplay{i}")}', fill=255, font=FONT_11)	1✔
139
140	# Fuel Totaliser
141	draw.text(xy=(36, 29), text=self.get_bios('FuelTotal'), fill=255, font=FONT_16)	1✔
142	return img	1✔
143
144	def set_bios(self, selector: str, value: str) -> None:	1✔
145	"""
146	Set new data.
147
148	:param selector:
149	:param value:
150	"""
151	if selector in ('ScratchpadStr1', 'ScratchpadStr2', 'COMM1', 'COMM2'):	1✔
152	value = value.replace('`', '1').replace('~', '2')	1✔
153	super().set_bios(selector, value)	1✔
154
155	def button_request(self, button: int, request: str = '\n') -> str:	1✔
156	"""
157	Prepare F/A-18 Hornet specific DCS-BIOS request for button pressed.
158
159	If button is out of scope new line is return.
160
161	:param button: possible values 1-4
162	:type: int
163	:param request: valid DCS-BIOS command as string
164	:type request: str
165	:return: ready to send DCS-BIOS request
166	:rtype: str
167	"""
168	action = {1: 'UFC_COMM1_CHANNEL_SELECT DEC\n',	1✔
169	2: 'UFC_COMM1_CHANNEL_SELECT INC\n',
170	3: 'UFC_COMM2_CHANNEL_SELECT DEC\n',
171	4: 'UFC_COMM2_CHANNEL_SELECT INC\n',
172	9: 'UFC_COMM1_CHANNEL_SELECT DEC\n',
173	10: 'UFC_COMM1_CHANNEL_SELECT INC\n',
174	14: 'UFC_COMM2_CHANNEL_SELECT DEC\n',
175	13: 'UFC_COMM2_CHANNEL_SELECT INC\n'}
176	return super().button_request(button, action.get(button, '\n'))	1✔
177
178
179	class F16C50(Aircraft):	1✔
180	def __init__(self, lcd_type: LcdSize) -> None:	1✔
181	"""
182	Basic constructor.
183
184	:param lcd_type: LCD type
185	"""
186	super().__init__(lcd_type)	1✔
187	self.bios_data: Dict[str, BIOS_VALUE] = {	1✔
188	'DED_LINE_1': {'class': 'StringBuffer', 'args': {'address': 0x4502, 'length': 25}, 'value': str()},
189	'DED_LINE_2': {'class': 'StringBuffer', 'args': {'address': 0x451c, 'length': 25}, 'value': str()},
190	'DED_LINE_3': {'class': 'StringBuffer', 'args': {'address': 0x4536, 'length': 25}, 'value': str()},
191	'DED_LINE_4': {'class': 'StringBuffer', 'args': {'address': 0x4550, 'length': 25}, 'value': str()},
192	'DED_LINE_5': {'class': 'StringBuffer', 'args': {'address': 0x456a, 'length': 25}, 'value': str()}}
193
194	def prepare_image(self) -> Image.Image:	1✔
195	"""
196	Prepare image for F-16C Viper to be send to LCD.
197
198	:return: image instance ready display on LCD
199	"""
200	img = super().prepare_image()	1✔
201	draw = ImageDraw.Draw(img)	1✔
202	for i in range(1, 6):	1✔
203	offset = (i - 1) * 8	1✔
204	# replace 'o' to degree sign and 'a' with up-down arrow
205	text = str(self.get_bios(f'DED_LINE_{i}')).replace('o', '\u00b0').replace('a', '\u2195')	1✔
206	draw.text(xy=(0, offset), text=text, fill=255, font=FONT_11)	1✔
207	return img	1✔
208
209
210	class Ka50(Aircraft):	1✔
211	def __init__(self, lcd_type: LcdSize) -> None:	1✔
212	"""
213	Basic constructor.
214
215	:param lcd_type: LCD type
216	"""
217	super().__init__(lcd_type)	1✔
218	self.bios_data: Dict[str, BIOS_VALUE] = {	1✔
219	'l1_apostr1': {'class': 'StringBuffer', 'args': {'address': 0x1934, 'length': 1}, 'value': str()},
220	'l1_apostr2': {'class': 'StringBuffer', 'args': {'address': 0x1936, 'length': 1}, 'value': str()},
221	'l1_point': {'class': 'StringBuffer', 'args': {'address': 0x1930, 'length': 1}, 'value': str()},
222	'l1_sign': {'class': 'StringBuffer', 'args': {'address': 0x1920, 'length': 1}, 'value': str()},
223	'l1_text': {'class': 'StringBuffer', 'args': {'address': 0x1924, 'length': 6}, 'value': str()},
224	'l2_apostr1': {'class': 'StringBuffer', 'args': {'address': 0x1938, 'length': 1}, 'value': str()},
225	'l2_apostr2': {'class': 'StringBuffer', 'args': {'address': 0x193a, 'length': 1}, 'value': str()},
226	'l2_point': {'class': 'StringBuffer', 'args': {'address': 0x1932, 'length': 1}, 'value': str()},
227	'l2_sign': {'class': 'StringBuffer', 'args': {'address': 0x1922, 'length': 1}, 'value': str()},
228	'l2_text': {'class': 'StringBuffer', 'args': {'address': 0x192a, 'length': 6}, 'value': str()},
229	'AP_ALT_HOLD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1936, 'mask': 0x8000, 'shift_by': 0xf}, 'value': int()},
230	'AP_BANK_HOLD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1936, 'mask': 0x200, 'shift_by': 0x9}, 'value': int()},
231	'AP_FD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1938, 'mask': 0x200, 'shift_by': 0x9}, 'value': int()},
232	'AP_HDG_HOLD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1936, 'mask': 0x800, 'shift_by': 0xb}, 'value': int()},
233	'AP_PITCH_HOLD_LED': {'class': 'IntegerBuffer', 'args': {'address': 0x1936, 'mask': 0x2000, 'shift_by': 0xd}, 'value': int()}}
234
235	def button_request(self, button: int, request: str = '\n') -> str:	1✔
236	"""
237	Prepare Ka-50 Black Shark specific DCS-BIOS request for button pressed.
238
239	If button is out of scope new line is return.
240
241	:param button: possible values 1-4
242	:type: int
243	:param request: valid DCS-BIOS command as string
244	:type request: str
245	:return: ready to send DCS-BIOS request
246	:rtype: str
247	"""
248	action = {1: 'PVI_WAYPOINTS_BTN 1\nPVI_WAYPOINTS_BTN 0\n',	1✔
249	2: 'PVI_FIXPOINTS_BTN 1\nPVI_FIXPOINTS_BTN 0\n',
250	3: 'PVI_AIRFIELDS_BTN 1\nPVI_AIRFIELDS_BTN 0\n',
251	4: 'PVI_TARGETS_BTN 1\nPVI_TARGETS_BTN 0\n',
252	9: 'PVI_WAYPOINTS_BTN 1\nPVI_WAYPOINTS_BTN 0\n',
253	10: 'PVI_FIXPOINTS_BTN 1\nPVI_FIXPOINTS_BTN 0\n',
254	14: 'PVI_AIRFIELDS_BTN 1\nPVI_AIRFIELDS_BTN 0\n',
255	13: 'PVI_TARGETS_BTN 1\nPVI_TARGETS_BTN 0\n'}
256	return super().button_request(button, action.get(button, '\n'))	1✔
257
258	def prepare_image(self) -> Image.Image:	1✔
259	"""
260	Prepare image for Ka-50 Black Shark to be send to LCD.
261
262	:return: image instance ready display on LCD
263	"""
264	img = super().prepare_image()	1✔
265	draw = ImageDraw.Draw(img)	1✔
266	text1, text2 = '', ''	1✔
267	draw.rectangle(xy=(0, 1, 85, 18), fill=0, outline=255)	1✔
268	draw.rectangle(xy=(0, 22, 85, 39), fill=0, outline=255)	1✔
269	draw.rectangle(xy=(88, 1, 103, 18), fill=0, outline=255)	1✔
270	draw.rectangle(xy=(88, 22, 103, 39), fill=0, outline=255)	1✔
271	l1_text = str(self.get_bios('l1_text'))	1✔
272	l2_text = str(self.get_bios('l2_text'))	1✔
273	if l1_text:	1✔
274	text1 = f'{l1_text[-6:-3]}{self.get_bios("l1_apostr1")}{l1_text[-3:-1]}{self.get_bios("l1_apostr2")}{l1_text[-1]}'	1✔
275	if l2_text:	1✔
276	text2 = f'{l2_text[-6:-3]}{self.get_bios("l2_apostr1")}{l2_text[-3:-1]}{self.get_bios("l2_apostr2")}{l2_text[-1]}'	1✔
277	line1 = f'{self.get_bios("l1_sign")}{text1} {self.get_bios("l1_point")}'	1✔
278	line2 = f'{self.get_bios("l2_sign")}{text2} {self.get_bios("l2_point")}'	1✔
279	draw.text(xy=(2, 3), text=line1, fill=255, font=FONT_16)	1✔
280	draw.text(xy=(2, 24), text=line2, fill=255, font=FONT_16)	1✔
281	self._auto_pilot_switch(draw)	1✔
282	return img	1✔
283
284	def _auto_pilot_switch(self, draw_obj: ImageDraw) -> None:	1✔
285	"""
286	Draw rectangle and add text for autopilot channels in correct coordinates.
287
288	:param draw_obj: ImageDraw object form PIL
289	"""
290	for c_rect, c_text, ap_channel, turn_on in (((111, 1, 124, 18), (114, 3), 'B', self.get_bios('AP_BANK_HOLD_LED')),	1✔
291	((128, 1, 141, 18), (130, 3), 'P', self.get_bios('AP_PITCH_HOLD_LED')),
292	((145, 1, 158, 18), (147, 3), 'F', self.get_bios('AP_FD_LED')),
293	((111, 22, 124, 39), (114, 24), 'H', self.get_bios('AP_HDG_HOLD_LED')),
294	((128, 22, 141, 39), (130, 24), 'A', self.get_bios('AP_ALT_HOLD_LED'))):
295	if turn_on:	1✔
296	draw_obj.rectangle(c_rect, fill=255, outline=255)	×
297	draw_obj.text(xy=c_text, text=ap_channel, fill=0, font=FONT_16)	×
298	else:
299	draw_obj.rectangle(xy=c_rect, fill=0, outline=255)	1✔
300	draw_obj.text(xy=c_text, text=ap_channel, fill=255, font=FONT_16)	1✔
301
302
303	class F14B(Aircraft):	1✔
304	def __init__(self, lcd_type: LcdSize) -> None:	1✔
305	"""
306	Basic constructor.
307
308	:param lcd_type: LCD type
309	"""
310	super().__init__(lcd_type)	1✔
311	self.bios_data: Dict[str, BIOS_VALUE] = {	1✔
312	'RIO_CAP_CLEAR': {'class': 'IntegerBuffer', 'args': {'address': 0x12d4, 'mask': 0x1, 'shift_by': 0x0}, 'value': int()},
313	'RIO_CAP_SW': {'class': 'IntegerBuffer', 'args': {'address': 0x12d2, 'mask': 0x8000, 'shift_by': 0xf}, 'value': int()},
314	'RIO_CAP_NE': {'class': 'IntegerBuffer', 'args': {'address': 0x12d2, 'mask': 0x4000, 'shift_by': 0xe}, 'value': int()},
315	'RIO_CAP_ENTER': {'class': 'IntegerBuffer', 'args': {'address': 0x12d4, 'mask': 0x2, 'shift_by': 0x1}, 'value': int()}}
316
317	def button_request(self, button: int, request: str = '\n') -> str:	1✔
318	"""
319	Prepare F-14 Tomcat specific DCS-BIOS request for button pressed.
320
321	If button is out of scope new line is return.
322
323	:param button: possible values 1-4
324	:type: int
325	:param request: valid DCS-BIOS command as string
326	:type request: str
327	:return: ready to send DCS-BIOS request
328	:rtype: str
329	"""
330	action = {1: 'RIO_CAP_CLEAR 1\nRIO_CAP_CLEAR 0\n',	1✔
331	2: 'RIO_CAP_SW 1\nRIO_CAP_SW 0\n',
332	3: 'RIO_CAP_NE 1\nRIO_CAP_NE 0\n',
333	4: 'RIO_CAP_ENTER 1\nRIO_CAP_ENTER 0\n',
334	9: 'RIO_CAP_CLEAR 1\nRIO_CAP_CLEAR 0\n',
335	10: 'RIO_CAP_SW 1\nRIO_CAP_SW 0\n',
336	14: 'RIO_CAP_NE 1\nRIO_CAP_NE 0\n',
337	13: 'RIO_CAP_ENTER 1\nRIO_CAP_ENTER 0\n'}
338	return super().button_request(button, action.get(button, '\n'))	1✔
339
340	def prepare_image(self) -> Image.Image:	1✔
341	"""
342	Prepare image for F-14B Tomcat to be send to LCD.
343
344	:return: image instance ready display on LCD
345	"""
346	img = super().prepare_image()	1✔
347	draw = ImageDraw.Draw(img)	1✔
348	draw.text(xy=(2, 3), text='F-14B Tomcat', fill=255, font=FONT_16)	1✔
349	return img	1✔

emcek / dcspy / 462

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous