full support of array, pointer and record stuffs

1 files changed, 20 insertions, 1 deletions

diff --git a/TypeIdentifier.cc b/TypeIdentifier.cc
index 8f6894b..3a249da 100644
--- a/TypeIdentifier.cc
+++ b/TypeIdentifier.cc
@@ -54,4 +54,23 @@ std::shared_ptr<Type> ArrayTypeIdentifier::GetType() {
     }
     return type_;
 }
-} // namespace pcc
\ No newline at end of file
+
+RecordTypeIdentifier::RecordTypeIdentifier(Context* context,
+                                   std::list<Declaration>&& members)
+    : TypeIdentifier(context)
+    , members_(std::move(members)) {}
+
+std::shared_ptr<Type> RecordTypeIdentifier::GetType() {
+    if (type_ == nullptr) {
+        std::vector<RecordMember> MemberTypes(members_.size());
+        std::transform(members_.cbegin(), members_.cend(), MemberTypes.begin(),
+                       [](const Declaration& member) {
+                           return std::make_pair(
+                               std::get<0>(member),
+                               std::get<1>(member)->GetType());
+                       });
+        type_ = context_->GetTypeManager()->CreateRecordType(MemberTypes);
+    }
+    return type_;
+}
+} // namespace pcc