在开关状态情况下,在模板功能内部分配不同类型

时间:2012-06-14 14:00:42

标签: c++

Apple.h

class Apple {
public:
    Apple(int);
    static int typeID;
private:
    int id_;
};

Apple.cpp

#include "Apple.h"
Apple::Apple(int pID) {
    id_ = pID;
}

Potato.h,Potato.cpp与Apple相同

storage.h定义

#pragma once
#include "Apple.h"
#include "Potato.h"
#include <vector>
class Storage {
public:
    Storage();
    template<typename foodName> void store(foodName * object){
        (*getBasket<foodName>()).push_back(object);
    };
    template<typename foodName> int countSize(){
        return (*getBasket<foodName>()).size();
    };

private:
    std::vector<Apple*> applebasket_;
    std::vector<Potato*> potatobasket_;
    template <typename foodName> std::vector<foodName*> * getBasket(){
        std::vector<foodName*> * result;
        switch(foodName::typeID){
            case 0:
                result = &applebasket_;
                break;
            case 1:
                //result = &potatobasket_;
                break;
        }
        return result;
    } 
};

Storage.cpp

#include "Storage.h"
int Apple::typeID;
int Potato::typeID;
Storage::Storage() {
    Apple::typeID = 0;
    Potato::typeID =1;
}

的main.cpp

#include "Storage.h"
#include <iostream>
int main() {
    Apple* apple;
    Potato* potato;
    Storage storage;
    int i;
    for(i = 0;i < 7;i++){
        apple = new Apple(i);
        storage.store<Apple>(apple);  
    }      
    std::cout<<storage.countSize<Apple>();
    return 0;
}

此代码工作并输出正确的向量大小,但如果取消注释switch语句(在Storage.h内)的case行,则编译器(g ++)抛出 “错误:无法在分配中将'std :: vector&lt; Potato *&gt; *'转换为'std :: vector&lt; Apple *&gt; *'。 就像编译器尝试两种情况一样,我无法找到它是否可能以及如何避免这种情况。 我需要这方面的帮助,也许需要一些关于整个事情的建议(一个接口用于不同类型的容器),我最近开始学习C ++,可能我在这里尝试这样做的方式是一团糟。

4 个答案:

答案 0 :(得分:2)

您的代码无法编译,因为两个case都应该编译,这是不可能的,因为case中的类型不同。

这个问题的一个解决方案是使用重载而不是函数模板(这意味着你的类中不需要typeID!):

std::vector<Apple*> * get_basket(Apple *)
{
   return &applebasket_;  //return pointer to the apple basket
}

std::vector<Potato*> * get_basket(Potato *)
{
   return &potatobasket_; //return pointer to the potate basket
}

并将其命名为:

template<typename FoodType> 
void store(FoodType * object)
{
    std::vector<FoodType> * basket = get_basket(static_cast<FoodType*>(0));
    basket->push_back(object);
}

这里的技巧是你有两个重载,每个都有一个不同类型的参数,所以你使用static_cast<FoodType*>(0)来帮助编译器根据<选择正确的重载表达式static_cast<FoodType*>(0)的em> type ,其类型为Apple*Potato*


@Gorpik said in the comment两者(这个以及其他解决方案)都是丑陋的,所以这是解决这个问题的另一种尝试。

base_storage类模板定义为:

template<typename FoodType>
class base_storage
{
    std::vector<FoodType*> m_storage;
    public:
        void store(FoodType *foodItem)
        {
            m_storage.push_back(foodItem);
        }
        size_t count() const
        {
            return m_storage.size();
        }
};

此基类仅存储一个类型的食物项目,但在问题中,我们需要存储两个类型的食物项目。因此,为了做到这一点,让我们定义另一个类Storage从上面的类模板派生为:

class storage : private base_storage<Apple>, private base_storage<Potato>
{
    public:
        template<typename FoodType> 
        void store(FoodType * foodItem)
        {
            base_storage<FoodType>::store(foodItem);
        }
        template<typename FoodType> 
        size_t count() const
        {
            return base_storage<FoodType>::count();
        }
};

请注意以下两点:

  • 班级storage现在没有任何会员资料。它只是将调用转发给基类,该基类是根据模板参数FoodType的类型选择的。
  • 它从基类派生私有。所以它不是is-a关系。

请在此处查看此解决方案的在线演示:http://ideone.com/Ykjo5

这个解决方案的优点在于,如果你想让它适用于三种类型的食物,那么你需要从三个基类派生它:

class storage : private base_storage<Apple>, 
                private base_storage<Potato>,
                private base_storage<Mango>   //added line!
{

     //same as before; no change at all !

};

演示:http://ideone.com/lnMds

答案 1 :(得分:2)

您甚至不需要使用typeID字段。让编译器完成所有工作。

class Storage {
public:
    Storage();
    template<typename foodName> void store(foodName * object){
        (*getBasket<foodName>()).push_back(object);
    };
    template<typename foodName> int countSize(){
        return (*getBasket<foodName>()).size();
    };

private:
    std::vector<Apple*> applebasket_;
    std::vector<Potato*> potatobasket_;

    template <typename foodName> std::vector<foodName*> * getBasket();
};


template <> std::vector<Apple*> * Storage::getBasket()
{
    return &applebasket_;
} 
template <> std::vector<Potato*> * Storage::getBasket()
{
    return &potatobasket_;
} 

答案 2 :(得分:2)

std::vector<Apple *>std::vector<Potato *>是不同的类型。 C ++模板在编译时生成类定义及其所有代码的完整副本,因此所得到的具体类型之间根本没有关系。

如果ApplePotato共享一个共同的父类,请说Food您可以将这两种内容存储在std::vector<Food *>中,但您仍然无法返回预期std::vector<Apple *>的{​​{1}}。

您的std::vector<Food *>方法试图通过模板化返回类型来解决这个问题,但它不能,因为模板是在编译时评估的,而在评估switch语句时运行时间。编译器必须在程序开始运行之前决定getBasket()参数是什么,因此它不知道该switch语句的结果可能是什么。然而,无论如何它必须做出决定,它可以从它在函数体中找到的第一个foodStuff语句得出结果,但不幸的结果是第二个语句无效。

因此,您需要单独存储和检索returnApple(可能使用模板,因为其他一些答案已经建议),或者使用继承以便您可以将它们存储在能够理解共同父类型的东西。

答案 3 :(得分:0)

对这两种情况进行专门化getBasket

或者更好的是,不要以这种方式混合抽象级别,只需提供单独的马铃薯和苹果容器。