#ifndef RASTER_H

#define RASTER_H

#include <cassert></cassert>

#include "track.h"

class TouchPoint;

typedef std::vector<touchpoint> TouchPointList;</touchpoint>

class Painter {

public:

    virtual ~Painter() { }

    virtual Real get_min_height() const = 0;

    virtual Vector2 get_radius_xy() const = 0;

    virtual Real get_height_xy(const Vector2 &offset) const = 0;

};

class Raster {

private:

    Real *data;

    int width;

    int height;

public:

    explicit Raster(int width = 0, int height = 0, Real value = 0):

        data(), width(), height()

        { resize(width, height, value); }

    ~Raster() { reset(); }

    void fill(Real value) {

        if (data)

            for(Real *i = data, *e = i + width*height; i < e; ++i)

                *i = value;

    }

    void resize(int width, int height, Real value = 0) {

        if (data) delete[] data;

        data = 0;

        this->width = this->height = 0;

        if (width > 0 && height > 0) {

            this->width = width;

            this->height = height;

            this->data = new Real[width*height];

            fill(value);

        }

    }

    void reset() { resize(0, 0); }

    int get_width() const { return width; }

    int get_height() const { return height; }

    const Real* get_data() const { return data; }

    const Real* operator[](int row) const

        { assert(data); return data + row*width; }

    Real* get_data() { return data; }

    Real* operator[](int row)

        { assert(data); return data + row*width; }

    Real get(int x, int y, Real def = 0) const

        { return x >= 0 && y >= 0 && x < width && y < height ? (*this)[y][x] : def; }

    void set(int x, int y, Real value)

        { if (x >= 0 && y >= 0 && x < width && y < height) (*this)[y][x] = value; }

    void touch(const Painter &painter, const Vector2 &position, int index, TouchPointList &points);

};

class TouchPoint {

public:

    int index;

    int row;

    int col;

    Real prev;

    Real next;

    TouchPoint(int index = 0, int row = 0, int col = 0, Real prev = 0, Real next = 0):

        index(index), row(row), col(col), prev(prev), next(next) { }

    void apply_next(Raster &raster) const

        { raster.set(col, row, next); }

    void apply_prev(Raster &raster) const

        { raster.set(col, row, prev); }

};

class FlatRaster {

public:

    Real x0;

    Real x1;

    Real y0;

    Real y1;

    Real min_value;

    Real max_value;

    Raster raster;

    explicit FlatRaster(

        int xpixels = 0,

        int ypixels = 0,

        Real x0 = 0,

        Real x1 = 1,

        Real y0 = 0,

        Real y1 = 1,

        Real min_value = 0,

        Real max_value = 1

    ):

        x0(x0),

        x1(x1),

        y0(y0),

        y1(y1),

        min_value(min_value),

        max_value(max_value),

        raster(xpixels, ypixels, min_value)

    { }

    Vector4 color_by_distance(Real d) const;

    void draw() const;

};

class PolarRaster {

public:

    Real x0;

    Real x1;

    Real min_value;

    Real max_value;

    Raster raster;

    explicit PolarRaster(

        int xpixels = 0,

        int ypixels = 0,

        Real x0 = 0,

        Real x1 = 1,

        Real min_value = 0,

        Real max_value = 1

    ):

        x0(x0),

        x1(x1),

        min_value(min_value),

        max_value(max_value),

        raster(xpixels, ypixels, min_value)

    { }

    Vector4 color_by_distance(Real d) const;

    void draw() const;

};

#endif