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.
79 lines
2.4 KiB
C++
79 lines
2.4 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 {
|
|
namespace datasetwr {
|
|
|
|
enum { kDataSet, kIDPrefix, kIDNumber, kBuffer };
|
|
|
|
constexpr std::initializer_list<index> idNumberDefaults{0, 0};
|
|
|
|
constexpr auto DataSetWrParams = defineParameters(
|
|
DataSetClientRef::makeParam("dataSet", "DataSet ID"),
|
|
StringParam("idPrefix", "ID Prefix"),
|
|
LongArrayParam("idNumber", "ID Counter Offset", idNumberDefaults),
|
|
BufferParam("buf", "Data Buffer"));
|
|
|
|
class DataSetWriterClient : public FluidBaseClient, OfflineIn, OfflineOut {
|
|
|
|
public:
|
|
using ParamDescType = decltype(DataSetWrParams);
|
|
|
|
using ParamSetViewType = ParameterSetView<ParamDescType>;
|
|
std::reference_wrapper<ParamSetViewType> mParams;
|
|
|
|
void setParams(ParamSetViewType &p) { mParams = p; }
|
|
|
|
template <size_t N> auto &get() const {
|
|
return mParams.get().template get<N>();
|
|
}
|
|
|
|
static constexpr auto &getParameterDescriptors() { return DataSetWrParams; }
|
|
|
|
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"};
|
|
}
|
|
};
|
|
} // namespace datasetwr
|
|
|
|
using NRTThreadedDataSetWriter =
|
|
NRTThreadingAdaptor<ClientWrapper<datasetwr::DataSetWriterClient>>;
|
|
} // namespace client
|
|
} // namespace fluid
|