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>
#include <Wt/WText>

class IconPair;

namespace Wt {
  class WTable;
  class WImage;
}


class TreeNode : public Wt::WCompositeWidget
{
public:
  TreeNode(const std::string labelText,
           Wt::TextFormat labelFormat,
           IconPair *labelIcon, Wt::WContainerWidget *parent = 0);

  void addChildNode(TreeNode *node);

  void removeChildNode(TreeNode *node);

  const std::vector<TreeNode *>& childNodes() const { return childNodes_; }

  void collapse();

  void expand();

private:
  std::vector<TreeNode *> childNodes_;

  TreeNode                 *parentNode_;

  Wt::WTable               *layout_;

  IconPair                 *expandIcon_;

  Wt::WImage               *noExpandIcon_;

  IconPair                 *labelIcon_;

  Wt::WText                *labelText_;

  Wt::WText                *childCountLabel_;

  Wt::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_