import {rs as polygonPP} from '/shape/polygon.mjs';
import {rs as linePP} from '/shape/line.mjs';
import {rs as basicsP} from '/generators/basics.mjs';
import {rs as addGridMethods} from '/mlib/grid.mjs';
import {rs as addRandomMethods} from '/mlib/boundedRandomGrids.mjs';

let rs = basicsP.instantiate();
addGridMethods(rs);
addRandomMethods(rs);
rs.setName('grid_distortion_field');
let nr = 40;
let dim = 400;
let topParams = {numRows:nr,numCols:nr,width:dim,height:dim,highJiggle:10,highJiggleStep:5};

Object.assign(rs,topParams);

rs.initProtos = function () {
  this.blineP = linePP.instantiate();
  this.blineP.stroke = 'rgb(255,255,255)';
  this.blineP['stroke-width'] = 0.5;
  this.polygonP = polygonPP.instantiate();
  this.polygonP.stroke = 'rgb(255,255,255)';
  this.polygonP['stroke-width'] = 0.5;
  this.polygonP.fill = 'red';
}  

rs.boundaryLineGenerator = function (end0,end1,rvs,cell) {
  let {blineP,lines} = this;
  let line = blineP.instantiate();
  lines.push(line);
  line.setEnds(end0,end1);
  line.show();
  return line;
}

rs.shapeGenerator = function (rvs,cell,cnt) {
  let {shapes,polygonP} = this;
  let corners = this.cellCorners(cell);
  let mcnt = cnt.minus();
  let rCorners = this.displaceArray(corners,mcnt);
  let sCorners = this.scaleArray(rCorners,0.5,0.25);
  let pgon = polygonP.instantiate();
  pgon.corners = sCorners;
  shapes.push(pgon);
  pgon.show();
  pgon.update();
  return pgon;
}


rs.initialize = function () {
  this.initProtos();
  let {numRows,numCols,lowJiggle,highJiggle,lowJiggleStep,highJiggleStep} = this;
  this.addFrame();
  let root2 = Math.sqrt(2);
  let hnr = numRows/2;
  const computeParams =  (i,j) => {
    let di = i - hnr;
    let dj = j - hnr;
    let maxFromCenter = root2*hnr;
    let fromCenter = Math.sqrt(di*di + dj*dj);
    let factor = 1 - fromCenter/maxFromCenter;// 0 at corners, up to 1 at center
    let jiggleMax = factor * highJiggle;
    let jiggleStep = factor * highJiggleStep;
    return {step:jiggleStep,min:-jiggleMax,max:jiggleMax,bias:i};
  }
  this.jiggleParams = computeParams;
  this.generateGrid();
}

export {rs};