code cleanup

include/Accessor.hpp

@@ -6,8 +6,8 @@
 #include <vector>
 #include <cstring>
 #include <string>
-#include <iostream>
 #include <initializer_list>
+#include <ostream>
 
 namespace cytnx {
 

include/Bond.hpp

@@ -1,14 +1,16 @@
 #ifndef CYTNX_BOND_H_
 #define CYTNX_BOND_H_
 
-#include "Type.hpp"
-#include "cytnx_error.hpp"
-#include "Symmetry.hpp"
-#include <initializer_list>
-#include <vector>
+#include <algorithm>
 #include <fstream>
+#include <initializer_list>
 #include <map>
-#include <algorithm>
+#include <ostream>
+#include <vector>
+
+#include "Symmetry.hpp"
+#include "Type.hpp"
+#include "cytnx_error.hpp"
 #include "intrusive_ptr_base.hpp"
 #include "utils/vec_clone.hpp"
 

include/Device.hpp

@@ -1,8 +1,9 @@
 #ifndef CYTNX_DEVICE_H_
 #define CYTNX_DEVICE_H_
 
-#include <vector>
+#include <iostream>
 #include <string>
+#include <vector>
 
 namespace cytnx {
 
@@ -18,7 +19,10 @@ namespace cytnx {
     int Ncpus;
     std::vector<std::vector<bool>> CanAccessPeer;
     Device_class();
-    void Print_Property();
+    void Print_Property() { this->Print_Property(std::cout); }
+    /// @cond
+    void Print_Property(std::ostream &os);
+    /// @endcond
     std::string getname(const int &device_id);
     ~Device_class();
     // void cudaDeviceSynchronize();

include/Gncon.hpp

@@ -1,16 +1,18 @@
 #ifndef CYTNX_GNCON_H_
 #define CYTNX_GNCON_H_
 
-#include "Type.hpp"
-#include "cytnx_error.hpp"
+#include <fstream>
 #include <initializer_list>
-#include <vector>
+#include <iostream>
 #include <map>
-#include <fstream>
-#include "intrusive_ptr_base.hpp"
-#include "utils/utils.hpp"
+#include <vector>
+
+#include "Type.hpp"
 #include "UniTensor.hpp"
 #include "contraction_tree.hpp"
+#include "cytnx_error.hpp"
+#include "intrusive_ptr_base.hpp"
+#include "utils/utils.hpp"
 
 #ifdef BACKEND_TORCH
 #else
@@ -69,7 +71,7 @@ namespace cytnx {
       return true;
     }
 
-    // void print_Gncon() const;
+    // void print_Gncon(std::ostream &os = std::cout) const;
 
     // void PreConstruct(bool force = true);
 
@@ -93,7 +95,7 @@ namespace cytnx {
     virtual void clear();
     virtual std::string getOptimalOrder();
     virtual UniTensor Launch(const bool &optimal = false, const std::string &contract_order = "");
-    virtual void PrintNet(std::ostream &os);
+    virtual void PrintNet(std::ostream &os) const;
     virtual boost::intrusive_ptr<Gncon_base> clone();
     virtual void Savefile(const std::string &fname);
     virtual ~Gncon_base(){};
@@ -137,7 +139,7 @@ namespace cytnx {
       boost::intrusive_ptr<Gncon_base> out(tmp);
       return out;
     }
-    void PrintNet(std::ostream &os);
+    void PrintNet(std::ostream &os) const;
     void Savefile(const std::string &fname);
     ~RegularGncon(){};
   };
@@ -182,7 +184,7 @@ namespace cytnx {
       boost::intrusive_ptr<Gncon_base> out(tmp);
       return out;
     }
-    void PrintNet(std::ostream &os){};
+    void PrintNet(std::ostream &os) const {};
     void Savefile(const std::string &fname){};
     ~FermionGncon(){};
   };
@@ -352,7 +354,10 @@ namespace cytnx {
       out._impl = this->_impl->clone();
       return out;
     }
-    void PrintNet() { this->_impl->PrintNet(std::cout); }
+    void PrintNet() const { this->_impl->PrintNet(std::cout); }
+    /// @cond
+    void PrintNet(std::ostream &os) const { this->_impl->PrintNet(os); }
+    /// @endcond
 
     void Savefile(const std::string &fname) { this->_impl->Savefile(fname); }
   };

include/LinOp.hpp

@@ -1,17 +1,19 @@
 #ifndef CYTNX_LINOP_H_
 #define CYTNX_LINOP_H_
 
-#include "Type.hpp"
-#include "cytnx_error.hpp"
-#include <vector>
+#include <algorithm>
 #include <fstream>
 #include <functional>
+#include <iostream>
 #include <map>
 #include <utility>
-#include <algorithm>
-#include "intrusive_ptr_base.hpp"
+#include <vector>
+
 #include "Tensor.hpp"
+#include "Type.hpp"
 #include "UniTensor.hpp"
+#include "cytnx_error.hpp"
+#include "intrusive_ptr_base.hpp"
 
 #ifdef BACKEND_TORCH
 #else
@@ -156,7 +158,10 @@ namespace cytnx {
     int dtype() const { return this->_dtype; };
     cytnx_uint64 nx() const { return this->_nx; };
 
-    void _print();
+    void _print() { this->_print(std::cout); };
+    /// @cond
+    void _print(std::ostream &os);
+    /// @endcond
 
     /// @cond
     // this expose to interitance:

include/Network.hpp

@@ -1,16 +1,18 @@
 #ifndef CYTNX_NETWORK_H_
 #define CYTNX_NETWORK_H_
 
-#include "Type.hpp"
-#include "cytnx_error.hpp"
+#include <fstream>
 #include <initializer_list>
-#include <vector>
+#include <iostream>
 #include <map>
-#include <fstream>
-#include "intrusive_ptr_base.hpp"
-#include "utils/utils.hpp"
+#include <vector>
+
+#include "Type.hpp"
 #include "UniTensor.hpp"
 #include "contraction_tree.hpp"
+#include "cytnx_error.hpp"
+#include "intrusive_ptr_base.hpp"
+#include "utils/utils.hpp"
 
 #ifdef BACKEND_TORCH
 #else
@@ -95,7 +97,7 @@ namespace cytnx {
       return true;
     }
 
-    // void print_network() const;
+    // void print_network(std::ostream &os = std::cout) const;
 
     // void PreConstruct(bool force = true);
 
@@ -135,7 +137,7 @@ namespace cytnx {
                            const std::vector<std::vector<std::string>> &labels,
                            const std::vector<std::string> &outlabel, const cytnx_int64 &outrk,
                            const std::string &order, const bool optim);
-    virtual void PrintNet(std::ostream &os);
+    virtual void PrintNet(std::ostream &os = std::cout) const;
     virtual boost::intrusive_ptr<Network_base> clone();
     virtual void Savefile(const std::string &fname);
     virtual ~Network_base(){};
@@ -194,7 +196,7 @@ namespace cytnx {
       boost::intrusive_ptr<Network_base> out(tmp);
       return out;
     }
-    void PrintNet(std::ostream &os);
+    void PrintNet(std::ostream &os = std::cout) const;
     void Savefile(const std::string &fname);
     ~RegularNetwork(){};
   };
@@ -245,7 +247,7 @@ namespace cytnx {
       boost::intrusive_ptr<Network_base> out(tmp);
       return out;
     }
-    void PrintNet(std::ostream &os){};
+    void PrintNet(std::ostream &os = std::cout) const {};
     void Savefile(const std::string &fname){};
     ~FermionNetwork(){};
   };
