TreeNode.h

Go to the documentation of this file.

// This may look like C code, but it's really -*- C++ -*-
/*
 * Copyright (C) 2008 Emweb bv, Herent, Belgium.
 *
 * See the LICENSE file for terms of use.
 */
#ifndef TREENODE_H_
#define TREENODE_H_
 
#include <Wt/WCompositeWidget.h>
#include <Wt/WText.h>
 
class IconPair;
 
using namespace Wt;
 
namespace Wt {
  class WTable;
  class WImage;
}
 
 
class TreeNode : public WCompositeWidget
{
public:
  TreeNode(const std::string labelText,
           TextFormat labelFormat,
           std::unique_ptr<IconPair> labelIcon);
 
  void addChildNode(std::unique_ptr<TreeNode> node);
 
  void removeChildNode(TreeNode *node, int index);
 
  const std::vector<TreeNode *>& childNodes() const { return childNodes_; }
 
  void collapse();
 
  void expand();
 
private:
  std::vector<TreeNode *>   childNodes_;
 
  TreeNode                 *parentNode_;
 
  WTable                   *layout_;
 
  IconPair                 *expandIcon_;
 
  WImage                   *noExpandIcon_;
 
  IconPair                 *labelIcon_;
 
  WText                    *childCountLabel_;
 
  WContainerWidget         *expandedContent_;
 
  void adjustExpandIcon();
 
  bool isLastChildNode() const;
 
  void childNodesChanged();
 
  bool wasCollapsed_;
 
  void undoCollapse();
 
  void undoExpand();
 
  enum ImageIndex { Middle = 0, Last = 1 };
 
  static std::string imageLine_[];
  static std::string imagePlus_[];
  static std::string imageMin_[];
}; //
 
#endif // WTREENODE_H_