blob: 440bad1ec65b9383b371ee28cf207e4438d7d14f (
plain)
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
|
# Copyright 2014 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Classes to do screen capture."""
import logging
from PIL import Image
from autotest_lib.client.cros.multimedia import image_generator
def _unlevel(p):
"""Unlevel a color value from TV level back to PC level
@param p: The color value in one character byte
@return: The color value in integer in PC level
"""
# TV level: 16~236; PC level: 0~255
p = (p - 126) * 128 // 110 + 128
if p < 0:
p = 0
elif p > 255:
p = 255
return p
class CommonChameleonScreenCapturer(object):
"""A class to capture the screen on Chameleon.
Calling its member method capture() captures the screen.
"""
TAG = 'Chameleon'
def __init__(self, chameleon_port):
"""Initializes the CommonChameleonScreenCapturer objects."""
self._chameleon_port = chameleon_port
def capture(self):
"""Captures the screen.
@return An Image object.
"""
logging.info('Capturing the screen on Chameleon...')
image = self._chameleon_port.capture_screen()
# unleveling from TV level [16, 235]
pmin, pmax = image_generator.ImageGenerator.get_extrema(image)
if pmin > 10 and pmax < 240:
logging.info(' (TV level: %d %d)', pmin, pmax)
image = Image.eval(image, _unlevel)
return image
class VgaChameleonScreenCapturer(object):
"""A class to capture the screen on a VGA port of Chameleon.
Calling its member method capture() captures the screen.
"""
TAG = 'Chameleon'
def __init__(self, chameleon_port):
"""Initializes the VgaChameleonScreenCapturer objects."""
self._chameleon_port = chameleon_port
def capture(self):
"""Captures the screen.
@return An Image object.
"""
logging.info('Capturing the screen on a VGA port of Chameleon...')
image = self._chameleon_port.capture_screen()
# Find the box containing white points on its boundary.
boundary = image.convert('L').point(
lambda x: 255 if x >= 220 else 0).getbbox()
logging.info('Boundary: %r', boundary)
image = image.crop(boundary)
return image
class CrosExternalScreenCapturer(object):
"""A class to capture the external screen on Chrome OS.
Calling its member method capture() captures the screen.
"""
TAG = 'CrOS-Ext'
def __init__(self, display_facade):
"""Initializes the CrosExternalScreenCapturer objects."""
self._display_facade = display_facade
def capture(self):
"""Captures the screen.
@return An Image object.
"""
logging.info('Capturing the external screen on CrOS...')
return self._display_facade.capture_external_screen()
class CrosInternalScreenCapturer(object):
"""A class to capture the internal screen on Chrome OS.
Calling its member method capture() captures the screen.
"""
TAG = 'CrOS-Int'
def __init__(self, display_facade):
"""Initializes the CrosInternalScreenCapturer objects."""
self._display_facade = display_facade
def capture(self):
"""Captures the screen.
@return An Image object.
"""
logging.info('Capturing the internal screen on CrOS...')
return self._display_facade.capture_internal_screen()
class CrosCalibrationImageCapturer(object):
"""A class to capture the calibration image on Chrome OS.
Calling its member method capture() captures the image.
"""
TAG = 'Calibration'
def __init__(self, display_facade):
"""Initializes the CrosCalibrationImageCapturer objects."""
self._display_facade = display_facade
def capture(self):
"""Captures the screen.
@return An Image object.
"""
logging.info('Capturing the calibration image on CrOS...')
return self._display_facade.capture_calibration_image()
|