@@ -450,7 +452,10 @@ namespace cytnx {
       out._impl = this->_impl->clone();
       return out;
     }
-    void PrintNet() { this->_impl->PrintNet(std::cout); }
+    void PrintNet() const { this->_impl->PrintNet(std::cout); }
+    /// @cond
+    void PrintNet(std::ostream &os) const { this->_impl->PrintNet(os); }
+    /// @endcond
 
     void Savefile(const std::string &fname) { this->_impl->Savefile(fname); }
   };

include/Symmetry.hpp

@@ -2,7 +2,7 @@
 #define CYTNX_SYMMETRY_H_
 
 #include <fstream>
-#include <ostream>
+#include <iostream>
 #include <string>
 #include <vector>
 
@@ -90,7 +90,7 @@ namespace cytnx {
     virtual fermionParity get_fermion_parity(const cytnx_int64 &in_qnum) const;
     virtual bool is_fermionic() const { return false; };
 
-    virtual void print_info() const;
+    virtual void print_info(std::ostream &os = std::cout) const;
     virtual std::string stype_str() const;
     // virtual std::vector<cytnx_int64>& combine_rule(const std::vector<cytnx_int64> &inL, const
     // std::vector<cytnx_int64> &inR);
@@ -118,7 +118,7 @@ namespace cytnx {
     void combine_rule_(cytnx_int64 &out, const cytnx_int64 &inL, const cytnx_int64 &inR,
                        const bool &is_reverse);
     void reverse_rule_(cytnx_int64 &out, const cytnx_int64 &in);
-    void print_info() const;
+    void print_info(std::ostream &os = std::cout) const;
     std::string stype_str() const override { return "U1"; };
   };
   ///@endcond
@@ -144,7 +144,7 @@ namespace cytnx {
     void combine_rule_(cytnx_int64 &out, const cytnx_int64 &inL, const cytnx_int64 &inR,
                        const bool &is_reverse);
     void reverse_rule_(cytnx_int64 &out, const cytnx_int64 &in);
-    void print_info() const;
+    void print_info(std::ostream &os = std::cout) const;
     std::string stype_str() const override { return "Z" + std::to_string(this->n); };
   };
   ///@endcond
@@ -169,7 +169,7 @@ namespace cytnx {
     void reverse_rule_(cytnx_int64 &out, const cytnx_int64 &in);
     fermionParity get_fermion_parity(const cytnx_int64 &in_qnum) const override;
     bool is_fermionic() const override { return true; };
-    void print_info() const;
+    void print_info(std::ostream &os = std::cout) const;
     std::string stype_str() const override { return "fP"; }
   };
   ///@endcond
@@ -194,7 +194,7 @@ namespace cytnx {
     void reverse_rule_(cytnx_int64 &out, const cytnx_int64 &in);
     fermionParity get_fermion_parity(const cytnx_int64 &in_qnum) const override;
     bool is_fermionic() const override { return true; };
-    void print_info() const;
+    void print_info(std::ostream &os = std::cout) const;
     std::string stype_str() const override { return "f#"; }
   };
   ///@endcond
@@ -524,7 +524,10 @@ namespace cytnx {
     /**
      * @brief Print the information of current Symmetry object.
      */
-    void print_info() const { this->_impl->print_info(); }
+    void print_info() const { this->_impl->print_info(std::cout); }
+    /// @cond
+    void print_info(std::ostream &os) const { this->_impl->print_info(os); }
+    /// @endcond
 
     /**
      * @brief the equality operator of the Symmetry object.