1 /*
   2  * Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 package org.graalvm.compiler.hotspot.aarch64;
  24 
  25 import static jdk.vm.ci.aarch64.AArch64.r0;
  26 import static jdk.vm.ci.aarch64.AArch64.r1;
  27 import static jdk.vm.ci.aarch64.AArch64.r10;
  28 import static jdk.vm.ci.aarch64.AArch64.r11;
  29 import static jdk.vm.ci.aarch64.AArch64.r12;
  30 import static jdk.vm.ci.aarch64.AArch64.r13;
  31 import static jdk.vm.ci.aarch64.AArch64.r14;
  32 import static jdk.vm.ci.aarch64.AArch64.r15;
  33 import static jdk.vm.ci.aarch64.AArch64.r16;
  34 import static jdk.vm.ci.aarch64.AArch64.r17;
  35 import static jdk.vm.ci.aarch64.AArch64.r18;
  36 import static jdk.vm.ci.aarch64.AArch64.r19;
  37 import static jdk.vm.ci.aarch64.AArch64.r2;
  38 import static jdk.vm.ci.aarch64.AArch64.r20;
  39 import static jdk.vm.ci.aarch64.AArch64.r21;
  40 import static jdk.vm.ci.aarch64.AArch64.r22;
  41 import static jdk.vm.ci.aarch64.AArch64.r23;
  42 import static jdk.vm.ci.aarch64.AArch64.r24;
  43 import static jdk.vm.ci.aarch64.AArch64.r25;
  44 import static jdk.vm.ci.aarch64.AArch64.r26;
  45 import static jdk.vm.ci.aarch64.AArch64.r27;
  46 import static jdk.vm.ci.aarch64.AArch64.r28;
  47 import static jdk.vm.ci.aarch64.AArch64.r3;
  48 import static jdk.vm.ci.aarch64.AArch64.r4;
  49 import static jdk.vm.ci.aarch64.AArch64.r5;
  50 import static jdk.vm.ci.aarch64.AArch64.r6;
  51 import static jdk.vm.ci.aarch64.AArch64.r7;
  52 import static jdk.vm.ci.aarch64.AArch64.r8;
  53 import static jdk.vm.ci.aarch64.AArch64.r9;
  54 import static jdk.vm.ci.aarch64.AArch64.v0;
  55 import static jdk.vm.ci.aarch64.AArch64.v1;
  56 import static jdk.vm.ci.aarch64.AArch64.v10;
  57 import static jdk.vm.ci.aarch64.AArch64.v11;
  58 import static jdk.vm.ci.aarch64.AArch64.v12;
  59 import static jdk.vm.ci.aarch64.AArch64.v13;
  60 import static jdk.vm.ci.aarch64.AArch64.v14;
  61 import static jdk.vm.ci.aarch64.AArch64.v15;
  62 import static jdk.vm.ci.aarch64.AArch64.v16;
  63 import static jdk.vm.ci.aarch64.AArch64.v17;
  64 import static jdk.vm.ci.aarch64.AArch64.v18;
  65 import static jdk.vm.ci.aarch64.AArch64.v19;
  66 import static jdk.vm.ci.aarch64.AArch64.v2;
  67 import static jdk.vm.ci.aarch64.AArch64.v20;
  68 import static jdk.vm.ci.aarch64.AArch64.v21;
  69 import static jdk.vm.ci.aarch64.AArch64.v22;
  70 import static jdk.vm.ci.aarch64.AArch64.v23;
  71 import static jdk.vm.ci.aarch64.AArch64.v24;
  72 import static jdk.vm.ci.aarch64.AArch64.v25;
  73 import static jdk.vm.ci.aarch64.AArch64.v26;
  74 import static jdk.vm.ci.aarch64.AArch64.v27;
  75 import static jdk.vm.ci.aarch64.AArch64.v28;
  76 import static jdk.vm.ci.aarch64.AArch64.v29;
  77 import static jdk.vm.ci.aarch64.AArch64.v3;
  78 import static jdk.vm.ci.aarch64.AArch64.v30;
  79 import static jdk.vm.ci.aarch64.AArch64.v31;
  80 import static jdk.vm.ci.aarch64.AArch64.v4;
  81 import static jdk.vm.ci.aarch64.AArch64.v5;
  82 import static jdk.vm.ci.aarch64.AArch64.v6;
  83 import static jdk.vm.ci.aarch64.AArch64.v7;
  84 import static jdk.vm.ci.aarch64.AArch64.v8;
  85 import static jdk.vm.ci.aarch64.AArch64.v9;
  86 
  87 import java.util.ArrayList;
  88 import java.util.BitSet;
  89 
  90 import org.graalvm.compiler.core.common.alloc.RegisterAllocationConfig;
  91 
  92 import jdk.vm.ci.code.Register;
  93 import jdk.vm.ci.code.RegisterArray;
  94 import jdk.vm.ci.code.RegisterConfig;
  95 
  96 public class AArch64HotSpotRegisterAllocationConfig extends RegisterAllocationConfig {
  97 
  98     // @formatter:off
  99     static final Register[] registerAllocationOrder = {
 100         r0,  r1,  r2,  r3,  r4,  r5,  r6,  r7,
 101         r8,  r9,  r10, r11, r12, r13, r14, r15,
 102         r16, r17, r18, r19, r20, r21, r22, r23,
 103         r24, r25, r26, r27, r28, /* r29, r30, r31 */
 104 
 105         v0,  v1,  v2,  v3,  v4,  v5,  v6,  v7,
 106         v8,  v9,  v10, v11, v12, v13, v14, v15,
 107         v16, v17, v18, v19, v20, v21, v22, v23,
 108         v24, v25, v26, v27, v28, v29, v30, v31
 109     };
 110     // @formatter:on
 111 
 112     public AArch64HotSpotRegisterAllocationConfig(RegisterConfig registerConfig, String[] allocationRestrictedTo) {
 113         super(registerConfig, allocationRestrictedTo);
 114     }
 115 
 116     @Override
 117     protected RegisterArray initAllocatable(RegisterArray registers) {
 118         BitSet regMap = new BitSet(registerConfig.getAllocatableRegisters().size());
 119         for (Register reg : registers) {
 120             regMap.set(reg.number);
 121         }
 122 
 123         ArrayList<Register> allocatableRegisters = new ArrayList<>(registers.size());
 124         for (Register reg : registerAllocationOrder) {
 125             if (regMap.get(reg.number)) {
 126                 allocatableRegisters.add(reg);
 127             }
 128         }
 129 
 130         return super.initAllocatable(new RegisterArray(allocatableRegisters));
 131     }
 132 }