[core] Add TGenericClassInfo::AdoptMemberStreamer()

Author: silverweed

core/meta/inc/TGenericClassInfo.h

@@ -21,6 +21,7 @@
 #include "TSchemaHelper.h"
 #include <vector>
 #include <string>
+#include <unordered_map>
 
 // Forward declarations
 class TVirtualIsAProxy;
@@ -72,6 +73,7 @@ namespace ROOT {
       std::vector<ROOT::Internal::TSchemaHelper>  fReadRules;
       std::vector<ROOT::Internal::TSchemaHelper>  fReadRawRules;
       std::vector<ROOT::TClassAlt*>               fAlternate;
+      std::unordered_map<std::string, TMemberStreamer *> fAdoptedMemberStreamers;
 
    public:
       TGenericClassInfo(const char *fullClassname,
@@ -118,6 +120,7 @@ namespace ROOT {
       TClass                           *IsA(const void *obj);
 
       void                              AdoptAlternate(ROOT::TClassAlt *alt);
+      void                              AdoptMemberStreamer(const char *name, TMemberStreamer *strm);
       Short_t                           AdoptStreamer(TClassStreamer*);
       Short_t                           AdoptCollectionProxy(TVirtualCollectionProxy*);
       void                              AdoptCollectionProxyInfo(Detail::TCollectionProxyInfo*);

core/meta/src/TGenericClassInfo.cxx

@@ -281,7 +281,11 @@ namespace Internal {
          fClass->SetConvStreamerFunc(fConvStreamerFunc);
          fClass->SetMerge(fMerge);
          fClass->SetResetAfterMerge(fResetAfterMerge);
-         fClass->AdoptStreamer(fStreamer); fStreamer = nullptr;
+         fClass->AdoptStreamer(fStreamer);
+         fStreamer = nullptr;
+         for (const auto &[name, strm] : fAdoptedMemberStreamers)
+            fClass->AdoptMemberStreamer(name.c_str(), strm);
+         fAdoptedMemberStreamers.clear();
          // If IsZombie is true, something went wrong and we will not be
          // able to properly copy the collection proxy
          if (!fClass->IsZombie()) {
@@ -434,6 +438,21 @@ namespace Internal {
       fAlternate.push_back(alt);
    }
 
+   void TGenericClassInfo::AdoptMemberStreamer(const char *name, TMemberStreamer *strm)
+   {
+      if (fClass) {
+         assert(fAdoptedMemberStreamers.empty());
+         fClass->AdoptMemberStreamer(name, strm);
+         return;
+      }
+
+      auto [it, inserted] = fAdoptedMemberStreamers.emplace(name, strm);
+      if (!inserted) {
+         delete it->second;
+         it->second = strm;
+      }
+   }
+
    void TGenericClassInfo::AdoptCollectionProxyInfo(TCollectionProxyInfo *info)
    {
       // Set the info for the CollectionProxy and take ownership of the object