温馨提示×

如何自定义c++ visit函数

c++
小樊
88
2024-09-16 13:37:40
栏目: 编程语言

在C++中,实现一个自定义的visit函数可以通过模板和重载技术来完成。这里是一个简单的例子,展示了如何创建一个visit函数,它可以接受不同类型的参数并根据参数类型执行相应的操作。

#include<iostream>
#include<string>

// 定义一个基类,用于多态
class Animal {
public:
    virtual ~Animal() = default;
};

// 定义两个派生类,分别表示狗和猫
class Dog : public Animal {
public:
    void bark() const {
        std::cout << "Woof!"<< std::endl;
    }
};

class Cat : public Animal {
public:
    void meow() const {
        std::cout << "Meow!"<< std::endl;
    }
};

// 定义一个访问者类,用于实现双重分派
class Visitor {
public:
    virtual void visit(const Dog& dog) = 0;
    virtual void visit(const Cat& cat) = 0;
};

// 定义一个具体的访问者类,实现对狗和猫的访问操作
class AnimalVisitor : public Visitor {
public:
    void visit(const Dog& dog) override {
        dog.bark();
    }

    void visit(const Cat& cat) override {
        cat.meow();
    }
};

// 修改Animal类,添加accept方法,用于接受访问者
class Animal {
public:
    virtual ~Animal() = default;

    virtual void accept(Visitor& visitor) const = 0;
};

// 更新Dog和Cat类,实现accept方法
class Dog : public Animal {
public:
    void bark() const {
        std::cout << "Woof!"<< std::endl;
    }

    void accept(Visitor& visitor) const override {
        visitor.visit(*this);
    }
};

class Cat : public Animal {
public:
    void meow() const {
        std::cout << "Meow!"<< std::endl;
    }

    void accept(Visitor& visitor) const override {
        visitor.visit(*this);
    }
};

int main() {
    Dog dog;
    Cat cat;

    AnimalVisitor visitor;

    dog.accept(visitor); // 输出 "Woof!"
    cat.accept(visitor); // 输出 "Meow!"

    return 0;
}

在这个例子中,我们首先定义了一个基类Animal,以及两个派生类DogCat。然后,我们定义了一个访问者类Visitor,它包含两个纯虚函数visit,分别用于访问DogCat。接下来,我们定义了一个具体的访问者类AnimalVisitor,它实现了Visitor接口,并在visit方法中调用了DogCat的相应方法。最后,我们修改了Animal类,添加了一个accept方法,用于接受访问者。在DogCat类中,我们实现了accept方法,并在其中调用了访问者的visit方法。

这样,我们就实现了一个自定义的visit函数,它可以根据传入的参数类型自动选择相应的操作。在main函数中,我们创建了一个AnimalVisitor对象,并使用它访问DogCat对象,分别输出"Woof!“和"Meow!”。

0