xl-public-utils 1.0.4 → 1.0.5

package/src/exportPLY.mjs

@@ -0,0 +1,285 @@
+import vtkPolyData from "@kitware/vtk.js/Common/DataModel/PolyData";
+import { mat4 } from "gl-matrix";
+var FormatTypes = {
+  ASCII: 'ascii',
+  BINARY: 'binary'
+};
+var TextureCoordinatesName = {
+  UV: ['u', 'v'],
+  TEXTURE_UV: ['texture_u', 'texture_v']
+};
+
+var DEFAULT_VALUES = {
+  format: FormatTypes.ASCII,
+  dataByteOrder: 0,
+  headerComments: [],
+  textureFileName: '',
+  textureCoordinatesName: TextureCoordinatesName.UV,
+  withNormals: false,
+  withUVs: false,
+  withColors: false,
+  withIndices: true
+}
+
+var _writeHeader = function writeHeader(polyData, fileFormat, fileType, headerComments, textureFileName, textureCoordinatesName, vertexCount, faceListLength, withNormals, withUVs, withColors, withIndices) {
+  var isBinary = fileFormat !== FormatTypes.ASCII;
+  var format;
+
+  if (isBinary) {
+    format = fileType ? 'binary_little_endian' : 'binary_big_endian';
+  } else format = 'ascii';
+
+  headerComments.unshift('VTK.js generated PLY File');
+
+  if (textureFileName) {
+    headerComments.push("TextureFile ".concat(textureFileName));
+  }
+
+  var commentElements = headerComments.map(function (comment) {
+    return "comment ".concat(comment);
+  }).join('\n');
+  var header = ['ply', "format ".concat(format, " 1.0"), "".concat(commentElements), "element vertex ".concat(vertexCount), 'property float x', 'property float y', 'property float z']; // normals
+
+  if (withNormals) {
+    header.push('property float nx', 'property float ny', 'property float nz');
+  } // uvs
+
+
+  if (withUVs) {
+    header.push("property float ".concat(textureCoordinatesName[0]), "property float ".concat(textureCoordinatesName[1]));
+  } // colors
+
+
+  if (withColors) {
+    header.push('property uchar red', 'property uchar green', 'property uchar blue');
+  } // faces
+
+
+  if (withIndices) {
+    header.push("element face ".concat(faceListLength), 'property list uchar int vertex_indices');
+  }
+
+  header.push('end_header\n');
+  return header.join('\n');
+};
+
+var binaryWriter = function binaryWriter() {
+  var output;
+  var vOffset;
+  var fOffset;
+  var indexByteCount = 4;
+  var ft;
+  return {
+    init: function init(polyData) {},
+    writeHeader: function writeHeader(polyData, fileFormat, fileType, headerComments, textureFileName, textureCoordinatesName, numPts, numPolys, withNormals, withUVs, withColors, withIndices) {
+      var vertexCount = polyData.getPoints().getNumberOfPoints();
+      ft = fileType; // 1 byte shape descriptor
+      // 3 vertex indices at ${indexByteCount} bytes
+
+      var faceListLength = withIndices ? numPolys * (indexByteCount * 3 + 1) : 0; // 3 position values at 4 bytes
+      // 3 normal values at 4 bytes
+      // 3 color channels with 1 byte
+      // 2 uv values at 4 bytes
+
+      var vertexListLength = vertexCount * (4 * 3 + (withNormals ? 4 * 3 : 0) + (withUVs ? 4 * 2 : 0) + (withColors ? 3 : 0));
+
+      var header = _writeHeader(polyData, fileFormat, fileType, headerComments, textureFileName, textureCoordinatesName, numPts, numPolys, withNormals, withUVs, withColors, withIndices);
+
+      var headerBin = new TextEncoder().encode(header);
+      output = new DataView(new ArrayBuffer(headerBin.length + vertexListLength + faceListLength));
+      new Uint8Array(output.buffer).set(headerBin, 0);
+      vOffset = headerBin.length;
+      fOffset = vOffset + vertexListLength;
+    },
+    writeVertice: function writeVertice(x, y, z, nx, ny, nz, u, v, r, g, b) {
+      // xyz
+      output.setFloat32(vOffset, x, ft);
+      vOffset += 4;
+      output.setFloat32(vOffset, y, ft);
+      vOffset += 4;
+      output.setFloat32(vOffset, z, ft);
+      vOffset += 4; // nxnynz
+
+      if (nx !== null && ny !== null && nz !== null) {
+        output.setFloat32(vOffset, nx, ft);
+        vOffset += 4;
+        output.setFloat32(vOffset, ny, ft);
+        vOffset += 4;
+        output.setFloat32(vOffset, nz, ft);
+        vOffset += 4;
+      } // uv
+
+
+      if (u !== null && v !== null) {
+        output.setFloat32(vOffset, u, ft);
+        vOffset += 4;
+        output.setFloat32(vOffset, v, ft);
+        vOffset += 4;
+      } // rgb
+
+
+      if (r !== null && g !== null && b !== null) {
+        output.setUint8(vOffset, r);
+        vOffset += 1;
+        output.setUint8(vOffset, g);
+        vOffset += 1;
+        output.setUint8(vOffset, b);
+        vOffset += 1;
+      }
+    },
+    writeFace: function writeFace(n, x, y, z) {
+      output.setUint8(fOffset, n);
+      fOffset += 1;
+      output.setUint32(fOffset, x, ft);
+      fOffset += indexByteCount;
+      output.setUint32(fOffset, y, ft);
+      fOffset += indexByteCount;
+      output.setUint32(fOffset, z, ft);
+      fOffset += indexByteCount;
+    },
+    writeFooter: function writeFooter(polyData) {},
+    getOutputData: function getOutputData() {
+      return output;
+    }
+  };
+};
+
+var asciiWriter = function asciiWriter() {
+  var fileContent = '';
+  return {
+    init: function init(polyData) {},
+    writeHeader: function writeHeader(polyData, fileFormat, fileType, headerComments, textureFileName, textureCoordinatesName, numPts, numPolys, withNormals, withUVs, withColors, withIndices) {
+      fileContent += _writeHeader(polyData, fileFormat, fileType, headerComments, textureFileName, textureCoordinatesName, numPts, numPolys, withNormals, withUVs, withColors, withIndices);
+    },
+    writeVertice: function writeVertice(x, y, z, nx, ny, nz, u, v, r, g, b) {
+      fileContent += "".concat(x, " ").concat(y, " ").concat(z);
+
+      if (nx !== null && ny !== null && nz !== null) {
+        fileContent += " ".concat(nx, " ").concat(ny, " ").concat(nz);
+      }
+
+      if (u !== null && v !== null) {
+        fileContent += " ".concat(u, " ").concat(v);
+      }
+
+      if (r !== null && g !== null && b !== null) {
+        fileContent += " ".concat(r, " ").concat(g, " ").concat(b);
+      }
+
+      fileContent += '\n';
+    },
+    writeFace: function writeFace(n, x, y, z) {
+      fileContent += "".concat(n, " ").concat(x, " ").concat(y, " ").concat(z, "\n");
+    },
+    writeFooter: function writeFooter(polyData) {},
+    getOutputData: function getOutputData() {
+      return fileContent;
+    }
+  };
+};
+
+/**
+ * 将polyData转换为ply文件
+ * @param {vtkPolyData} polyData 需要导出的polydata
+ * @param {'ascii' | 'binary'} format 导出文件格式
+ * @param {0 | 1} dataByteOrder=0 字节序,默认为0, 0为大端模式， 1小端模式
+ * @param {string[]} headerComments=[] 文件头描述
+ * @param {string} textureFileName='' header描述文件名 
+ * @param {['u', 'v']} textureCoordinatesName=['u', 'v'] header添加u和v
+ * @param {boolean} withNormals=false 是否需要法向量
+ * @param {boolean} withUVs=false 是否需要UV
+ * @param {boolean} withColors=false 是否需要Colors
+ * @param {boolean} withColors=true 面使用点索引
+ * @returns {string | DataView} 返回文件内容
+ */
+export function writePLY(polyData, format, dataByteOrder = DEFAULT_VALUES.dataByteOrder, headerComments = DEFAULT_VALUES.headerComments, textureFileName = DEFAULT_VALUES.textureFileName, textureCoordinatesName = DEFAULT_VALUES.textureCoordinatesName, withNormals = DEFAULT_VALUES.withNormals, withUVs = DEFAULT_VALUES.withUVs, withColors = DEFAULT_VALUES.withColors, withIndices = DEFAULT_VALUES.withIndices) {
+  var inPts = polyData.getPoints();
+  var polys = polyData.getPolys();
+
+  if (inPts === null || polys === null) {
+    console.error('输入错误！')
+  }
+
+  var writer = null;
+
+  if (format === 'binary') {
+    writer = binaryWriter();
+  } else if (format === 'ascii') {
+    writer = asciiWriter();
+  } else {
+    console.error('Invalid type format');
+  }
+
+  var tCoordsName = textureCoordinatesName;
+
+  if (typeof textureCoordinatesName === 'undefined') {
+    console.log('Invalid TextureCoordinatesName value, fallback to default uv values');
+    tCoordsName = TextureCoordinatesName.UV;
+  }
+
+  writer.init(polyData);
+  var numPts = inPts.getNumberOfPoints();
+  var numPolys = polys.getNumberOfCells(); // textureCoords / uvs
+
+  var textureCoords = polyData.getPointData().getTCoords(); // eslint-disable-next-line no-param-reassign
+
+  // withUVs = !(textureCoords === null); // scalars / colors
+
+  var scalars = polyData.getPointData().getScalars(); // eslint-disable-next-line no-param-reassign
+
+  withColors = !(scalars === null);
+  var fileType = dataByteOrder ? 0 : 1;
+  writer.writeHeader(polyData, format, fileType, headerComments, textureFileName, tCoordsName, numPts, numPolys, withNormals, withUVs, withColors, withIndices);
+  var normals = polyData.getPointData().getNormals(); // points / vertices
+
+  for (var i = 0; i < numPts; i++) {
+    // eslint-disable-next-line prefer-const
+    var p = inPts.getPoint(i); 
+    // divide by 1 to remove trailing zeros
+
+
+    var x = p[0].toFixed(8);
+    var y = p[1].toFixed(8);
+    var z = p[2].toFixed(8); // normals
+
+    var nx = null;
+    var ny = null;
+    var nz = null; // uvs
+
+    var u = null;
+    var v = null; // colors
+
+    var r = null;
+    var g = null;
+    var b = null;
+
+    // if (textureCoords) {
+    //   u = textureCoords.getData()[i * 2];
+    //   v = textureCoords.getData()[i * 2 + 1];
+    // }
+
+    if (scalars) {
+      r = scalars.getData()[i * 3];
+      g = scalars.getData()[i * 3 + 1];
+      b = scalars.getData()[i * 3 + 2];
+    }
+
+    if (normals) {
+      nx = normals.getData()[i * 2];
+      ny = normals.getData()[i * 2 + 1];
+      nz = normals.getData()[i * 2 + 2];
+    }
+
+    writer.writeVertice(x, y, z, nx, ny, nz, u, v, r, g, b);
+  } // polys / indices
+
+
+  var pd = polys.getData();
+
+  for (var _i = 0, l = pd.length; _i < l; _i += 4) {
+    writer.writeFace(pd[_i + 0], pd[_i + 1], pd[_i + 2], pd[_i + 3]);
+  }
+
+  writer.writeFooter(polyData);
+  return writer.getOutputData();
+}

