CompressedStorage.h

Go to the documentation of this file.

#pragma once
 
#include "quantities/Quantity.h"
#include "quantities/Storage.h"
 
NAMESPACE_SPH_BEGIN
 
class CompressedVector {
    StaticArray<float, 3> data;
 
public:
    CompressedVector() = default;
 
    CompressedVector(const Vector& v) {
        data[X] = float(v[X]);
        data[Y] = float(v[Y]);
        data[Z] = float(v[Z]);
    }
 
    explicit operator Vector() const {
        return Vector(Float(data[X]), Float(data[Y]), Float(data[Z]));
    }
};
 
class CompressedStorage : public Noncopyable {
private:
    Array<CompressedVector> positions;
    Array<CompressedVector> velocities;
    Array<float> radii;
 
public:
    CompressedStorage() = default;
 
    CompressedStorage(const Storage& storage) {
        ArrayView<const Vector> r, v, dv;
        tie(r, v, dv) = storage.getAll<Vector>(QuantityId::POSITION);
 
        positions.resize(r.size());
        velocities.resize(r.size());
        radii.resize(r.size());
 
        for (Size i = 0; i < r.size(); ++i) {
            positions[i] = r[i];
            velocities[i] = v[i];
            radii[i] = float(r[i][H]);
        }
    }
 
    explicit operator Storage() const {
        Storage storage;
        Array<Vector> r(positions.size());
        for (Size i = 0; i < r.size(); ++i) {
            r[i] = Vector(positions[i]);
            r[i][H] = Float(radii[i]);
        }
        storage.insert<Vector>(QuantityId::POSITION, OrderEnum::FIRST, std::move(r));
 
        Array<Vector> v(velocities.size());
        for (Size i = 0; i < v.size(); ++i) {
            v[i] = Vector(velocities[i]);
        }
        storage.getDt<Vector>(QuantityId::POSITION) = std::move(v);
 
        return storage;
    }
};
 
NAMESPACE_SPH_END