From c0931400a1c5f02cf145c90fd7e34216836efd88 Mon Sep 17 00:00:00 2001
From: Boris Kolpackov <boris@codesynthesis.com>
Date: Wed, 12 Sep 2012 14:28:03 +0200
Subject: Implement --output-name, --generate-schema-only, and --at-once
 options

---
 odb/header.cxx | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'odb/header.cxx')

diff --git a/odb/header.cxx b/odb/header.cxx
index 8e6718d..8bf5453 100644
--- a/odb/header.cxx
+++ b/odb/header.cxx
@@ -17,8 +17,12 @@ namespace header
     ostream& os (ctx.os);
 
     os << "#include <memory>" << endl
-       << "#include <cstddef>" << endl // std::size_t
-       << endl;
+       << "#include <cstddef>" << endl; // std::size_t
+
+    if (ctx.features.polymorphic_object)
+      os << "#include <string>" << endl; // For discriminator.
+
+    os << endl;
 
     os << "#include <odb/core.hxx>" << endl
        << "#include <odb/traits.hxx>" << endl
-- 
cgit v1.1