matfile/matfile_concepts/src/Writers/ArraySparseWrapper.hh

Go to the documentation of this file.

/*
    This file is part of matfile.

    Copyright (C) 2010-2011 Andrea Arteaga <arteagaa@student.ethz.ch>

    matfile is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published
    by the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    matfile 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 Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with matfile.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef ARRAYSPARSEWRAPPERWRITER_HPP_
#define ARRAYSPARSEWRAPPERWRITER_HPP_

template<class WrapperClass> class Writer<WrapperClass, ArraySparseWrapper>
{
public:

  static void write(std::ostream& out, const std::string& name, const WrapperClass& wrapper)
  {
    
   
    int beginpos = out.tellp();
  
    // Parameter werden gelesen
      bool isComplex = wrapper.isComplex();
      miINT32_t nnz = wrapper.nnz();
      miINT32_t rows, cols;
      wrapper.size(rows, cols);
      const MATtype realType = wrapper.realType();
      const MATtype imagType = wrapper.imagType();
      const miINT32_t realScalarSize = typeSize(realType);
      const miINT32_t imagScalarSize = typeSize(imagType);
     

    // Write tag
    miINT32_t tag[2];
    tag[0] = miMATRIX;
    tag[1] = 0;
    out.write(reinterpret_cast<char*>(tag), 8);
  

    // Write flags
    miINT32_t flagstag[2];
    flagstag[0] = miUINT32;
    flagstag[1] = 8;
    miUINT32_t flags[2];
    flags[0] = mxSPARSE_CLASS;
    flags[0] += 2048 * isComplex;
    flags[1] = nnz;
    out.write(reinterpret_cast<char*>(flagstag), 8).write(reinterpret_cast<char*>(flags), 8);


    // Write dimensions
    miINT32_t dimensions[4];
    dimensions[0] = miINT32;
    dimensions[1] = 8;
    dimensions[2] = rows;
    dimensions[3] = cols;
    out.write(reinterpret_cast<char*>(dimensions), 16);


    // Write name
    miINT32_t nametag[2];
    nametag[0] = miINT8;
    nametag[1] = name.size();
    out.write(reinterpret_cast<char*>(nametag), 8).write(name.data(), name.size());
    for (miINT32_t i = name.size(); i % 8 != 0; ++i)
      out.put(' ');

  
    // Write ir
    miINT32_t irtag[2];
    irtag[0] = miINT32;
    irtag[1] = 4*nnz;
    out.write(reinterpret_cast<char*>(irtag), 8).write(wrapper.ir(), irtag[1]);
    if (irtag[1] % 8 != 0)
      out.write("    ", 4);

    // Write jc
    miINT32_t jctag[2];
    jctag[0] = miINT32;
    jctag[1] = 4*(cols+1);
    out.write(reinterpret_cast<char*>(jctag), 8).write(wrapper.jc(), jctag[1]);
    if (jctag[1] % 8 != 0)
      out.write("    ", 4);

    // Write pr
    miINT32_t prtag[2];
    prtag[0] = realType;
    prtag[1] = nnz * realScalarSize;
    out.write(reinterpret_cast<char*>(prtag), 8).write(wrapper.pr(), prtag[1]);
    for (miINT32_t i = prtag[1]; i % 8 != 0; ++i)
      out.put(' ');

    // Write pi
    if (isComplex) {
      miINT32_t pitag[2];
      pitag[0] = imagType;
      pitag[1] = nnz * imagScalarSize;
      out.write(reinterpret_cast<char*>(pitag), 8).write(wrapper.pi(), pitag[1]);
      for (miINT32_t i = pitag[1]; i % 8 != 0; ++i)
        out.put(' ');
    }

    // Seek back and write size
    int endpos = out.tellp();
    out.seekp(beginpos+4);
    miINT32_t size = endpos-beginpos-8;
    out.write(reinterpret_cast<char*>(&size), 4);
    out.seekp(endpos);
    
  }
};

#endif /* ARRAYSPARSEWRAPPERWRITER_HPP_ */