混合模板和非模板访问者方法

时间:2018-12-12 20:55:23

标签: c++ templates design-patterns visitor-pattern

目前,我正在学习访客模式并尝试各种想法。 下面是我当前设置的代码,希望以某种方式起作用。

我希望有两个访问者,一个访问者分别计算RedBlu的实例,另一个访问者计数任何东西(一个可以假设,它是一个Color

通过简单地实现类似于第一个访客的第二个访客当然可以解决,但是不使用单独的变量进行计数,而只是使用一个变量。 但是我认为这是不必要的-例如,如果我有很多不同的颜色,则代码将是非常重复的:该访问者中的所有函数都是相同的,它们只会增加一个变量。当然,有一种更简单的方法,但是如何? 根据标准的访问者图案,我必须为每种颜色类别实现访问功能,因此这似乎不是正确的方法。

有人会如何解决这个问题?

#include <iostream>

class Color
{
public:
    virtual void accept(class Visitor*) = 0;
};

class Red: public Color
{
public:
    /*virtual*/
    void accept(Visitor*);
    void eye()
    {
        std::cout << "Red::eye\n";
    }
};
class Blu: public Color
{
public:
    /*virtual*/
    void accept(Visitor*);
    void sky()
    {
        std::cout << "Blu::sky\n";
    }
};

class Visitor
{
public:
    virtual void visit(Red*) = 0;
    virtual void visit(Blu*) = 0;
};

class CountVisitor: public Visitor
{
public:
    CountVisitor()
    {
        m_num_red = m_num_blu = 0;
    }
    /*virtual*/
    void visit(Red*)
    {
        ++m_num_red;
    }
    /*virtual*/void visit(Blu*)
    {
        ++m_num_blu;
    }
    void report_num()
    {
        std::cout << "Reds " << m_num_red << ", Blus " << m_num_blu << '\n';
    }
private:
    int m_num_red, m_num_blu;
};

class TemplateVisitor: public Visitor
{
public:
    TemplateVisitor() : num_of_colours(0) {}

    /*virtual*/
    template<class C>
    void visit(C* c)
    {
        ++num_of_colours;
    }
    void report_num()
    {
        std::cout << "Colours " << num_of_colours << '\n';
    }

private:
    int num_of_colours;

};


void Red::accept(Visitor *v)
{
    v->visit(this);
}

void Blu::accept(Visitor *v)
{
    v->visit(this);
}

int main()
{
    Color *set[] =
    {
        new Red, new Blu, new Blu, new Red, new Red, nullptr
    };
    CountVisitor count_operation;
    TemplateVisitor template_visitor;
    for (int i = 0; set[i]; i++)
    {
        set[i]->accept(&count_operation);
        set[i]->accept(&template_visitor);
    }
    count_operation.report_num();
    template_visitor.report_num();
}

2 个答案:

答案 0 :(得分:1)

不幸的是,虚拟方法和模板方法无法匹配。

我的意思是……如果您的基类Visitor需要

virtual void visit(Red*) = 0;
virtual void visit(Blu*) = 0;

在派生类中实现两个虚拟方法,您不能使用单个模板方法解决此义务

template<class C>
void visit(C* c)
{
    ++num_of_colours;
}

您必须编写两个方法(绝对不是模板),并带有确切的签名。也许还添加override,以减少出错的风险。

  void visit (Red * r) override
   { ++num_of_colours; }

  void visit (Blu * b) override
   { ++num_of_colours; }

显然,您可以定义两个虚拟重写方法都调用的模板方法(可能使用其他名称,也可以使用visit()

  template <typename C>
  void visit (C * c)
   { ++num_of_colours; }

  void visit (Red * r) override
   { visit<Red>(r); }

  void visit (Blu * b) override
   { visit<Blu>(b); }

这样,您可以在单个模板方法中实现访问者的逻辑并由所有虚拟方法调用

答案 1 :(得分:0)

为什么不只使用地图并为颜色添加一些功能以用作标识符?

class Color
{    
public:
    virtual void accept(class Visitor*) = 0;
    virtual std::string color_name() = 0; 

};

class Visitor
{
public:
    virtual void visit(Color* c);
};

class CountVisitor: public Visitor
{
    std::unordered_map<std::string, int> map; 

public:
    /*virtual*/
    void visit(Color* c)
    {
        map[c.color_name()]++;
    }
};