ControlNetwork/Back/WD/WDcompute.py

import sqlite3
import os
import tkinter as tk
from tkinter import messagebox
import math
import numpy as np
import math
import tkinter as tk
import numpy as np
from tkinter import messagebox
from tkinter import *

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}' 计算成功！")
    except Exception as e:
        QMessageBox.critical(None, "错误", f"文件 '{file_name}' 计算失败！错误信息: {str(e)}")