transform.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <memory>
#include <utility>
#include <vector>
 
#include "pulsatrix/dataset.hpp"
 
namespace pulsatrix {
 
class Transform {
public:
    virtual ~Transform() = default;
 
    [[nodiscard]] virtual Sample apply(Sample sample) const = 0;
};
 
class Compose : public Transform {
public:
    explicit Compose(std::vector<std::shared_ptr<Transform>> steps) : steps_(std::move(steps)) {}
 
    [[nodiscard]] Sample apply(Sample sample) const override {
        for (const auto& step : steps_) {
            sample = step->apply(std::move(sample));
        }
        return sample;
    }
 
private:
    std::vector<std::shared_ptr<Transform>> steps_;
};
 
class TransformDataset : public Dataset {
public:
    TransformDataset(std::shared_ptr<Dataset> base, std::shared_ptr<Transform> transform)
        : base_(std::move(base)), transform_(std::move(transform)) {}
 
    [[nodiscard]] int64_t size() const override { return base_->size(); }
 
    [[nodiscard]] Sample get(int64_t index) const override { return transform_->apply(base_->get(index)); }
 
private:
    std::shared_ptr<Dataset> base_;
    std::shared_ptr<Transform> transform_;
};
 
}  // namespace pulsatrix