我想确定哪些子类在向量中初始化指针。
我准备了一个包含不同类型的覆盖类的向量。之后,我将其设置为同一函数中的参数。在函数中,我想确定哪些类会初始化此参数,并为每个类应用不同的过程。
这是代码和当前输出。
#include <memory>
#include <iostream>
#include <vector>
class Parent
{
public:
Parent() {};
virtual ~Parent() {};
virtual void print()
{
std::cout << "I am parent!" << std::endl;
}
};
class ChildA : public Parent
{
public:
ChildA() {};
~ChildA() {};
void print() override
{
std::cout << "I am child A!" << std::endl;
}
};
class ChildB : public Parent
{
public:
ChildB() {};
~ChildB() {};
void print() override
{
std::cout << "I am child B!" << std::endl;
}
};
void func(std::unique_ptr<Parent>& pa)
{
pa->print();
//if (pa is initialized from Parent)
//{
//}
//if (pa is initialized from ChildA)
//{
//}
//if (pa is initialized from ChildB)
//{
//}
return;
}
int main(int argc, char** argv)
{
std::unique_ptr<Parent> pa = std::make_unique<Parent>();
std::unique_ptr<ChildA> chA = std::make_unique<ChildA>();
std::unique_ptr<ChildB> chB = std::make_unique<ChildB>();
std::vector<std::unique_ptr<Parent>> mList;
mList.push_back(std::move(pa));
mList.push_back(std::move(chA));
mList.push_back(std::move(chB));
for (auto& l : mList)
func(l);
return 0;
}
输出
I am parent!
I am child A!
I am child B!
你能告诉我任何想法吗?
谢谢。
最佳答案
通常,使用虚函数执行多态代码的目的是为了避免执行“这是X”样式的测试。而是在基类中创建虚拟函数doSpecialStuff
,然后通过基类指针(如代码中的print
)调用它,然后该指针对每个派生类进行适当的操作。
如果您确实要测试类型,请使用dynamic_cast
:
if(auto child_ptr=dynamic_cast<ChildA*>(pa.get())){
child_ptr->child_A_function();
}