1/*
2 * Copyright (C) 2019 Apple Inc. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions
6 * are met:
7 * 1. Redistributions of source code must retain the above copyright
8 * notice, this list of conditions and the following disclaimer.
9 * 2. Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 *
13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#pragma once
27
28#if ENABLE(B3_JIT)
29
30#include "AirLiveness.h"
31#include "AirTmpMap.h"
32
33namespace JSC {
34
35class CCallHelpers;
36
37namespace B3 { namespace Air {
38
39class Code;
40
41class GenerateAndAllocateRegisters {
42 struct TmpData {
43 StackSlot* spillSlot;
44 Reg reg;
45 };
46
47public:
48 GenerateAndAllocateRegisters(Code&);
49
50 void prepareForGeneration();
51 void generate(CCallHelpers&);
52
53private:
54 void insertBlocksForFlushAfterTerminalPatchpoints();
55 void flush(Tmp, Reg);
56 void spill(Tmp, Reg);
57 void alloc(Tmp, Reg, bool isDef);
58 void freeDeadTmpsIfNeeded();
59 bool assignTmp(Tmp&, Bank, bool isDef);
60 void buildLiveRanges(UnifiedTmpLiveness&);
61 bool isDisallowedRegister(Reg);
62
63 Code& m_code;
64 CCallHelpers* m_jit { nullptr };
65
66 TmpMap<TmpData> m_map;
67
68#if !ASSERT_DISABLED
69 Vector<Tmp> m_allTmps[numBanks];
70#endif
71
72 Vector<Reg> m_registers[numBanks];
73 RegisterSet m_availableRegs[numBanks];
74 size_t m_globalInstIndex;
75 IndexMap<Reg, Tmp>* m_currentAllocation { nullptr };
76 bool m_didAlreadyFreeDeadSlots;
77 TmpMap<size_t> m_liveRangeEnd;
78 RegisterSet m_namedUsedRegs;
79 RegisterSet m_namedDefdRegs;
80 RegisterSet m_allowedRegisters;
81
82 struct PatchSpillData {
83 CCallHelpers::Jump jump;
84 CCallHelpers::Label continueLabel;
85 HashMap<Tmp, Arg*> defdTmps;
86 };
87
88 HashMap<BasicBlock*, PatchSpillData> m_blocksAfterTerminalPatchForSpilling;
89};
90
91} } } // namespace JSC::B3::Air
92
93#endif // ENABLE(B3_JIT)
94