tsg-web/src/utils/tools.js

import moment from 'moment';
import centerOfMass from '@turf/center-of-mass';
import bbox from '@turf/bbox';
import turfLength from '@turf/length';
import along from '@turf/along';

const class2type = {};
const { toString } = class2type;
'Boolean Number String Function Array Date RegExp Object Error'.split(' ').forEach((name) => {
  class2type[`[object ${name}]`] = name.toLowerCase();
});

export const fileDownload = (url,params) => {
  let downloadLink = document.createElement("a");
      downloadLink.href = `http://local.gunshiiot.com:18083/gunshiApp/${url}/${params?.fileId}`;
      downloadLink.download = `${params.fileName}`;
      downloadLink.style.display = "none";
      // 将链接添加到页面中
      document.body.appendChild(downloadLink);

      // 模拟点击事件，开始下载
      downloadLink.click();
}
/**
 * 对象类型
 * @param {object} obj
 */
export function objType(obj) {
  if (obj == null) {
    return `${obj}`;
  }
  return typeof obj === 'object' || typeof obj === 'function' ? class2type[toString.call(obj)] || 'object' : typeof obj;
}
/**
 * 是否数组
 * @param {object} obj 需要判断的对象
 * @return {boolean} 返回 true 或 false
 */
export function isArray(obj) {
  return objType(obj) === 'array';
}

export function numberFormat(value) {
  let param = {}
  let k = 10000
  let sizes = ['', '万', '亿', '万亿']
  let i
  if (value < k) {
      param.value = value
      param.unit = ''
  } else {
      i = Math.floor(Math.log(value) / Math.log(k));
      param.value = ((value / Math.pow(k, i))).toFixed(2);
      param.unit = sizes[i];
  }
  console.log(param)
  return param;
}

export const handleData = (array, key) => {
  if (array.length === 0) return
  // 1、对array进行升序排序，按name排序
  const newArr = array.slice().sort((a, b) => {
    return array.findIndex(item => item.name === a.name) - array.findIndex(item => item.name === b.name);
  })
  
  let arr = [...newArr]
  // 1、startItem(默认rowSpan = 1)记录开始计数的对象
  let startItem = arr[0]
  startItem.rowSpan = 1
  // 2、遍历数组，取下一项进行比较，当name相同则startItem的rowSpan+1, 否则设置新的startItem为下一项
  arr.forEach((item, index) => {
    let nextItem = arr[index+1] || {}
    if(item[key] === nextItem[key]){
      startItem.rowSpan++
    }else{
      startItem = nextItem
      startItem.rowSpan = 1
    }
  })
  return arr

}
export const exportFile = (name, res) => {
  // 创建a标签
  let blob = new Blob([res],{ type: 'application/octet-stream' })
  // 创建下载链接
  const downloadUrl = URL.createObjectURL(blob);
  // 创建下载链接的 <a> 元素
  const downloadLink = document.createElement("a");
  downloadLink.href = downloadUrl;
  downloadLink.download = name;
  // 模拟点击下载链接
  downloadLink.click();
  // 清理创建的 URL 对象
  URL.revokeObjectURL(downloadUrl);
 
 }
/**
 * 将简单结构数据转换成树状结构
 * @param {object} setting 设置节点的 idKey\pIdKey\children
 * @param {array} sNodes 数据数组
 * @return {array} 树状解构数据
 */
export function transformToTreeFormat(setting, sNodes) {
  setting = setting || {};
  let i;
  let l;
  const key = setting.idKey || 'id'; // 当前节点的唯一key名
  const parentKey = setting.pIdKey || 'pid'; // 指向的父节点唯一key名
  const childKey = setting.children || 'children'; // 子元素的key名

  if (!key || key === '' || !sNodes) return [];

  if (isArray(sNodes)) {
    const r = [];
    const tmpMap = [];
    for (i = 0, l = sNodes.length; i < l; i += 1) {
      tmpMap[sNodes[i][key]] = sNodes[i];
    }
    for (i = 0, l = sNodes.length; i < l; i += 1) {
      if (tmpMap[sNodes[i][parentKey]] && sNodes[i][key] !== sNodes[i][parentKey]) {
        if (!tmpMap[sNodes[i][parentKey]][childKey]) {
          tmpMap[sNodes[i][parentKey]][childKey] = [];
        }
        tmpMap[sNodes[i][parentKey]][childKey].push(sNodes[i]);
      } else {
        r.push(sNodes[i]);
      }
    }
    return r;
  }

  return [sNodes];
}

/**
 * 获取上级adcd
 * @param {string} adcd 15位
 * @return {string} 上级行政区划 adcd 15 位
 */
