/home/brandes/workspace/LAMA/src/lama/StorageIO.hpp

Go to the documentation of this file.

#ifndef LAMA_STORAGE_IO_HPP_
#define LAMA_STORAGE_IO_HPP_

// for dll_import
#include <lama/config.hpp>

// others
#include <lama/LAMAArray.hpp>
#include <lama/io/FileType.hpp>

#include <fstream>

namespace lama
{

/* -------------------------------------------------------------------------- */

class LAMA_DLL_IMPORTEXPORT _StorageIO 
{
public:

    static void writeCSRHeader( const IndexType numRows,
                                const IndexType numValues,
                                const File::FileType& fileType,
                                const std::string& fileName,
                                const PartitionId size,
                                const PartitionId rank );

    static void readCSRHeader( IndexType& numRows,
                               IndexType& numColumns,
                               IndexType& numValues,
                               PartitionId& size,
                               PartitionId& rank,
                               File::FileType& fileType,
                               const std::string& fileName );

    static size_t getIndexDataTypeSize( const File::IndexDataType indexDataType );

protected:

    LAMA_LOG_DECL_STATIC_LOGGER( logger );  

private:

    static const int mIversion;
};

template<typename T>
class LAMA_DLL_IMPORTEXPORT StorageIO : public _StorageIO
{
public:

    typedef T ValueType;    

    static void writeCSRToFile( const PartitionId size,
                                const PartitionId rank,
                                const LAMAArray<IndexType>& csrIA,
                                const IndexType numColumns,
                                const LAMAArray<IndexType>& csrJA,
                                const LAMAArray<ValueType>& csrValues,
                                const std::string& fileName,
                                const File::FileType& fileType,
                                const File::DataType& dataType,
                                const File::IndexDataType indexDataTypeIA /*=LONG*/,
                                const File::IndexDataType indexDataTypeJA /*=LONG*/
                              );

    static void writeCSRToFormattedFile( const LAMAArray<IndexType>& csrIA,
                                         const LAMAArray<IndexType>& csrJA,
                                         const LAMAArray<ValueType>& csrValues,
                                         const std::string& fileName );

    static void writeCSRToBinaryFile( const LAMAArray<IndexType>& csrIA,
                                      const LAMAArray<IndexType>& csrJA,
                                      const LAMAArray<ValueType>& csrValues,
                                      const std::string& fileName,
                                      const long indexDataTypeSizeIA,
                                      const long indexDataTypeSizeJA,
                                      const long dataTypeSize );

    static void writeCSRToXDRFile( const LAMAArray<IndexType>& csrIA,
                                   const LAMAArray<IndexType>& csrJA,
                                   const LAMAArray<ValueType>& csrValues,
                                   const std::string& fileName,
                                   const long indexDataTypeSizeIA,
                                   const long indexDataTypeSizeJA,
                                   const long dataTypeSize );

    static void writeCSRToMMFile( const LAMAArray<IndexType>& csrIA,
                                  const IndexType numColumns,
                                  const LAMAArray<IndexType>& csrJA,
                                  const LAMAArray<ValueType>& csrValues,
                                  const std::string& fileName,
                                  const File::DataType& dataType );

    static void readCSRFromFile( LAMAArray<IndexType>& csrIA,
                                 IndexType& numColumns,
                                 LAMAArray<IndexType>& csrJA,
                                 LAMAArray<ValueType>& csrValues,
                     const std::string& fileName );

    static void readCSRFromFormattedFile( LAMAArray<IndexType>& csrIA,
                                          LAMAArray<IndexType>& csrJA,
                                          LAMAArray<ValueType>& csrValues,
                          const std::string& fileName,
                          const IndexType numRows );

    static void readCSRFromBinaryFile( LAMAArray<IndexType>& csrIA,
                                       LAMAArray<IndexType>& csrJA,
                                       LAMAArray<ValueType>& csrValues,
                       const std::string& fileName,
                       const IndexType numRows );

    static void readCSRFromXDRFile( LAMAArray<IndexType>& csrIA,
                                    LAMAArray<IndexType>& csrJA,
                                    LAMAArray<ValueType>& csrValues,
                                    const std::string& fileName,
                                    const IndexType numRows );

    static void readCSRFromMMFile( LAMAArray<IndexType>& csrIA,
                                   IndexType& numColumns,
                                   LAMAArray<IndexType>& csrJA,
                                   LAMAArray<ValueType>& csrValues,
                                   const std::string& fileName );

    static size_t getDataTypeSize( const File::DataType dataType );
};

/* -------------------------------------------------------------------------- */

}
#endif // LAMA_CSRSTORAGE_HPP_