HilbertTriplet.hpp

/*
 * Copyright (c) 2007-2014, A. N. Yzelman,   Utrecht University 2007-2011;
 *                                                    KU Leuven 2011-2014.
 *                          R. H. Bisseling, Utrecht University 2007-2014.
 * 
 * This file is part of the Sparse Library.
 * 
 * This library was developed under supervision of Prof. dr. Rob H. Bisseling at
 * Utrecht University, from 2007 until 2011. From 2011-2014, development continued 
 * at KU Leuven, where Prof. dr. Dirk Roose contributed significantly to the ideas 
 * behind the newer parts of the library code.
 * 
 *     The Sparse Library 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.
 * 
 *     The Sparse Library 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 the Sparse Library. If not, see <http://www.gnu.org/licenses/>.
 */


/*
 * File created by:
 *     A. N. Yzelman, Dept. of Mathematics, Utrecht University, 2007.
 */


#include "Matrix2HilbertCoordinates.hpp"
#include<vector>
#include<string>
#include<fstream>

#ifndef _H_HILBERT_TRIPLET
#define _H_HILBERT_TRIPLET

template< typename T >
class HilbertTriplet {

   protected:

        size_t row;

        size_t column;

        size_t hilbert1;

        size_t hilbert2;

   public:

        unsigned long int i() const { return row; }

        unsigned long int j() const { return column; }
        
        T value;

        HilbertTriplet( unsigned long int i, unsigned long int j, T val ): row( i ), column( j ), hilbert1( 0 ), hilbert2( 0 ), value( val ) {}

        HilbertTriplet(): row( 0 ), column( 0 ), hilbert1( 0 ), hilbert2( 0 ), value( 0 ) {}

        void calculateHilbertCoordinate() {
                Matrix2HilbertCoordinates::IntegerToHilbert( row, column, hilbert1, hilbert2 );
        }

        void getHilbertCoordinate( size_t &h1, size_t &h2 ) {
                h1 = hilbert1;
                h2 = hilbert2;
        }

        size_t getMostSignificantHilbertBits() {
                return hilbert1;
        }

        size_t getLeastSignificantHilbertBits() {
                return hilbert2;
        }

        static void save( std::string fn, HilbertTriplet< T >* toWrite, const unsigned long int m, const unsigned long int n, const size_t s ) {
                std::fstream myFile ( fn.c_str(), std::ios::out | std::ios::binary);
                myFile.write( (char*) &m, sizeof( unsigned long int ) );
                myFile.write( (char*) &n, sizeof( unsigned long int ) );
                for( size_t i = 0; i<s; i++ ) {
                        const unsigned long int wi = toWrite[ i ].i();
                        const unsigned long int wj = toWrite[ i ].j();
                        const double wv = toWrite[ i ].value;
#ifdef _DEBUG
                        std::cout << "Wrote: ( " << wi << " , " << wj << " , " << wv << " ) " << std::endl;
#endif
                        myFile.write( (char*) &( wi ), sizeof( unsigned long int ) );
                        myFile.write( (char*) &( wj ), sizeof( unsigned long int ) );
                        myFile.write( (char*) &( wv ), sizeof( T ) );
                }
                myFile.close();
        }

        static void save( std::string fn, std::vector< HilbertTriplet< T > > &toWrite, const unsigned long int m, const unsigned long int n ) {
                save( fn, &( toWrite[ 0 ] ), m, n, toWrite.size() );
        }

};

#endif