mirror of
https://github.com/triqs/dft_tools
synced 2024-12-25 22:03:43 +01:00
88 lines
4.4 KiB
C++
88 lines
4.4 KiB
C++
/*******************************************************************************
|
|
*
|
|
* TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
|
*
|
|
* Copyright (C) 2012 by O. Parcollet
|
|
*
|
|
* TRIQS is free software: you can redistribute it and/or modify it under the
|
|
* terms of the GNU General Public License as published by the Free Software
|
|
* Foundation, either version 3 of the License, or (at your option) any later
|
|
* version.
|
|
*
|
|
* TRIQS is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
* details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with
|
|
* TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
******************************************************************************/
|
|
#ifndef TRIQS_CLEF_IO_H
|
|
#define TRIQS_CLEF_IO_H
|
|
#include "../clef.hpp"
|
|
#include <iostream>
|
|
|
|
namespace triqs { namespace clef {
|
|
|
|
/* ---------------------------------------------------------------------------------------------------
|
|
* Printing of the expressions
|
|
* --------------------------------------------------------------------------------------------------- */
|
|
|
|
template<int N> std::ostream &operator <<(std::ostream &sout, placeholder<N> ){return sout << "_"<<N ; }
|
|
template<typename T> std::ostream & operator<<(std::ostream & out, std::reference_wrapper<T> const & x) { return out<< x.get(); }
|
|
|
|
inline std::ostream & variadic_print(std::ostream& out) { return out; }
|
|
template<typename T0, typename... T> std::ostream & variadic_print(std::ostream& out, T0&& t0, T&&... t)
|
|
{ out<< t0<<( sizeof...(T)>0 ? ", ": ""); variadic_print (out, t...); return out;}
|
|
|
|
template<int c, int m > struct print_tuple_impl {
|
|
template<typename Tuple> void operator() (std::ostream& out, Tuple const & t) { out << std::get<c>(t) << (c!=m-1 ? ", ": ""); print_tuple_impl<c+1,m>()(out, t); }
|
|
};
|
|
template<int m> struct print_tuple_impl<m,m> {
|
|
template<typename Tuple> void operator() (std::ostream& out, Tuple const & t) {}
|
|
};
|
|
template<typename Tuple> std::ostream & print_tuple(std::ostream& out, Tuple const & t) {
|
|
print_tuple_impl<1,std::tuple_size<Tuple>::value>()(out,t); return out;
|
|
}
|
|
template<typename Tag, typename L>
|
|
typename std::enable_if<std::is_base_of<tags::unary_op,Tag>::value, std::ostream &>::type
|
|
operator <<(std::ostream &sout, expr<Tag,L> const &ex){
|
|
return sout << "("<< Tag::name() <<" "<<std::get<0>(ex.childs)<<")" ;
|
|
}
|
|
template<typename Tag, typename L, typename R>
|
|
typename std::enable_if<std::is_base_of<tags::binary_op,Tag>::value, std::ostream &>::type
|
|
operator <<(std::ostream &sout, expr<Tag,L,R> const &ex){
|
|
return sout << "("<<std::get<0>(ex.childs) <<" "<< Tag::name() <<" "<<std::get<1>(ex.childs)<<")" ;
|
|
}
|
|
template<typename C, typename A, typename B> std::ostream &operator <<(std::ostream &sout, expr<tags::if_else,C,A,B> const &ex){
|
|
return sout << "(" << std::get<0>(ex.childs) << "?"<< std::get<1>(ex.childs) <<" : "<< std::get<2>(ex.childs) <<")";
|
|
}
|
|
template<typename... T> std::ostream &operator <<(std::ostream &sout, expr<tags::function,T...> const &ex){
|
|
sout << std::get<0>(ex.childs) << "("; print_tuple(sout,ex.childs); return sout<<")" ;
|
|
}
|
|
template<typename... T> std::ostream &operator <<(std::ostream &sout, expr<tags::subscript,T...> const &ex){
|
|
sout << std::get<0>(ex.childs) << "["; print_tuple(sout,ex.childs); return sout<<"]" ;
|
|
}
|
|
|
|
template<typename T> std::ostream &operator <<(std::ostream &sout,
|
|
expr<tags::terminal,T> const &ex) {return sout << std::get<0>(ex.childs) ; }
|
|
|
|
template<typename T> std::ostream &operator <<(std::ostream &sout,
|
|
expr<tags::subscript,T> const &ex) {return sout << std::get<0>(ex.childs) << "["<<std::get<1>(ex.childs)<<"]" ; }
|
|
|
|
template<typename T> std::ostream &operator <<(std::ostream &sout,
|
|
expr<tags::negate,T> const &ex) {return sout << "-("<<std::get<0>(ex.childs) <<")" ; }
|
|
|
|
template< typename Expr, int... Is>
|
|
std::ostream &operator <<(std::ostream &sout, make_fun_impl<Expr,Is...> const & x) {
|
|
sout << "lazy function : ("; variadic_print(sout, placeholder<Is>()...); return sout<< ") --> "<< x.ex ;
|
|
}
|
|
|
|
template<typename F> std::ostream & operator<<(std::ostream & out, function<F> const & x) { return out<<"function ";}
|
|
|
|
}}
|
|
|
|
#endif
|
|
|