compiler/cpp/src/parse/t_struct.h

   1 // Copyright (c) 2006- Facebook
   2 // Distributed under the Thrift Software License
   3 //
   4 // See accompanying file LICENSE or visit the Thrift site at:
   5 // http://developers.facebook.com/thrift/
   6
   7 #ifndef T_STRUCT_H
   8 #define T_STRUCT_H
   9
  10 #include <vector>
  11 #include <string>
  12
  13 #include "t_type.h"
  14 #include "t_field.h"
  15
  16 // Forward declare that puppy
  17 class t_program;
  18
  19 /**
  20  * A struct is a container for a set of member fields that has a name. Structs
  21  * are also used to implement exception types.
  22  *
  23  * @author Mark Slee <mcslee@facebook.com>
  24  */
  25 class t_struct : public t_type {
  26  public:
  27   t_struct(t_program* program) :
  28     t_type(program),
  29     is_xception_(false),
  30     xsd_all_(false) {}
  31
  32   t_struct(t_program* program, const std::string& name) :
  33     t_type(program, name),
  34     is_xception_(false),
  35     xsd_all_(false) {}
  36
  37   void set_name(const std::string& name) {
  38     name_ = name;
  39   }
  40
  41   void set_xception(bool is_xception) {
  42     is_xception_ = is_xception;
  43   }
  44
  45   void set_xsd_all(bool xsd_all) {
  46     xsd_all_ = xsd_all;
  47   }
  48
  49   bool get_xsd_all() const {
  50     return xsd_all_;
  51   }
  52
  53   void append(t_field* elem) {
  54     members_.push_back(elem);
  55   }
  56
  57   const std::vector<t_field*>& get_members() {
  58     return members_;
  59   }
  60
  61   bool is_struct() const {
  62     return !is_xception_;
  63   }
  64
  65   bool is_xception() const {
  66     return is_xception_;
  67   }
  68
  69   virtual std::string get_fingerprint_material() const {
  70     std::string rv = "{";
  71     std::vector<t_field*>::const_iterator m_iter;
  72     for (m_iter = members_.begin(); m_iter != members_.end(); ++m_iter) {
  73       rv += (*m_iter)->get_fingerprint_material();
  74       rv += ";";
  75     }
  76     rv += "}";
  77     return rv;
  78   }
  79
  80   virtual void generate_fingerprint() {
  81     t_type::generate_fingerprint();
  82     std::vector<t_field*>::const_iterator m_iter;
  83     for (m_iter = members_.begin(); m_iter != members_.end(); ++m_iter) {
  84       (*m_iter)->get_type()->generate_fingerprint();
  85     }
  86   }
  87
  88   bool validate_field(t_field* field) {
  89     int key = field->get_key();
  90     std::vector<t_field*>::const_iterator m_iter;
  91     for (m_iter = members_.begin(); m_iter != members_.end(); ++m_iter) {
  92       if ((*m_iter)->get_key() == key) {
  93         return false;
  94       }
  95     }
  96     return true;
  97   }
  98
  99  private:
 100
 101   std::vector<t_field*> members_;
 102   bool is_xception_;
 103
 104   bool xsd_all_;
 105 };
 106
 107 #endif