Line data    Source code

       1              : /*
       2              :  * Copyright (c) 2012-2015:  G-CSC, Goethe University Frankfurt
       3              :  * Author: Arne Nägel
       4              :  * 
       5              :  * This file is part of UG4.
       6              :  * 
       7              :  * UG4 is free software: you can redistribute it and/or modify it under the
       8              :  * terms of the GNU Lesser General Public License version 3 (as published by the
       9              :  * Free Software Foundation) with the following additional attribution
      10              :  * requirements (according to LGPL/GPL v3 §7):
      11              :  * 
      12              :  * (1) The following notice must be displayed in the Appropriate Legal Notices
      13              :  * of covered and combined works: "Based on UG4 (www.ug4.org/license)".
      14              :  * 
      15              :  * (2) The following notice must be displayed at a prominent place in the
      16              :  * terminal output of covered works: "Based on UG4 (www.ug4.org/license)".
      17              :  * 
      18              :  * (3) The following bibliography is recommended for citation and must be
      19              :  * preserved in all covered files:
      20              :  * "Reiter, S., Vogel, A., Heppner, I., Rupp, M., and Wittum, G. A massively
      21              :  *   parallel geometric multigrid solver on hierarchically distributed grids.
      22              :  *   Computing and visualization in science 16, 4 (2013), 151-164"
      23              :  * "Vogel, A., Reiter, S., Rupp, M., Nägel, A., and Wittum, G. UG4 -- a novel
      24              :  *   flexible software system for simulating pde based models on high performance
      25              :  *   computers. Computing and visualization in science 16, 4 (2013), 165-179"
      26              :  * 
      27              :  * This program is distributed in the hope that it will be useful,
      28              :  * but WITHOUT ANY WARRANTY; without even the implied warranty of
      29              :  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
      30              :  * GNU Lesser General Public License for more details.
      31              :  */
      32              : 
      33              : /*
      34              :  *      (based on connection_viewer_output.h by mrupp)
      35              :  */
      36              : 
      37              : #ifndef CSVGNUPLOTOUTPUT_H_
      38              : #define CSVGNUPLOTOUTPUT_H_
      39              : 
      40              : #include <iostream>
      41              : #include <fstream>
      42              : #include <vector>
      43              : 
      44              : namespace ug
      45              : {
      46              : 
      47              : 
      48              : 
      49              : // WriteVectorCSV (derived from ConnectionViewer output)
      50              : //--------------------------------------------------
      51              : /**
      52              :  * \brief writes to a file in somewhat SparseMatrix-market format (for connection viewer)
      53              :  * \param filename Filename to write matrix to
      54              :  * \param b Vector
      55              :  * \param positions Positions, there has to be one position for each i in (0, ..., max(A.num_rows(), A.num_cols())).
      56              :  * \param dimensions    Dimensions of Positions
      57              :  */
      58              : template<typename Vector_type, typename postype>
      59            0 : void WriteVectorCSV(const char *filename, const Vector_type &b, postype *positions, int dimensions)
      60              : {
      61            0 :         std::fstream file(filename, std::ios::out);
      62              :         size_t rows = b.size();
      63              : 
      64              :         // write status
      65            0 :         file << "# dim=" << dimensions<<  ", rows="<< rows <<std::endl;
      66              :         // write data
      67            0 :         if(dimensions == 1)
      68            0 :                 for(size_t i=0; i < rows; i++)
      69            0 :                                 file << positions[i][0] << ", " << b[i] <<std::endl;
      70              : 
      71            0 :         else if(dimensions == 2)
      72            0 :                 for(size_t i=0; i < rows; i++)
      73            0 :                         file << positions[i][0] << ", " << positions[i][1] << ", " << b[i]<< std::endl;
      74              :         else
      75            0 :                 for(size_t i=0; i < rows; i++)
      76            0 :                   file << positions[i][0] << ", " << positions[i][1] << ", " << positions[i][2] << ", " << b[i] << std::endl;
      77              : 
      78            0 : }
      79              : 
      80              : 
      81              : 
      82              : 
      83              : } // namespace ug
      84              : #endif /* CSVOUTPUT_H_ */