import Feature from '../../../../src/ol/Feature.js';
import ImageLayer from '../../../../src/ol/layer/Image.js';
import Map from '../../../../src/ol/Map.js';
import Point from '../../../../src/ol/geom/Point.js';
import Projection from '../../../../src/ol/proj/Projection.js';
import RasterSource, {
  Processor,
  newImageData,
} from '../../../../src/ol/source/Raster.js';
import Source from '../../../../src/ol/source/Source.js';
import Static from '../../../../src/ol/source/ImageStatic.js';
import TileSource from '../../../../src/ol/source/Tile.js';
import TileState from '../../../../src/ol/TileState.js';
import VectorImageLayer from '../../../../src/ol/layer/VectorImage.js';
import VectorSource from '../../../../src/ol/source/Vector.js';
import View from '../../../../src/ol/View.js';
import XYZ from '../../../../src/ol/source/XYZ.js';
import {Circle, Fill, Style} from '../../../../src/ol/style.js';

const red =
  'data:image/gif;base64,R0lGODlhAQABAPAAAP8AAP///yH5BAAAAAAALAAAAAA' +
  'BAAEAAAICRAEAOw==';

const green =
  'data:image/gif;base64,R0lGODlhAQABAPAAAAD/AP///yH5BAAAAAAALAAAA' +
  'AABAAEAAAICRAEAOw==';

