TreeNode.h

Go to the documentation of this file.

// This may look like C code, but it's really -*- C++ -*-
/*
 * Copyright (C) 2008 Emweb bvba, Kessel-Lo, 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_;

  std::unique_ptr<WText>    labelText_;

  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_