1 //===- SPIRVBasicBlock.h � SPIR-V Basic Block --------------------*- C++ -*-===//
2 //
3 //                     The LLVM/SPIRV Translator
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 // Copyright (c) 2014 Advanced Micro Devices, Inc. All rights reserved.
9 //
10 // Permission is hereby granted, free of charge, to any person obtaining a
11 // copy of this software and associated documentation files (the "Software"),
12 // to deal with the Software without restriction, including without limitation
13 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
14 // and/or sell copies of the Software, and to permit persons to whom the
15 // Software is furnished to do so, subject to the following conditions:
16 //
17 // Redistributions of source code must retain the above copyright notice,
18 // this list of conditions and the following disclaimers.
19 // Redistributions in binary form must reproduce the above copyright notice,
20 // this list of conditions and the following disclaimers in the documentation
21 // and/or other materials provided with the distribution.
22 // Neither the names of Advanced Micro Devices, Inc., nor the names of its
23 // contributors may be used to endorse or promote products derived from this
24 // Software without specific prior written permission.
25 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
26 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
27 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
28 // CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
29 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
30 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS WITH
31 // THE SOFTWARE.
32 //
33 //===----------------------------------------------------------------------===//
34 /// \file
35 ///
36 /// This file defines Basic Block class for SPIR-V.
37 ///
38 //===----------------------------------------------------------------------===//
39 
40 #ifndef SPIRVBASICBLOCK_HPP_
41 #define SPIRVBASICBLOCK_HPP_
42 
43 #include "SPIRVValue.h"
44 #include <algorithm>
45 
46 namespace SPIRV{
47 class SPIRVFunction;
48 class SPIRVInstruction;
49 class SPIRVDecoder;
50 class SPIRVBasicBlock: public SPIRVValue {
51 
52 public:
53   SPIRVBasicBlock(SPIRVId TheId, SPIRVFunction *Func);
54 
SPIRVBasicBlock()55   SPIRVBasicBlock():SPIRVValue(OpLabel), ParentF(NULL){
56     setAttr();
57   }
58 
59   SPIRVDecoder getDecoder(std::istream &IS);
getParent()60   SPIRVFunction *getParent() const { return ParentF;}
getNumInst()61   size_t getNumInst() const { return InstVec.size();}
getInst(size_t I)62   SPIRVInstruction *getInst(size_t I) const { return InstVec[I];}
getPrevious(const SPIRVInstruction * I)63   SPIRVInstruction *getPrevious(const SPIRVInstruction *I) const {
64     auto Loc = find(I);
65     if (Loc == InstVec.end() || Loc == InstVec.begin())
66       return nullptr;
67     return *(--Loc);
68   }
getNext(const SPIRVInstruction * I)69   SPIRVInstruction *getNext(const SPIRVInstruction *I) const {
70     auto Loc = find(I);
71     if (Loc == InstVec.end())
72       return nullptr;
73     ++Loc;
74     if (Loc == InstVec.end())
75       return nullptr;
76     return *Loc;
77   }
78 
79   void setScope(SPIRVEntry *Scope);
setParent(SPIRVFunction * F)80   void setParent(SPIRVFunction *F) { ParentF = F;}
81   SPIRVInstruction *addInstruction(SPIRVInstruction *I);
82 
setAttr()83   void setAttr() { setHasNoType();}
84   _SPIRV_DCL_ENCDEC
85   void encodeChildren(spv_ostream &)const;
validate()86   void validate()const {
87     SPIRVValue::validate();
88     assert(ParentF && "Invalid parent function");
89   }
90 
91 private:
92   SPIRVFunction *ParentF;
93   typedef std::vector<SPIRVInstruction *> SPIRVInstructionVector;
94   SPIRVInstructionVector InstVec;
95 
find(const SPIRVInstruction * Inst)96   SPIRVInstructionVector::const_iterator find(const SPIRVInstruction *Inst)
97     const {
98     return std::find(InstVec.begin(), InstVec.end(), Inst);
99   }
100 };
101 
102 typedef SPIRVBasicBlock SPIRVLabel;
103 }
104 
105 #endif
106