120 lines
2.8 KiB
C++
120 lines
2.8 KiB
C++
|
|
#include "containers/LedStripRGBW.h"
|
|
|
|
#include "effects/AlexaSwipe.h"
|
|
#include "effects/ReverseSwipe.h"
|
|
#include "helpers/ColorConversions.h"
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
#include <fstream>
|
|
|
|
template <typename T>
|
|
std::ostream &operator<<(std::ostream &os, const std::vector<T> &vec)
|
|
{
|
|
os << "[";
|
|
for (const auto &e : vec)
|
|
{
|
|
if (std::is_same<uint8_t, T>::value)
|
|
os << int(e) << ",";
|
|
else
|
|
os << e << ",";
|
|
}
|
|
os << "]";
|
|
return os;
|
|
}
|
|
|
|
template <typename T>
|
|
std::ostream &operator<<(std::ostream &os, const std::vector<std::vector<T>> &vec)
|
|
{
|
|
os << "[";
|
|
for (const auto &e : vec)
|
|
os << e << ",";
|
|
os << "]";
|
|
return os;
|
|
}
|
|
|
|
template <typename TEffect, int NLeds>
|
|
void effectToFile(const std::string &filename, TEffect &effect, LedStripRGBW<NLeds> &strip, int calls = 100)
|
|
{
|
|
std::vector<std::vector<uint8_t>> vr(calls);
|
|
std::vector<std::vector<uint8_t>> vg(calls);
|
|
std::vector<std::vector<uint8_t>> vb(calls);
|
|
|
|
std::vector<std::vector<float>> vh(calls);
|
|
std::vector<std::vector<float>> vs(calls);
|
|
std::vector<std::vector<float>> vv(calls);
|
|
|
|
for (int time = 0; time < calls; ++time)
|
|
{
|
|
effect();
|
|
vr[time].resize(NLeds);
|
|
vg[time].resize(NLeds);
|
|
vb[time].resize(NLeds);
|
|
|
|
vh[time].resize(NLeds);
|
|
vs[time].resize(NLeds);
|
|
vv[time].resize(NLeds);
|
|
|
|
for (int i = 0; i < NLeds; ++i)
|
|
{
|
|
uint8_t r, g, b, w;
|
|
strip.getRGBW(i, r, g, b, w);
|
|
vr[time][i] = r;
|
|
vg[time][i] = g;
|
|
vb[time][i] = b;
|
|
|
|
auto hsv = rgb2hsv(ColorRGBW{r, g, b, w});
|
|
vh[time][i] = hsv.h;
|
|
vs[time][i] = hsv.s;
|
|
vv[time][i] = hsv.v;
|
|
}
|
|
}
|
|
|
|
std::ofstream fs(filename.c_str());
|
|
fs << "r = " << vr << "\n";
|
|
fs << "g = " << vg << "\n";
|
|
fs << "b = " << vb << "\n";
|
|
fs << "h = " << vh << "\n";
|
|
fs << "s = " << vs << "\n";
|
|
fs << "v = " << vv << "\n";
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
{
|
|
auto cfg = EffectAlexaSwipeConfig{20.f, 20.f, 90.f, 5.f, 180, true, ColorRGBW{255, 0, 0, 0}, ColorRGBW{255, 0, 0, 0}};
|
|
LedStripRGBW<51> strip;
|
|
|
|
EffectAlexaSwipe<decltype(strip)> effect(cfg, strip);
|
|
effectToFile("swipe.py", effect, strip, 200);
|
|
}
|
|
|
|
{
|
|
auto cfg = EffectReverseSwipeConfig{360.f, 5.f, 180};
|
|
LedStripRGBW<51> strip;
|
|
for (int i = 0; i < strip.numLeds(); ++i)
|
|
setLedRGBW(strip, i, ColorRGBW{255, 255, 255, 0});
|
|
|
|
EffectReverseSwipe<decltype(strip)> effect(cfg, strip);
|
|
effectToFile("reverse_swipe.py", effect, strip, 200);
|
|
}
|
|
|
|
/*
|
|
effect.currentPosition_ = 150;
|
|
|
|
const auto numLeds = strip.numLeds() / 2;
|
|
|
|
std::vector<float> brightness(numLeds, 0);
|
|
std::vector<float> interpolation(numLeds, 0);
|
|
|
|
for (int i = 0; i < numLeds; ++i)
|
|
effect.getParams(float(i), interpolation[i], brightness[i]);
|
|
|
|
printVec(brightness);
|
|
printVec(interpolation);
|
|
|
|
effect();
|
|
*/
|
|
return 0;
|
|
} |