Matteo Franchin | 7c6c2ac | 2014-07-01 18:03:08 +0100 | [diff] [blame] | 1 | # Copyright (C) 2014 The Android Open Source Project |
| 2 | # |
| 3 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | # you may not use this file except in compliance with the License. |
| 5 | # You may obtain a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | # See the License for the specific language governing permissions and |
| 13 | # limitations under the License. |
| 14 | |
| 15 | all_tests = [ |
| 16 | ({'@INT@': 'int', '@SUFFIX@':''}, |
| 17 | [('CheckDiv', 'idiv_by_pow2_', [2**i for i in range(31)]), |
| 18 | ('CheckDiv', 'idiv_by_small_', [i for i in range(3, 16) if i not in (4, 8)]), |
| 19 | ('CheckRem', 'irem_by_pow2_', [2**i for i in range(31)])]), |
| 20 | ({'@INT@': 'long', '@SUFFIX@': 'l'}, |
| 21 | [('CheckDiv', 'ldiv_by_pow2_', [2**i for i in range(63)]), |
| 22 | ('CheckDiv', 'ldiv_by_small_', [i for i in range(3, 16) if i not in (4, 8)]), |
| 23 | ('CheckRem', 'lrem_by_pow2_', [2**i for i in range(63)])]) |
| 24 | ] |
| 25 | |
| 26 | def subst_vars(variables, text): |
| 27 | '''Substitute variables in text.''' |
| 28 | for key, value in variables.iteritems(): |
| 29 | text = text.replace(str(key), str(value)) |
| 30 | return text |
| 31 | |
| 32 | # Generate all the function bodies (in decls) and all the function calls (in calls). |
| 33 | decls, calls = '', {} |
| 34 | for default_vars, tests in all_tests: |
| 35 | local_vars = default_vars.copy() |
| 36 | int_type = local_vars['@INT@'] |
| 37 | for checker, name, values in tests: |
| 38 | local_vars['@CHECKER@'] = checker |
| 39 | for i, value in enumerate(values): |
| 40 | local_vars['@NAME@'] = name + str(i) |
| 41 | local_vars['@VALUE@'] = value |
| 42 | local_vars['@OP@'] = '/' if 'div' in name else '%' |
| 43 | |
| 44 | # Function body. |
| 45 | decls += subst_vars(local_vars, ''' |
| 46 | public static @INT@ @NAME@(@INT@ x) {return x @OP@ @VALUE@@SUFFIX@;}''') |
| 47 | |
| 48 | # Function call and test. |
| 49 | calls[int_type] = calls.get(int_type, '') + subst_vars(local_vars, ''' |
| 50 | @INT@@CHECKER@("@NAME@", @NAME@(x), x, @VALUE@@SUFFIX@);''') |
| 51 | |
| 52 | # Generate the checkers. |
| 53 | checkers = '' |
| 54 | local_vars = {} |
| 55 | for int_type in ('int', 'long'): |
| 56 | local_vars['@INT@'] = int_type |
| 57 | for op, op_name in (('/', 'Div'), ('%', 'Rem')): |
| 58 | local_vars['@OP@'] = op |
| 59 | local_vars['@OP_NAME@'] = op_name |
| 60 | checkers += subst_vars(local_vars, ''' |
| 61 | public static void @INT@Check@OP_NAME@(String desc, @INT@ result, @INT@ dividend, @INT@ divisor) { |
| 62 | @INT@ correct_result = dividend @OP@ divisor; |
| 63 | if (result != correct_result) { |
| 64 | reportError(desc + "(" + dividend + ") == " + result + |
| 65 | " should be " + correct_result); |
| 66 | } |
| 67 | }''') |
| 68 | |
| 69 | |
| 70 | code = \ |
| 71 | '''/* |
| 72 | * Copyright (C) 2014 The Android Open Source Project |
| 73 | * |
| 74 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 75 | * you may not use this file except in compliance with the License. |
| 76 | * You may obtain a copy of the License at |
| 77 | * |
| 78 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 79 | * |
| 80 | * Unless required by applicable law or agreed to in writing, software |
| 81 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 82 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 83 | * See the License for the specific language governing permissions and |
| 84 | * limitations under the License. |
| 85 | */ |
| 86 | |
| 87 | public class Main { |
| 88 | public static int num_errors = 0; |
| 89 | |
| 90 | public static void reportError(String message) { |
| 91 | if (num_errors == 10) { |
| 92 | System.out.println("Omitting other error messages..."); |
| 93 | } else if (num_errors < 10) { |
| 94 | System.out.println(message); |
| 95 | } |
| 96 | num_errors += 1; |
| 97 | } |
| 98 | %s |
| 99 | %s |
| 100 | |
| 101 | public static void intCheckAll(int x) {%s |
| 102 | } |
| 103 | |
| 104 | public static void longCheckAll(long x) {%s |
| 105 | } |
| 106 | |
| 107 | public static void main(String[] args) { |
| 108 | int i; |
| 109 | long l; |
| 110 | |
| 111 | System.out.println("Begin"); |
| 112 | |
| 113 | System.out.println("Int: checking some equally spaced dividends..."); |
| 114 | for (i = -1000; i < 1000; i += 300) { |
| 115 | intCheckAll(i); |
| 116 | intCheckAll(-i); |
| 117 | } |
| 118 | |
| 119 | System.out.println("Int: checking small dividends..."); |
| 120 | for (i = 1; i < 100; i += 1) { |
| 121 | intCheckAll(i); |
| 122 | intCheckAll(-i); |
| 123 | } |
| 124 | |
| 125 | System.out.println("Int: checking big dividends..."); |
| 126 | for (i = 0; i < 100; i += 1) { |
| 127 | intCheckAll(Integer.MAX_VALUE - i); |
| 128 | intCheckAll(Integer.MIN_VALUE + i); |
| 129 | } |
| 130 | |
| 131 | System.out.println("Long: checking some equally spaced dividends..."); |
| 132 | for (l = 0l; l < 1000000000000l; l += 300000000000l) { |
| 133 | longCheckAll(l); |
| 134 | longCheckAll(-l); |
| 135 | } |
| 136 | |
| 137 | System.out.println("Long: checking small dividends..."); |
| 138 | for (l = 1l; l < 100l; l += 1l) { |
| 139 | longCheckAll(l); |
| 140 | longCheckAll(-l); |
| 141 | } |
| 142 | |
| 143 | System.out.println("Long: checking big dividends..."); |
| 144 | for (l = 0l; l < 100l; l += 1l) { |
| 145 | longCheckAll(Long.MAX_VALUE - l); |
| 146 | longCheckAll(Long.MIN_VALUE + l); |
| 147 | } |
| 148 | |
| 149 | System.out.println("End"); |
| 150 | } |
| 151 | } |
| 152 | ''' % (checkers, decls, calls['int'], calls['long']) |
| 153 | |
| 154 | with open('src/Main.java', 'w') as f: |
| 155 | f.write(code) |