mirror of
https://github.com/JakeHillion/object-introspection.git
synced 2024-09-19 19:19:05 +01:00
0330ef8945
As we now store ContainerInfo objects in OICodeGen::Config, we can not copy it any more. Change all places that took copies to take const references instead. The copy in OICodeGen modified membersToStub, the contents of which form part of OICache's hash. However, as OICache also previously had its own copy, it would not have been OICodeGen's modifications.
76 lines
2.4 KiB
C++
76 lines
2.4 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
#pragma once
|
|
|
|
#include <filesystem>
|
|
#include <functional>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
#include <vector>
|
|
|
|
#include "ContainerInfo.h"
|
|
#include "OICodeGen.h"
|
|
|
|
struct drgn_type;
|
|
|
|
class SymbolService;
|
|
|
|
namespace type_graph {
|
|
class Class;
|
|
struct Member;
|
|
class TypeGraph;
|
|
} // namespace type_graph
|
|
|
|
class CodeGen {
|
|
public:
|
|
CodeGen(const OICodeGen::Config& config, SymbolService& symbols)
|
|
: config_(config), symbols_(symbols) {
|
|
}
|
|
|
|
/*
|
|
* Helper function to perform all the steps required for code generation for a
|
|
* single drgn_type.
|
|
*/
|
|
bool codegenFromDrgn(struct drgn_type* drgnType, std::string& code);
|
|
|
|
void registerContainer(const std::filesystem::path& path);
|
|
void addDrgnRoot(struct drgn_type* drgnType,
|
|
type_graph::TypeGraph& typeGraph);
|
|
void transform(type_graph::TypeGraph& typeGraph);
|
|
void generate(type_graph::TypeGraph& typeGraph,
|
|
std::string& code,
|
|
struct drgn_type*
|
|
drgnType /* TODO: this argument should not be required */
|
|
);
|
|
|
|
private:
|
|
const OICodeGen::Config& config_;
|
|
SymbolService& symbols_;
|
|
std::vector<ContainerInfo> containerInfos_;
|
|
std::unordered_set<const ContainerInfo*> definedContainers_;
|
|
std::unordered_map<const type_graph::Class*, const type_graph::Member*>
|
|
thriftIssetMembers_;
|
|
|
|
void genDefsThrift(const type_graph::TypeGraph& typeGraph, std::string& code);
|
|
void addGetSizeFuncDefs(const type_graph::TypeGraph& typeGraph,
|
|
std::string& code);
|
|
void getClassSizeFuncDef(const type_graph::Class& c, std::string& code);
|
|
void getClassSizeFuncConcrete(std::string_view funcName,
|
|
const type_graph::Class& c,
|
|
std::string& code) const;
|
|
};
|