chooserecfunc.cpp

Go to the documentation of this file.

#include <grace_np.h>
#include <boost/filesystem/operations.hpp>
#include <boost/filesystem/path.hpp>
#include <boost/filesystem/convenience.hpp>
#include <boost/version.hpp>
#include <iostream>
#include <gsl/gsl_math.h>
#include <string>
#include "SeismicDataComp.h"

namespace fs = boost::filesystem;
using namespace std;
using namespace gplib;

int main(int argc, char* argv[])
  {
    SeismicDataComp RecData;

    string version = "$Id: chooserecfunc.cpp 1837 2009-11-27 14:17:58Z mmoorkamp $";
    cout << endl << endl;
    cout << "Program " << version << endl;
    cout
        << " Display all the receiver functions in a directory and select good and bad ones "
        << endl;
    cout << " Receiver functions are identified by file ending '.rec' " << endl;
    cout
        << " Type 'g' for good or 'b' for bad after receiver function is displayed "
        << endl;
    cout << endl << endl;

    fs::path full_path(fs::initial_path());

    if ( !fs::exists(full_path) )
      {
        std::cout << "\nNot found: " << full_path.native_file_string()
            << std::endl;
        return 1;
      }
    std::string statid;
    cout << "Enter station identifier: ";
    cin >> statid;
    if (GraceOpen(2048) == -1)
      {
        fprintf(stderr, "Can't run Grace. \n");
        exit(EXIT_FAILURE);
      }
    fs::directory_iterator end_iter;
    std::string response;
    for (fs::directory_iterator dir_itr(full_path); dir_itr != end_iter; ++dir_itr)
      {
        //leaf is marked as depreciated, but we are still using 1.35 on ubuntu 10/08
        //change to filename() for later Versions of boost
        if (extension(*dir_itr) == ".rec" && dir_itr->leaf().find(statid) != string::npos)
          {
            std::string currentname(dir_itr->leaf());
            RecData.ReadData(currentname);
            const double dt = RecData.GetDt();
            for (size_t i = 0; i < RecData.GetData().size() && GraceIsOpen(); i++)
              {
                if (finite(RecData.GetData().at(i))) // if we have a valid value we display it

                      {
                        GracePrintf("g0.s0 point %f, %f", i *dt, RecData.GetData().at(i));
                      }
                    else //otherwise we show junk to signal bad data without upsetting xmgrace

                      {
                        GracePrintf("g0.s0 point %f, %f", i *dt, 10000.0);
                      }
                  }
            double ymax = *max_element(RecData.GetData().begin(), RecData.GetData().end());
            double ymin = *min_element(RecData.GetData().begin(), RecData.GetData().end());
            GracePrintf("world xmax %f", RecData.GetData().size()*dt);
            GracePrintf("world ymax %f", ymax);
            GracePrintf("world ymin %f", ymin);
            GracePrintf("xaxis tick major %f", RecData.GetData().size()*dt/5.0);
            GracePrintf("xaxis tick minor %f", RecData.GetData().size()*dt/10.0);
            GracePrintf("yaxis tick major %f", ymax/5);
            GracePrintf("yaxis tick minor %f", ymax/10);
            GraceCommand(("title \" "+ currentname +" \" ").c_str());
            GracePrintf("redraw");
            std::cin >> response;
            if (response == "b")
              fs::rename(currentname, fs::path("bad") / currentname);
            else if (response == "g")
              fs::rename(currentname, fs::path("good") / currentname);
            else
              cout <<"Invalid reply, ignoring file !" << endl;
            GracePrintf("KILL g0.s0");
            //GraceFlush();
            std::cout << "Data size: " << RecData.GetData().size() << std::endl;
          }
        //else
        //      std::cout << "Filename: " << dir_itr->leaf() << " " << extension(*dir_itr) <<  std::endl;

      }
    if (GraceIsOpen())
      GraceClose();
  }