Home | History | Annotate | Download | only in slang
      1 /*
      2  * Copyright 2010, The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *     http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 #ifndef _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_EXPORT_FUNC_H_  // NOLINT
     18 #define _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_EXPORT_FUNC_H_
     19 
     20 #include <list>
     21 #include <string>
     22 
     23 #include "llvm/ADT/StringRef.h"
     24 #include "llvm/Support/raw_ostream.h"
     25 
     26 #include "clang/AST/Decl.h"
     27 
     28 #include "slang_assert.h"
     29 #include "slang_rs_export_type.h"
     30 #include "slang_rs_exportable.h"
     31 
     32 namespace llvm {
     33   class StructType;
     34 }
     35 
     36 namespace clang {
     37   class FunctionDecl;
     38 }   // namespace clang
     39 
     40 namespace slang {
     41 
     42 class RSContext;
     43 
     44 class RSExportFunc : public RSExportable {
     45   friend class RSContext;
     46 
     47  private:
     48   std::string mName;
     49   std::string mMangledName;
     50   bool mShouldMangle;
     51   RSExportRecordType *mParamPacketType;
     52 
     53   RSExportFunc(RSContext *Context, const llvm::StringRef &Name,
     54                const clang::FunctionDecl *FD)
     55     : RSExportable(Context, RSExportable::EX_FUNC),
     56       mName(Name.data(), Name.size()),
     57       mMangledName(),
     58       mShouldMangle(false),
     59       mParamPacketType(nullptr) {
     60 
     61     mShouldMangle = Context->getMangleContext().shouldMangleDeclName(FD);
     62 
     63     if (mShouldMangle) {
     64       llvm::raw_string_ostream BufStm(mMangledName);
     65       Context->getMangleContext().mangleName(FD, BufStm);
     66       BufStm.flush();
     67     }
     68   }
     69 
     70  public:
     71   static RSExportFunc *Create(RSContext *Context,
     72                               const clang::FunctionDecl *FD);
     73 
     74   typedef RSExportRecordType::const_field_iterator const_param_iterator;
     75 
     76   inline const_param_iterator params_begin() const {
     77     slangAssert((mParamPacketType != nullptr) &&
     78                 "Get parameter from export function having no parameter!");
     79     return mParamPacketType->fields_begin();
     80   }
     81   inline const_param_iterator params_end() const {
     82     slangAssert((mParamPacketType != nullptr) &&
     83                 "Get parameter from export function having no parameter!");
     84     return mParamPacketType->fields_end();
     85   }
     86 
     87   inline const std::string &getName(bool mangle = true) const {
     88     return (mShouldMangle && mangle) ? mMangledName : mName;
     89   }
     90 
     91   inline bool hasParam() const
     92     { return (mParamPacketType && !mParamPacketType->getFields().empty()); }
     93   inline size_t getNumParameters() const
     94     { return ((mParamPacketType) ? mParamPacketType->getFields().size() : 0); }
     95 
     96   inline const RSExportRecordType *getParamPacketType() const
     97     { return mParamPacketType; }
     98 
     99   // Check whether the given ParamsPacket type (in LLVM type) is "size
    100   // equivalent" to the one obtained from getParamPacketType(). If the @Params
    101   // is nullptr, means there must be no any parameters.
    102   bool checkParameterPacketType(llvm::StructType *ParamTy) const;
    103 };  // RSExportFunc
    104 
    105 
    106 }   // namespace slang
    107 
    108 #endif  // _FRAMEWORKS_COMPILE_SLANG_SLANG_RS_EXPORT_FUNC_H_  NOLINT
    109