wrong_shared_ptr_visitor.cpp

#include <iostream>
#include <memory>

class Visitable1;
class Visitable2;
class BaseVisitable;

class BaseVisitor {
public:
  virtual int visit(std::shared_ptr<BaseVisitable> c) { std::cout << __PRETTY_FUNCTION__ << std::endl; }
  virtual int visit(std::shared_ptr<Visitable1> c) { std::cout << __PRETTY_FUNCTION__ << std::endl; }
  virtual int visit(std::shared_ptr<Visitable2> c) { std::cout << __PRETTY_FUNCTION__ << std::endl; }
  virtual ~BaseVisitor()= default;
};

class Visitor1 : public BaseVisitor{
public:
  int visit(std::shared_ptr<Visitable1> c) override { std::cout << __PRETTY_FUNCTION__ << std::endl; }
  int visit(std::shared_ptr<Visitable2> c) override { std::cout << __PRETTY_FUNCTION__ << std::endl; }
};

class Visitor2 : public BaseVisitor{
public:
  int visit(std::shared_ptr<Visitable2> c) override { std::cout << __PRETTY_FUNCTION__ << std::endl; }
};

/*************************************************************************/

class BaseVisitable : public std::enable_shared_from_this<BaseVisitable> {
public:
  virtual ~BaseVisitable()= default;
  virtual int accept(BaseVisitor *v) { return v->visit(shared_from_this()); }
};

class Visitable1 : public BaseVisitable {
  /* Visitable accept one BaseVisitor */
public:
  int accept(BaseVisitor *v) { return v->visit(shared_from_this()); }
};

class Visitable2 : public BaseVisitable {
  /* Visitable accept one BaseVisitor */
public:
  int accept(BaseVisitor *v) { return v->visit(shared_from_this()); }
};

int main(int argc, char **argv) {
  Visitor1 visitor1;
  Visitor2 visitor2;
  std::shared_ptr<Visitable1> visitable1 = std::make_shared<Visitable1>();
  std::shared_ptr<Visitable2> visitable2 = std::make_shared<Visitable2>();

  std::shared_ptr<BaseVisitable> visitable = visitable1;
  visitable->accept(&visitor1);
  visitable->accept(&visitor2);
  visitable = visitable2;
  visitable->accept(&visitor1);
  visitable->accept(&visitor2);

}