#pragma once #include "Layer.h" class AddEdgeLayer : public Layer { private: int mode; intArray coolWarm(int xo, int yo, int w, int h); intArray heatIce(int xo, int yo, int w, int h); intArray introduceSpecial(int xo, int yo, int w, int h); public: AddEdgeLayer(int32_t seed, int64_t seedMixup, std::shared_ptr parent,int mode); virtual ~AddEdgeLayer() {} intArray getArea(int xo, int yo, int w, int h) override; };