where('Uint8ClampedArray').describe('ol.source.Raster', function () {
  let map, target, redSource, greenSource, blueSource, raster;

  beforeEach(function () {
    target = document.createElement('div');

    const style = target.style;
    style.position = 'absolute';
    style.left = '-1000px';
    style.top = '-1000px';
    style.width = '2px';
    style.height = '2px';
    document.body.appendChild(target);

    const extent = [-1, -1, 1, 1];

    redSource = new Static({
      url: red,
      imageExtent: extent,
      attributions: ['red raster source'],
    });

    greenSource = new Static({
      url: green,
      imageExtent: extent,
      attributions: ['green raster source'],
    });

    blueSource = new VectorImageLayer({
      source: new VectorSource({
        features: [new Feature(new Point([0, 0]))],
      }),
      style: new Style({
        image: new Circle({
          radius: 3,
          fill: new Fill({color: 'blue'}),
        }),
      }),
    });

    raster = new RasterSource({
      threads: 0,
      sources: [redSource, greenSource, blueSource],
      operation: function (inputs) {
        return inputs[0];
      },
    });

    map = new Map({
      target: target,
      view: new View({
        resolutions: [1],
        projection: new Projection({
          code: 'image',
          units: 'pixels',
          extent: extent,
        }),
      }),
      layers: [
        new ImageLayer({
          source: raster,
        }),
      ],
    });
  });

  afterEach(function () {
    if (map) {
      disposeMap(map);
    }
    map = null;
    raster.dispose();
    greenSource.dispose();
    redSource.dispose();
    blueSource.dispose();
  });

  describe('constructor', function () {
    it('returns a raster source', function () {
      const source = new RasterSource({
        threads: 0,
        sources: [new TileSource({})],
      });
      expect(source).to.be.a(Source);
      expect(source).to.be.a(RasterSource);
    });

    it('defaults to "pixel" operation', function (done) {
      const log = [];

      const source = new RasterSource({
        threads: 0,
        sources: [redSource, greenSource, blueSource],
        operation: function (inputs) {
          log.push(inputs);
          return inputs[0];
        },
      });

      source.once('afteroperations', function () {
        expect(log.length).to.equal(4);
        const inputs = log[0];
        const pixel = inputs[0];
        expect(pixel).to.be.an('array');
        done();
      });

      map.getLayers().item(0).setSource(source);
      const view = map.getView();
      view.setCenter([0, 0]);
      view.setZoom(0);
    });

    it('disposes the processor when disposed', function () {
      const source = new RasterSource({
        threads: 0,
        sources: [redSource, greenSource, blueSource],
        operation: function (inputs) {
          return inputs[0];
        },
      });

      source.dispose();

      expect(source.processor_.disposed).to.be(true);
    });

    it('allows operation type to be set to "image"', function (done) {
      const log = [];

      const source = new RasterSource({
        operationType: 'image',
        threads: 0,
        sources: [redSource, greenSource, blueSource],
        operation: function (inputs) {
          log.push(inputs);
          return inputs[0];
        },
      });

      source.once('afteroperations', function () {
        expect(log.length).to.equal(1);
        const inputs = log[0];
        const imageData = inputs[0];
        expect(imageData.data).to.be.a(Uint8ClampedArray);
        expect(imageData.width).to.be(2);
        expect(imageData.height).to.be(2);
        done();
      });

      map.getLayers().item(0).setSource(source);
      const view = map.getView();
      view.setCenter([0, 0]);
      view.setZoom(0);
    });
  });

  describe('config option `attributions`', function () {
    it('handles empty attributions', function () {
      const blue = new RasterSource({
        operationType: 'image',
        threads: 0,
        sources: [blueSource],
        operation: function (inputs) {
          return inputs[0];
        },
      });
      const blueAttributions = blue.getAttributions();
      expect(blueAttributions()).to.be(null);
    });

    it('shows single attributions', function () {
      const red = new RasterSource({
        operationType: 'image',
        threads: 0,
        sources: [redSource],
        operation: function (inputs) {
          return inputs[0];
        },
      });
      const redAttribtuions = red.getAttributions();

      expect(redAttribtuions()).to.not.be(null);
      expect(typeof redAttribtuions).to.be('function');
      expect(redAttribtuions()).to.eql(['red raster source']);
    });

    it('concatinates multiple attributions', function () {
      const redGreen = new RasterSource({
        operationType: 'image',
        threads: 0,
        sources: [redSource, greenSource],
        operation: function (inputs) {
          return inputs[0];
        },
      });
      const redGreenAttributions = redGreen.getAttributions();

      expect(redGreenAttributions()).to.not.be(null);
      expect(typeof redGreenAttributions).to.be('function');
      expect(redGreenAttributions()).to.eql([
        'red raster source',
        'green raster source',
      ]);
    });
  });

  describe('#setOperation()', function () {
    it('allows operation to be set', function (done) {
      let count = 0;
      raster.setOperation(function (pixels) {
        ++count;
        const redPixel = pixels[0];
        const greenPixel = pixels[1];
        const bluePixel = pixels[2];
        expect(redPixel).to.eql([255, 0, 0, 255]);
        expect(greenPixel).to.eql([0, 255, 0, 255]);
        expect(bluePixel).to.eql([0, 0, 255, 255]);
        return pixels[0];
      });

      const view = map.getView();
      view.setCenter([0, 0]);
      view.setZoom(0);

      raster.once('afteroperations', function (event) {
        expect(count).to.equal(4);
        done();
      });
    });

    it('updates and re-runs the operation', function (done) {
      const view = map.getView();
      view.setCenter([0, 0]);
      view.setZoom(0);

      let count = 0;
      raster.on('afteroperations', function (event) {
        ++count;
        if (count === 1) {
          raster.setOperation(function (inputs) {
            return inputs[0];
          });
        } else {
          done();
        }
      });
    });

    it('disposes the previous processor', function () {
      const previousProcessor = raster.processor_;

      raster.setOperation(function (pixels) {
        return pixels[0];
      });

      expect(previousProcessor.disposed).to.be(true);
      expect(raster.processor_.disposed).to.be(false);
    });
  });

  describe('beforeoperations', function () {
    it('gets called before operations are run', function (done) {
      let count = 0;
      raster.setOperation(function (inputs) {
        ++count;
        return inputs[0];
      });

      raster.once('beforeoperations', function (event) {
        expect(count).to.equal(0);
        expect(!!event).to.be(true);
        expect(event.extent).to.be.an('array');
        expect(event.resolution).to.be.a('number');
        expect(event.data).to.be.an('object');
        done();
      });

      const view = map.getView();
      view.setCenter([0, 0]);
      view.setZoom(0);
    });

    it('allows data to be set for the operation', function (done) {
      raster.setOperation(function (inputs, data) {
        ++data.count;
        return inputs[0];
      });

      raster.on('beforeoperations', function (event) {
        event.data.count = 0;
      });

      raster.once('afteroperations', function (event) {
        expect(event.data.count).to.equal(4);
        done();
      });

      const view = map.getView();
      view.setCenter([0, 0]);
      view.setZoom(0);
    });
  });

  describe('afteroperations', function () {
    it('gets called after operations are run', function (done) {
      let count = 0;
      raster.setOperation(function (inputs) {
        ++count;
        return inputs[0];
      });

      raster.once('afteroperations', function (event) {
        expect(count).to.equal(4);
        expect(!!event).to.be(true);
        expect(event.extent).to.be.an('array');
        expect(event.resolution).to.be.a('number');
        expect(event.data).to.be.an('object');
        done();
      });

      const view = map.getView();
      view.setCenter([0, 0]);
      view.setZoom(0);
    });

    it('receives data set by the operation', function (done) {
      raster.setOperation(function (inputs, data) {
        data.message = 'hello world';
        return inputs[0];
      });

      raster.once('afteroperations', function (event) {
        expect(event.data.message).to.equal('hello world');
        done();
      });

      const view = map.getView();
      view.setCenter([0, 0]);
      view.setZoom(0);
    });
  });

  describe('tile loading', function () {
    let map2;
    afterEach(function () {
      disposeMap(map2);
      map2 = null;
    });

    it('is initiated on the underlying source', function (done) {
      const source = new XYZ({
        url: 'spec/ol/data/osm-{z}-{x}-{y}.png',
      });

      raster = new RasterSource({
        threads: 0,
        sources: [source],
        operation: function (inputs) {
          return inputs[0];
        },
      });

      map2 = new Map({
        target: target,
        view: new View({
          center: [0, 0],
          zoom: 0,
        }),
        layers: [
          new ImageLayer({
            source: raster,
          }),
        ],
      });

      const tileCache = source.tileCache;

      expect(tileCache.getCount()).to.equal(0);

      map2.once('moveend', function () {
        expect(tileCache.getCount()).to.equal(1);
        const state = tileCache.peekLast().getState();
        expect(state === TileState.LOADING || state === TileState.LOADED).to.be(
          true
        );
        done();
      });
    });
  });
});

