LensFlare.h

Go to the documentation of this file.

 
#ifndef LENS_FLARE_H
#define LENS_FLARE_H
 
#include "Context.h"
#include <complex>
#include <map>
#include <string>
#include <vector>
 
// Forward declaration - full definition is in RadiationModel.h
// LensFlareProperties must be defined before this header is included
struct LensFlareProperties;
 
 
class LensFlare {
public:
 
    LensFlare(const LensFlareProperties &lens_flare_props, helios::int2 resolution);
 
 
    void apply(std::map<std::string, std::vector<float>> &pixel_data, helios::int2 resolution);
 
 
    std::vector<std::tuple<int, int, float>> findBrightPixels(const std::map<std::string, std::vector<float>> &pixel_data, helios::int2 resolution, float threshold) const;
 
private:
 
    void generateStarburstKernel();
 
 
    void generateApertureMask(std::vector<float> &mask) const;
 
 
    void applyStarburst(std::map<std::string, std::vector<float>> &pixel_data, helios::int2 resolution, const std::vector<std::tuple<int, int, float>> &bright_pixels);
 
 
    void applyGhosts(std::map<std::string, std::vector<float>> &pixel_data, helios::int2 resolution, const std::vector<std::tuple<int, int, float>> &bright_pixels);
 
 
    static float fresnelReflectance(float cos_theta, float n1 = 1.0f, float n2 = 1.5f);
 
 
    static void fft2D(const std::vector<float> &input, std::vector<std::complex<float>> &output, int size);
 
 
    static void fft1D(std::vector<std::complex<float>> &data, int size, bool inverse = false);
 
 
    void renderSoftDisc(std::vector<float> &channel, helios::int2 resolution, float center_x, float center_y, float radius, float intensity) const;
 
    LensFlareProperties props_;
 
    helios::int2 resolution_;
 
    std::vector<float> starburst_kernel_;
 
    int kernel_size_ = 128;
 
    static constexpr float ghost_scales_[] = {0.3f, 0.5f, 0.8f, 1.2f, 1.6f};
 
    static constexpr float ghost_sizes_[] = {0.02f, 0.03f, 0.015f, 0.025f, 0.04f};
};
 
#endif // LENS_FLARE_H