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.
49 lines
1.4 KiB
C++
49 lines
1.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 {
|
|
|
|
class DataSetWriterClient : public FluidBaseClient, OfflineIn, OfflineOut
|
|
{
|
|
public:
|
|
FLUID_DECLARE_PARAMS(StringParam("label", "Label"),
|
|
BufferParam("buf", "Data Buffer"),
|
|
DataSetClientRef::makeParam("dataSet", "DataSet Name"));
|
|
|
|
DataSetWriterClient(ParamSetViewType& p) : mParams(p) {}
|
|
|
|
template <typename T>
|
|
Result process(FluidContext&)
|
|
{
|
|
auto& idx = get<0>();
|
|
auto buf = get<1>();
|
|
auto dataset = get<2>().get();
|
|
if (auto datasetPtr = dataset.lock())
|
|
return datasetPtr->addPoint(idx, buf);
|
|
else
|
|
return {Result::Status::kError, "No dataset"};
|
|
}
|
|
};
|
|
|
|
using NRTThreadedDataSetWriter =
|
|
NRTThreadingAdaptor<ClientWrapper<DataSetWriterClient>>;
|
|
} // namespace client
|
|
} // namespace fluid
|