export function getAdcdPid(adcd) {
  if (!adcd || adcd.length !== 15) {
    return '';
  }

  if (adcd === '429021000000000') { // 神农架
    return '429021000000000';
  } else if (adcd === '420821450000000') { // 屈家岭
    return '420800000000000'; // 荆门
  } else if (adcd === '420381450000000') { // 武当山
    return '420300000000000'; // 十堰
  } else if (/00000000000$/.test(adcd)) { // 市
    return '420000000000000';
  } else if (/000000000$/.test(adcd)) { // 县\区\县级市
    return `${adcd.substring(0, 4)}00000000000`;
  } else if (/000000$/.test(adcd)) { // 乡\镇
    return `${adcd.substring(0, 6)}000000000`;
  } else if (/000$/.test(adcd)) { // 行政村
    return `${adcd.substring(0, 9)}000000`;
  }

  return `${adcd.substring(0, 12)}000`; // 自然村
}
/**
 * 根据行政区划设定放缩至的级别
 * @param {string} adcd 行政区划
 */
export function getAdcdZoom(adcd) {
  if (!adcd || adcd.length !== 15) {
    return -1;
  }

  if (adcd === '420821450000000') { // 屈家岭
    return 3;
  } else if (adcd === '420381450000000') { // 武当山
    return 3;
  } else if (/00000000000$/.test(adcd)) { // 市
    return 2;
  } else if (/000000000$/.test(adcd)) { // 县\区\县级市
    return 3;
  } else if (/000000$/.test(adcd)) { // 乡\镇
    return 4;
  } else if (/000$/.test(adcd)) { // 行政村
    return 5;
  }

  return 6; // 自然村
}



/**
 * int 属性值排序
 * @param {object} o1
 * @param {object} o2
 * @param {string} key
 */
export function orderByInt(o1, o2, key) {
  const v1 = o1[key] || 0;
  const v2 = o2[key] || 0;

  if (parseInt(v1, 10) >= parseInt(v2, 10)) {
    return 1;
  }

  return -1;
}
/**
 * 检查用户是否登陆
 */
export function checkUL() {
  return true;
  // const token = cookies.get('token');
  // const uid = cookies.get('uid');

  // return token != null && uid != null;
}


export function isEmptyObject(e) {
  if (e === undefined || e === null) {
    return true;
  }
  for (const t in e) {
    return false;
  }

  return true;
}

export function strIsEmpty(v) {
  // return v === undefined || v == null || v === '';
    return v === undefined;
}

export function complexUrlParams(resultObj, key, obj) {
  if (!resultObj) {
    resultObj = {};
  }
  if (typeof obj !== 'object') {
    resultObj[key] = obj;
    return resultObj;
  }

  for (const subkey in obj) {
    complexUrlParams(resultObj, `${key}[${subkey}]`, obj[subkey]);
  }
  return resultObj;
}

export function arr2map(arrData) {
  if (!Array.isArray(arrData)) {
    return null;
  }
  const ret = {};
  for (const str of arrData) {
    ret[str] = true;
  }
  return ret;
}

export function map2arr(mapData) {
  if (!mapData) {
    return null;
  }

  const ret = [];
  for (const key in mapData) {
    if (mapData[key]) {
      ret.push(key);
    }
  }
  return ret;
}

export function isDebug() {
  return window.location.host.indexOf('dev') > 0 || window.location.host.indexOf('localhost') >= 0;
}


export function normalizeSearchTmRange(tm, type) {
  if (!tm || !tm[0] || !tm[1]) {
    return undefined;
  }

  if (tm[0] instanceof Date) {
    tm[0] = moment(tm[0]);
  }
  if (tm[1] instanceof Date) {
    tm[1] = moment(tm[1]);
  }

  if (type === 'd') {
    return [
      tm[0].format('YYYY-MM-DD 00:00:00'),
      tm[1].format('YYYY-MM-DD 00:00:00'),
    ];
  }
  if (type === 'm') {
    return [
      tm[0].format('YYYY-MM-01 00:00:00'),
      tm[1].format('YYYY-MM-01 00:00:00'),
    ];
  }
  if (type === 't') {
    return [
      tm[0].format('YYYY-MM-01 00:00:00'),
      tm[1].format('YYYY-MM-21 00:00:00'),
    ];
  }
  if (type === 'y') {
    return [
      tm[0].format('YYYY-01-01 00:00:00'),
      tm[1].format('YYYY-01-01 00:00:00'),
    ];
  }
  if (type === 'h') {
    return [
      tm[0].format('YYYY-MM-DD HH:00:00'),
      tm[1].format('YYYY-MM-DD HH:00:00'),
    ];
  }

  return [
    tm[0].format('YYYY-MM-DD HH:mm:00'),
    tm[1].format('YYYY-MM-DD HH:mm:00'),
  ];
}

