RailwayRouteOptimization/scripts/Common/DrawPM.js

/**
 * 读取PM文件格式
 * 曲线类型：
 *     (ZH)直线：分别为起点、中点、终点（ZH点）三个点坐标值，最后一个点为0；
 *     (HY)入缓和曲线：分别为起点（ZH点）、中点、终点（HY点）、圆曲线圆心四个点坐标值；
 *     (YH)圆曲线：分别为起点（HY点）、中点、终点（YH点）、圆曲线圆心四个点坐标值；
 *     (HZ)出缓和曲线：分别为起点（YH点）、中点、终点（HZ点）三个点坐标值，最后一个点为0。
 * 曲线类型编号 => | 直线：1 | 入缓和曲线：21 | 圆曲线：3 | 出缓和曲线：22 |
 */
include("scripts/EAction.js");
include("scripts/simple.js");
include("scripts/Common/DrawingUtils.js")

/** 外部调用入口
 * file 要加载的文件名称 需为PM文件
 * ratio 默认值1
 * colorIndex 线的颜色
 *            0-优化后；1-初始PM文件；2-csv转pm；3-调整；4-表格手动输入
 */
function DrawPM(file, ratio, colorIndex) {
  this.colorIndex = colorIndex;
  // 比例尺 实际尺寸/绘画尺寸
  this.ratio = ratio || 1 / 1;
  // 里程距离（用于计算百里标和公里标）
  this.mileage = 0.0;
  // 图层名称
  this.layername = getLayername("PM", file);
  // PM数据 {曲线类型，开始点坐标，结束点坐标，半径, 曲线的方位, 第一行， 第二行， 第三行}
  this.dataArray = new Array();
  // 交点的信息 [JD_x,JD_y,R,ls]
  this.csvArray = new Array();
  this.file = file;
  this.readFile(this.file);
}

