ControlNetwork/Front/back/WD/WDcompute.py

import sqlite3
import os
import math
import numpy as np
import math
import numpy as np

from PySide6.QtWidgets import QMessageBox


# region 各种方法
def to_utf8(text):
    str1 = text.encode('utf-8')
    return str1


def jzys1(listy, zxzby, listx, zxzbx, sf):
    rlist = []
    listlen = len(listy)
    ll = 0
    while ll < listlen:
        # 第一个的值
        # num1 = round(((listx[ll] - zxzbx) / sf),0)
        # num2 = round(((listy[ll] - zxzby) / sf),0)
        num1 = listx[ll] - zxzbx
        num2 = listy[ll] - zxzby
        rlist.append(num1)
        rlist.append(num2)
        ll = ll + 1
    return rlist


def jzys2(listy, zxzby, listx, zxzbx, sf):
    rlist = []
    listlen = len(listy)
    ll = 0
    while ll < listlen:
        # 第一个的值
        num2 = (-1) * (listx[ll] - zxzbx) / sf
        num1 = (listy[ll] - zxzby) / sf
        rlist.append(num1)
        rlist.append(num2)
        ll = ll + 1
    return rlist


def jzys3(listy, aa, bb):
    rlist = []
    listlen = len(listy)
    ll = 0
    while ll < listlen:
        # 第一个的值
        num1 = aa
        num2 = bb
        rlist.append(num1)
        rlist.append(num2)
        ll = ll + 1
    return rlist


def jzys4(listpasty, zxzbpasty, listpastx, zxzbpastx, listnewy, zxzbnewy, listnewx, zxzbnewx, sf):
    rlist = []
    listlen = len(listnewx)
    ll = 0
    while ll < listlen:
        num1 = (listpastx[ll] - listnewx[ll] - zxzbpastx + zxzbnewx) / sf
        num2 = (listpasty[ll] - listnewy[ll] - zxzbpasty + zxzbnewy) / sf
        rlist.append(num1)
        rlist.append(num2)
        ll = ll + 1
    return rlist


def jzys5(listp, zxzbp, sf):
    rlist = []
    listlen = len(listp)
    ll = 0
    while ll < listlen:
        num1 = (listp[ll] - zxzbp) / sf
        rlist.append(num1)
        ll = ll + 1
    return rlist


def jzys6(listp, num):
    rlist = []
    listlen = len(listp)
    ll = 0
    while ll < listlen:
        num1 = listp[ll] * num
        rlist.append(num1)
        ll = ll + 1
    return rlist


def cjh(listp):
    rp = 0
    listlen = len(listp)
    ll = 0
    while ll < listlen:
        num1 = listp[ll] * listp[ll]
        rp = rp + num1
        ll = ll + 1
    return rp


def cjh2(lista, listb):
    rp = 0
    listlen = len(lista)
    ll = 0
    while ll < listlen:
        num1 = lista[ll] * listb[ll]
        rp = rp + num1
        ll = ll + 1
    return rp


def gsys(listnewp, jxlist1, jylist1, lxlist, lylist, arrz, sf, zxzbnewp, zxzbpastp, ii):
    # 新x+（矩阵[jx1 jy1 lx1 ly1]*矩阵z）*缩放-新重心坐标x+旧重心坐标x
    rlist = []
    listlen = len(listnewp)
    ll = 0
    while ll < listlen:
        arr0 = np.array((jxlist1[ll], jylist1[ll],
                         lxlist[2 * ll + ii], lylist[2 * ll + ii]))
        arr1 = np.matmul(arr0, arrz)
        arr3 = sf * arr1
        newp = listnewp[ll] + arr3 - zxzbnewp + zxzbpastp
        rlist.append(newp)
        ll = ll + 1
    return rlist


def xcys(newlist, pastlist):
    listlen = len(newlist)
    ll = 0
    rlist = []
    while ll < listlen:
        num1 = newlist[ll] - pastlist[ll]
        num2 = num1 * 1000
        rlist.append(num2)
        ll = ll + 1
    return rlist


def takeFirst(elem):
    return elem[0]


