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.
62 lines
2.1 KiB
C++
62 lines
2.1 KiB
C++
#pragma once
|
|
|
|
#include <algorithms/KDTree.hpp>
|
|
#include <clients/common/FluidBaseClient.hpp>
|
|
#include <clients/common/FluidNRTClientWrapper.hpp>
|
|
#include <clients/common/MessageSet.hpp>
|
|
#include <clients/common/OfflineClient.hpp>
|
|
#include <clients/common/ParameterSet.hpp>
|
|
#include <clients/common/ParameterTypes.hpp>
|
|
#include <clients/common/Result.hpp>
|
|
#include <clients/common/SharedClientUtils.hpp>
|
|
#include <clients/nrt/CommonResults.hpp>
|
|
#include <clients/nrt/DataSetClient.hpp>
|
|
#include <data/FluidDataSet.hpp>
|
|
#include <data/FluidIndex.hpp>
|
|
#include <data/FluidTensor.hpp>
|
|
#include <data/TensorTypes.hpp>
|
|
#include <string>
|
|
|
|
namespace fluid {
|
|
namespace client {
|
|
|
|
class DataSetWriterClient : public FluidBaseClient, OfflineIn, OfflineOut {
|
|
enum { kDataSet, kIDPrefix, kIDNumber, kBuffer };
|
|
static constexpr std::initializer_list<index> idNumberDefaults{0, 0};
|
|
|
|
public:
|
|
FLUID_DECLARE_PARAMS(DataSetClientRef::makeParam("dataSet", "DataSet ID"),
|
|
StringParam("idPrefix", "ID Prefix"),
|
|
LongArrayParam("idNumber", "ID Counter Offset",
|
|
idNumberDefaults),
|
|
BufferParam("buf", "Data Buffer"));
|
|
|
|
DataSetWriterClient(ParamSetViewType &p) : mParams(p) {}
|
|
|
|
template <typename T> Result process(FluidContext &) {
|
|
auto dataset = get<kDataSet>().get();
|
|
if (auto datasetPtr = dataset.lock()) {
|
|
std::string &idPrefix = get<kIDPrefix>();
|
|
auto &idNumberArr = get<kIDNumber>();
|
|
if (idNumberArr.size() != 2)
|
|
return {Result::Status::kError, "ID number malformed"};
|
|
if (idPrefix.size() == 0 && idNumberArr[0] == 0)
|
|
return {Result::Status::kError, "No ID supplied"};
|
|
|
|
std::string id = idPrefix;
|
|
|
|
if (idNumberArr[0] > 0)
|
|
id += std::to_string(idNumberArr[1]);
|
|
|
|
auto buf = get<kBuffer>();
|
|
return datasetPtr->setPoint(id, buf);
|
|
} else
|
|
return {Result::Status::kError, "No DataSet"};
|
|
}
|
|
};
|
|
|
|
using NRTThreadedDataSetWriter =
|
|
NRTThreadingAdaptor<ClientWrapper<DataSetWriterClient>>;
|
|
} // namespace client
|
|
} // namespace fluid
|