ScalarFieldIO.h

Go to the documentation of this file.

/*-------------------------------------------------------------------
Copyright 2011 Ravishankar Sundararaman

This file is part of JDFTx.

JDFTx 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.

JDFTx 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 JDFTx.  If not, see <http://www.gnu.org/licenses/>.
-------------------------------------------------------------------*/

#ifndef JDFTX_CORE_SCALARFIELDIO_H
#define JDFTX_CORE_SCALARFIELDIO_H


#include <core/ScalarField.h>
#include <core/vector3.h>
#include <core/Util.h>

#define Tptr std::shared_ptr<T>

template<typename T> void saveRawBinary(const Tptr& X, FILE* fp)
{       int nWrote = fwriteLE(X->data(), sizeof(typename T::DataType), X->nElem, fp);
        if(nWrote < X->nElem) die("Write failed after %d of %d records.\n", nWrote, X->nElem)
}
template<typename T> void saveRawBinary(const Tptr& X, const char* filename)
{       FILE* fp = fopen(filename, "wb");
        if(!fp) die("Could not open '%s' for writing.\n", filename)
        saveRawBinary(X, fp);
        fclose(fp);
}

template<typename T> void loadRawBinary(Tptr& X, FILE* fp)
{       int nRead = freadLE(X->data(), sizeof(typename T::DataType), X->nElem, fp);
        if(nRead < X->nElem) die("Read failed after %d of %d records.\n", nRead, X->nElem)
}
template<typename T> void loadRawBinary(Tptr& X, const char* filename)
{       FILE* fp = fopen(filename, "rb");
        if(!fp) die("Could not open '%s' for reading.\n", filename)
        
        off_t fLen = fileSize(filename);
        off_t expectedLen = sizeof(typename T::DataType) * X->nElem;
        if(fLen != expectedLen)
        {       die("\nLength of '%s' was %ld instead of the expected %ld bytes.\n"
                                "Hint: Are you really reading the correct file?\n\n",
                                filename, (unsigned long)fLen, (unsigned long)expectedLen);
        }
        
        loadRawBinary(X, fp);
        fclose(fp);
}

#undef Tptr

void saveDX(const ScalarField&, const char* filenamePrefix);

std::vector< std::vector<double> > sphericalize(const ScalarField* dataR, int nColumns, double drFac=1.0, vector3<>* center=0);

void saveSphericalized(const ScalarField* dataR, int nColumns, const char* filename, double drFac=1.0, vector3<>* center=0);


void saveSphericalized(const ScalarFieldTilde* dataG, int nColumns, const char* filename, double dGFac=1.0);

#endif // JDFTX_CORE_SCALARFIELDIO_H