package/src/exportSTL.mjs

@@ -0,0 +1,222 @@
+import vtkPolyData from '@kitware/vtk.js/Common/DataModel/PolyData';
+import { vec3 } from 'gl-matrix';
+
+/**
+ * 离散stl重复的点并格式化输出
+ * @param {vtkPolyData} polydata 网格
+ * @returns 离散化之后的PolyData
+ */
+var getResult = function (polydata) {
+  var points = [...polydata.getPoints().getData()];
+  // 离散化
+  dispersedPoints2Stl(points)
+  var vAr = new Float32Array(points);
+  var iAr = new Uint32Array([...polydata.getPolys().getData()]);
+  const nbTriangles = iAr.length / 4;
+  const { set, sub, cross, normalize, create} = vec3
+  var v1 = create();
+  var v2 = create();
+  var v3 = create();
+  var nAr =  new Float32Array(nbTriangles * 3);
+  for (var i = 0; i < nbTriangles; ++i) {
+    var id = i * 4;
+    var i1 = iAr[id + 1] * 3;
+    var i2 = iAr[id + 2] * 3;
+    var i3 = iAr[id + 3] * 3;
+    set(v1, vAr[i1], vAr[i1 + 1], vAr[i1 + 2]);
+    set(v2, vAr[i2], vAr[i2 + 1], vAr[i2 + 2]);
+    set(v3, vAr[i3], vAr[i3 + 1], vAr[i3 + 2]);
+
+    sub(v2, v2, v1); // v2 = v2 - v1
+    sub(v3, v3, v2); // v3 = v3 - v1
+    cross(v1, v2, v3); // v1 = v2 ^ v3
+    normalize(v1, v1);
+
+    nAr[id] = v1[0];
+    nAr[id + 1] = v1[1];
+    nAr[id + 2] = v1[2];
+  }
+
+  return {
+    nbTriangles,
+    vertices: vAr,
+    triangles: iAr,
+    faceNormals: nAr,
+    nbVertices: vAr.length / 3,
+  };
+};
+
+/**
+ * 获取网格的AsciiSTL文件类容
+ * @param {vtkPolyData} polydata 网格
+ * @returns {string} 网格的asciiSTL文件内容
+ */
+const exportAsciiSTL = function (polydata) {
+  var res = getResult(polydata);
+  var nbTriangles = res.nbTriangles;
+  var vAr = res.vertices;
+  var iAr = res.triangles;
+  var fnAr = res.faceNormals;
+
+  var data = 'solid mesh\n';
+  for (var i = 0; i < nbTriangles; ++i) {
+    var id = i * 3;
+    const fid = i * 4;
+    data += ' facet normal ' + fnAr[id] + ' ' + fnAr[id + 1] + ' ' + fnAr[id + 2] + '\n';
+    data += '  outer loop\n';
+    var iv1 = iAr[fid + 1] * 3;
+    var iv2 = iAr[fid + 2] * 3;
+    var iv3 = iAr[fid + 3] * 3;
+    data += '   vertex ' + vAr[iv1] + ' ' + vAr[iv1 + 1] + ' ' + vAr[iv1 + 2] + '\n';
+    data += '   vertex ' + vAr[iv2] + ' ' + vAr[iv2 + 1] + ' ' + vAr[iv2 + 2] + '\n';
+    data += '   vertex ' + vAr[iv3] + ' ' + vAr[iv3 + 1] + ' ' + vAr[iv3 + 2] + '\n';
+    data += '  endloop\n';
+    data += ' endfacet\n';
+  }
+  data += 'endsolid mesh\n';
+  return new Blob([data]);
+};
+
+/**
+ * 获取网格的BinarySTL文件类容
+ * @param {vtkPolyData} polydata 网格
+ * @param {boolean} colorMagic=false 颜色
+ * @returns {Blob} 网格的BinarySTL文件内容
+ */
+const exportBinarySTL = function (polydata, colorMagic = false) {
+  var res = getResult(polydata);
+  var nbTriangles = res.nbTriangles;
+  var vAr = res.vertices;
+  // var cAr = res.colors;
+  var iAr = res.triangles;
+  // var fnAr = res.faceNormals;
+  var i, k;
+
+  var data = new Uint8Array(84 + nbTriangles * 50);
+  if (colorMagic) {
+    // COLOR=255,255,255,255
+    var hdr = [67, 79, 76, 79, 82, 61, 255, 255, 255, 255];
+    for (i = 0; i < hdr.length; ++i) {
+      data[i] = hdr[i];
+    }
+  }
+
+  (new DataView(data.buffer)).setUint32(80, nbTriangles, true);
+  // var verBuffer = new Uint8Array((vAr as any).buffer);
+  // var norBuffer = new Uint8Array(fnAr.buffer);
+  var offset = 84;
+  var inc = 0;
+
+  // var colorActivate = colorMagic ? 0 : (1 << 15);
+
+  const dataView = new DataView(data.buffer)
+  // var mulc = 31 / 3;
+  for (i = 0; i < nbTriangles; ++i) {
+    k = i * 12;
+    // for (inc = 0; inc < 12; ++inc) {
+    //   offset++
+    //   // data[offset++] = norBuffer[k++];
+    // }
+    offset += 12
+    k = i * 3;
+    const fid = i * 4;
+    var iv1 = iAr[fid + 1] * 3;
+    var iv2 = iAr[fid + 2] * 3;
+    var iv3 = iAr[fid + 3] * 3;
+    var id1 = iv1 * 4;
+    // for (inc = 0; inc < 12; ++inc) {
+      // data[offset++] = verBuffer[id1++];
+    // }
+    // var id2 = iv2 * 4;
+    // for (inc = 0; inc < 12; ++inc) {
+    //   data[offset++] = verBuffer[id2++];
+    // }
+    // var id3 = iv3 * 4;
+    // for (inc = 0; inc < 12; ++inc) {
+    //   data[offset++] = verBuffer[id3++];
+    // }
+    for(inc =0; inc < 3; ++inc){
+      dataView.setFloat32(offset, vAr[iv1 + inc], true);
+      offset = offset + 4;
+    }
+    for(inc =0; inc < 3; ++inc){
+      dataView.setFloat32(offset, vAr[iv2 + inc], true);
+      offset = offset + 4;
+    }
+    for(inc =0; inc < 3; ++inc){
+      dataView.setFloat32(offset, vAr[iv3 + inc], true);
+      offset = offset + 4;
+    }
+    
+    offset++;
+    offset++
+    
+  }
+  return new Blob([data]);
+};
+/**
+ * 对数值进行处理，保留几位小数
+ * @param {number} num 数值
+ * @param {number} fix=4 保留几位小数 
+ * @returns 格式化之后的数值
+ */
+function getFixed4(num, fix = 4) {
+  return +num.toFixed(fix)
+}
+/**
+ * 根据点生成key
+ * @param {number[]} point 三位点
+ * @returns 根据点xyz生成key，取小数位前4位置
+ */
+function getPointKey(point) {
+  return `${getFixed4(point[0])},${getFixed4(point[1])},${getFixed4(point[2])}`;
+}
+/**
+ * 返回噪点随机数
+ * @returns {number} 早点随机数
+ */
+function randNum() {
+  return (
+    (Math.random() > 0.5 ? 1 : -1) * (Math.random() * (1e-4 - 1e-5)) + 1e-5
+  );
+}
+/**
+ * 导出stl模型的时候 需要离散点 防止被认为是同一个点 精度到e-7就够了
+ * @param {number[]} points 网格点 
+ */
+export function dispersedPoints2Stl(points) {
+  let pointMap = new Map();
+  for (let i = 0; i < points.length; i += 3) {
+    let transformPoint = [points[i], points[i + 1], points[i + 2]];
+    const key = getPointKey(transformPoint);
+    if (pointMap.has(key)) {
+      let randPoint = [
+        transformPoint[0] + randNum(),
+        transformPoint[1] + randNum(),
+        transformPoint[2] + randNum(),
+      ];
+      let randKey = getPointKey(randPoint);
+      let attemptTimes = 0;
+      while (pointMap.has(randKey) && attemptTimes < 1000) {
+        randPoint = [
+          transformPoint[0] + randNum(),
+          transformPoint[1] + randNum(),
+          transformPoint[2] + randNum(),
+        ];
+        randKey = getPointKey(randPoint);
+        attemptTimes++;
+      }
+      points[i] = randPoint[0];
+      points[i + 1] = randPoint[1];
+      points[i + 2] = randPoint[2];
+      pointMap.set(randKey, true);
+      // import.meta.env.DEV &&  console.log('重复点', [points[i], points[i + 1], points[i + 2]])
+    } else {
+      pointMap.set(key, true);
+    }
+  }
+}
+export {
+  exportBinarySTL,
+  exportAsciiSTL,
+}

