CMTStationList.cpp

Go to the documentation of this file.

#include "CMTStationList.h"
#include "../Station Parser/StationLexer.hpp"
#include "../Station Parser/StationParser.hpp"
#include "CFatalException.h"
#include "convert.h"
#include "stringcompare.h"
#include <fstream>
#include <iostream>
#include <utility>
#include <gsl/gsl_math.h>
#include <boost/bind.hpp>
using namespace std;
using namespace antlr;

CMTStationList::CMTStationList()
{}
CMTStationList::~CMTStationList()
{}

void CMTStationList::GetData(const std::string filename)
{
        ifstream infile(filename.c_str());
        

        if (infile)
        {
                StationLexer lexer(infile);
                StationParser parser(lexer);
                parser.stationfile();
                for (unsigned int i = 0; i < parser.Stationnames.size(); ++i)
                {
                        cout << "Working on file " << parser.Stationnames.at(i) << endl;
                        try
                        {
                          CMTStation CurrentStation;
                           CurrentStation.GetData(parser.Stationnames.at(i)); 
                                if (parser.HasLatLong.at(i))
                                {
                                        CurrentStation.SetLongitude(parser.Longitudes.at(i));
                                        CurrentStation.SetLatitude(parser.Latitudes.at(i));
                                }
                                StationData.push_back(CurrentStation);
                        }
                        catch(ANTLRException& e)
                        {
                                cerr << "File " << parser.Stationnames.at(i) << " not found !" << endl;
                        }
                }
        }
        else
        {
                throw CFatalException("File not found: " +filename);
        }
        FindCommon();
}
        
CMTStation& CMTStationList::at(int loc)
{       
        return (StationData.at(loc));
}

void CMTStationList::WriteList(const std::string filename)
{
        ofstream outfile(filename.c_str());
        for (vector<CMTStation>::iterator CurrentStation = StationData.begin(); CurrentStation != StationData.end(); CurrentStation++)
        {
                outfile << CurrentStation->GetName();
                outfile <<  setfill(' ') << setw(15) <<resetiosflags(ios::fixed);       
                outfile << CurrentStation->GetLongitude() << " ";
                outfile <<  setfill(' ') << setw(15) <<resetiosflags(ios::fixed);
                outfile << CurrentStation->GetLatitude();
                outfile << endl;                
        }
}

void CMTStationList::WriteAllData()
{
        for (std::vector<CMTStation>::iterator CurrentStation = StationData.begin(); CurrentStation != StationData.end(); CurrentStation++)
                CurrentStation->WriteBack();
}

void CMTStationList::FindCommon(void)
{
        trealdata masterfrequencies(StationData.begin()->GetFrequencies());
        const double tolerance = 0.1;
        double bestfit;
        bool foundfrequency, iscommon;
        unsigned int k;
        commonfrequencies.clear();
        for (unsigned int i = 0; i < masterfrequencies.size(); ++i)
        {
                iscommon = true;
                for (unsigned int j = 0; j < StationData.size(); ++j)
                {
                        foundfrequency = false;
                        k = 0;
                        while ( (foundfrequency == false) && (k < StationData.at(j).GetFrequencies().size()))
                        {
                                if ( gsl_fcmp(masterfrequencies.at(i), StationData.at(j).GetFrequencies().at(k), tolerance) == 0)
                                        foundfrequency = true;          
                                ++k;
                        }
                        iscommon = (iscommon &&  foundfrequency); // if foundfrequency is false once, iscommon will be false for the rest of the loop
                }
                if (iscommon)
                        commonfrequencies.push_back(masterfrequencies.at(i));
        }
        tindexvector indices(commonfrequencies.size(),-1);
        cfindices.assign(StationData.size(),indices);
        for (unsigned int i = 0; i < commonfrequencies.size(); ++i)
        {
                for (unsigned int j = 0; j < StationData.size(); ++j)
                {
                        bestfit = 1e32;
                        for (k = 0; k < StationData.at(j).GetFrequencies().size(); ++k)
                        {
                                if ( (gsl_fcmp(commonfrequencies.at(i), StationData.at(j).GetFrequencies().at(k), tolerance) == 0) && ( abs(commonfrequencies.at(i) - StationData.at(j).GetFrequencies().at(k)) < bestfit) )
                                {
                                        cfindices.at(j).at(i) = k;
                                        bestfit =  commonfrequencies.at(i) - StationData.at(j).GetFrequencies().at(k);
                                }
                        }
                }
        }
        if (commonfrequencies.empty())
        {
                throw CFatalException("No common frequencies below a tolerance of "+ stringify(tolerance));
        }
}


//here starts the implementation of some free functions that don't belong to the class itself, but are associated with it

//! returns if station a and b have the same name
class HasSameName: public std::binary_function<CMTStation,CMTStation,bool>{
        public:
        bool inline operator()(CMTStation a, CMTStation b)
        {
                // if a does not precede b and b does not preced a, they have the same name , concept of equivalence, Effective STL Item 19, page 84
                return !ciStringCompare(a.GetName(),b.GetName()) && !ciStringCompare(b.GetName(),a.GetName());
        }
};


tStatSyncPair FindCorrespondingSites(tStationList &MasterList, tStationList &SlaveList)
{
        std::vector<std::pair<CMTStation*,CMTStation*> > CorrespondingSites;
        for (size_t i = 0; i < MasterList.size(); ++i)
        {
                tStationList::iterator slaveit = find_if(SlaveList.begin(),SlaveList.end(),
                        boost::bind(HasSameName(),MasterList.at(i),_1)); //find site with same name in SlaveList
                if (slaveit != SlaveList.end())         
                        CorrespondingSites.push_back(std::make_pair(&MasterList.at(i),&*slaveit));
                else
                        std::cerr << "Not found: " << MasterList.at(i).GetName() << std::endl;
        }
        return CorrespondingSites;
}

---