You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
154 lines
6.4 KiB
C++
154 lines
6.4 KiB
C++
/***************************************************************************
|
|
* Copyright (C) gempa GmbH *
|
|
* All rights reserved. *
|
|
* Contact: gempa GmbH (seiscomp-dev@gempa.de) *
|
|
* *
|
|
* GNU Affero General Public License Usage *
|
|
* This file may be used under the terms of the GNU Affero *
|
|
* Public License version 3.0 as published by the Free Software Foundation *
|
|
* and appearing in the file LICENSE included in the packaging of this *
|
|
* file. Please review the following information to ensure the GNU Affero *
|
|
* Public License version 3.0 requirements will be met: *
|
|
* https://www.gnu.org/licenses/agpl-3.0.html. *
|
|
* *
|
|
* Other Usage *
|
|
* Alternatively, this file may be used in accordance with the terms and *
|
|
* conditions contained in a signed written agreement between you and *
|
|
* gempa GmbH. *
|
|
***************************************************************************/
|
|
|
|
|
|
#ifndef SEISCOMP_PROCESSING_STREAMPROCESSOR_APPLICATION_H
|
|
#define SEISCOMP_PROCESSING_STREAMPROCESSOR_APPLICATION_H
|
|
|
|
|
|
#include <seiscomp/client/streamapplication.h>
|
|
#include <seiscomp/datamodel/waveformstreamid.h>
|
|
#include <seiscomp/processing/waveformprocessor.h>
|
|
#include <seiscomp/processing/timewindowprocessor.h>
|
|
#include <seiscomp/processing/streambuffer.h>
|
|
|
|
|
|
namespace Seiscomp {
|
|
namespace Processing {
|
|
|
|
|
|
// >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
|
|
/** \brief Application class for stream processing commandline applications.
|
|
*/
|
|
class SC_SYSTEM_CLIENT_API Application : public Client::StreamApplication {
|
|
// ----------------------------------------------------------------------
|
|
// X'truction
|
|
// ----------------------------------------------------------------------
|
|
public:
|
|
//! C'tor
|
|
Application(int argc, char **argv);
|
|
|
|
//! D'tor
|
|
~Application();
|
|
|
|
|
|
// ----------------------------------------------------------------------
|
|
// Public interface
|
|
// ----------------------------------------------------------------------
|
|
public:
|
|
StreamBuffer& streamBuffer();
|
|
const StreamBuffer& streamBuffer() const;
|
|
|
|
void addProcessor(const std::string& networkCode,
|
|
const std::string& stationCode,
|
|
const std::string& locationCode,
|
|
const std::string& channelCode,
|
|
WaveformProcessor *wp);
|
|
void addProcessor(const DataModel::WaveformStreamID &wfid,
|
|
WaveformProcessor *wp);
|
|
|
|
void addProcessor(const std::string& networkCode,
|
|
const std::string& stationCode,
|
|
const std::string& locationCode,
|
|
const std::string& channelCode,
|
|
TimeWindowProcessor *twp);
|
|
void addProcessor(const DataModel::WaveformStreamID &wfid,
|
|
TimeWindowProcessor *twp);
|
|
|
|
void removeProcessors(const std::string& networkCode,
|
|
const std::string& stationCode,
|
|
const std::string& locationCode,
|
|
const std::string& channelCode);
|
|
void removeProcessors(const DataModel::WaveformStreamID &wfid);
|
|
|
|
void removeProcessor(WaveformProcessor *wp);
|
|
|
|
size_t processorCount() const;
|
|
|
|
|
|
// ----------------------------------------------------------------------
|
|
// Protected methods
|
|
// ----------------------------------------------------------------------
|
|
protected:
|
|
void addObject(const std::string& parentID, DataModel::Object* o) override;
|
|
void removeObject(const std::string& parentID, DataModel::Object* o) override;
|
|
void updateObject(const std::string& parentID, DataModel::Object* o) override;
|
|
|
|
void handleRecord(Record *rec) override;
|
|
|
|
void enableStation(const std::string& code, bool enabled);
|
|
void enableStream(const std::string& code, bool enabled);
|
|
|
|
virtual void handleNewStream(const Record *) {}
|
|
virtual void processorFinished(const Record *, WaveformProcessor *) {}
|
|
|
|
void done() override;
|
|
|
|
|
|
// ----------------------------------------------------------------------
|
|
// Private methods
|
|
// ----------------------------------------------------------------------
|
|
private:
|
|
void registerProcessor(const std::string& networkCode,
|
|
const std::string& stationCode,
|
|
const std::string& locationCode,
|
|
const std::string& channelCode,
|
|
WaveformProcessor *wp);
|
|
void registerProcessor(const DataModel::WaveformStreamID &wfid,
|
|
WaveformProcessor *twp);
|
|
|
|
void registerProcessor(const std::string& networkCode,
|
|
const std::string& stationCode,
|
|
const std::string& locationCode,
|
|
const std::string& channelCode,
|
|
TimeWindowProcessor *twp);
|
|
void registerProcessor(const DataModel::WaveformStreamID &wfid,
|
|
TimeWindowProcessor *twp);
|
|
|
|
|
|
// ----------------------------------------------------------------------
|
|
// Private members
|
|
// ----------------------------------------------------------------------
|
|
private:
|
|
typedef std::multimap<std::string, WaveformProcessorPtr> StationProcessors;
|
|
typedef std::multimap<std::string, WaveformProcessorPtr> ProcessorMap;
|
|
typedef DataModel::WaveformStreamID WID;
|
|
typedef std::pair<WID, WaveformProcessorPtr> WaveformProcessorItem;
|
|
typedef std::pair<WID, TimeWindowProcessorPtr> TimeWindowProcessorItem;
|
|
typedef std::list<WaveformProcessorItem> WaveformProcessorQueue;
|
|
typedef std::list<WaveformProcessorPtr> WaveformProcessorRemovalQueue;
|
|
typedef std::list<TimeWindowProcessorItem> TimeWindowProcessorQueue;
|
|
|
|
ProcessorMap _processors;
|
|
StationProcessors _stationProcessors;
|
|
|
|
StreamBuffer _waveformBuffer;
|
|
|
|
WaveformProcessorQueue _waveformProcessorQueue;
|
|
WaveformProcessorRemovalQueue _waveformProcessorRemovalQueue;
|
|
TimeWindowProcessorQueue _timeWindowProcessorQueue;
|
|
bool _registrationBlocked;
|
|
};
|
|
|
|
|
|
}
|
|
}
|
|
|
|
#endif
|