地圖存儲派生的對象

Question

我有父類：

class Data 
{ 
    public: 
     Data (void) { } 
     Virtual int Size (void) 
     { 
      return 100; 
     } 
    protected: 
     map<string, Data*> m; 

}; 

類從類繼承的數據：

class Struct : public Data 
{ 
    public: 
     Struct (void) { } 
     Struct & Add (const string & name, Data x) 
     { 
      Data * tmp = new Data (x); 
      m[name] = tmp; 
      return *this; 
     } 
     void Print (void) 
     { 
      for (const auto & tmp : m) 
       cout << tmp . first << " " << tmp . second -> Size() << endl; 
     } 
}; 

class IntData : public Data 
{ 
    public: 
     IntData (void) { } 
     int Size (void) 
     { 
      return 4; 
     } 
}; 

class DoubleData : public Data 
{ 
    public: 
     DoubleData (void) { } 
     int Size (void) 
     { 
      return 8; 
     } 

}; 

主：

int main (void) 
{ 
    Struct a; 
    a . Add ("Integer",IntData()); 
    a . Print(); 
    return 0; 
} 

---

Current output : Integer 100 
Expected output : Integer 4 

我想創建一個映射，它可以存放從Data類派生的各種類型的對象。但是當我想調用方法大小存儲對象在映射（在這種情況下IntData）應該返回4它總是從父類數據返回值。我該如何解決這個問題？

Answer 1

有你的問題：

 Data * tmp = new Data (x); 

你把到地圖的實際指針是Data父類的一個實例。您正在拷貝構建一個新的Data父類的實例，該實例來自您按值傳遞的參數作爲參數。

您需要更改這個整體功能：

Struct & Add (const string & name, Data *x) 
    { 
     m[name] = x; 
     return *this; 
    } 

，主叫方現在是負責構建任何子類的新實例：

a . Add ("Integer",new IntData); 

那麼，這將作爲您預期。

當然，這種方法帶來了內存泄漏等各種問題，所以你最好使用std::shared_ptr。但是，這將是一個不同的問題...

Answer 2

我已經爲你重寫你的代碼。

#include <unordered_map> 
#include <memory> 
#include <iostream> 

class Data { 
    public: 
     virtual ~Data(){} 
     virtual int Size() = 0; 
}; 

class Struct : public Data { 
      std::unordered_map<std::string, std::unique_ptr<Data>> m; 
    public: 
     Struct& Add(const std::string& name, std::unique_ptr<Data> x) { 
      m[name] = std::move(x); 
      return *this; 
     } 
     void Print() { 
      for(const auto& tmp : m) 
       std::cout << tmp.first << " " << tmp.second->Size() << "\n"; 
     } 
     int Size() override { 
      int sum = 0; 
      for (const auto& tmp : m) 
       sum += tmp.second->Size(); 
      return sum; 
     } 
}; 

class IntData : public Data { 
    public: 
     int Size() override { return 4; } 
}; 

class DoubleData : public Data { 
    public: 
     DoubleData() { } 
     int Size() override { return 8; } 
}; 

int main() { 
    Struct a; 
    a.Add("Integer", std::make_unique<IntData>()); 
    a.Print(); 
} 

Your welcome.