DrawPM.prototype.readFile = function (file) {
  var reader = new QFile(file);
  var flags = makeQIODeviceOpenMode(QIODevice.ReadOnly, QIODevice.Text);
  if (!reader.open(flags)) {
    EAction.handleUserWarning("cannot open file: " + file);
    return;
  }
  EAction.handleUserMessage("Starting to load PM file: " + file);
  var ts = new QTextStream(reader);
  setUtf8Codec(ts);
  var lineIndex = 0;
  var lineOneFields = undefined;
  var lineTwoFields = undefined;
  var lineThrerFields = undefined;
  this.dataArray = new Array();
  do {
    lineIndex = lineIndex + 1;
    var textLine = ts.readLine().trim().replace(/[\t]+/g, " ").replace(/[\s]+/g, " ").replace(/\r\n/g, "");
    var fields = textLine.split(" ");
    if (lineIndex < 4) {
      // 前三行无效
      continue;
    }
    if (fields.length < 4) {
      // 第一行 7列 | 第二行 8列 | 第三列 4列
      continue;
    }
    if (lineIndex % 3 == 1) {
      lineOneFields = fields;
      lineTwoFields = undefined;
      lineThrerFields = undefined;
    } else if (lineIndex % 3 == 2) {
      lineTwoFields = fields;
      lineThrerFields = undefined;
    } else if (lineIndex % 3 == 0) {
      lineThrerFields = fields;
    }
    if (lineOneFields && lineTwoFields && lineThrerFields) {
      var type = parseInt(lineOneFields[6]);
      var start_x = parseFloat(lineTwoFields[0]) / this.ratio;
      var start_y = parseFloat(lineTwoFields[1]) / this.ratio;
      var end_x = parseFloat(lineTwoFields[4]) / this.ratio;
      var end_y = parseFloat(lineTwoFields[5]) / this.ratio;
      // TODO 纠正CSV文件导入的问题 X和Y相反
      if (this.colorIndex == 1) {
        start_x = parseFloat(lineTwoFields[1]) / this.ratio;
        start_y = parseFloat(lineTwoFields[0]) / this.ratio;
        end_x = parseFloat(lineTwoFields[5]) / this.ratio;
        end_y = parseFloat(lineTwoFields[4]) / this.ratio;
      }
      var round_radius = parseFloat(lineOneFields[4]) / this.ratio;
      var reversed = parseInt(lineOneFields[0]);
      // TODO 纠正CSV文件导入的问题 左右曲线相反
      if (this.colorIndex == 1) {
        reversed = reversed == 1 ? -1 : 1;
      }
      // PM数据 {曲线类型，开始点坐标，结束点坐标，半径, 曲线的方位, 第一行， 第二行， 第三行}
      this.dataArray.push([type, [start_x, start_y], [end_x, end_y], round_radius, reversed, lineOneFields, lineTwoFields, lineThrerFields]);
    }
  } while (!ts.atEnd());
};
/** 绘画PM */
DrawPM.prototype.applyDrawing = function () {
  // 添加图层
  addLayerToView(this.layername, "white", "CONTINUOUS", RLineweight.Weight025, 1);
  setCurrentLayer(this.layername);
  // 交点的序号
  var jdIndex = 0;
  // 颜色
  var array = new Array(
    [new RColor(255, 0, 0), new RColor(255, 0, 0), new RColor(160, 32, 240), new RColor(0, 255, 0),],
    [new RColor(128, 0, 128), new RColor(77, 255, 77), new RColor(200, 96, 255), new RColor(160, 32, 240),],
    [new RColor(255, 153, 77), new RColor(255, 153, 77), new RColor(128, 255, 255), new RColor(0, 255, 255),],
    [new RColor(77, 255, 255), new RColor(77, 255, 255), new RColor(255, 255, 128), new RColor(255, 255, 0),],
    [new RColor(148, 121, 253), new RColor(148, 121, 253), new RColor(255, 192, 128), new RColor(255, 165, 0),]
  );
  var colorArray = array[this.colorIndex % array.length];
  for (var i = 0; i < this.dataArray.length; i++) {
    var data = this.dataArray[i];
    var type = data[0];
    var start_x = data[1][0];
    var start_y = data[1][1];
    var end_x = data[2][0];
    var end_y = data[2][1];
    var round_radius = data[3];
    var reversed = data[4];
    switch (type) {
      case 1:  // ZH
        addPolylinetoView([[start_x, start_y], [end_x, end_y]], false, false, 1, colorArray[0]);
        this.mileage_line(1, [start_x, start_y], [end_x, end_y], colorArray[3], true);
        if (i > 3) {
          jdIndex += 1;
          this.applyDrawingIntersection(jdIndex, this.dataArray[i - 4], this.dataArray[i - 3], this.dataArray[i - 2], this.dataArray[i - 1], this.dataArray[i], colorArray[0]);
        }
        break;
      case 21: // HY
      case 3:  // YH
      case 22: // HZ
        var round_center = getRoundCenter(start_x, start_y, end_x, end_y, round_radius, reversed);
        var start_angle = calculateAngle(round_center[0], round_center[1], start_x, start_y);
        var end_angle = calculateAngle(round_center[0], round_center[1], end_x, end_y)
        addArctoView(round_center[0], round_center[1], round_radius, start_angle, end_angle, reversed > 0 ? false : true, 1, type == 3 ? colorArray[1] : colorArray[2]);
        this.mileage_circle(type, round_center, round_radius, start_angle, end_angle, reversed > 0 ? false : true, colorArray[3], true);
        break;
      default:
        EAction.handleUserMessage("Unrecognized curve type number, type: " + type.toString());
    }
  }
  // 居中显示
  var di = getAssignDocumentInterface(1);
  if (!isNull(di)) {
    di.autoZoom();
  }
}
/** 保存PM数据到CSV */
DrawPM.prototype.saveFileCSV = function () {
  if (this.dataArray.length < 2) {
    EAction.handleUserMessage("Not find PM data...");
    return;
  }
  var fname = RSettings.getStringValue("OpenProject/Current", RSettings.getDocumentsLocation()) + "/JD_R_ls.csv";
  var file = new QFile(fname);
  if (file.exists()) {
    EAction.handleUserMessage("csv file exists...");
  }
  file.open(makeQIODeviceOpenMode(QIODevice.WriteOnly, QIODevice.Text));
  var out = new QTextStream(file);
  //写表头
  out.writeString("JD_x,JD_y,R,ls\n");
  //写数据
  out.writeString(this.dataArray[0][1][0] + "," + this.dataArray[0][1][1] + "," + 0 + "," + 0 + "\n");
  for (i = 0; i < this.csvArray.length; i++) {
    var csvItem = this.csvArray[i];
    out.writeString(csvItem[0] + "," + csvItem[1] + "," + csvItem[2] + "," + csvItem[3] + "\n");
  }
  out.writeString(this.dataArray[this.dataArray.length - 1][2][0] + "," + this.dataArray[this.dataArray.length - 1][2][1] + "," + 0 + "," + 0 + "\n");
  out.flush();
  //写完数据需要关闭文件
  file.close();
  EAction.handleUserMessage("save csv file success!");
}
/** 绘画交点JD 每一项都是{曲线类型，开始点坐标，结束点坐标，半径, 曲线的方位, 第一行， 第二行， 第三行} */
DrawPM.prototype.applyDrawingIntersection = function (index, ZH_start, HY, YH, HZ, ZH_end, color) {
  var JD = lineCrossoverPoint(ZH_start[1], ZH_start[2], ZH_end[1], ZH_end[2]);
  addPolylinetoView([JD, ZH_start[2]], false, false, 1, color);
  addPolylinetoView([JD, ZH_end[1]], false, false, 1, color);

  // YH的圆心坐标
  var round_center = getRoundCenter(YH[1][0], YH[1][1], YH[2][0], YH[2][1], YH[3], YH[4]);
  var offset = computeOffset(JD, round_center);
  // JD->round 的偏移量
  var drift = offset.drift;
  // 文字坐标的偏移量
  var vertical_angle = offset.vertical_angle;
  // 圆和线（JD->round）的交点坐标
  var cross_x = round_center[0] - YH[3] * drift[0];
  var cross_y = round_center[1] - YH[3] * drift[1];
  // JD标注
  addSimpleTextToView("JD" + index.toString(), [JD[0] - 20 * drift[0], JD[1] - 20 * drift[1]], 15, vertical_angle + (YH[4] > 0 ? 0 : 180), "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
  var textArray = new Array();
  textArray.push("JD" + index.toString());
  var start_angle = calculateAngle(round_center[0], round_center[1], YH[1][0], YH[1][1]);
  var end_angle = calculateAngle(round_center[0], round_center[1], YH[2][0], YH[2][1])
  var round_angle = Math.abs(start_angle - end_angle);
  var deg = Math.floor(round_angle);
  var min = Math.floor((round_angle - deg) * 60);
  var sec = Math.floor(((round_angle - deg) * 60 - min) * 60);
  textArray.push("az-" + deg.toString() + "°" + min.toString() + "'" + sec.toString() + "''");
  textArray.push("R-" + YH[3].toString());
  textArray.push("l-" + (round(Math.abs(HY[5][2]) * 1000) / 1000).toString());
  textArray.push("T-" + (round(HY[5][3] * 1000) / 1000).toString());
  textArray.push("L-" + (round(calculateArcLength(YH[3], round_angle) * 1000) / 1000).toString());

  // 标注和交叉点的距离
  var distance = 150;
  for (var i = 0; i < textArray.length; i++) {
    var text_x = (distance + 25 * i) * drift[0] + cross_x;
    var text_y = (distance + 25 * i) * drift[1] + cross_y;
    addSimpleTextToView(textArray[i], [text_x, text_y], 15, vertical_angle + (YH[4] > 0 ? 0 : 180), "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
  }
  this.csvArray.push([JD[0], JD[1], YH[3], Math.abs(parseFloat(HY[5][2]))]);
}
DrawPM.prototype.compute_pm_offset = function (origin, target) {
  var origin_x = origin[0];
  var origin_y = origin[1];
  var target_x = target[0];
  var target_y = target[1];
  // 计算向量AB与x轴的夹角
  var angleRad = Math.atan2(origin_y - target_y, origin_x - target_x);
  // 将弧度转换为角度
  var angleDeg = rad2deg(angleRad);
  // 保证角度在0到360度之间
  if (angleDeg < 0) {
    angleDeg += 360.0;
  }
  // target向origin偏移量
  var drift_x = Math.cos(deg2rad(angleDeg));
  var drift_y = Math.sin(deg2rad(angleDeg));
  // JD文本的旋转角度
  var vertical_angle = 0;
  var vertical_drift_x = 0;
  var vertical_drift_y = 0;
  if (0 <= angleDeg && angleDeg < 90) {
    drift_x = -Math.abs(drift_x);
    drift_y = -Math.abs(drift_y);
    vertical_angle = -(90 - (angleDeg % 90));
    vertical_drift_x = Math.abs(Math.cos(deg2rad(vertical_angle)));
    vertical_drift_y = - Math.abs(Math.sin(deg2rad(vertical_angle)));
  } else if (90 <= angleDeg && angleDeg < 180) {
    drift_x = Math.abs(drift_x);
    drift_y = -Math.abs(drift_y);
    vertical_angle = angleDeg % 90;
    vertical_drift_x = Math.abs(Math.cos(deg2rad(vertical_angle)));
    vertical_drift_y = Math.abs(Math.sin(deg2rad(vertical_angle)));
  } else if (180 <= angleDeg && angleDeg < 270) {
    drift_x = Math.abs(drift_x);
    drift_y = Math.abs(drift_y);
    vertical_angle = -(90 - (angleDeg % 90));
    vertical_drift_x = Math.abs(Math.cos(deg2rad(vertical_angle)));
    vertical_drift_y = - Math.abs(Math.sin(deg2rad(vertical_angle)));
  } else {
    drift_x = -Math.abs(drift_x);
    drift_y = Math.abs(drift_y);
    vertical_angle = angleDeg % 90;
    vertical_drift_x = Math.abs(Math.cos(deg2rad(vertical_angle)));
    vertical_drift_y = Math.abs(Math.sin(deg2rad(vertical_angle)));
  }
  return new Array(drift_x, drift_y, angleDeg, vertical_drift_x, vertical_drift_y, vertical_angle);
}

DrawPM.prototype.mileage_line = function (CurCurveType, origin, target, ZXColor, isPrime) {
  var origin_x = origin[0];
  var origin_y = origin[1];
  var target_x = target[0];
  var target_y = target[1];
  var distance = Math.sqrt(Math.pow(target_x - origin_x, 2) + Math.pow(target_y - origin_y, 2));
  if (isPrime) {
    var offset = computeOffset(origin, target);
    var drift = offset["drift"];
    var vertical = offset["vertical"];
    var angle_prime = offset["vertical_angle"];
    addPolylinetoView(new Array([target_x, target_y], [target_x + 150 * vertical[0], target_y + 150 * vertical[1]]), false, false, 1, ZXColor);
    var text_x = target_x + 18 * vertical[0] - 10 * drift[0];
    var text_y = target_y + 18 * vertical[1] - 10 * drift[1];
    addSimpleTextToView("ZH +" + (round((this.mileage + distance) % 1000 * 1000) / 1000).toString(), [text_x, text_y], 15, angle_prime, "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
  }
  var distanceNext = 100 - (this.mileage % 100);
  if (distance < distanceNext) {
    // 不足下一个百米标的位置直接返回
    this.mileage += distance;
    return;
  }
  this.mileage += distanceNext;
  var offset = this.compute_pm_offset(origin, target);
  var offset_x = offset[0];
  var offset_y = offset[1];
  var angle = offset[2];
  var offset_vertical_x = offset[3];
  var offset_vertical_y = offset[4];
  var angle_vertical = offset[5];
  // 下一个百米标的坐标
  var x = origin_x + distanceNext * offset_x;
  var y = origin_y + distanceNext * offset_y;
  var text = this.mileage % 1000 == 0 ? "K " + this.mileage / 1000 : (this.mileage / 100) % 10;
  addPolylinetoView(new Array([x, y], [x + 10 * offset_vertical_x, y + 10 * offset_vertical_y]), false, false, 1, ZXColor);
  addSimpleTextToView(text.toString(), [x + 20 * offset_vertical_x, y + 20 * offset_vertical_y], 15, angle_vertical, "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
  this.mileage_line(CurCurveType, new Array(x, y), target, ZXColor, false);
};

DrawPM.prototype.mileage_circle = function (CurCurveType, center, radius, startAngle, endAngle, reversed, ZXColor, isPrime) {
  // reversed  true:顺时针 false：逆时针
  var center_x = center[0];
  var center_y = center[1];
  var angle_gap = startAngle - endAngle;
  var distance = radius * deg2rad(Math.abs(angle_gap));
  if (reversed && angle_gap < 0) {
    distance = radius * deg2rad(360 - Math.abs(angle_gap));
  }
  if (isPrime) {
    var prime_x = center_x + radius * Math.cos(deg2rad(endAngle));
    var prime_y = center_y + radius * Math.sin(deg2rad(endAngle));
    var angle_prime_text = (endAngle - 90 + 360) % 180 - 90;
    addPolylinetoView(new Array([prime_x, prime_y], [prime_x + 150 * Math.cos(deg2rad(angle_prime_text)), prime_y + 150 * Math.sin(deg2rad(angle_prime_text))]), false, false, 1, ZXColor);
    var prime_text_x = prime_x + 18 * Math.abs(Math.cos(deg2rad(angle_prime_text + 90))) + 10 * Math.cos(deg2rad(angle_prime_text));
    var prime_text_y = prime_y + 18 * Math.abs(Math.sin(deg2rad(angle_prime_text + 90))) + 10 * Math.sin(deg2rad(angle_prime_text));
    if (1 == CurCurveType) {
      addSimpleTextToView("ZH +" + (round((this.mileage + distance) % 1000 * 1000) / 1000).toString(), [prime_text_x, prime_text_y], 15, angle_prime_text, "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
    } else if (21 == CurCurveType) {
      addSimpleTextToView("HY +" + (round((this.mileage + distance) % 1000 * 1000) / 1000).toString(), [prime_text_x, prime_text_y], 15, angle_prime_text, "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
    } else if (3 == CurCurveType) {
      addSimpleTextToView("YH +" + (round((this.mileage + distance) % 1000 * 1000) / 1000).toString(), [prime_text_x, prime_text_y], 15, angle_prime_text, "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
    } else {
      addSimpleTextToView("HZ +" + (round((this.mileage + distance) % 1000 * 1000) / 1000).toString(), [prime_text_x, prime_text_y], 15, angle_prime_text, "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
    }
  }
  var distanceNext = 100 - (this.mileage % 100);
  if (distance < distanceNext) {
    // 不足下一个百米标的位置直接返回
    this.mileage += distance;
    return;
  }
  this.mileage += distanceNext;
  var newAngle = startAngle + (reversed ? -1 : 1) * rad2deg(distanceNext / radius);
  var new_offset_x = Math.abs(Math.cos(deg2rad(newAngle)));
  var new_offset_y = Math.abs(Math.sin(deg2rad(newAngle)));
  // 计算坐标偏移方向
  var tmp_angle = (newAngle + 360) % 360;
  if (tmp_angle <= 90) {
    new_offset_x = new_offset_x;
    new_offset_y = new_offset_y;
  } else if (tmp_angle > 90 && tmp_angle <= 180) {
    new_offset_x = - new_offset_x;
    new_offset_y = new_offset_y;
  } else if (tmp_angle > 180 && tmp_angle <= 270) {
    new_offset_x = - new_offset_x;
    new_offset_y = - new_offset_y;
  } else {
    new_offset_x = new_offset_x;
    new_offset_y = - new_offset_y;
  }
  var new_x = center_x + radius * new_offset_x;
  var new_y = center_y + radius * new_offset_y;
  // 开始画线
  var offset = this.compute_pm_offset(new Array(new_x, new_y), new Array(center_x, center_y));
  var offset_x = offset[0];
  var offset_y = offset[1];
  var text = this.mileage % 1000 == 0 ? "K " + this.mileage / 1000 : (this.mileage / 100) % 10;
  var isReverse = reversed; // tmp_angle > 90 && tmp_angle < 270;
  //   EAction.handleUserMessage("--mileage_circle: "+this.mileage.toString()+", CurCurveType: " + CurCurveType.toString()+ ", text: " + text.toString());
  //   EAction.handleUserMessage("newAngle: " + newAngle.toString()+ ", endAngle: " + endAngle.toString());
  addPolylinetoView(new Array([new_x, new_y], [new_x + (isReverse ? 1 : -1) * 10 * offset_x, new_y + (isReverse ? 1 : -1) * 10 * offset_y]), false, false, 1, ZXColor);
  addSimpleTextToView(text.toString(), [new_x + (isReverse ? 1 : -1) * 20 * offset_x, new_y + (isReverse ? 1 : -1) * 20 * offset_y], 15, isReverse ? tmp_angle - 180 : tmp_angle, "Standard", RS.VAlignTop, RS.HAlignLeft, false, false, 1);
  this.mileage_circle(CurCurveType, center, radius, newAngle, endAngle, reversed, ZXColor, false);
};