/**
 * PANDA 3D SOFTWARE
 * Copyright (c) Carnegie Mellon University.  All rights reserved.
 *
 * All use of this software is subject to the terms of the revised BSD
 * license.  You should have received a copy of this license along
 * with this source code in a file named "LICENSE."
 *
 * @file mayaNodeTree.h
 * @author drose
 * @date 2003-06-06
 */

#ifndef MAYANODETREE_H
#define MAYANODETREE_H

#include "pandatoolbase.h"

#include "mayaNodeDesc.h"
#include "mayaBlendDesc.h"
#include "globPattern.h"
#include "indirectCompareNames.h"
#include "ordered_vector.h"
#include "pset.h"
#include "pmap.h"

class MayaToEggConverter;
class EggData;
class EggGroupNode;
class EggTable;
class EggXfmSAnim;
class EggSAnimData;

/**
 * Describes a complete tree of maya nodes for conversion.
 */
class MayaNodeTree {
public:
  MayaNodeTree(MayaToEggConverter *converter);
  MayaNodeDesc *build_node(const MDagPath &dag_path);
  bool build_hierarchy();

  void tag_joint_all();
  // bool tag_joint_selected();
  bool tag_joint_named(const GlobPattern &glob);

  void tag_all();
  bool tag_selected();
  bool tag_named(const GlobPattern &glob);
  bool untag_named(const GlobPattern &glob);

  int get_num_nodes() const;
  MayaNodeDesc *get_node(int n) const;

  void clear();
  void clear_egg(EggData *egg_data, EggGroupNode *egg_root,
                 EggGroupNode *skeleton_node, EggGroupNode *morph_node);
  EggGroup *get_egg_group(MayaNodeDesc *node_desc);
  EggTable *get_egg_table(MayaNodeDesc *node_desc);
  EggXfmSAnim *get_egg_anim(MayaNodeDesc *node_desc);
  EggSAnimData *get_egg_slider(MayaBlendDesc *blend_desc);

  bool ignore_slider(const std::string &name) const;
  void report_ignored_slider(const std::string &name);

  MayaBlendDesc *add_blend_desc(MayaBlendDesc *blend_desc);
  int get_num_blend_descs() const;
  MayaBlendDesc *get_blend_desc(int n) const;

  void reset_sliders();

public:
  std::string _subroot_parent_name;
  PT(MayaNodeDesc) _root;
  PN_stdfloat _fps;

private:
  MayaNodeDesc *r_build_node(const std::string &path);

  MayaToEggConverter *_converter;

  EggData *_egg_data;
  EggGroupNode *_egg_root;
  EggGroupNode *_skeleton_node;
  EggGroupNode *_morph_node;

  typedef pmap<std::string, MayaNodeDesc *> NodesByPath;
  NodesByPath _nodes_by_path;

  typedef pvector<MayaNodeDesc *> Nodes;
  Nodes _nodes;

  typedef ov_set<PT(MayaBlendDesc), IndirectCompareNames<MayaBlendDesc> > BlendDescs;
  BlendDescs _blend_descs;

  typedef pset<std::string> Strings;
  Strings _ignored_slider_names;
};

#endif