# endregion

def insert_into_database1(database, pastname, newname, beforename, excelname, listname1, listpastx1, listpasty1,
                          listcgcs1, listbex, listbey, sxylist, wypd, gsx, gsy, listcgcs):
    # 转换下汉字
    utf_tn = to_utf8(excelname)
    utf_nn = to_utf8(newname)
    utf_pn = to_utf8(pastname)
    utf_bn = to_utf8(beforename)
    utf_bx = to_utf8('变形')
    utf_wd = to_utf8('稳定')
    # 将结果输出到数据库
    db1 = sqlite3.connect(database)
    # 获取游标
    cursor1 = db1.cursor()
    # 先清除已有数据，用来更新
    try:
        sqlstr3 = 'delete from WD_Result_Param WHERE TableName = ?'
        sqlstr4 = 'delete from WD_Result_Point WHERE TableName = ?'
        cursor1.execute(sqlstr3, (utf_tn,))
        cursor1.execute(sqlstr4, (utf_tn,))
    except:
        pass
    cursor1.execute(
        'INSERT INTO WD_Result_Param(TableName,Last_ResultName,New_ResultName,Formula_X1,Formula_X2,Formula_X3,Formula_Y1,Formula_Y2,Formula_Y3) VALUES(?,?,?,?,?,?,?,?,?)',
        (utf_tn, utf_pn, utf_nn, sxylist[0], sxylist[1], sxylist[2], sxylist[3], sxylist[4], sxylist[5],))
    # WD_Result_Point的输入
    kk = -1
    kk1 = -1
    if (-1) in listcgcs:
        kk = 1
    # 记位移点个数
    wyd = 0
    for lname in listname1:
        # 获取对应索引号
        index1 = listname1.index(lname)
        PointName = lname
        First_X = listbex[index1]
        First_Y = listbey[index1]
        Last_X = listpastx1[index1]
        Last_Y = listpasty1[index1]
        Last_Wight = listcgcs1[index1]
        New_Wight = listcgcs[index1]
        # 位移点
        if New_Wight == -1:
            # 获取他原本的坐标和初始改算值进行对比
            for smx in jslist:
                if First_X == smx[0] and First_Y == smx[1]:
                    numa3 = sxylist[0] * jslist1[index1][0]
                    numa4 = sxylist[1] * jslist1[index1][1]
                    numa5 = numa3 + numa4 + sxylist[2]
                    numa6 = sxylist[3] * jslist1[index1][0]
                    numa7 = sxylist[4] * jslist1[index1][1]
                    numa8 = numa6 + numa7 + sxylist[5]
                    Result_X = numa5
                    Result_Y = numa8
                    New_Wight = 1
                    numb1 = (numa5 - First_X) * 1000
                    numb2 = (numa8 - First_Y) * 1000
                    numb3 = numb1 * numb1
                    numb4 = numb2 * numb2
                    numb5 = math.sqrt(numb3 + numb4)
                    New_FirstX = numb1
                    New_FirstY = numb2
                    New_FirstP = numb5
                    # 新-首有个位移判定
                    if numb5 > wypd:
                        NFDis_Ass = utf_bx
                    else:
                        NFDis_Ass = utf_wd

                    numc1 = (numa5 - Last_X) * 1000
                    numc2 = (numa8 - Last_Y) * 1000
                    numc3 = numc1 * numc1
                    numc4 = numc2 * numc2
                    numc5 = math.sqrt(numc3 + numc4)
                    New_LastX = numc1
                    New_LastY = numc2
                    New_LastP = numc5
                    NLDis_Ass = utf_bx
                    kk1 = 1
                    wyd = wyd + 1
                    break
        # 不是位移点
        else:
            # 全部都是稳定点
            if kk == -1:
                num3 = Last_X * Last_Wight
                num4 = num3 + gsx[index1]
                num5 = Last_Wight + 1
                num6 = num4 / num5
                num7 = Last_Y * Last_Wight
                num8 = num7 + gsy[index1]
                num10 = num8 / num5
                Result_X = num6
                Result_Y = num10
                New_Wight = num5
                numd1 = (num6 - First_X) * 1000
                numd2 = (num10 - First_Y) * 1000
                numd3 = numd1 * numd1
                numd4 = numd2 * numd2
                numd5 = math.sqrt(numd3 + numd4)
                New_FirstX = numd1
                New_FirstY = numd2
                New_FirstP = numd5
                # 新-首有个位移判定
                if numd5 > wypd:
                    NFDis_Ass = utf_bx
                else:
                    NFDis_Ass = utf_wd
                nume1 = (num6 - Last_X) * 1000
                nume2 = (num10 - Last_Y) * 1000
                nume3 = nume1 * nume1
                nume4 = nume2 * nume2
                nume5 = math.sqrt(nume3 + nume4)
                New_LastX = nume1
                New_LastY = nume2
                New_LastP = nume5
                NLDis_Ass = utf_wd
            # 有位移点，且已经走过了
            if kk == 1 and kk1 == 1:
                num3 = Last_X * Last_Wight
                num4 = num3 + gsx[index1 - wyd]
                num5 = Last_Wight + 1
                num6 = num4 / num5
                num7 = Last_Y * Last_Wight
                num8 = num7 + gsy[index1 - wyd]
                num10 = num8 / num5
                Result_X = num6
                Result_Y = num10
                New_Wight = num5
                numd1 = (num6 - First_X) * 1000
                numd2 = (num10 - First_Y) * 1000
                numd3 = numd1 * numd1
                numd4 = numd2 * numd2
                numd5 = math.sqrt(numd3 + numd4)
                New_FirstX = numd1
                New_FirstY = numd2
                New_FirstP = numd5
                # 新-首有个位移判定
                if numd5 > wypd:
                    NFDis_Ass = utf_bx
                else:
                    NFDis_Ass = utf_wd
                nume1 = (num6 - Last_X) * 1000
                nume2 = (num10 - Last_Y) * 1000
                nume3 = nume1 * nume1
                nume4 = nume2 * nume2
                nume5 = math.sqrt(nume3 + nume4)
                New_LastX = nume1
                New_LastY = nume2
                New_LastP = nume5
                NLDis_Ass = utf_wd
            # 有位移点，但没有走过
            if kk == 1 and kk1 == -1:
                num3 = Last_X * Last_Wight
                num4 = num3 + gsx[index1]
                num5 = Last_Wight + 1
                num6 = num4 / num5
                num7 = Last_Y * Last_Wight
                num8 = num7 + gsy[index1]
                num10 = num8 / num5
                Result_X = num6
                Result_Y = num10
                New_Wight = num5
                numd1 = (num6 - First_X) * 1000
                numd2 = (num10 - First_Y) * 1000
                numd3 = numd1 * numd1
                numd4 = numd2 * numd2
                numd5 = math.sqrt(numd3 + numd4)
                New_FirstX = numd1
                New_FirstY = numd2
                New_FirstP = numd5
                # 新-首有个位移判定
                if numd5 > wypd:
                    NFDis_Ass = utf_bx
                else:
                    NFDis_Ass = utf_wd
                nume1 = (num6 - Last_X) * 1000
                nume2 = (num10 - Last_Y) * 1000
                nume3 = nume1 * nume1
                nume4 = nume2 * nume2
                nume5 = math.sqrt(nume3 + nume4)
                New_LastX = nume1
                New_LastY = nume2
                New_LastP = nume5
                NLDis_Ass = utf_wd
        # 在这里整体输出
        cursor1.execute(
            'INSERT INTO WD_Result_Point(TableName,First_ResultName,Last_ResultName,New_ResultName,First_X,First_Y,Last_X,Last_Y,Result_X,Result_Y,Last_Wight,New_Wight,New_FirstX,New_FirstY,New_FirstP,NFDis_Ass,New_LastX,New_LastY,New_LastP,NLDis_Ass,PointName) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)',
            (
                utf_tn, utf_bn, utf_pn, utf_nn, First_X, First_Y, Last_X, Last_Y, Result_X, Result_Y, Last_Wight,
                New_Wight,
                New_FirstX, New_FirstY, New_FirstP, NFDis_Ass, New_LastX, New_LastY, New_LastP, NLDis_Ass, PointName,))
    # 数据库执行
    db1.commit()
    # 做完一切后，先关闭游标，再关闭数据库
    cursor1.close()
    db1.close()


