1 /*
   2  * Copyright (c) 2015, 2018, 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 
  24 
  25 package org.graalvm.compiler.hotspot.aarch64;
  26 
  27 import static jdk.vm.ci.aarch64.AArch64.r0;
  28 import static jdk.vm.ci.aarch64.AArch64.r1;
  29 import static jdk.vm.ci.aarch64.AArch64.r10;
  30 import static jdk.vm.ci.aarch64.AArch64.r11;
  31 import static jdk.vm.ci.aarch64.AArch64.r12;
  32 import static jdk.vm.ci.aarch64.AArch64.r13;
  33 import static jdk.vm.ci.aarch64.AArch64.r14;
  34 import static jdk.vm.ci.aarch64.AArch64.r15;
  35 import static jdk.vm.ci.aarch64.AArch64.r16;
  36 import static jdk.vm.ci.aarch64.AArch64.r17;
  37 import static jdk.vm.ci.aarch64.AArch64.r18;
  38 import static jdk.vm.ci.aarch64.AArch64.r19;
  39 import static jdk.vm.ci.aarch64.AArch64.r2;
  40 import static jdk.vm.ci.aarch64.AArch64.r20;
  41 import static jdk.vm.ci.aarch64.AArch64.r21;
  42 import static jdk.vm.ci.aarch64.AArch64.r22;
  43 import static jdk.vm.ci.aarch64.AArch64.r23;
  44 import static jdk.vm.ci.aarch64.AArch64.r24;
  45 import static jdk.vm.ci.aarch64.AArch64.r25;
  46 import static jdk.vm.ci.aarch64.AArch64.r26;
  47 import static jdk.vm.ci.aarch64.AArch64.r28;
  48 import static jdk.vm.ci.aarch64.AArch64.r3;
  49 import static jdk.vm.ci.aarch64.AArch64.r4;
  50 import static jdk.vm.ci.aarch64.AArch64.r5;
  51 import static jdk.vm.ci.aarch64.AArch64.r6;
  52 import static jdk.vm.ci.aarch64.AArch64.r7;
  53 import static jdk.vm.ci.aarch64.AArch64.r8;
  54 import static jdk.vm.ci.aarch64.AArch64.r9;
  55 import static jdk.vm.ci.aarch64.AArch64.v0;
  56 import static jdk.vm.ci.aarch64.AArch64.v1;
  57 import static jdk.vm.ci.aarch64.AArch64.v10;
  58 import static jdk.vm.ci.aarch64.AArch64.v11;
  59 import static jdk.vm.ci.aarch64.AArch64.v12;
  60 import static jdk.vm.ci.aarch64.AArch64.v13;
  61 import static jdk.vm.ci.aarch64.AArch64.v14;
  62 import static jdk.vm.ci.aarch64.AArch64.v15;
  63 import static jdk.vm.ci.aarch64.AArch64.v16;
  64 import static jdk.vm.ci.aarch64.AArch64.v17;
  65 import static jdk.vm.ci.aarch64.AArch64.v18;
  66 import static jdk.vm.ci.aarch64.AArch64.v19;
  67 import static jdk.vm.ci.aarch64.AArch64.v2;
  68 import static jdk.vm.ci.aarch64.AArch64.v20;
  69 import static jdk.vm.ci.aarch64.AArch64.v21;
  70 import static jdk.vm.ci.aarch64.AArch64.v22;
  71 import static jdk.vm.ci.aarch64.AArch64.v23;
  72 import static jdk.vm.ci.aarch64.AArch64.v24;
  73 import static jdk.vm.ci.aarch64.AArch64.v25;
  74 import static jdk.vm.ci.aarch64.AArch64.v26;
  75 import static jdk.vm.ci.aarch64.AArch64.v27;
  76 import static jdk.vm.ci.aarch64.AArch64.v28;
  77 import static jdk.vm.ci.aarch64.AArch64.v29;
  78 import static jdk.vm.ci.aarch64.AArch64.v3;
  79 import static jdk.vm.ci.aarch64.AArch64.v30;
  80 import static jdk.vm.ci.aarch64.AArch64.v31;
  81 import static jdk.vm.ci.aarch64.AArch64.v4;
  82 import static jdk.vm.ci.aarch64.AArch64.v5;
  83 import static jdk.vm.ci.aarch64.AArch64.v6;
  84 import static jdk.vm.ci.aarch64.AArch64.v7;
  85 import static jdk.vm.ci.aarch64.AArch64.v8;
  86 import static jdk.vm.ci.aarch64.AArch64.v9;
  87 
  88 import java.util.ArrayList;
  89 import java.util.BitSet;
  90 
  91 import org.graalvm.compiler.core.common.alloc.RegisterAllocationConfig;
  92 
  93 import jdk.vm.ci.code.Register;
  94 import jdk.vm.ci.code.RegisterArray;
  95 import jdk.vm.ci.code.RegisterConfig;
  96 
  97 public class AArch64HotSpotRegisterAllocationConfig extends RegisterAllocationConfig {
  98 
  99     // @formatter:off
 100     static final Register[] registerAllocationOrder = {
 101         r0,  r1,  r2,  r3,  r4,  r5,  r6,  r7,
 102         r8,  r9,  r10, r11, r12, r13, r14, r15,
 103         r16, r17, r18, r19, r20, r21, r22, r23,
 104         r24, r25, r26, /* r27, */ r28, /* r29, r30, r31 */
 105 
 106         v0,  v1,  v2,  v3,  v4,  v5,  v6,  v7,
 107         v8,  v9,  v10, v11, v12, v13, v14, v15,
 108         v16, v17, v18, v19, v20, v21, v22, v23,
 109         v24, v25, v26, v27, v28, v29, v30, v31
 110     };
 111     // @formatter:on
 112 
 113     public AArch64HotSpotRegisterAllocationConfig(RegisterConfig registerConfig, String[] allocationRestrictedTo) {
 114         super(registerConfig, allocationRestrictedTo);
 115     }
 116 
 117     @Override
 118     protected RegisterArray initAllocatable(RegisterArray registers) {
 119         BitSet regMap = new BitSet(registerConfig.getAllocatableRegisters().size());
 120         for (Register reg : registers) {
 121             regMap.set(reg.number);
 122         }
 123 
 124         ArrayList<Register> allocatableRegisters = new ArrayList<>(registers.size());
 125         for (Register reg : registerAllocationOrder) {
 126             if (regMap.get(reg.number)) {
 127                 allocatableRegisters.add(reg);
 128             }
 129         }
 130 
 131         return super.initAllocatable(new RegisterArray(allocatableRegisters));
 132     }
 133 }