Site.h

Go to the documentation of this file.

/*****************************************
 * This file is documented for Doxygen.
 * If you modify this file please update
 * the comments so that Doxygen will still
 * be able to work.
 ****************************************/
#ifndef SITE_H
#define SITE_H

#include <json.h>
#include <geo.h>
#include <set>
#include <string>
#include <vector>
#include <memory>
#include <utility>
#include <tuple>
#include <mutex>
#include <atomic>
#include "Link.h"
#include "Pick.h"

namespace glasscore {

// forward declarations
class CNode;
class CTrigger;
class CHypo;
class CPickList;

struct SitePickCompare {
    bool operator()(const std::shared_ptr<CPick> &lhs,
                    const std::shared_ptr<CPick> &rhs) const {
        if (lhs->getTSort() < rhs->getTSort()) {
            return (true);
        }
        return (false);
    }
};

class CSite {
 public:
    CSite();

    CSite(std::string sta, std::string comp, std::string net, std::string loc,
            double lat, double lon, double elv, double qual, bool enable,
            bool useTele);

    explicit CSite(std::shared_ptr<json::Object> site);

    ~CSite();

    void clear();

    void update(CSite *site);

    bool initialize(std::string sta, std::string comp, std::string net,
                    std::string loc, double lat, double lon, double elv,
                    double qual, bool enable, bool use, bool useTele);

    void setLocation(double lat, double lon, double z);

    double getDelta(glass3::util::Geo *geo2);

    double getDistance(std::shared_ptr<CSite> site);

    void addPick(std::shared_ptr<CPick> pck);

    void removePick(std::shared_ptr<CPick> pck);

    std::vector<std::shared_ptr<CPick>> getPicks(double t1, double t2);

    void addNode(std::shared_ptr<CNode> node, double distDeg,
                    double travelTime1, std::string phase1,
                    double travelTime2 = -1, std::string phase2 = "");

    void removeNode(std::string nodeID);

    std::vector<std::shared_ptr<CTrigger>> nucleate(double tpick,
        CPickList* parentThread);

    void addTriggerToList(std::vector<std::shared_ptr<CTrigger>> *vTrigger,
                            std::shared_ptr<CTrigger> trigger);

    int getNodeLinksCount() const;

    bool getEnable() const;

    void setEnable(bool enable);

    bool getIsUsed() const;

    bool getUse() const;

    void setUse(bool use);

    bool getUseForTeleseismic() const;

    void setUseForTeleseismic(bool useForTele);

    double getQuality() const;

    void setQuality(double qual);

    double getRawLatitude() const;

    double getRawLongitude() const;

    double getRawElevation() const;

    glass3::util::Geo &getGeo();

    const std::string& getSCNL() const;

    const std::string& getSite() const;

    const std::string& getComponent() const;

    const std::string& getNetwork() const;

    const std::string& getLocation() const;

    time_t getTLastPickAdded() const;

    double * getUnitVectors(double * vec);

    int getPickCountSinceCheck() const;

    void setPickCountSinceCheck(int count);

    int getPickCount() const;

    std::multiset<std::shared_ptr<CPick>, SitePickCompare>::iterator getLower(
            double min);

    std::multiset<std::shared_ptr<CPick>, SitePickCompare>::iterator getUpper(
            double max);

    std::multiset<std::shared_ptr<CPick>, SitePickCompare>::iterator getEnd();

    std::mutex & getPickMutex();

    void updatePosition(std::shared_ptr<CPick> pick);

 private:
    void eraseFromMultiset(std::shared_ptr<CPick> pick);

    mutable std::mutex vPickMutex;

    std::multiset<std::shared_ptr<CPick>, SitePickCompare> m_msPickList;

    std::string m_sSCNL;

    std::string m_sSite;

    std::string m_sComponent;

    std::string m_sNetwork;

    std::string m_sLocation;

    glass3::util::Geo m_Geo;

    std::atomic<double> m_dRawLatitude;

    std::atomic<double> m_dRawLongitude;

    std::atomic<double> m_dRawElevation;

    double m_daUnitVectors[3];

    std::atomic<bool> m_bEnable;

    std::atomic<bool> m_bUse;

    std::atomic<bool> m_bUseForTeleseismic;

    std::atomic<double> m_dQuality;

    std::atomic<int> m_iPickCountSinceCheck;

    mutable std::mutex m_vNodeMutex;

    std::vector<NodeLink> m_vNode;

    mutable std::recursive_mutex m_SiteMutex;

    std::atomic<double> m_tLastPickAdded;

    std::shared_ptr<CPick> m_LowerValue;

    std::shared_ptr<CPick> m_UpperValue;

    // constants
    static const int k_iUnitVectorXCoordinateIndex = 0;

    static const int k_iUnitVectorYCoordinateIndex = 1;

    static const int k_iUnitVectorZCoordinateIndex = 2;
};
}  // namespace glasscore
#endif  // SITE_H