1 /*
2  * Copyright (c) 2024 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #ifndef OHOS_IDL_SA_MAPTYPE_EMITTER_H
17 #define OHOS_IDL_SA_MAPTYPE_EMITTER_H
18 
19 #include "codegen/SA/sa_type_emitter.h"
20 
21 namespace OHOS {
22 namespace Idl {
23 class SaMapTypeEmitter : public SaTypeEmitter {
24 public:
SetKeyEmitter(const AutoPtr<SaTypeEmitter> & keyEmitter)25     inline void SetKeyEmitter(const AutoPtr<SaTypeEmitter> &keyEmitter)
26     {
27         keyEmitter_ = keyEmitter;
28     }
29 
SetValueEmitter(const AutoPtr<SaTypeEmitter> & valueEmitter)30     inline void SetValueEmitter(const AutoPtr<SaTypeEmitter> &valueEmitter)
31     {
32         valueEmitter_ = valueEmitter;
33     }
34 
35     TypeKind GetTypeKind() override;
36 
37     std::string EmitCppType(TypeMode mode) const override;
38 
39     std::string EmitRustType(bool pt = false, TypeMode mode = TypeMode::NO_MODE) const override;
40 
41     std::string EmitTsType(TypeMode mode = TypeMode::NO_MODE) const override;
42 
43     void EmitCppWriteVar(const std::string &parcelName, const std::string &name, StringBuilder &sb,
44         const std::string &prefix) const override;
45 
46     void EmitCppReadVar(const std::string &parcelName, const std::string &name, StringBuilder &sb,
47         const std::string &prefix, bool emitType) const override;
48 
49     void EmitRustReadVar(const std::string &result, const std::string &name, StringBuilder &sb,
50         const std::string &prefix) const override;
51 
52     void EmitRustWriteVar(const std::string &result, const std::string &name, StringBuilder &sb,
53         const std::string &prefix) const override;
54 
55     void EmitTsWriteVar(const std::string &parcelName, const std::string &name, StringBuilder &sb,
56         const std::string &prefix) const override;
57 
58     void EmitTsReadVar(const std::string &parcelName, const std::string &name, StringBuilder &sb,
59         const std::string &prefix, TypeMode mode = TypeMode::NO_MODE) const override;
60 private:
61     AutoPtr<SaTypeEmitter> keyEmitter_;
62     AutoPtr<SaTypeEmitter> valueEmitter_;
63 };
64 } // namespace Idl
65 } // namespace OHOS
66 
67 #endif // OHOS_IDL_SA_MAPTYPE_EMITTER_H