diff options
| -rw-r--r-- | binparse/output.hpp | 70 |
1 files changed, 48 insertions, 22 deletions
diff --git a/binparse/output.hpp b/binparse/output.hpp index d942fca..4d17175 100644 --- a/binparse/output.hpp +++ b/binparse/output.hpp @@ -49,14 +49,23 @@ template <typename ValueType> std::ostream& operator<<(std::ostream& os, std::vector<ValueType> const& vec); template <typename Key, typename Value> -std::ostream& operator<<(std::ostream& os, std::map<Key, Value> const& map) { - try { - indent scoped(os); - os << std::endl; - for(auto&& entry : map) { - os << entry.first << ": " << entry.second << std::endl; - } - } catch(std::bad_cast&) { +std::ostream& operator<<(otreestream& os, std::map<Key, Value> const& map) { + indent scoped(os); + os << std::endl; + for(auto&& entry : map) { + os << entry.first << ": " << entry.second << std::endl; + } + + return os; +} + +template <typename Key, typename Value> +std::ostream& operator<<(std::ostream& os, std::map<Key, Value> const& map) { + auto* tree_os = dynamic_cast<otreestream*>(&os); + + if(tree_os) { + *tree_os << map; + } else { os << std::endl; for(auto&& entry : map) { os << entry.first << ": " << entry.second << std::endl; @@ -67,24 +76,31 @@ std::ostream& operator<<(std::ostream& os, std::map<Key, Value> const& map) { } template <typename ValueType> +std::ostream& operator<<(otreestream& os, std::vector<ValueType> const& vec) { + indent scoped(os); + os << std::endl; + for(auto&& entry : vec) { + os << entry << std::endl; + } + + return os; +} + +template <typename ValueType> std::ostream& operator<<(std::ostream& os, std::vector<ValueType> const& vec) { - try { - indent scoped(os); - os << std::endl; - for(auto&& entry : vec) { - os << entry << std::endl; - } - } catch(std::bad_cast&) { + auto* tree_os = dynamic_cast<otreestream*>(&os); + if(tree_os) { + *tree_os << vec; + } else { os << std::endl; for(auto&& entry : vec) { os << entry << std::endl; } } + return os; } - - template<typename T> std::ostream& output_impl(std::ostream& os, T const&, indices<>) { return os; @@ -99,14 +115,24 @@ std::ostream& output_impl(std::ostream& os, T const& t, indices<I, Indices...>) } template <typename T, typename std::enable_if<boost::fusion::traits::is_sequence<T>::value>::type*> +std::ostream& operator<<(otreestream& os, T const& t) { + typedef typename build_indices<boost::fusion::result_of::size<T>::value>::type indices; + + indent scoped(os); + os << std::endl; + output_impl<T>(os, t, indices{}); + + return os; +} + +template <typename T, typename std::enable_if<boost::fusion::traits::is_sequence<T>::value>::type*> std::ostream& operator<<(std::ostream& os, T const& t) { typedef typename build_indices<boost::fusion::result_of::size<T>::value>::type indices; - try { - indent scoped(os); - os << std::endl; - output_impl<T>(os, t, indices{}); - } catch(std::bad_cast&) { + auto* tree_os = dynamic_cast<otreestream*>(&os); + if(tree_os) { + *tree_os << t; + } else { os << std::endl; output_impl<T>(os, t, indices{}); } |