// Plotter models, handling modes, and pen-lift motors (port of

// nextdrawcore/nextdraw_options/models.py).

// Servo PWM limits are in units of 1/12 MHz (~83.3 ns).

export const zMotors = [

  { motorName: 'Standard servo', pin: 1, max: 27831, min: 9855, sweepTime: 200, moveMin: 45, moveSlope: 2.69 },

  { motorName: 'Narrow-band brushless servo', pin: 2, max: 12600, min: 5400, sweepTime: 70, moveMin: 20, moveSlope: 1.28 },

];

// resolution 1: high (2032 steps/in per motor axis), 2: low (1016 steps/in).

// jerk 0 means constant-speed mode.

export const handlers = [

  { name: 'No handler selected' },

  { name: 'Technical drawing', resolution: 1, jerk: 20000, speed: 8.6979, speedUp: 8.6979, tolerance: 0.002 },

  { name: 'Handwriting', resolution: 2, jerk: 90000, speed: 7, speedUp: 12, tolerance: 0.008 },

  { name: 'Sketching', resolution: 2, jerk: 14000, speed: 12, speedUp: 12, tolerance: 0.005 },

  { name: 'Constant speed', resolution: 1, jerk: 0, speed: 3.0, speedUp: 8.6979, tolerance: 0.002 },

];

function plotter(modelName, travelX, travelY, jerkPenUpHi, jerkPenUpLo, jerkDerate, autoHome, zMotor) {

  return { modelName, travelX, travelY, jerkPenUpHi, jerkPenUpLo, jerkDerate, autoHome, zMotor };

}

export const plotters = [

  plotter('No model selected', 0, 0, 0, 0, 1, false, 1),

  plotter('AxiDraw V3 or SE/A4', 11.81, 8.58, 19000, 13000, 1.0, false, 0),

  plotter('AxiDraw V3/A3 or SE/A3', 16.93, 11.69, 17000, 12000, 0.9, false, 0),

  plotter('AxiDraw V3 XLX', 23.42, 8.58, 18000, 13000, 1.0, false, 0),

  plotter('AxiDraw MiniKit', 6.30, 4.00, 6000, 6000, 0.6, false, 0),

  plotter('AxiDraw SE/A1', 34.02, 23.39, 13000, 8000, 0.6, false, 0),

  plotter('AxiDraw SE/A2', 23.39, 17.01, 16000, 10000, 0.7, false, 0),

  plotter('AxiDraw V3/B6', 7.48, 5.51, 14000, 14000, 1.0, false, 0),

  plotter('Bantam Tools NextDraw™ 8511', 11.81, 8.58, 19000, 13000, 1.0, true, 1),

  plotter('Bantam Tools NextDraw™ 1117', 16.93, 11.69, 17000, 12000, 0.9, true, 1),

  plotter('Bantam Tools NextDraw™ 2234', 34.02, 23.39, 13000, 8000, 0.6, true, 1),

];

const OVERRIDE_KEYS = {

  modelName: 'modelName', travelX: 'travelX', travelY: 'travelY', jerkPenUp: 'jerkPenUp',

  autoHome: 'autoHome', resolution: 'resolution', curveTolerance: 'curveTolerance',

  constSpeed: 'constSpeed', jerkPenDown: 'jerkPenDown', speedLimit: 'speedLimit', zMotor: 'zMotor',

};

const SERVO_OVERRIDE_KEYS = ['servoPin', 'servoMax', 'servoMin', 'servoSweepTime', 'servoMoveMin', 'servoMoveSlope'];

/**

 * Apply model-, handling- and servo-specific defaults to nd.params, then any

 * non-null nd.params.overrides (models.apply_model_and_handling).

 */

export function applyModelAndHandling(nd, initialize = false) {

  const p = nd.params;

  if (initialize) {

    p.modelOld = -1;

    p.handlingOld = -1;

    p.zMotorOld = -1;

  }

  const model = nd.options.model;

  if (model === 0) return;

  if (!(model >= 1 && model < plotters.length)) throw new RangeError(`Invalid model: ${model}`);

  if (model !== p.modelOld) {

    p.modelOld = model;

    const m = plotters[model];

    p.modelName = m.modelName;

    p.travelX = m.travelX;

    p.travelY = m.travelY;

    p.jerkDerate = m.jerkDerate;

    p.autoHome = m.autoHome;

    p.zMotor = m.zMotor;

    if (nd.options.penlift === 3) p.zMotor = 1; // Brushless upgrade

    // Handling values depend on the model's jerk derating. (The Python driver

    // keeps the old values until the handling mode changes; we recompute.)

    p.handlingOld = -1;

  }

  const handling = nd.options.handling;

  if (handling !== 0 && handling !== p.handlingOld) {

    if (!(handling >= 1 && handling <= 4)) throw new RangeError(`Invalid handling: ${handling}`);

    p.handlingOld = handling;

    const h = handlers[handling];

    p.resolution = h.resolution;

    p.jerkPenDown = h.jerk * p.jerkDerate;

    p.constSpeed = h.jerk === 0;

    p.curveTolerance = h.tolerance;

    const lim = p.resolution === 2 ? p.speedLimXyLr : p.speedLimXyHr;

    p.speedLimit = Math.min(h.speed, lim);

    p.speedUp = Math.min(h.speedUp, lim);

  }

  p.jerkPenUp = p.resolution === 2 ? plotters[model].jerkPenUpLo : plotters[model].jerkPenUpHi;

  for (const key of Object.keys(OVERRIDE_KEYS)) {

    if (p.overrides[key] != null) p[key] = p.overrides[key];

  }

  if (p.zMotor !== p.zMotorOld) {

    p.zMotorOld = p.zMotor;

    const z = zMotors[p.zMotor];

    p.servoPin = z.pin;

    p.servoMax = z.max;

    p.servoMin = z.min;

    p.servoSweepTime = z.sweepTime;

    p.servoMoveMin = z.moveMin;

    p.servoMoveSlope = z.moveSlope;

  }

  for (const key of SERVO_OVERRIDE_KEYS) {

    if (p.overrides[key] != null) p[key] = p.overrides[key];

  }

}