google-research

1
# coding=utf-8
2
# Copyright 2024 The Google Research Authors.
3
#
4
# Licensed under the Apache License, Version 2.0 (the "License");
5
# you may not use this file except in compliance with the License.
6
# You may obtain a copy of the License at
7
#
8
#     http://www.apache.org/licenses/LICENSE-2.0
9
#
10
# Unless required by applicable law or agreed to in writing, software
11
# distributed under the License is distributed on an "AS IS" BASIS,
12
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
# See the License for the specific language governing permissions and
14
# limitations under the License.
15

16
"""Configurations for all word2act data generation global configs."""
17

18
from __future__ import absolute_import
19
from __future__ import division
20
from __future__ import print_function
21

22

23
# Android Emulator Config
24
ANDROID_LOG_MAX_ABS_X = 32676
25
ANDROID_LOG_MAX_ABS_Y = 32676
26
SCREEN_WIDTH = 540
27
SCREEN_HEIGHT = 960
28
SCREEN_CHANNEL = 4
29
# Rico dataset screen config
30
RICO_SCREEN_WIDTH = 1440
31
RICO_SCREEN_HEIGHT = 2560
32

33
# Data Generation Config
34
LABEL_DEFAULT_VALUE_INT = 0
35
LABEL_DEFAULT_VALUE_STRING = ''
36
LABEL_DEFAULT_INVALID_INT = -1
37
LABEL_DEFAULT_INVALID_STRING = ''
38

39
FEATURE_ANCHOR_PADDING_INT = -1
40
FEATURE_DEFAULT_PADDING_INT = 0
41
FEATURE_DEFAULT_PADDING_FLOAT = -0.0
42
FEATURE_DEFAULT_PADDING_STR = ''
43
TOKEN_DEFAULT_PADDING_INT = 0
44

45
MAX_WORD_NUM_UPPER_BOUND = 30
46
MAX_WORD_LENGTH_UPPER_BOUND = 50
47
SHARD_NUM = 10
48
MAX_INPUT_WORD_NUMBER = 5
49

50
# synthetic action config
51
MAX_OBJ_NAME_WORD_NUM = 3
52
MAX_WIN_OBJ_NAME_WORD_NUM = 10
53
MAX_INPUT_STR_LENGTH = 10
54
NORM_VERTICAL_NEIGHBOR_MARGIN = 0.01
55
NORM_HORIZONTAL_NEIGHBOR_MARGIN = 0.01
56
INPUT_ACTION_UPSAMPLE_RATIO = 1
57

58
# Windows data dimension Config. The numbers are set based on real data
59
# dimension distribution.
60
MAX_UI_OBJ_WORD_NUM_UPPER_BOUND = 20
61
MAX_UI_OBJ_WORD_LENGTH_UPPER_BOUND = 21
62

63
# view hierarchy config
64
MAX_PER_OBJECT_WORD_NUM = 10
65
MAX_WORD_LENGTH = 100
66
TRAINING_BATCH_SIZE = 2
67
UI_OBJECT_TYPE_NUM = 15
68
ADJACENT_BOUNDING_BOX_THRESHOLD = 3
69