package/src/qrcode.mjs

@@ -2,22 +2,22 @@ import { qrcode, datamatrix} from './bwip-js.mjs';
 
 /**
  * @typedef {Object} QrCodeOptions
- * @property {'qrcode' | 'dataMatrix'} [bcid='qrcode'] - 二维码类型，具体支持请查看文档
- * @property {number} [width=1056] - 最终图片宽度
- * @property {number} [height=342] - 最终图片高度
- * @property {number} [qrcodeW=300] - 二维码宽度
- * @property {number} [qrcodeH=300] - 二维码高度
- * @property {boolean} [addText=true] - 是否在右侧添加文字
- * @property {number} [top=10] - 顶部距离
- * @property {number} [left=10] - 左侧距离
- * @property {number} [lineHeight=1] - 文字行高，换行后下一行文字的y位置计算为fontSize * lineHeight
- * @property {number} [fontSize=60] - 文字大小
- * @property {string} [fontFamily='黑体'] - 字体类型
- * @property {string} [bgColor='#fff'] - 背景颜色
- * @property {number} [textLeftMargin=20] - 文字距离二维码间隔
- * @property {boolean} [fontWeight=700] - 字体加粗
- * @property {string} [textColor='#000'] - 字体颜色
- * @property {string} [textContent] - 右侧文字内容
+ * @property {'qrcode' | 'dataMatrix'} bcid='qrcode' - 二维码类型，具体支持请查看文档
+ * @property {number} width=1056 - 最终图片宽度
+ * @property {number} height=342 - 最终图片高度
+ * @property {number} qrcodeW=300 - 二维码宽度
+ * @property {number} qrcodeH=300 - 二维码高度
+ * @property {boolean} addText=true - 是否在右侧添加文字
+ * @property {number} top=10 - 顶部距离
+ * @property {number} left=10 - 左侧距离
+ * @property {number} lineHeight=1 - 文字行高，换行后下一行文字的y位置计算为fontSize * lineHeight
+ * @property {number} fontSize=60 - 文字大小
+ * @property {string} fontFamily='黑体' - 字体类型
+ * @property {string} bgColor='#fff' - 背景颜色
+ * @property {number} textLeftMargin=20 - 文字距离二维码间隔
+ * @property {boolean} fontWeight=700 - 字体加粗
+ * @property {string} textColor='#000' - 字体颜色
+ * @property {string} textContent - 右侧文字内容
  */
 
 /**

package/src/utils.mjs

@@ -29,4 +29,46 @@ export function getRatio() {
  * Alias for {@link getRatio}
  * @function
  */
