1f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project/* 2f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Copyright (C) 2007 The Android Open Source Project 3f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 4f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Licensed under the Apache License, Version 2.0 (the "License"); 5f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * you may not use this file except in compliance with the License. 6f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * You may obtain a copy of the License at 7f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 8f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * http://www.apache.org/licenses/LICENSE-2.0 9f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 10f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Unless required by applicable law or agreed to in writing, software 11f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * distributed under the License is distributed on an "AS IS" BASIS, 12f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * See the License for the specific language governing permissions and 14f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * limitations under the License. 15f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 16f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 17f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectpackage com.android.dx.ssa.back; 18f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 19fe107fb6e3f308ac5174ebdc5a794ee880c741d9Jesse Wilsonimport com.android.dx.rop.code.PlainInsn; 20f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.rop.code.RegOps; 21f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.rop.code.RegisterSpec; 22fe107fb6e3f308ac5174ebdc5a794ee880c741d9Jesse Wilsonimport com.android.dx.rop.code.RegisterSpecList; 23f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.rop.code.Rops; 24f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.rop.code.SourcePosition; 25f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.ssa.NormalSsaInsn; 26f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.ssa.RegisterMapper; 27fe107fb6e3f308ac5174ebdc5a794ee880c741d9Jesse Wilsonimport com.android.dx.ssa.SsaBasicBlock; 28f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.ssa.SsaInsn; 29f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.ssa.SsaMethod; 30f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport com.android.dx.util.IntIterator; 31fe107fb6e3f308ac5174ebdc5a794ee880c741d9Jesse Wilsonimport com.android.dx.util.IntSet; 32f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectimport java.util.ArrayList; 33f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 34f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project/** 35d3190a0566518c28656cf5e6f41a8e8697775e26Dan Bornstein * Base class of all register allocators. 36f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 37f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Projectpublic abstract class RegisterAllocator { 38f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** method being processed */ 39f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project protected final SsaMethod ssaMeth; 40f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 41f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** interference graph, indexed by register in both dimensions */ 42f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project protected final InterferenceGraph interference; 43f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 44f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 4599409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * Creates an instance. Call {@code allocateRegisters} to run. 46f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param ssaMeth method to process. 47f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param interference Interference graph, indexed by register in both 48f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * dimensions. 49f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 5099409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project public RegisterAllocator(SsaMethod ssaMeth, 5199409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project InterferenceGraph interference) { 52f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project this.ssaMeth = ssaMeth; 53f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project this.interference = interference; 54f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 55f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 56f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 57f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Indicates whether the method params were allocated at the bottom 58f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * of the namespace, and thus should be moved up to the top of the 59f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * namespace after phi removal. 60f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 6199409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * @return {@code true} if params should be moved from low to high 62f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 63f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public abstract boolean wantsParamsMovedHigh(); 64f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 65f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 66f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Runs the algorithm. 67de75089fb7216d19e9c22cce4dc62a49513477d3Carl Shapiro * 6899409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * @return a register mapper to apply to the {@code SsaMethod} 69f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 70f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project public abstract RegisterMapper allocateRegisters(); 71f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 72f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 73f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Returns the category (width) of the definition site of the register. 7499409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * Returns {@code 1} for undefined registers. 75f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 76f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param reg register 7799409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * @return {@code 1..2} 78f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 7999409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project protected final int getCategoryForSsaReg(int reg) { 80d3190a0566518c28656cf5e6f41a8e8697775e26Dan Bornstein SsaInsn definition = ssaMeth.getDefinitionForRegister(reg); 81f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 82f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project if (definition == null) { 83f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project // an undefined reg 84f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project return 1; 85f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } else { 86f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project return definition.getResult().getCategory(); 87f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 88f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 89f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 90f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 91f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Returns the RegisterSpec of the definition of the register. 92f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 9399409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * @param reg {@code >= 0;} SSA register 94f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @return definition spec of the register or null if it is never defined 9599409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * (for the case of "version 0" SSA registers) 96f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 9799409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project protected final RegisterSpec getDefinitionSpecForSsaReg(int reg) { 9899409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project SsaInsn definition = ssaMeth.getDefinitionForRegister(reg); 99f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 100f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project return definition == null ? null : definition.getResult(); 101f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 102f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 103f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 104f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Returns true if the definition site of this register is a 10599409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * move-param (ie, this is a method parameter). 106de75089fb7216d19e9c22cce4dc62a49513477d3Carl Shapiro * 107f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * @param reg register in question 108d3190a0566518c28656cf5e6f41a8e8697775e26Dan Bornstein * @return {@code true} if this is a method parameter 109f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 110f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project protected boolean isDefinitionMoveParam(int reg) { 111f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project SsaInsn defInsn = ssaMeth.getDefinitionForRegister(reg); 11299409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project 113f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project if (defInsn instanceof NormalSsaInsn) { 114f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project NormalSsaInsn ndefInsn = (NormalSsaInsn) defInsn; 115f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 116f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project return ndefInsn.getOpcode().getOpcode() == RegOps.MOVE_PARAM; 117f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 118f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 119f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project return false; 120f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 121f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 122f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /** 123f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Inserts a move instruction for a specified SSA register before a 124f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * specified instruction, creating a new SSA register and adjusting the 125f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * interference graph in the process. The insn currently must be the 126f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * last insn in a block. 127f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * 12899409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * @param insn {@code non-null;} insn to insert move before, must 12999409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * be last insn in block 13099409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * @param reg {@code non-null;} SSA register to duplicate 13199409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * @return {@code non-null;} spec of new SSA register created by move 132f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 133f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project protected final RegisterSpec insertMoveBefore(SsaInsn insn, 134f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project RegisterSpec reg) { 135f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project SsaBasicBlock block = insn.getBlock(); 136f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project ArrayList<SsaInsn> insns = block.getInsns(); 137f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project int insnIndex = insns.indexOf(insn); 138f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 139d3190a0566518c28656cf5e6f41a8e8697775e26Dan Bornstein if (insnIndex < 0) { 140f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project throw new IllegalArgumentException ( 141f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project "specified insn is not in this block"); 142f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 143f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 144f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project if (insnIndex != insns.size() - 1) { 145f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /* 146f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Presently, the interference updater only works when 147f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * adding before the last insn, and the last insn must have no 148f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * result 149f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 150f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project throw new IllegalArgumentException( 151f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project "Adding move here not supported:" + insn.toHuman()); 152f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 153f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 154f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /* 15599409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project * Get new register and make new move instruction. 156f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 157f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 15899409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project // The new result must not have an associated local variable. 159f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project RegisterSpec newRegSpec = RegisterSpec.make(ssaMeth.makeNewSsaReg(), 160f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project reg.getTypeBearer()); 161f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 16299409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project SsaInsn toAdd = SsaInsn.makeFromRop( 16399409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project new PlainInsn(Rops.opMove(newRegSpec.getType()), 16499409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project SourcePosition.NO_INFO, newRegSpec, 16599409883d9c4c0ffb49b070ce307bb33a9dfe9f1The Android Open Source Project RegisterSpecList.make(reg)), block); 166f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 167f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project insns.add(insnIndex, toAdd); 168f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 169f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project int newReg = newRegSpec.getReg(); 170f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 171f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project /* 172f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * Adjust interference graph based on what's live out of the current 173f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project * block and what's used by the final instruction. 174f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project */ 175f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 176f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project IntSet liveOut = block.getLiveOutRegs(); 177f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project IntIterator liveOutIter = liveOut.iterator(); 178f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 179d3190a0566518c28656cf5e6f41a8e8697775e26Dan Bornstein while (liveOutIter.hasNext()) { 180f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project interference.add(newReg, liveOutIter.next()); 181f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 182f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 183d3190a0566518c28656cf5e6f41a8e8697775e26Dan Bornstein // Everything that's a source in the last insn interferes. 184f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project RegisterSpecList sources = insn.getSources(); 185f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project int szSources = sources.size(); 186f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 187f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project for (int i = 0; i < szSources; i++) { 188f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project interference.add(newReg, sources.get(i).getReg()); 189f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 190f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 191f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project ssaMeth.onInsnsChanged(); 192f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project 193f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project return newRegSpec; 194f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project } 195f6c387128427e121477c1b32ad35cdcaa5101ba3The Android Open Source Project} 196