gszbx = []
gszby = []
jslist = []
jslist1 = []


def bhjs(listcgcs1, listname1, listpastx1, listpasty1, wypd, listname, listnewx, listnewy, listpastx, listpasty, points,
         listcgcs, listbex, listbey, sf, dbpath, befname, finalname, lastname, js, file_name):
    np.set_printoptions(suppress=False)
    # pt用于给点数point计数
    # point会随着回递函数而减少
    pt = 0
    sumnewx = 0
    sumnewy = 0
    sumpastx = 0
    sumpasty = 0
    while pt < points:
        sumnewx = sumnewx + listnewx[pt]
        sumnewy = sumnewy + listnewy[pt]
        sumpastx = sumpastx + listpastx[pt]
        sumpasty = sumpasty + listpasty[pt]
        pt = pt + 1
    # 计算4个重心坐标（最新一期和上一期的）
    zxzbnewx = round((sumnewx / points), 7)
    zxzbnewy = round((sumnewy / points), 7)
    zxzbpastx = round((sumpastx / points), 7)
    zxzbpasty = round((sumpasty / points), 7)
    # 先计算平移参数
    # pycsx = abs(zxzbnewx - zxzbpastx) * 1000
    # pycsy = abs(zxzbnewy - zxzbpasty) * 1000
    pycsx = abs(zxzbnewx - zxzbpastx)
    pycsy = abs(zxzbnewy - zxzbpasty)
    # 分别得k，sita的值
    klist = jzys1(listnewy, zxzbnewy, listnewx, zxzbnewx, sf)
    slist = jzys2(listnewy, zxzbnewy, listnewx, zxzbnewx, sf)
    # 得dx，dy
    dxlist = jzys3(listnewy, 1, 0)
    dylist = jzys3(listnewy, 0, 1)
    # 矩阵太难用了，还是手动计算
    # 分别计算k、s、dx、dy的乘积和
    k2 = cjh(klist)
    s2 = cjh(slist)
    dx2 = cjh(dxlist)
    dy2 = cjh(dylist)
    # 这里再创建矩阵1
    arr1 = np.array(((k2, 0, 0, 0), (0, s2, 0, 0),
                     (0, 0, dx2, 0), (0, 0, 0, dy2)))
    # 矩阵1的逆矩阵矩阵2
    arr2 = np.linalg.inv(arr1)
    # 求矩阵l
    llist = jzys4(listpasty, zxzbpasty, listpastx, zxzbpastx,
                  listnewy, zxzbnewy, listnewx, zxzbnewx, sf)
    # 得到数列e1，e2，e3，e4
    e1 = cjh2(klist, llist)
    e2 = cjh2(slist, llist)
    e3 = cjh2(dxlist, llist)
    e4 = cjh2(dylist, llist)
    arrl = np.array(((e1), (e2), (e3), (e4)))
    # 得到矩阵z
    arrz = np.matmul(arr2, arrl)
    # 求转换矩阵
    jxlist1 = jzys5(listnewx, zxzbnewx, sf)
    jylist1 = jzys5(listnewy, zxzbnewy, sf)
    jxlist2 = jylist1
    # jylist2 = jxlist1 * (-1)
    jylist2 = jzys6(jxlist1, -1)
    lxlist = jzys3(jylist1, 1, 0)
    lylist = jzys3(jylist1, 0, 1)
    # 求改算坐标
    gsx = gsys(listnewx, jxlist1, jylist1, lxlist,
               lylist, arrz, sf, zxzbnewx, zxzbpastx, 0)
    gsy = gsys(listnewy, jxlist2, jylist2, lxlist,
               lylist, arrz, sf, zxzbnewy, zxzbpasty, 1)
    # 还是要先求较差
    xcx = xcys(gsx, listpastx)
    xcy = xcys(gsy, listpasty)
    # 这里记录下第一次所有点的改算坐标
    if js == 0:
        lengs = len(gsx)
        xx = 0
        while xx < lengs:
            gszbx.append(gsx[xx])
            gszby.append(gsy[xx])
            zblist1 = []
            zblist1.append(listbex[xx])
            zblist1.append(listbey[xx])
            zblist1.append(gsx[xx])
            zblist1.append(gsy[xx])
            zblist1.append(xcx[xx])
            zblist1.append(xcy[xx])
            jslist.append(zblist1)
            zblist2 = []
            zblist2.append(listnewx[xx])
            zblist2.append(listnewy[xx])
            jslist1.append(zblist2)
            xx = xx + 1
        # 后面不需要经过这个if
    # 判定方式是用差值平方和和位移对比
    gxpd = 0
    lenxc = len(xcx)
    yy = 0
    while yy < lenxc:
        sumxcx = xcx[yy] * xcx[yy]
        sumxcy = xcy[yy] * xcy[yy]
        sumxc = sumxcx + sumxcy
        xcpd = math.sqrt(sumxc)
        if xcpd > wypd:
            gxpd = -1
            break
        yy = yy + 1
    if gxpd == 0:
        # 输出之前把转换参数都算出来
        # X=(z1+1)*x+z2*y+(旧重心坐标x-新重心坐标x-新重心坐标x*z1-新重心坐标y*z2+z3)
        # Y=(-z2)*x+(z1+1)*y+(z1*新重心坐标y+新重心坐标x*z2+z4+旧重心坐标y-新重心坐标y)
        n1 = float(arrz[0]) + 1
        n2 = float(arrz[1])
        s1 = zxzbnewx * float(arrz[0])
        s2 = zxzbnewy * float(arrz[1])
        n3 = zxzbpastx - zxzbnewx - s1 - s2 + float(arrz[2])
        n4 = (-1) * float(arrz[1])
        n5 = n1
        s3 = (-1) * float(arrz[0]) * zxzbnewy
        s4 = float(arrz[1]) * zxzbnewx
        n6 = s3 + s4 + float(arrz[3]) + zxzbpasty - zxzbnewy
        sxylist = []
        sxylist.append(n1)
        sxylist.append(n2)
        sxylist.append(n3)
        sxylist.append(n4)
        sxylist.append(n5)
        sxylist.append(n6)
        # 输出
        relist1 = []
        relist1.append(listname)
        relist1.append(xcx)
        relist1.append(xcy)
        relist1.append(listcgcs)
        relist1.append(gszbx)
        relist1.append(gszby)
        relist1.append(sxylist)
        # 存入数据库
        insert_into_database1(dbpath, lastname, finalname, befname, file_name, listname1, listpastx1, listpasty1,
                              listcgcs1, listbex, listbey, sxylist, wypd, gsx, gsy, listcgcs)
    else:
        # 先把所有的合在一起，方便删除
        lenlist1 = len(xcx)
        ii = 0
        relist1 = []
        while ii < lenlist1:
            smlist1 = []
            nn2 = xcx[ii] * xcx[ii]
            mm2 = xcy[ii] * xcy[ii]
            nm2 = math.sqrt(nn2 + mm2)
            smlist1.append(nm2)
            smlist1.append(listname[ii])
            smlist1.append(listnewx[ii])
            smlist1.append(listnewy[ii])
            smlist1.append(listpastx[ii])
            smlist1.append(listpasty[ii])
            relist1.append(smlist1)
            ii = ii + 1
        # 直接删除最大的那个
        relist1.sort(key=takeFirst, reverse=True)
        num1 = relist1[0]
        # 获取name
        num2 = num1[1]
        ii2 = 0
        mm2 = 0
        while ii2 < lenlist1:
            if listname[ii2] == num2:
                del listname[ii2]
                del listnewx[ii2]
                del listnewy[ii2]
                del listpastx[ii2]
                del listpasty[ii2]
                points = points - 1
                break
            else:
                ii2 = ii2 + 1
        while mm2 < len(listcgcs):
            if listname1[mm2] == num2:
                listcgcs[mm2] = -1
                break
            else:
                mm2 = mm2 + 1
        print(" Finish!")
        js1 = 1
        bhjs(listcgcs1, listname1, listpastx1, listpasty1, wypd, listname, listnewx, listnewy, listpastx,
             listpasty, points, listcgcs, listbex, listbey, sf, dbpath, befname, finalname, lastname, js1, file_name)