-export const detectZoom = getRatio;
+export const detectZoom = getRatio;
+
+/**
+ * 将Base64转换为ArrayBuffer
+ * @param {string} base64String
+ * @returns {ArrayBuffer} base64的ArrayBuffer
+ */
+
+export function base64ToArrayBuffer(base64String) {
+  const binary_string = window.atob(base64String);
+  const len = binary_string.length;
+  const bytes = new Uint8Array(len);
+  for (var i = 0; i < len; i++) {
+    bytes[i] = binary_string.charCodeAt(i);
+  }
+  return bytes.buffer
+}
+
+/**
+ * 将Int8Array转换为Base64格式的字符串
+ * @param {Int8Array} int8 
+ * @returns {string} base64格式的字符串
+ */
+export function int8ArrayToBase64(int8) {
+  const data = new Uint8Array(int8)
+  let base64String = '';
+  for (let i = 0; i < data.length; i++) {
+    base64String += String.fromCharCode(data[i])
+  }
+  return window.btoa(base64String)
+}
+
+
+/**
+ * 将ArrayBuffer转换为Base64格式的字符串
+ * @param {ArrayBuffer} outputBuffer 
+ * @returns {string} base64格式的字符串
+ */
+export function arrayBufferToBase64(outputBuffer) {
+  const outputData = new Int8Array(outputBuffer);
+
+  return int8ArrayToBase64(outputData)
+}