/home/julian/svn/gpucpumd/mdgpu2/data.cpp

Go to the documentation of this file.

#include "data.h"
bool loadPosVelFromFile(std::string filename, int &N, FLOAT_ARRAY_TYPE *pos, FLOAT_ARRAY_TYPE *vel)
{
    std::ifstream indata(filename.c_str());
    assert(indata.good());
    for (int i=0; i<N; ++i)
    {
        if (!indata.eof())
        {
            indata >> pos[i].x >> pos[i].y >> pos[i].z >> vel[i].x >> vel[i].y >> vel[i].z;
        }
        else
        {
            std::cout<<"Not enough input data is provieded for "<<N<<"particles!\n";
            indata.close();
            return false;
        }
    }
    indata.close();
    return true;
}


bool loadInitData(std::string filename, FLOAT_ARRAY_TYPE *pos, FLOAT_ARRAY_TYPE *vel, int* cnn, int &N, int &numNN)
{
    std::ifstream indata(filename.c_str());
    if (!indata.good())
    {
        std::cout<<filename<<" could not be read!\n";
        return false;
    }
    for (int i=0; i<N; ++i)
    {
        if (!indata.eof())
        {
            indata >> pos[i].x >> pos[i].y >> pos[i].z >> vel[i].x >> vel[i].y >> vel[i].z;
            for (int j=0; j<numNN; ++j)
            {
                //read neighbor data
                indata >>cnn[i+j*N];
            }
        }
        else
        {
            std::cout<<"Not enough input data is provieded for "<<N<<"particles!\n";
            return false;
        }
    }
    indata.close();
    return true;
}


bool writeData(std::string &basename, FLOAT_ARRAY_TYPE *pos, FLOAT_ARRAY_TYPE *vel, int *cnn, int &N, int &numNN, int timestamp)
{
    std::ostringstream ss_file;
    std::string file;
    std::ofstream out;
    ss_file.fill('0');
    ss_file<<basename<<"";
    ss_file.width(10);
    ss_file<<timestamp;
    ss_file.width(1);
    ss_file<<".dat";
    file = ss_file.str();
    out.open(file.c_str());
    if (!out.good())
    {
        std::cout<<"File not open!\n";
        return false;
    }
    //set output precision to agree with float type (machine dependent!)
    out.setf(std::ios_base::scientific);
    out.precision(std::numeric_limits< float >::digits10);
    for (int i=0; i<N; ++i)
    {
        out<<pos[i].x<< " "<<pos[i].y<<" "<<pos[i].z<<" "<<vel[i].x<<" "<<vel[i].y<<" "<<vel[i].z<<" ";
        for (int j=0; j<numNN; ++j)
        {
            out<<cnn[i+j*N]<<" ";
        }
        out<<"\n";
    }
    out.close();
    return true;
}

---