PsDoublModel.cpp

Go to the documentation of this file.

#include "PsDoublModel.h"
#include <fstream>
#include <stdio.h>
#include <iostream>
#include <cstdlib>
#include "CFatalException.h"

using namespace std;
PsDoublModel::PsDoublModel()
{}
PsDoublModel::~PsDoublModel()
{}


int PsDoublModel::SetupInsertPoints(const int nexpected)
{
        InsertPoints.clear();
        const int size = GetPVelocity().size();
        for (int i = 0; i < size; ++i)
        {
                if (GetThickness().at(i) < 0)
                        InsertPoints.push_back(&(SetThickness().at(i)));
                if (GetPVelocity().at(i) < 0)
                        InsertPoints.push_back(&(SetPVelocity().at(i)));
                if (GetSVelocity().at(i) < 0)
                        InsertPoints.push_back(&(SetSVelocity().at(i)));
                if (GetDensity().at(i) < 0)
                        InsertPoints.push_back(&(SetDensity().at(i)));
                if (GetQp().at(i) < 0)
                        InsertPoints.push_back(&(SetQp().at(i)));
                if (GetQs().at(i) < 0)
                        InsertPoints.push_back(&(SetQs().at(i)));
        }
        return 0;
}

void PsDoublModel::GetData(const std::string filename)
{
        double thick,pvel,svel,dens,qp,qs;
        int flag;

        if (!GetPVelocity().empty())
        {
                SetPVelocity().clear();
                SetThickness().clear();
                SetSVelocity().clear();
                SetDensity().clear();
                Flags.clear();
                SetQp().clear();
                SetQs().clear();
        }
        ifstream infile(filename.c_str());
        if (!infile)
        {
                throw CFatalException("Error opening file: "+filename);
        }
        while (infile.good())
        {
                infile >> thick >> pvel >> svel >> dens >> qp >> qs >>flag;
                if (infile.good())
                {
                        SetThickness().push_back(thick);
                        SetPVelocity().push_back(pvel);
                        SetSVelocity().push_back(svel);
                        SetDensity().push_back(dens);
                        SetQp().push_back(qp);
                        SetQs().push_back(qs);
                        Flags.push_back(flag);
                }
        }
}
void PsDoublModel::WriteData(const std::string filename)
{
        const unsigned int size = GetPVelocity().size();

        FILE *outfile;

        outfile = fopen((filename+".mod").c_str(),"w");

        for (int i =0; i < size; ++i)
        fprintf(outfile,"%10.3lf %10.3lf %10.3lf %10.3lf %10.3lf %10.3lf %5d \n",
                GetThickness().at(i),GetPVelocity().at(i),GetSVelocity().at(i),GetDensity().at(i),GetQp().at(i),GetQs().at(i),Flags.at(i));
        fclose(outfile);
}
void PsDoublModel::WriteRunFile(const std::string &filename)
{
        ofstream runfile;

        runfile.open(filename.c_str());
        runfile << "#!/bin/bash" << endl;
        runfile << "cp psfui_file "<< filename+"_dir" <<endl;
        runfile << "cd " << filename+"_dir" << endl;
        runfile << "../psfui -f ./psfui_file > ./psdoubl.dat" << endl;
        runfile << "../inf2zero ./psdoubl.dat" << endl;
        runfile << "../psdoubl13" << endl;
        runfile << "cd .." << endl;
        runfile.close();
        system(("chmod u+x "+filename).c_str());
}