odb/source.cxx


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

// file      : odb/source.cxx
// author    : Boris Kolpackov <boris@codesynthesis.com>
// copyright : Copyright (c) 2009-2010 Code Synthesis Tools CC
// license   : GNU GPL v2; see accompanying LICENSE file

#include <odb/source.hxx>

namespace
{
  struct class_: traversal::class_, context
  {
    class_ (context& c)
        : context (c)
    {
    }

    virtual void
    traverse (type& c)
    {
      if (c.file () != unit.file () || !odb_class (c))
        return;

      string const& name (c.name ());

      os << "// " << c.name () << endl
         << "//" << endl;

      os << name << "::" << endl
         << name << " (::odb::image& i)"
         << "{"
         << "}";
    }

  private:
    bool
    odb_class (type& c)
    {
      // See if this class defines the ODB-specific c-tor.
      //
      tree t (c.tree_node ());

      for (tree f (TYPE_METHODS (t)); f != 0; f = TREE_CHAIN (f))
      {
        if (DECL_CONSTRUCTOR_P (f))
        {
          // Get the argument list and skip the first (this) argument.
          //
          tree a (TREE_CHAIN (DECL_ARGUMENTS (f)));

          if (a == 0)
            continue;

          tree at (TREE_TYPE (a));

          // Check that it is ::odb::image&.
          //
          if (TREE_CODE (at) != REFERENCE_TYPE)
            continue;

          tree rt (TREE_TYPE (at));
          tree mt (TYPE_MAIN_VARIANT (rt));

          semantics::node* node (unit.find (mt));

          if (node == 0)
            continue;

          semantics::type* t_node (dynamic_cast<semantics::type*> (node));

          if (t_node == 0)
            continue;

          if (t_node->anonymous () || t_node->fq_name () != "::odb::image")
            continue;

          // Make sure it is unqualified.
          //
          if (cp_type_quals (rt) != TYPE_UNQUALIFIED)
            continue; // @@ Should probably be an error/warning.

          // Check that it is the only argument.
          //
          if (TREE_CHAIN (a) != 0)
            continue; // @@ Should probably be an error/warning.

          return true;
        }
      }

      return false;
    }
  };
}

void
generate_source (context& ctx)
{
  traversal::unit unit;
  traversal::defines unit_defines;
  namespace_ ns (ctx);
  class_ c (ctx);

  unit >> unit_defines >> ns;
  unit_defines >> c;

  traversal::defines ns_defines;

  ns >> ns_defines >> ns;
  ns_defines >> c;

  unit.dispatch (ctx.unit);
}