def tablein(dbpath, file_name):
    # 收集每一列的数据，方便后期计算
    listname1 = []
    listpastx1 = []
    listpasty1 = []
    listcgcs1 = []
    listname = []
    listnewx = []
    listnewy = []
    listpastx = []
    listpasty = []
    # 成果次数
    listcgcs = []
    # 最早的
    listbex = []
    listbey = []
    points = 0
    # 查询，提取需要的数据
    db2 = sqlite3.connect(dbpath)
    cursor2 = db2.cursor()
    utfstr1 = to_utf8(file_name)
    sqlstr1 = 'SELECT * FROM WD_Input_Point WHERE TableName = ?'
    cursor2.execute(sqlstr1, (utfstr1,))
    all_da = cursor2.fetchall()
    for da in all_da:
        listbex.append(da[4])
        listbey.append(da[5])
        listpastx.append(da[6])
        listpastx1.append(da[6])
        listpasty.append(da[7])
        listpasty1.append(da[7])
        listnewx.append(da[8])
        listnewy.append(da[9])
        listcgcs.append(da[10])
        listcgcs1.append(da[10])
        points = points + 1
        listname.append(da[11])
        listname1.append(da[11])
    sqlstr2 = 'SELECT * FROM WD_Input_Param WHERE TableName = ?'
    cursor2.execute(sqlstr2, (utfstr1,))
    all_par = cursor2.fetchall()
    befname = all_par[0][0].decode('utf-8')
    lastname = all_par[0][1].decode('utf-8')
    finalname = all_par[0][2].decode('utf-8')
    # pjbc是平均边长
    # fxzwc是方向中误差，zrbzwc是最弱边边长相对中误差，pjbcs是平均边数，pjfxs是平均方向数，sf是缩放值
    # 新增总边数zbs，总方向数zfxs
    # 先计算位移判断值
    pjbc = all_par[0][3]
    fxzwc = all_par[0][4]
    zbs = all_par[0][5]
    zfxs = all_par[0][7]
    sf = all_par[0][9]
    pjbcs = zbs / points
    pjfxs = zfxs / points
    # 再拆细点
    wy1 = pjbc / 1000
    wy2 = wy1 + 1
    wypd1 = wy2 * wy2 / pjbcs
    # 好像206.265是常数系数
    wy3 = pjbc * fxzwc / 206.265
    wypd2 = wy3 * wy3 / pjfxs
    wypd3 = math.sqrt(wypd1 + wypd2)
    wypd = round((wypd3 * 2 * math.sqrt(2)), 9)
    wypd0 = wypd
    js = 0
    return listcgcs1, listname1, listpastx1, listpasty1, wypd0, listname, listnewx, listnewy, listpastx, listpasty, points, listcgcs, listbex, listbey, sf, befname, finalname, lastname, js


# 主函数 计算
def main_function(file_name, dbpath):
    try:
        # 从数据库中调用
        listcgcs1, listname1, listpastx1, listpasty1, wypd0, listname, listnewx, listnewy, listpastx, listpasty, points, listcgcs, listbex, listbey, sf, befname, finalname, lastname, js = tablein(
            dbpath, file_name)
        # 计算
        bhjs(listcgcs1, listname1, listpastx1, listpasty1, wypd0, listname, listnewx, listnewy, listpastx, listpasty,
             points, listcgcs, listbex, listbey, sf, dbpath, befname, finalname, lastname, js, file_name)
        QMessageBox.information(None, "提示", f"文件 '{file_name}' 计算成功！")
        return 1
    except Exception as e:
        QMessageBox.critical(None, "错误", f"文件 '{file_name}' 计算失败！请检查表格是否规范。")
        return -1