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 WTF_MAKE_FAST_ALLOCATED;
43 WTF_MAKE_NONMOVABLE(GenerateAndAllocateRegisters);
44
45 struct TmpData {
46 StackSlot* spillSlot;
47 Reg reg;
48 };
49
50public:
51 GenerateAndAllocateRegisters(Code&);
52
53 void prepareForGeneration();
54 void generate(CCallHelpers&);
55
56private:
57 void insertBlocksForFlushAfterTerminalPatchpoints();
58 void release(Tmp, Reg);
59 void flush(Tmp, Reg);
60 void spill(Tmp, Reg);
61 void alloc(Tmp, Reg, bool isDef);
62 void freeDeadTmpsIfNeeded();
63 bool assignTmp(Tmp&, Bank, bool isDef);
64 void buildLiveRanges(UnifiedTmpLiveness&);
65 bool isDisallowedRegister(Reg);
66
67 void checkConsistency();
68
69 Code& m_code;
70 CCallHelpers* m_jit { nullptr };
71
72 TmpMap<TmpData> m_map;
73
74#if !ASSERT_DISABLED
75 Vector<Tmp> m_allTmps[numBanks];
76#endif
77
78 Vector<Reg> m_registers[numBanks];
79 RegisterSet m_availableRegs[numBanks];
80 size_t m_globalInstIndex;
81 IndexMap<Reg, Tmp>* m_currentAllocation { nullptr };
82 bool m_didAlreadyFreeDeadSlots;
83 TmpMap<size_t> m_liveRangeEnd;
84 RegisterSet m_namedUsedRegs;
85 RegisterSet m_namedDefdRegs;
86 RegisterSet m_allowedRegisters;
87
88 struct PatchSpillData {
89 CCallHelpers::Jump jump;
90 CCallHelpers::Label continueLabel;
91 HashMap<Tmp, Arg*> defdTmps;
92 };
93
94 HashMap<BasicBlock*, PatchSpillData> m_blocksAfterTerminalPatchForSpilling;
95};
96
97} } } // namespace JSC::B3::Air
98
99#endif // ENABLE(B3_JIT)
100