export function normalizeRzMinMax_Dangyang(min, max) {
  let max50cm = Math.ceil(max * 2);
  let min50cm = Math.floor(min * 2);
  let av50cm = Math.ceil(max + min);

  if (av50cm < 3) {
    av50cm = 3;
  }
  if (min50cm < 0) {
    min50cm = 0
  }

  if (max50cm < av50cm + 3) {
    max50cm = av50cm + 3;
  }
  if (min50cm > av50cm - 3) {
    min50cm = av50cm - 3;
  }

  return {
    min: min50cm / 2,
    max: max50cm / 2,
  }
}

export const MomentTypeFromTmType = {
  h: 'hour',
  m: 'month',
  d: 'day',
}

export function parseGeoJSONFeature(data) {
  if (typeof data.lgtd === 'number' && typeof data.lttd === 'number') {
    return {
      type: "Feature",
      properties: data,
      geometry: {
        type: "Point",
        coordinates: [data.lgtd, data.lttd]
      }
    };
  } else if (data.geometry) {
    let geometryObj = data.geometry;
    if (typeof geometryObj === 'string') {
      try {
        geometryObj = JSON.parse(geometryObj);
      } catch (e) {
        geometryObj = null;
      }
    }
    if (geometryObj?.type) {
      return {
        type: "Feature",
        properties: data,
        geometry: geometryObj
      }
    }
  }
  return null;
}

export function parseGeoJSON(data) {
  const ret = {
    type: "FeatureCollection",
    name: "drpreal",
    crs: {
      type: "name",
      properties: {
        name: "urn:ogc:def:crs:OGC:1.3:CRS84"
      }
    },
    features: []
  };

  if (Array.isArray(data)) {
    ret.features = data.map(parseGeoJSONFeature).filter(Boolean);
  }

  return ret;
}

export function wait(tm) {
  return new Promise((resolve) => {
    setTimeout(() => {
      resolve(true);
    }, tm || 0);
  });
}


export function strNumber(val, def) {
  if (typeof val === "number") {
    return val;
  }

  return def;
}

export function KrM100Render(val) {
  if (typeof val !== "number") {
    return '-';
  }

  return (val * 100).toFixed(2);
}


/**
 * 深度优先遍历树
 * 一个递归方法
 * @params tree: 要转换的树结构数据
 * @params list: 保存结果的列表结构数据，初始传list = []
 * @params parentId: 当前遍历节点的父级节点id，初始为null(因为根节点无parentId)
 **/
function toListDF(tree, list) {
    for (let node of tree) { //遍历最上层
        list.push({
            id: node.id,
            key: node.key,
            title: node.title,
            type: node?.type,
            parentId: node?.parentId
        });
        //如果有子结点,再遍历子结点
        if (node.children.length !== 0) {
            toListDF(node.children, list)  //递归
        }
    }
}

/**
 * 树转list
 */
export function treeToList(tree){
    const list = [];  // 结果list
    for(let node of tree){
        if (node?.children?.length !== 0) {  //遍历树的第一层,只有一个根结点
            toListDF(node.children, list);  //遍历子树,并加入到list中.
        }
    }
    return list;
}

/**
 * 将时间戳转换成年月日
 */
export function timeChange(val){
  let time = moment(val).format()

  let d = new Date(time);

  let batchDate = d.getFullYear() + '-' + (d.getMonth() + 1) + '-' + d.getDate() + ' ' + d.getHours() + ':' + d.getMinutes() + ':' + d.getSeconds();

  return batchDate;
}


function childNodesDeep(nodes, arr) {
    if (nodes)
        nodes.forEach((ele) => {
            arr.push(ele.key);
            if (ele.children) {
                childNodesDeep(ele.children, arr);
            }
        });
  }

/**
 * 获取父节点下所有子节点的key
 * @param nodes
 * @param key
 * @param arr
 * @returns {Array}
 */
export function getChildKysByParentKey(nodes, key, arr) {
    for (let el of nodes) {
        if (el.key === key) {
            arr.push(el.key);
            if (el.children) {
                childNodesDeep(el.children, arr);
            }
        } else if (el.children) {
            getChildKysByParentKey(el.children, key, arr);
        }
    }
    return arr;
}

