ResPkModel.cpp

Go to the documentation of this file.

#include "ResPkModel.h"
#include "FatalException.h"
#include <fstream>
#include <iomanip>
#include <iostream>
#include <algorithm>
#include <boost/cast.hpp>
using namespace std;

namespace gplib
  {
    ResPkModel::ResPkModel() :
        slowness(-1.0), InputWave(PWave)
      {
      }
    ResPkModel::~ResPkModel()
      {
      }

    ResPkModel::ResPkModel(const int nlayers) :
        slowness(-1.0), InputWave(PWave), SeismicModel(nlayers)
      {

      }

    ResPkModel::ResPkModel(const SeismicModel& source) :
        slowness(-1.0), InputWave(PWave)
      {
        this->SeismicModel::operator=(source);
      }

    ResPkModel::ResPkModel(const ResPkModel& source)
      {
        this->SeismicModel::operator=(source);
        this->slowness = source.slowness;
        this->InputWave = source.InputWave;
        copy(source.Strike.begin(), source.Strike.end(), back_inserter(this->Strike));
        copy(source.Dip.begin(), source.Dip.end(), back_inserter(this->Dip));
      }

    ResPkModel& ResPkModel::operator=(const ResPkModel& source)
      {
        if (this == &source)
          return *this;
        this->SeismicModel::operator=(source);
        this->slowness = source.slowness;
        this->InputWave = source.InputWave;
        Strike.assign(source.Strike.size(), 0);
        Dip.assign(source.Dip.size(), 0);
        copy(source.Strike.begin(), source.Strike.end(), Strike.begin());
        copy(source.Dip.begin(), source.Dip.end(), Dip.begin());
        return *this;
      }

    void ResPkModel::WriteModel(const std::string filename)
      {
        ofstream neu; //the file to write the model to

        neu.open(filename.c_str()); //open file for writing
        const unsigned int nlayers = GetPVelocity().size();
        if (Dip.size() != nlayers)
          {
            Dip.assign(nlayers, 0.0);
          }
        if (Strike.size() != nlayers)
          {
            Strike.assign(nlayers, 0.0);
          }
        neu << setw(3) << nlayers << " " << filename << endl;

        for (unsigned int i = 0; i < nlayers; ++i) //write all layers
          {
            neu.setf(ios::fixed);
            neu << setw(3);
            //we write out everything with a precision of 4 digits to make
            //sure that we represent the values well
            //only thickness only has a precision of 2 to allow value > 100
            //also 10m precision is sufficient for layer thickness
            //respktn is picky about formatting
            neu << i + 1 << setw(8) << setprecision(4) << GetPVelocity().at(i);
            neu << setw(8) << setprecision(4) << GetSVelocity().at(i) << setw(8)
                << setprecision(4) << GetDensity().at(i);
            neu << setw(8) << setprecision(2) << GetThickness().at(i) << setw(8)
                << setprecision(4) << GetQp().at(i);
            neu << setw(8) << setprecision(4) << GetQs().at(i);
            neu << setw(8) << setprecision(4) << GetStrike().at(i) << setw(8)
                << setprecision(4) << GetDip().at(i) << "  0.2500" << endl;
          }
        neu.close(); //close file
      }

    void ResPkModel::ReadModel(const std::string filename)
      {
        ifstream infile(filename.c_str());
        char dummy[255];
        double number;
        int nlayers = -1;
        infile >> nlayers;
        infile.getline(dummy, 255);
        if (nlayers < 1)
          throw FatalException("Problem reading file: " + filename);
        Init(nlayers);
        int i = 0;
        while (infile.good())
          {
            //we throw away the layer number
            infile >> number;
            //if the layer number read succeeded we read in the rest
            if (infile.good())
              {
                infile >> SetPVelocity().at(i) >> SetSVelocity().at(i)
                    >> SetDensity().at(i) >> SetThickness().at(i) >> SetQp().at(i)
                    >> SetQs().at(i);
                infile >> SetStrike().at(i) >> SetDip().at(i) >> number;
                ++i;
              }
          }
        if (i != nlayers)
          throw FatalException("Problem reading file: " + filename);
      }

    void ResPkModel::WriteRunFile(const std::string &filename)
      {
        const unsigned int mintime = 200; //the minimum time the seismogram has to have to be correct
        ofstream runfile;
        runfile.open(filename.c_str());
        runfile << "#!/bin/bash" << endl;
        runfile << "respknt 2>&1> /dev/null << eof" << endl;
        runfile << filename + ".mod" << endl;
        runfile << "n" << endl;
        if (InputWave == PWave)
          {
            runfile << "1" << endl;
          }
        else
          {
            runfile << "2" << endl;
          }
        runfile << GetDt() << endl;
        runfile
            << max(boost::numeric_cast<unsigned int>((GetNpts() - 1) * GetDt()), mintime)
            << endl;
        runfile << GetSlowness() << endl;
        runfile << "f" << endl;
        runfile << "y" << endl;
        runfile << "eof" << endl;
        runfile.close();
      }
  }