Twig.hpp

/*
 * Twig
 * Universal Accelerator Parser
 * Copyright (C) 2007 David Sagan
 * 
 * This library is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or (at
 * your option) any later version. 
 * 
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
 * for more details. 
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc., 
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 * 
 * Direct questions, comments, etc to:
 *   David Sagan (dcs16@cornell.edu)
 */

#ifndef Twig_hpp
#define Twig_hpp 1

#include <iostream>
#include <string>
#include <map>
#include <list>
#include <cassert>

#include "UAP/UAPNode.hpp"

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------

struct AttributeID {

  AttributeID ();
  AttributeID (std::string name, std::string value);

  std::string name;
  std::string value;
};

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------

typedef std::vector<AttributeID>                     AttributeIDVec;
typedef std::vector<AttributeID>::iterator           AttributeIDVecIter;
typedef std::vector<AttributeID>::const_iterator     AttributeIDVecCIter;

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------
class NodeID {

public:

  void addAttribute(std::string name, std::string value);

  UAPNode* getChild(UAPNode* parent, bool create = false);

  std::string toString () const;

  bool operator== (const NodeID& n) const;
  bool operator== (UAPNode* u);

  // Data

  std::string  name;   // NOTE: Right now name is input name
  std::string  xml_local_name;
  std::string  xml_prefix;
  std::string  xml_uri;
  AttributeIDVec attribute;
};

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------

typedef std::vector<NodeID>                     NodeIDVec;
typedef std::vector<NodeID>::iterator           NodeIDVecIter;
typedef std::vector<NodeID>::reverse_iterator   NodeIDVecRIter;
typedef std::vector<NodeID>::const_iterator     NodeIDVecCIter;

bool operator== (const NodeIDVec& x, const NodeIDVec& y);

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------

class Twig {

public:

  Twig ();

  Twig (std::string twig_str, bool has_ele_name = false);

  bool fromString (std::string trig_str, bool has_ele_name = false);

  std::string toString ();

  std::string toNodeString (bool include_target_attrib = false);

  UAPNode* getLocalSubNode (UAPNode* root, bool create = false);

  static UAPNode* getLocalSubNode (UAPNode* root, std::string twig_str, bool create = false);

  UAPNode* getSubNode (UAPNode* root);

  static NodeVec getSubNodes (UAPNode* root, std::string twig_str);

  NodeVec getSubNodes (UAPNode* root);

  bool matchNameAndPrefix (UAPNode* named_node);

  bool operator== (Twig t);

public:

  // Data

  std::string name;
  StrList prefix;
  NodeIDVec nodeid_vec;
  std::string target_attribute;
  bool has_brackets;

private:

  StrList break_line_into_words(std::string line, bool& ok);
  static UAPNode* get_sub_node (UAPNode* root, NodeID node0, NodeIDVec id_vec);
  static void get_sub_nodes (UAPNode* root, NodeID node0, 
                                             NodeIDVec id_vec, NodeVec& node_vec);
  static void get_local_sub_nodes (UAPNode* root, NodeIDVec id_vec, NodeVec& node_vec);
  bool matchName (std::string twig_name, std::string node_name);
  bool matchPrefix (StrList& twig_prefix, StrListCIter twig_iter,
                    StrList& node_prefix, StrListCIter node_iter);

  void nodeid_vec_push_back (NodeID& node, const std::string& twig_str);

};

#endif

---