export function geometryCenter(geometry) {
  if (!geometry) {
    return null;
  }

  try {
    if (typeof geometry === 'string') {
      geometry = JSON.parse(geometry);
    }

    // todo
    if (geometry.type === 'MultiLineString') {
      geometry = {
        type: 'LineString',
        coordinates: geometry.coordinates[0]
      }
    } else if (geometry.type === 'MultiPolygon') {
      geometry = {
        type: 'Polygon',
        coordinates: geometry.coordinates[0]
      }
    }

    let ptgeom = null;
    if (geometry.type === 'Polygon') {
      ptgeom = centerOfMass(geometry);
    } else if (geometry.type === 'LineString') {
      const length = turfLength(geometry);
      ptgeom = along(geometry, length * 0.5);
    }
    return ptgeom?.geometry?.coordinates;
  } catch (e) { return null; }

}

export function geometryBound(geometry) {
  if (!geometry) {
    return null;
  }

  try {
    if (typeof geometry === 'string') {
      geometry = JSON.parse(geometry);
    }

    const result = bbox(geometry);
    return result;
  } catch (e) { return null; }
}

export const dealValue = (data = [])=>{
  let linshiObj = {}
  data?.forEach((item)=>{
    linshiObj[item.value]=item.label
  })
 return linshiObj
}

export const ArrToObj = (data = [])=>{
  let linshiObj = {}
  data?.forEach((item)=>{
    linshiObj[item.value]=item.label
  })
 return linshiObj
}

  //前端筛选
export const myFiltrate = (data,params)=>{
    const {bgTm,endTm,code,dataSource,ly} = params
    const myBgTm = moment(bgTm).format('YYYY-MM-DD')+" 00:00:00"
    const myEndTm = moment(endTm).format('YYYY-MM-DD')+" 23:59:59"
    const newData = []
    data?.map((item,index)=>{
      let flag = 0
      if(bgTm){
        if(item?.tm && moment(bgTm).isBefore(item.tm)&&moment(item.tm).isBefore(myEndTm)){
        }else{
          flag = flag+1
        }
      }
      if(code){
        const str = (item.stnm?item.stnm:"")+(item.stcd?item.stcd:"")+(item.wscd?item.wscd:"")
        if(str.indexOf(code)>(-1)){
        }else{
          flag = flag+1
        }
      }
      if(dataSource){
        if(item?.type&&item?.type===dataSource){
        }else{
          flag = flag+1
        }
      }
      if(ly){
        if(item?.wscd&&item?.wscd.indexOf(ly)>0){
        }else{
          flag = flag+1
        }
      }
      if(flag === 0){
          newData.push(item)
      }
    })
    return newData
  }

  export const GetInterval=(min,max,step=10)=>{
    const distance = Math.ceil(max) - Math.floor(min)
    console.log(distance);
    let s = (distance / step);
    console.log(s);
    if (s > 5){
      s = Math.ceil(s / 5)  *5;
      return s;
    }else{
      return Math.ceil(s)
    }
}
  
export const convertQuarterToDate = (quarterStr) => {
  if (!quarterStr) {
      return null
  }
  // 将字符串按"-"分割成数组
  const [year, quarter] = quarterStr.split("-")

  // 判断季度并计算月份和日期
  switch (quarter) {
      case "Q1":
      return {
            startDate:`${year}-01-01`,
            endDate:`${year}-03-31`
          }
      case "Q2":
      return {
            startDate:`${year}-04-01`,
            endDate:`${year}-06-30`
          }
      case "Q3":
      return {
            startDate:`${year}-07-01`,
            endDate:`${year}-09-30`
          }
      case "Q4":
      return {
            startDate:`${year}-10-01`,
            endDate:`${year}-12-31`
          }
      default:
          return null
  }
}

export const getCurrentQuarter = (data) => {
  const currentMonth = data ? moment(data).month() : moment().month(); // 获取当前月份，注意月份是从0开始的
  const currentYear = data ? moment(data).year() : moment().year(); // 获取当前年份
  if (currentMonth >= 0 && currentMonth <= 2) {
    return {
      name: `${currentYear}年第1季度考核`,
      value: 1
    };
  } else if (currentMonth >= 3 && currentMonth <= 5) {
    return {
      name: `${currentYear}年第2季度考核`,
      value: 2
    };
  } else if (currentMonth >= 6 && currentMonth <= 8) {
    return {
      name: `${currentYear}年第3季度考核`,
      value: 3
    };
  } else {
    return {
      name: `${currentYear}年第4季度考核`,
      value: 4
    };
  }
};


export const getQuarterStartEndDates = (quarter, year) => {
  const quarterStartMonth = (quarter - 1) * 3; // 计算季度起始月份
  const quarterStartDate = moment({ year, month: quarterStartMonth }).startOf('month');
  const quarterEndDate = quarterStartDate.clone().endOf('quarter');

  return { startDate: quarterStartDate.format('YYYY-MM-DD'), endDate: quarterEndDate.format('YYYY-MM-DD') };
};