fork download
  1. #include <string>
  2. #include <memory>
  3. #include <vector>
  4. #include <iostream>
  5. #include <functional>
  6. class File;
  7. class Directory;
  8. class Visitor {
  9. public:
  10. virtual ~Visitor() {}
  11. virtual void visit(File &) = 0;
  12. virtual void visit(Directory &) = 0;
  13. virtual void leave(Directory &) = 0;
  14. };
  15. class Entry {
  16. public:
  17. virtual ~Entry() {};
  18. virtual std::string getName() const = 0;
  19. virtual size_t getSize() = 0;
  20. virtual void accept(Visitor &) = 0;
  21. };
  22. class File : public Entry {
  23. std::string name;
  24. size_t size;
  25. public:
  26. File(std::string name, size_t size) : name(name), size(size) {}
  27. std::string getName() const override {return name;}
  28. size_t getSize() override {return size;}
  29. void accept(Visitor &v) override {v.visit(*this);}
  30. };
  31. class Directory : public Entry {
  32. std::string name;
  33. std::vector<std::shared_ptr<Entry>> es;
  34. public:
  35. Directory(std::string name) : name(name) {}
  36. std::string getName() const override {return name;}
  37. size_t getSize() override {
  38. size_t sum = 0;
  39. for (auto &e : es) sum += e->getSize();
  40. return sum;
  41. }
  42. void accept(Visitor &v) override {
  43. v.visit(*this);
  44. for (auto &e : es) e->accept(v);
  45. v.leave(*this);
  46. }
  47. void add(std::shared_ptr<Entry> entry) {es.push_back(entry);}
  48. };
  49. // こっから↑は安定したクラス階層を想定
  50. //////////////////////////////////////////////////////////////////
  51. // こっから↓はVisitor派生クラスを作ることで「新しいオペレーションを簡単に追加できる」とのこと。
  52. class SizeVisitor : public Visitor {
  53. size_t size;
  54. public:
  55. SizeVisitor() : size(0) {};
  56. void visit(File &file) {size += file.getSize();}
  57. void visit(Directory &) {}
  58. void leave(Directory &) {}
  59. size_t getTotalSize() {return size;}
  60. };
  61. class ListVisitor : public Visitor {
  62. std::string cd;
  63. public:
  64. ListVisitor() : cd("") {}
  65. void visit(File &file) {
  66. std::cout << cd << "/" << file.getName() << " (" << file.getSize() << ")" << std::endl;
  67. }
  68. void visit(Directory &dir) {
  69. std::cout << cd << "/" << dir.getName() << " (" << dir.getSize() << ")" << std::endl;
  70. cd += "/" + dir.getName();
  71. }
  72. void leave(Directory &dir) {
  73. cd.resize(cd.length() - 1 - dir.getName().length());
  74. }
  75. };
  76. class FunctionalVisitor : public Visitor {
  77. std::function<void(File &)> vf;
  78. std::function<void(Directory &)> vd, ld;
  79. public:
  80. FunctionalVisitor(
  81. std::function<void(File &)> vf, std::function<void(Directory &)> vd, std::function<void(Directory &)> ld
  82. ) : vf(vf), vd(vd), ld(ld) {}
  83. void visit(File &file) {vf(file);}
  84. void visit(Directory &dir) {vd(dir);}
  85. void leave(Directory &dir) {ld(dir);}
  86. };
  87. int main() {
  88. auto a = std::make_shared<Directory>("a");
  89. auto b = std::make_shared<File>("b.txt", 100);
  90. auto c = std::make_shared<File>("c.txt", 200);
  91. auto d = std::make_shared<Directory>("d");
  92. auto e = std::make_shared<File>("e.txt", 300);
  93. a->add(b);
  94. a->add(c);
  95. a->add(d);
  96. d->add(e);
  97. SizeVisitor sv;
  98. a->accept(sv);
  99. std::cout << "total size: " << sv.getTotalSize() << std::endl;
  100.  
  101. ListVisitor lv;
  102. std::cout << "\nlist: " << std::endl;
  103. a->accept(lv);
  104.  
  105. std::vector<std::string> dirs;
  106. FunctionalVisitor fv(
  107. [&dirs](File &file) {
  108. if (file.getName().find(".txt") != std::string::npos) {
  109. for (auto &d : dirs) std::cout << "/" << d;
  110. std::cout << "/" << file.getName() << std::endl;
  111. }
  112. }
  113. , [&dirs](Directory &dir) {dirs.push_back(dir.getName());}
  114. , [&dirs](Directory &dir) {dirs.pop_back();}
  115. );
  116. std::cout << "\n.txt: " << std::endl;
  117. a->accept(fv);
  118. return 0;
  119. }
  120. /*
  121. ・https://i...content-available-to-author-only...e.com/oYzkxh を元に若干の整理を行った
  122.  
  123. ・他の人と同様shared_ptrを削除
  124.  値で持てるところは単に値で持つほうがC++っぽいと思う
  125.  ただ「Entry を値で持つのはいやだなあ」とのことなので部分的に残してる
  126.  Javaの参照型変数をshared_ptrに置き換えようとして困るのは
  127.  size_t File::accept(std::shared_ptr<Visitor> v) { return v->visit(std::make_shared<File>(this)); }
  128.  ここがJavaだと単にvisit(this)で済むからスッキリするんだけど
  129.  しかもこれmake_shared(this)だと多重開放するよね??
  130.  
  131. ・オリジナルのイテレータ実装は削除
  132.  Directory クラスがイテレータを公開したくなるのをぐっと堪えて我慢
  133.  
  134. ・インタフェースを整理
  135.  そもそもVisitorパターンってのはGoF本構造の段落にある
  136.  Visitor, Element, ConcreteElementクラスに変更がなさげでクラス階層の安定が前提で
  137.  ConcreteVisitorを追加したり変更したりすることでなんとかしていくパターン。
  138.  だから、それを念頭に依存関係、呼び出し関係を整理していく
  139.  
  140. ・抽象クラスをより抽象的に
  141.  Entry::accept, Visitor::visitは抽象的なままにしておきたい
  142.  この先どんなConcreteVisitorが来ても自然に受け入れられるように真っ白にしておきたいから
  143.  size_tを返すようなことはなんとなくしたくない、色をつけたくない。
  144.  GoF本のサンプルコードもvoidを返すようになってるからそれに従った
  145.  
  146. ・依存関係の正常化
  147.  ConcreteElementがConcreteVisitorを知っているようなのは気持ちが悪い
  148.  void Directory::treePrint() {auto visitor = std::make_shared<ListVisitor>();}
  149.  こういうところが循環すると見通しが悪くなってくるし保守性も下がると思う
  150.  ConcreteElementとConcreteVisitorを組合わせて使うのはクライアントだけでいい
  151.  ここでクライアントとはクラス群を使うユーザ、使う文脈みたいな意味、たとえばこのmain文を書く人。
  152.  SizeVisitorを知っているクライアントは、具象クラス固有の関数SizeVisitor::getTotalSize()も平然と使う。
  153.  このへんの依存関係の整理は徹底したいところ
  154.  
  155. ・visit()側で要素ケアしてたのをaccept()側に移動
  156.  これはGoF本実装の段、「CompositeElementクラスはAcceptオペレーションを次のように実装する」以下のテストコードに準拠。
  157.  void CompositeElement::Accept(Visitor &v) {
  158.   ListIterator<Element *> i(_children);
  159.   for (i.First(); !i.IsDone(); i.Next()) {i.CurrentItem()->Accept(v);
  160.   v.VisitCompositeElement(this);
  161.  }
  162.  これをこっちでするからこそDirectoryクラスがイテレータを外部に公開しなくて済んでる
  163.  
  164. ・Visitor::leave()を追加
  165.  これはVisitorパターンにとって本質的じゃない
  166.  勝手にこちらの実装の都合で用意しただけなんで気にしないで欲しい
  167.  
  168. ・オブジェクト構造を走査する責任をどのオブジェクトに持たせるのか
  169.  これは「GoF本実装の段2.」に20行以上説明があるので興味があったら読んでみてほしい
  170.  「visitorの中に走査のアルゴリズムを入れることもできる。しかし…」とある
  171.  ちなみに今回の実装ではConcreteElement::Accept側で走査を行うようにした
  172. */
  173.  
Success #stdin #stdout 0.01s 5280KB
stdin
Standard input is empty
stdout
total size: 600

list: 
/a (600)
/a/b.txt (100)
/a/c.txt (200)
/a/d (300)
/a/d/e.txt (300)

.txt: 
/a/b.txt
/a/c.txt
/a/d/e.txt