Mark Slee | e9ce01c | 2007-05-16 02:29:53 +0000 | [diff] [blame] | 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 | |
Mark Slee | 3198572 | 2006-05-24 21:45:31 +0000 | [diff] [blame] | 7 | #ifndef T_STRUCT_H |
| 8 | #define T_STRUCT_H |
| 9 | |
| 10 | #include <vector> |
| 11 | #include <string> |
David Reiss | 18bf22d | 2007-08-28 20:49:17 +0000 | [diff] [blame] | 12 | #include <cstring> |
Mark Slee | 3198572 | 2006-05-24 21:45:31 +0000 | [diff] [blame] | 13 | |
| 14 | #include "t_type.h" |
Mark Slee | e854063 | 2006-05-30 09:24:40 +0000 | [diff] [blame] | 15 | #include "t_field.h" |
Mark Slee | 3198572 | 2006-05-24 21:45:31 +0000 | [diff] [blame] | 16 | |
Mark Slee | f0712dc | 2006-10-25 19:03:57 +0000 | [diff] [blame] | 17 | // Forward declare that puppy |
| 18 | class t_program; |
| 19 | |
Mark Slee | f5377b3 | 2006-10-10 01:42:59 +0000 | [diff] [blame] | 20 | /** |
| 21 | * A struct is a container for a set of member fields that has a name. Structs |
| 22 | * are also used to implement exception types. |
| 23 | * |
| 24 | * @author Mark Slee <mcslee@facebook.com> |
| 25 | */ |
Mark Slee | 3198572 | 2006-05-24 21:45:31 +0000 | [diff] [blame] | 26 | class t_struct : public t_type { |
| 27 | public: |
Mark Slee | f0712dc | 2006-10-25 19:03:57 +0000 | [diff] [blame] | 28 | t_struct(t_program* program) : |
| 29 | t_type(program), |
Mark Slee | 782abbb | 2007-01-19 00:17:02 +0000 | [diff] [blame] | 30 | is_xception_(false), |
David Reiss | 18bf22d | 2007-08-28 20:49:17 +0000 | [diff] [blame] | 31 | xsd_all_(false) |
| 32 | { |
| 33 | memset(fingerprint_, 0, sizeof(fingerprint_)); |
| 34 | } |
Mark Slee | b15a68b | 2006-06-07 06:46:24 +0000 | [diff] [blame] | 35 | |
Mark Slee | f0712dc | 2006-10-25 19:03:57 +0000 | [diff] [blame] | 36 | t_struct(t_program* program, const std::string& name) : |
| 37 | t_type(program, name), |
Mark Slee | 782abbb | 2007-01-19 00:17:02 +0000 | [diff] [blame] | 38 | is_xception_(false), |
David Reiss | 18bf22d | 2007-08-28 20:49:17 +0000 | [diff] [blame] | 39 | xsd_all_(false) |
| 40 | { |
| 41 | memset(fingerprint_, 0, sizeof(fingerprint_)); |
| 42 | } |
Mark Slee | 3198572 | 2006-05-24 21:45:31 +0000 | [diff] [blame] | 43 | |
Mark Slee | 9cb7c61 | 2006-09-01 22:17:45 +0000 | [diff] [blame] | 44 | void set_name(const std::string& name) { |
| 45 | name_ = name; |
| 46 | } |
| 47 | |
Mark Slee | 9cb7c61 | 2006-09-01 22:17:45 +0000 | [diff] [blame] | 48 | void set_xception(bool is_xception) { |
| 49 | is_xception_ = is_xception; |
| 50 | } |
Mark Slee | e854063 | 2006-05-30 09:24:40 +0000 | [diff] [blame] | 51 | |
Mark Slee | 782abbb | 2007-01-19 00:17:02 +0000 | [diff] [blame] | 52 | void set_xsd_all(bool xsd_all) { |
| 53 | xsd_all_ = xsd_all; |
| 54 | } |
| 55 | |
| 56 | bool get_xsd_all() const { |
| 57 | return xsd_all_; |
| 58 | } |
| 59 | |
Mark Slee | 9cb7c61 | 2006-09-01 22:17:45 +0000 | [diff] [blame] | 60 | void append(t_field* elem) { |
| 61 | members_.push_back(elem); |
| 62 | } |
Mark Slee | e854063 | 2006-05-30 09:24:40 +0000 | [diff] [blame] | 63 | |
Mark Slee | 9cb7c61 | 2006-09-01 22:17:45 +0000 | [diff] [blame] | 64 | const std::vector<t_field*>& get_members() { |
| 65 | return members_; |
| 66 | } |
| 67 | |
| 68 | bool is_struct() const { |
| 69 | return !is_xception_; |
| 70 | } |
| 71 | |
| 72 | bool is_xception() const { |
| 73 | return is_xception_; |
| 74 | } |
Mark Slee | 3198572 | 2006-05-24 21:45:31 +0000 | [diff] [blame] | 75 | |
David Reiss | 18bf22d | 2007-08-28 20:49:17 +0000 | [diff] [blame] | 76 | virtual std::string get_fingerprint_material() const { |
| 77 | std::string rv = "{"; |
| 78 | std::vector<t_field*>::const_iterator m_iter; |
| 79 | for (m_iter = members_.begin(); m_iter != members_.end(); ++m_iter) { |
| 80 | rv += (**m_iter).get_fingerprint_material(); |
| 81 | rv += ";"; |
| 82 | } |
| 83 | rv += "}"; |
| 84 | return rv; |
| 85 | } |
| 86 | |
David Reiss | 9885c68 | 2007-08-30 23:12:37 +0000 | [diff] [blame^] | 87 | virtual void generate_fingerprint() { |
| 88 | t_type::generate_fingerprint(); |
| 89 | std::vector<t_field*>::const_iterator m_iter; |
| 90 | for (m_iter = members_.begin(); m_iter != members_.end(); ++m_iter) { |
| 91 | (**m_iter).get_type()->generate_fingerprint(); |
David Reiss | 18bf22d | 2007-08-28 20:49:17 +0000 | [diff] [blame] | 92 | } |
| 93 | } |
| 94 | |
Mark Slee | 3198572 | 2006-05-24 21:45:31 +0000 | [diff] [blame] | 95 | private: |
David Reiss | 18bf22d | 2007-08-28 20:49:17 +0000 | [diff] [blame] | 96 | |
Mark Slee | e854063 | 2006-05-30 09:24:40 +0000 | [diff] [blame] | 97 | std::vector<t_field*> members_; |
Mark Slee | 9cb7c61 | 2006-09-01 22:17:45 +0000 | [diff] [blame] | 98 | bool is_xception_; |
Mark Slee | 782abbb | 2007-01-19 00:17:02 +0000 | [diff] [blame] | 99 | |
| 100 | bool xsd_all_; |
Mark Slee | 3198572 | 2006-05-24 21:45:31 +0000 | [diff] [blame] | 101 | }; |
| 102 | |
| 103 | #endif |