where('Uint8ClampedArray').describe('Processor', function () {
  const identity = function (inputs) {
    return inputs[0];
  };

  describe('constructor', function () {
    it('creates a new processor', function () {
      const processor = new Processor({
        operation: identity,
      });

      expect(processor).to.be.a(Processor);
    });
  });

  describe('#process()', function () {
    it('calls operation with input pixels', function (done) {
      const processor = new Processor({
        operation: function (inputs, meta) {
          ++meta.count;
          const pixel = inputs[0];
          for (let i = 0, ii = pixel.length; i < ii; ++i) {
            meta.sum += pixel[i];
          }
          return pixel;
        },
      });

      const array = new Uint8ClampedArray([1, 2, 3, 4, 5, 6, 7, 8]);
      const input = newImageData(array, 1, 2);

      processor.process([input], {count: 0, sum: 0}, function (err, output, m) {
        if (err) {
          done(err);
          return;
        }
        expect(m.count).to.equal(2);
        expect(m.sum).to.equal(36);
        done();
      });
    });

    it('calls callback with processed image data', function (done) {
      const processor = new Processor({
        operation: function (inputs) {
          const pixel = inputs[0];
          pixel[0] *= 2;
          pixel[1] *= 2;
          pixel[2] *= 2;
          pixel[3] *= 2;
          return pixel;
        },
      });

      const array = new Uint8ClampedArray([1, 2, 3, 4, 5, 6, 7, 8]);
      const input = newImageData(array, 1, 2);

      processor.process([input], {}, function (err, output, m) {
        if (err) {
          done(err);
          return;
        }
        expect(output).to.be.a(ImageData);
        expect(output.data).to.eql(
          new Uint8ClampedArray([2, 4, 6, 8, 10, 12, 14, 16])
        );
        done();
      });
    });

    it('allows library functions to be called', function (done) {
      const lib = {
        sum: function (a, b) {
          return a + b;
        },
        diff: function (a, b) {
          return a - b;
        },
      };

      const normalizedDiff = function (pixels) {
        const pixel = pixels[0];
        const r = pixel[0];
        const g = pixel[1];
        /* eslint-disable */
        var nd = diff(r, g) / sum(r, g);
        /* eslint-enable */
        const index = Math.round((255 * (nd + 1)) / 2);
        return [index, index, index, pixel[3]];
      };

      const processor = new Processor({
        operation: normalizedDiff,
        lib: lib,
      });

      const array = new Uint8ClampedArray([10, 2, 0, 0, 5, 8, 0, 1]);
      const input = newImageData(array, 1, 2);

      processor.process([input], {}, function (err, output, m) {
        if (err) {
          done(err);
          return;
        }
        expect(output).to.be.a(ImageData);
        const v0 = Math.round((255 * (1 + 8 / 12)) / 2);
        const v1 = Math.round((255 * (1 + -3 / 13)) / 2);
        expect(output.data).to.eql(
          new Uint8ClampedArray([v0, v0, v0, 0, v1, v1, v1, 1])
        );

        done();
      });
    });

    it('calls callbacks for each call', function (done) {
      const processor = new Processor({
        operation: identity,
      });

      let calls = 0;

      function createCallback(index) {
        return function (err, output, meta) {
          if (err) {
            done(err);
            return;
          }
          expect(output).to.be.a(ImageData);
          ++calls;
        };
      }

      for (let i = 0; i < 5; ++i) {
        const input = newImageData(new Uint8ClampedArray([1, 2, 3, 4]), 1, 1);
        processor.process([input], {}, createCallback(i));
      }

      setTimeout(function () {
        expect(calls).to.be(5);
        done();
      }, 1000);
    });

    it('respects max queue length', function (done) {
      const processor = new Processor({
        queue: 1,
        operation: identity,
      });

      const log = [];

      function createCallback(index) {
        return function (err, output, meta) {
          if (err) {
            done(err);
            return;
          }
          log.push(output);
        };
      }

      for (let i = 0; i < 5; ++i) {
        const input = newImageData(new Uint8ClampedArray([1, 2, 3, 4]), 1, 1);
        processor.process([input], {}, createCallback(i));
      }

      setTimeout(function () {
        expect(log).to.have.length(5);
        expect(log[0]).to.be(null);
        expect(log[1]).to.be(null);
        expect(log[2]).to.be(null);
        expect(log[3]).to.be.a(ImageData);
        expect(log[4]).to.be.a(ImageData);
        done();
      }, 1000);
    });

    it('can run on multiple threads', function (done) {
      const processor = new Processor({
        threads: 2,
        operation: identity,
      });

      const input = newImageData(new Uint8ClampedArray([1, 2, 3, 4]), 1, 1);
      processor.process([input], {}, function (err) {
        if (err) {
          done(err);
        }
      });

      processor.dispose();
      setTimeout(done, 20);
    });
  });

  describe('#process() - faux worker', function () {
    let identitySpy;
    beforeEach(function () {
      identitySpy = sinon.spy(identity);
    });

    it('calls operation with input pixels', function (done) {
      const processor = new Processor({
        threads: 0,
        operation: identitySpy,
      });

      const array = new Uint8ClampedArray([1, 2, 3, 4, 5, 6, 7, 8]);
      const input = newImageData(array, 1, 2);

      processor.process([input], {}, function (err, output, m) {
        if (err) {
          done(err);
          return;
        }
        expect(identitySpy.callCount).to.be(2);
        const first = identitySpy.getCall(0);
        expect(first.args).to.have.length(2);
        done();
      });
    });

    it('passes meta object to operations', function (done) {
      const processor = new Processor({
        threads: 0,
        operation: identitySpy,
      });

      const array = new Uint8ClampedArray([1, 2, 3, 4]);
      const input = newImageData(array, 1, 1);
      const meta = {foo: 'bar'};

      processor.process([input], meta, function (err, output, m) {
        if (err) {
          done(err);
          return;
        }
        expect(m).to.eql(meta);
        expect(identitySpy.callCount).to.be(1);
        done();
      });
    });
  });

  describe('#dispose()', function () {
    it('stops callbacks from being called', function (done) {
      const processor = new Processor({
        operation: identity,
      });

      const array = new Uint8ClampedArray([1, 2, 3, 4, 5, 6, 7, 8]);
      const input = newImageData(array, 1, 2);

      processor.process([input], {}, function () {
        done(new Error('Expected abort to stop callback from being called'));
      });

      processor.dispose();
      setTimeout(done, 500);
    });
  });

  describe('#dispose() - faux worker', function () {
    it('stops callbacks from being called', function (done) {
      const processor = new Processor({
        threads: 0,
        operation: identity,
      });

      const array = new Uint8ClampedArray([1, 2, 3, 4, 5, 6, 7, 8]);
      const input = newImageData(array, 1, 2);

      processor.process([input], {}, function () {
        done(new Error('Expected abort to stop callback from being called'));
      });

      processor.dispose();
      setTimeout(done, 20);
    });
  });
});