cimforreduceloss/Script/FileData.py

__author__ = 'dmy'

## store iPso file data

import os
import sys
import distutils.dir_util
import shutil


class iPsoData:
    def __init__(self, filePath):
        self.__basicInfo = []
        self.__balanceInfo = []
        self.__line = []
        self.__trans = []
        self.__nodePwer = []
        self.__gen = []
        self.__reGen = []
        self.__voltLimit = []
        self.__ground = []
        self.__isAvailable = False
        self.__cap=[]
        self.__read__(filePath)
        pass


    def __read__(self, filePath):
        if not os.path.exists(filePath):
            self.__isAvailable = True
        cur = 0
        with open(filePath) as f:  # 把iPso文件的数据读到内存中
            line = f.readline()
            while line:
                formatedLine = line.strip()
                if line == '':
                    line = f.readline()
                    continue
                sep = formatedLine.split('	')
                if len(sep) < 1:
                    line = f.readline()
                    continue
                if sep[0] == '0':
                    cur += 1
                    line = f.readline()
                    continue
                try:
                    floatSep = [float(s) for s in sep]
                except:
                    print(sep)
                if cur == 0:
                    self.__basicInfo.append(floatSep)
                if cur == 1:
                    self.__balanceInfo.append(floatSep)
                if cur == 2:
                    self.__line.append(floatSep)
                if cur == 3:
                    self.__trans.append(floatSep)
                if cur == 5:
                    self.__ground.append(floatSep)
                if cur==6:
                    self.__cap.append(floatSep)
                if cur == 8:
                    self.__nodePwer.append(floatSep)
                if cur == 9:
                    self.__gen.append(floatSep)
                if cur == 11:
                    self.__reGen.append(floatSep)
                if cur == 13:
                    self.__voltLimit.append(floatSep)
                line = f.readline()

    def GetCapcitor(self):
        return self.__cap

    def GetBasicInfo(self):
        return self.__basicInfo

    def GetBalance(self):
        return self.__balanceInfo

    def GetLine(self):
        return self.__line

    def GetTrans(self):
        return self.__trans

    def GetNodePower(self):
        return self.__nodePwer

    def GetGen(self):
        return self.__gen

    def GetReactiveGen(self):
        return self.__reGen

    def GetVoltLimit(self):
        return self.__voltLimit

    def GetGround(self):
        return self.__ground


class MergeReconfigFile:  # 合并重构用文件
    def __init__(self, fileList):
        self.__basicInfo = [[0, 0]]  #节点数   支路数
        self.__line = []
        self.__trans = []
        self.__node = []
        self.__reverseDic = {}
        self.__mergedIpsoBalance = []
        self.__mergedIpsoLine = []
        self.__mergedIpsoTrans = []
        self.__mergedCapcitor=[]
        self.__mergedIpsoLoad = []
        self.__mergedIpsoGround = []
        self.__mergedNodeNameDic = {}
        # self.__addNumDic = {}  #计算每个文件在新文件中的偏移量
        self.__balanceNodeOffset=[]#记录每条线路头节点在整个合并文件中的位置
        self.__nodePlace= dict()
        self.__merge(fileList)


    def __merge(self, fileList):
        indLine = 1  #序号
        indTrans=1
        indNode = 2
        indGround = 1
        indCap=1
        addNum = 1
        # offsetNodeNum = {}  #记录每个一个文件在新文件里节点的偏移量
        count=1
        for f in fileList:
            # if count>10:
            #     print(f)
            #     break
            count+=1
            ipsodata = iPsoData(f)
            #节点数   可重构支路数
            self.__basicInfo[0][0] += len(ipsodata.GetNodePower())
            self.__basicInfo[0][1] += len(ipsodata.GetLine())
            self.__basicInfo[0][1] += len(ipsodata.GetTrans())
            if  len(ipsodata.GetNodePower())-len(ipsodata.GetLine())-len(ipsodata.GetTrans())>1:
                print('%s is not radial.'%(f) )
            #序号 变电站节点编号

            for s in ipsodata.GetBalance():
                self.__balanceNodeOffset.append(s[1]+addNum)
            #序号 支路节点I 支路节点J
            for l in ipsodata.GetLine():
                self.__mergedIpsoLine.append([indLine, l[1] + addNum, l[2] + addNum, l[3], l[4], l[5], l[6]])
                self.__reverseDic[l[1] + addNum] = l[1]#反过来查是哪个节点
                self.__reverseDic[l[2] + addNum] = l[2]
                indLine += 1
            for t in ipsodata.GetTrans():
                self.__mergedIpsoTrans.append([indTrans,1, t[2] + addNum, t[3] + addNum, t[4], t[5], 1, 1, 1, 1])
                self.__reverseDic[t[2] + addNum] = t[2]
                self.__reverseDic[t[3] + addNum] = t[3]
                indTrans += 1
            #接地支路
            for g in ipsodata.GetGround():
                self.__mergedIpsoGround.append([indGround, g[1] + addNum, g[2], g[3]])
                indGround+=1
            #电容器
            for c in ipsodata.GetCapcitor():
                self.__mergedCapcitor.append([indCap,c[1],c[2]+addNum,c[3],c[4],c[5],c[6],c[7]])
                indCap+=1
            #先给PG和QG赋值
            node = ipsodata.GetNodePower()
            #序号        节点编号        节点有功        节点无功     负荷有功     负荷无功
            for n in ipsodata.GetNodePower():
                self.__mergedIpsoLoad.append([indNode, n[1], n[2] + addNum, n[3], n[4], n[5], n[6], n[7], 0, 0])
                self.__nodePlace[n[2] + addNum]=f
                indNode += 1
            # self.__addNumDic[lineName] = len(ipsodata.GetNodePower())  #每条线路的节点数
            # offsetNodeNum[lineName] = addNum  #记录的是与之前线路的偏移量
            addNum += len(ipsodata.GetNodePower())
        #用一个总的头节点把之前所有线路的头节点连接起来
        for b in self.__balanceNodeOffset:
            self.__mergedIpsoLine.append ([indLine, 1 , b, 0.00001, 0.01, 0, 0])
            self.__basicInfo[0][1]+=1
            indLine+=1
        self.__mergedIpsoLoad.insert(0,[1, n[1], 1,n[3], 0, 0, 0, 0, 0, 0])
        self.__basicInfo[0][0]+=1

    def __GetLineName(self, filePath):
        baseName = os.path.basename(filePath)
        fileName = os.path.splitext(baseName)[0]
        lineName = fileName.split('_')[1]
        return lineName


    def __Swap(self, a, b):
        c = b
        b = a
        a = c
        return (a, b)

    def Output(self, fileOutDir):  #输出到文件
        #输出节点映射
        with open(fileOutDir, 'w') as f:
            for b in self.__basicInfo:
                f.write('%d %d 1 1e-5 1\n' % (b[0], b[1]))
            f.write('0\n')
            f.write('1 1 1.05\n')
            f.write('0\n')
            for b in self.__mergedIpsoLine:
                f.write('%d %d %d %f %f %f %f\n' % (b[0], b[1], b[2],b[3],b[4],b[5],b[6]))
            f.write('0\n')
            for b in self.__mergedIpsoTrans:
                f.write('%d %d %d %d %f %f %f %f %f %f\n' % (b[0], b[1], b[2],b[3],b[4],b[5],b[6],b[7],b[8],b[9]))
            f.write('0\n')
            f.write('0\n')
            for g in self.__mergedIpsoGround:
                f.write('%d %d %f %f\n' % (g[0], g[1], g[2],g[3]))
            f.write('0\n')
            for c in self.__mergedCapcitor:
                #[indCap,c[1],c[2]+addNum,c[3],c[4],c[5],c[6],c[7]]
                f.write('%d %d %d %f %f %d %d %d\n' % (c[0], c[1], c[2],c[3],c[4],c[5],c[6],c[7]))
            f.write('0\n')
            f.write('0\n')
            for n in self.__mergedIpsoLoad:
                f.write('%d %d %d %d %.10f %.10f %.10f %.10f %.10f %.10f\n' % (n[0], n[1], n[2], n[3], n[4], n[5], n[6], n[7], n[8], n[9]))  #负荷单位是kW
            f.write('0\n')
            f.write('1 1 1 1 -100 100 0 0 0 0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('1 1 1 1 1.05 -100 100 0 0 0 0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('1 1 0.999 1.001\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
        # with open(fileOutDir, 'w') as f:
        #     for b in self.__basicInfo:
        #         f.write('%d,%d' % (b[0], b[1]))
        #     f.write(',1,1,1\n')
        #     f.write('0\n')
        #     for s in self.__mergedIpsoBalance:
        #         f.write('%d,%d,%.10f\n' % (s[0], s[1], s[2]))
        #     f.write('0\n')
        #     for b in self.__mergedIpsoLine:
        #         f.write('%d,%d,%d,%.10f,%.10f,%.10f,%.10f\n' % (b[0], b[1], b[2], b[3], b[4], b[5], b[6]))
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     for g in self.__mergedIpsoGround:
        #         f.write('%d,%d,%.10f,%.10f\n' % (g[0], g[1], g[2], g[3]))
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     for b in self.__mergedIpsoLoad:
        #         f.write('%d,%d,%d,%d,%.10f,%.10f,%.10f,%.10f,%.10f,%.10f\n' % (b[0], b[1], b[2], b[3], b[4], b[5], b[6], b[7], b[8], b[9]))
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     for b in self.__mergedIpsoBalance:
        #         f.write('%d,%d,%d,%d,%.10f,%.10f,%.10f\n' % (b[0], 1, b[1], 1, 1, -100, 100))
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')
        #     f.write('0\n')

    def GetReverseMap(self):  #通过新的节点号返回原来的节点号
        return self.__reverseDic;

    def GetCanReconfig(self):
        return self.__canReconfig


class AddLoadToReconfigFile:  # 把负荷给加上
    def __init__(self, ReconfigFilepath):
        dirPath = os.path.dirname(ReconfigFilepath)
        reconfigData = ReconfigFileData(ReconfigFilepath)
        node = reconfigData.GetNodePowerInfo()
        PDFilePath = dirPath + '/PD.txt'
        with open(PDFilePath) as f:
            for i, line in enumerate(f):
                newLine = line.split()
                if newLine == '':
                    continue
                node[i][4] = float(newLine[-1])
        QDFilePath = dirPath + '/QD.txt'
        with open(QDFilePath) as f:
            for i, line in enumerate(f):
                newLine = line.split()
                if newLine == '':
                    continue
                node[i][5] = float(newLine[-1])
        dirName = os.path.basename(dirPath)
        newFilePath = dirPath + '/' + dirName + '_Reconfigure_newFile.txt'
        with open(newFilePath, 'w') as f:
            for b in reconfigData.GetBasicInfo():
                f.write('%d %d\n' % (b[0], b[1]))
            f.write('0\n')
            for b in reconfigData.GetSubstation():
                f.write('%d %d\n' % (b[0], b[1]))
            f.write('0\n')
            for b in reconfigData.GetBranchInfo():
                f.write('%d %d %d\n' % (b[0], b[1], b[2]))
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            #在发电机处加负荷
            for i in reconfigData.GetSubstation():
                reconfigData.GetNodePowerInfo()[i[1] - 1][2] = 100
                reconfigData.GetNodePowerInfo()[i[1] - 1][3] = 100
            for b in reconfigData.GetNodePowerInfo():
                f.write(
                    '%d %d %.10f %.10f %.10f %.10f\n' % (b[0], b[1], b[2] * 1e4, b[3] * 1e4, b[4] * 1e4, b[5] * 1e4))
            f.write('0\n')
            f.write('0\n')
            for b in reconfigData.GetInterLineInfo():
                f.write('%d %d %d %d\n' % (b[0], b[1], b[2], b[3]))
            f.write('0\n')
            f.write('0\n')


class AddLoadToIpsoFile:  # 把负荷给加上
    def __init__(self, ipsoFilepath):
        dirPath = os.path.dirname(ipsoFilepath)
        ipsoData = iPsoData(ipsoFilepath)
        node = ipsoData.GetNodePower()
        PDFilePath = dirPath + '/PD.txt'
        with open(PDFilePath) as f:
            for i, line in enumerate(f):
                newLine = line.split()
                if newLine == '':
                    continue
                node[i][6] = float(newLine[-1])
        QDFilePath = dirPath + '/QD.txt'
        with open(QDFilePath) as f:
            for i, line in enumerate(f):
                newLine = line.split()
                if newLine == '':
                    continue
                node[i][7] = float(newLine[-1])
        dirName = os.path.basename(dirPath)
        newFilePath = dirPath + '/' + dirName + '_iPso_newFile.txt'
        with open(newFilePath, 'w') as f:
            for b in ipsoData.GetBasicInfo():
                f.write('%d\t%d\t%d\t%.10f\t%.10f\n' % (b[0], b[1], 1, b[3], b[4]))
            f.write('0\n')
            for b in ipsoData.GetBalance():
                f.write('%d\t%d\t%.10f\n' % (b[0], b[1], b[2]))
            f.write('0\n')
            for b in ipsoData.GetLine():
                f.write('%d\t%d\t%d\t%.10f\t%.10f\t%.10f\t%.10f\n' % (b[0], b[1], b[2], b[3], b[4], b[5], b[6]))
            f.write('0\n')
            for b in ipsoData.GetTrans():
                f.write('%d\t%d\t%d\t%d\t%.10f\t%.10f\t%.10f\t%.10f\t%.10f\t%.10f\n' % (
                    b[0], b[1], b[2], b[3], b[4], b[5], b[6], b[7], b[8], b[9]))
            f.write('0\n')
            f.write('0\n')
            for b in ipsoData.GetGround():
                f.write('%d\t%d\t%.10f\t%.10f\n' % (b[0], b[1], b[2], b[3]))
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            for b in ipsoData.GetNodePower():
                f.write('%d\t%d\t%d\t%d\t%.10f\t%.10f\t%.10f\t%.10f\t%f\t%.10f\n' % (
                    b[0], b[1], b[2], b[3], b[4], b[5], b[6], b[7], b[8], b[9]))
            f.write('0\n')
            for b in ipsoData.GetGen():
                f.write('%d\t%d\t%d\t%d\t%.10f\t%.10f\t%.10f\t%.10f\t%.10f\t%.10f\n' % (
                    b[0], b[1], b[2], b[3], b[4], b[5], b[6], b[7], b[8], b[9]))
            f.write('0\n')
            f.write('0\n')
            for b in ipsoData.GetReactiveGen():
                f.write('%d\t%d\t%d\t%d\t%.10f\t%.10f\t%.10f\n' % (b[0], b[1], b[2], b[3], b[4], b[5], b[6]))
            f.write('0\n')
            f.write('0\n')
            for b in ipsoData.GetVoltLimit():
                f.write('%d\t%d\t%.10f\t%.10f\n' % (b[0], b[1], b[2], b[3]))
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')


class NodeInfo:
    def __init__(self, lineInfoPath, transInfoPath):
        self.__infoDic = {}
        with open(lineInfoPath) as f:
            for line in f:
                newLine = line.strip()
                if newLine == '':
                    continue
                sep = newLine.split(',')
                if len(sep) < 4:
                    continue
                nodeI = int(sep[1])
                nodeJ = int(sep[2])
                info = sep[3].split('-')
                if len(info) < 2:
                    continue
                nodeIinfo = info[0].strip()
                nodeJinfo = info[1].strip()
                if not nodeI in self.__infoDic.keys():
                    self.__infoDic[nodeI] = nodeIinfo
                if not nodeJ in self.__infoDic.keys():
                    self.__infoDic[nodeJ] = nodeJinfo
        with open(transInfoPath) as f:
            for line in f:
                newLine = line.strip()
                if newLine == '':
                    continue
                sep = newLine.split(',')
                if len(sep) < 4:
                    continue
                nodeI = int(sep[1])
                info = sep[4].strip()
                if not nodeI in self.__infoDic.keys():
                    self.__infoDic[nodeI] = info
                nodeJ = int(sep[2])
                if not nodeJ in self.__infoDic.keys():
                    self.__infoDic[nodeJ] = info

    def GetNodeName(self, nodeI):
        if nodeI in self.__infoDic.keys():
            return self.__infoDic[nodeI]
        else:
            return ""


class AddLoadtoFileBatch:
    def __init__(self, baseDir, AddLoadInterface=AddLoadToIpsoFile):
        dirList = os.listdir(baseDir)
        for d in dirList:
            absDir = baseDir + '/' + d
            if not os.path.isdir(absDir):
                continue
            # 看有没有PD.txt   和 QD.txt
            PDFilePath = absDir + '/PD.txt'
            QDFilePath = absDir + '/QD.txt'
            if not os.path.exists(PDFilePath) or not os.path.exists(QDFilePath):
                continue
            if AddLoadInterface is AddLoadToReconfigFile:
                absDatFile = absDir + '/iPso_' + d + '_DisnetReconfig.txt'
            if AddLoadInterface is AddLoadToIpsoFile:
                absDatFile = absDir + '/iPso_' + d + '.txt'
            if not os.path.exists(absDatFile):
                continue
            AddLoadInterface(absDatFile)
            print('完成 %s' % (absDatFile))


class LoadMapSingleton:
    PloadDic = {}
    QloadDic = {}


class NodeLoadMap:  # 依据变压器串号对负荷进行映射
    def __init__(self, manyFileDir, singleFileDir,
                 AddLoadInterface=AddLoadToIpsoFile):  #manyFileDir是分开的计算文件目录的上级目录  singleFileDir是合并后并形成重构的文件的目录
        dirList = os.listdir(manyFileDir)
        PloadDic = {}
        QloadDic = {}
        if len(LoadMapSingleton.PloadDic.keys()) == 0 or len(LoadMapSingleton.QloadDic.keys()) == 0:
            for d in dirList:
                absDir = manyFileDir + '/' + d
                if not os.path.exists(absDir):
                    continue
                #看看有没有负荷文件
                PDFilePath = absDir + '/PD.txt'
                QDFilePath = absDir + '/QD.txt'
                if not os.path.exists(PDFilePath) or not os.path.exists(QDFilePath):
                    continue
                print('利用了%s中的负荷' % absDir)
                #把负荷信息读出来
                PD = []
                with open(PDFilePath) as f:
                    for line in f:
                        newLine = line.strip()
                        if len(newLine) == 0:
                            continue
                        PD.append(float(newLine))
                QD = []
                with open(QDFilePath) as f:
                    for line in f:
                        newLine = line.strip()
                        if len(newLine) == 0:
                            continue
                        #if(float(newLine)<0):
                        #	print(newLine)

                        QD.append(float(newLine))
                #找到变压器信息文件
                manyTransFilePath = absDir + '/iPso_' + d + '_TransformerInfor.txt'
                if not os.path.exists(manyTransFilePath):
                    continue
                with open(manyTransFilePath) as f:
                    for line in f:
                        newLine = line.strip()
                        if len(newLine) == 0:
                            continue
                        sep = newLine.split(',')
                        if len(sep) >= 6:
                            ID = sep[5].strip()
                            loadSideNum = int(sep[2])
                            PloadDic[ID] = PD[loadSideNum - 1]
                            QloadDic[ID] = QD[loadSideNum - 1]
                LoadMapSingleton.PloadDic = PloadDic
                LoadMapSingleton.QloadDic = QloadDic
        else:
            PloadDic = LoadMapSingleton.PloadDic
            QloadDic = LoadMapSingleton.QloadDic
        #开始处理合并红的文件。方法是在合并的文件夹下面生成PD.txt和QD.txt，然后利用AddLoadToIpso类来添加。
        singleTransFile = singleFileDir + '/iPso_' + os.path.basename(singleFileDir) + '_TransformerInfor.txt'
        if not os.path.exists(singleTransFile):
            print('合并后的变压器信息文件不存在')
            return
        if AddLoadInterface is AddLoadToIpsoFile:
            absDatFile = singleFileDir + '/iPso_' + os.path.basename(singleFileDir) + '.txt'
            data = iPsoData(absDatFile)
            basicInfo = data.GetBasicInfo()
            totalNum = int(basicInfo[0][0])
        if AddLoadInterface is AddLoadToReconfigFile:
            absDatFile = singleFileDir + '/iPso_' + os.path.basename(singleFileDir) + '_DisnetReconfig.txt'
            data = ReconfigFileData(absDatFile)
            basicInfo = data.GetBasicInfo()
            totalNum = int(basicInfo[0][0])
        singlePD = [0 for i in range(totalNum)]
        singleQD = [0 for i in range(totalNum)]
        totalPD = 0
        with open(singleTransFile) as f:
            for line in f:
                newLine = line.strip()
                if len(newLine) == 0:
                    continue
                sep = newLine.split(',')
                if len(sep) >= 6:
                    ID = sep[5].strip()
                    loadSideNum = int(sep[2])
                    if ID in PloadDic.keys():
                        singlePD[loadSideNum - 1] = PloadDic[ID]
                        singleQD[loadSideNum - 1] = QloadDic[ID]
                    else:
                        singlePD[loadSideNum - 1] = 1e-8
                        singleQD[loadSideNum - 1] = 1e-8
                    totalPD += singlePD[loadSideNum - 1]
        singlePDFilePath = singleFileDir + '/PD.txt'
        singleQDFilePath = singleFileDir + '/QD.txt'
        PDf = open(singlePDFilePath, 'w')
        QDf = open(singleQDFilePath, 'w')
        if not PDf or not QDf:
            print('无法创建文件')
            return
        for i in range(len(singlePD)):
            PDf.write('%.10f\n' % singlePD[i])
            QDf.write('%.10f\n' % singleQD[i])
        if PDf:
            PDf.close()
        if QDf:
            QDf.close()
            AddLoadInterface(absDatFile)
        print('完成%s中负荷的添加' % (absDatFile))
        print('总的有功负荷为%f' % (totalPD))


class ReconfigFileData:  # 读入祝靖用的重构文件格式
    def __init__(self, filePath):
        currBlock = 1
        self.__basicInfo = []
        self.__subStationInfo = []
        self.__branchInfo = []
        self.__nodePowerInfo = []
        self.__interLineInfo = []
        with open(filePath) as f:
            for line in f:
                newLine = line.strip()
                sep = line.split(' ')
                if len(sep) > 0:
                    if int(sep[0]) == 0:
                        currBlock += 1
                        continue
                if currBlock == 1:
                    self.__basicInfo.append([int(sep[0]), int(sep[1])])
                    continue
                if currBlock == 2:
                    self.__subStationInfo.append([int(s) for s in sep])
                    continue
                if currBlock == 3:
                    self.__branchInfo.append([int(s) for s in sep])
                    continue
                if currBlock == 7:
                    self.__nodePowerInfo.append(
                        [int(sep[0]), int(sep[1]), float(sep[2]), float(sep[3]), float(sep[4]), float(sep[5])])
                    continue
                if currBlock == 9:
                    t = [int(s) for s in sep[:-1]]
                    t.append(float(sep[-1]))
                    self.__interLineInfo.append(t)

    def GetBasicInfo(self):
        return self.__basicInfo

    def GetSubstation(self):
        return self.__subStationInfo

    def GetBranchInfo(self):
        return self.__branchInfo

    def GetNodePowerInfo(self):
        return self.__nodePowerInfo

    def GetInterLineInfo(self):
        return self.__interLineInfo

    def SaveTo(self, filePath):
        with open(filePath, 'w') as f:
            for i in self.__basicInfo:
                f.write('%d %d\n' % (i[0], i[1]))
            f.write('0\n')
            for i in self.__subStationInfo:
                f.write('%d %d\n' % (i[0], i[1]))
            f.write('0\n')
            for i in self.__branchInfo:
                f.write('%d %d %d\n' % (i[0], i[1], i[2]))
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            f.write('0\n')
            for i in self.__nodePowerInfo:
                f.write('%d %d %f %f %f %f\n' % (i[0], i[1], i[2], i[3], i[4], i[5]))
            f.write('0\n')
            f.write('0\n')
            for i in self.__interLineInfo:
                f.write('%d %d %d %d\n' % (i[0], i[1], i[2], i[3]))
            f.write('0\n')


class StdiPsoToZJ:  # 把iPso的文件转换为祝靖要的格式
    def __init__(self, iPsoDatFile):
        iPsoDat = iPsoData(iPsoDatFile)
        self.__madeStructure = []
        madeStructure = self.__madeStructure
        basicInfo = iPsoDat.GetBasicInfo()
        madeStructure.append([basicInfo[0][0], basicInfo[0][1]])
        madeStructure.append([0])
        balanceInfo = iPsoDat.GetBalance()
        for i in balanceInfo:
            madeStructure.append([i[0], i[1]])
        madeStructure.append([0])
        lineInfo = iPsoDat.GetLine()
        for i in lineInfo:
            madeStructure.append([i[0], i[1], i[2]])
        transInfo = iPsoDat.GetTrans()
        for i in transInfo:
            madeStructure.append([i[0], i[2], i[3]])
        madeStructure.append([0])
        madeStructure.append([0])
        madeStructure.append([0])
        madeStructure.append([0])
        loadInfo = iPsoDat.GetNodePower()
        loadInfo[int(basicInfo[0][1])][4] = 100
        loadInfo[int(basicInfo[0][1])][5] = 100
        for i in loadInfo:
            madeStructure.append([i[0], i[2], i[4], i[5], i[6], i[7]])
        madeStructure.append([0])
        madeStructure.append([0])
        #可重构支路
        for i in lineInfo:
            madeStructure.append([i[0], i[1], i[2], 1])
        madeStructure.append([0])

    def Save(self, saveTo):
        with open(saveTo, 'w') as f:
            seperator = '\t'
            madeStructure = self.__madeStructure
            for i in madeStructure:
                strI = [str(j) for j in i]
                newLine = seperator.join(strI)
                f.write(newLine)
                f.write('\n')


class NodeLoadMapBatch:
    def __init__(self, mappingFileDir, mappedFileDir,
                 AddLoadInterface=AddLoadToIpsoFile):  # mappingFileDir是有负荷的文件夹的上级目录,mappedFileDir是被填充的文件夹的上级目录
        dirList = os.listdir(mappedFileDir)
        for d in dirList:
            absDir = mappedFileDir + '/' + d
            if os.path.isdir(absDir):
                NodeLoadMap(mappingFileDir, absDir, AddLoadInterface)


class ConnectionSwitch:
    def __init__(self, switchFile):
        if not os.path.exists(switchFile):
            raise Exception('swtich file not exists!')
        self.__switch = []
        with open(switchFile) as f:
            for line in f:
                newLine = line.strip()
                sep = newLine.split(',')
                if len(sep) == 0:
                    continue
                if len(sep) == 4:
                    self.__switch.append(sep)

    def GetSwtich(self):
        return self.__switch


class BuildReconfigurationArea:  # 依据重构开关吧，把需要重构的文件准备好，拷贝到一个目录下
    def __init__(self, baseDir, switchDir,
                 distFilePoolDir):  #baseDir    是拷贝到的文件夹,switchDir是重构开关的文件夹，都是txt文件 distFilePoolDir是所有配网文件的目录
        self.__getDistFiles(distFilePoolDir)
        allSwitchFile = self.__GetAllSwtichFile(switchDir)
        for a in allSwitchFile:
            print('正在利用%s' % a)
            absDir = baseDir + '/' + os.path.splitext(os.path.basename(a))[0]
            #先删除原来就有的目录
            #if os.path.exists(absDir):
            #	pass
            #shutil.rmtree(absDir)
            #建立一个新文件夹
            if not os.path.exists(absDir):
                distutils.dir_util.mkpath(absDir)
            switch = ConnectionSwitch(a)
            dupLineNam = []  #有重复的线路文件名称
            for s in switch.GetSwtich():
                dupLineNam.append(s[0])
                dupLineNam.append(s[2])
            uniqueLineName = set(dupLineNam)
            for u in uniqueLineName:
                try:
                    copyTargetDir = absDir + '/' + os.path.basename(self.__distFileDic[u])
                    if os.path.exists(copyTargetDir):
                        shutil.rmtree(copyTargetDir)
                except Exception as e:
                    print(a)
                    print(e)
                srcDirPath = self.__distFileDic[u]
                shutil.copytree(srcDirPath, copyTargetDir, True)
                #然后再考新文件
                #拷贝联络开关文件

                shutil.copy(a, absDir + '/breaker.txt')
        print('Build ReconfigurationArea Finished!')

    def __getDistFiles(self, distFilePoolDir):
        dirList = os.listdir(distFilePoolDir)
        self.__distFileDic = {}
        for d in dirList:
            absDir = distFilePoolDir + '/' + d
            if not os.path.isdir(absDir):
                continue
            lineName = self.__getLineName(absDir)
            self.__distFileDic[lineName] = absDir

    def __getLineName(self, linePath):
        baseName = os.path.basename(linePath)
        fileName = os.path.splitext(baseName)[0]
        lineName = fileName.split('_')[0]
        return lineName

    def __GetAllSwtichFile(self, switchDir):
        dirList = os.listdir(switchDir)
        for d in dirList:
            absFilePath = switchDir + '/' + d
            if os.path.isfile(absFilePath):
                yield absFilePath


class TargetDir:
    def __init__(self, dir):
        dirList = os.listdir(dir)
        opFileList = []
        for d in dirList:
            opDirPath = dir + '/' + d  # 操作文件夹
            if os.path.isdir(opDirPath):
                opFile = opDirPath + '/iPso_' + d + '.txt'
                if not os.path.exists(opFile):
                    print('文件 %s   不存在' % (opFile))
                opFileList.append(opFile)
        self.__opFileList = opFileList

    def GetOpFileList(self):
        print('文件顺序为:')
        for o in self.__opFileList:
            print(o)
        return self.__opFileList


class MergeReconfigBatch:
    def __init__(self, baseDir):
        listDir = os.listdir(baseDir)
        for l in listDir:
            absDir = baseDir + '/' + l
            if not os.path.isdir(absDir):
                continue
            # 清除旧文件
            self.__delOldFiles(absDir)
            opDirList = TargetDir(absDir)
            a = MergeReconfigFile(opDirList.GetOpFileList())
            if a.GetCanReconfig():
                a.Output(absDir + '/合并后iPso文件.txt')
            else:
                print('%s不满足重构条件' % absDir)

    def __delOldFiles(self, dir):
        files = os.listdir(dir)
        for f in files:
            absFile = dir + '/' + f
            if not os.path.isfile(absFile):
                continue
            if f == 'breaker.txt':
                continue
            if f == 'output.txt':
                continue
            os.remove(absFile)
        # opDirList=TargetDir(r'C:\lz/test3')
        #a=MergeReconfigFile(opDirList.GetOpFileList())
        #a.Output('c:/lz/test3/合并后iPso文件.txt')


class IslandedReconfig:  # 有孤岛的重构文件
    def __init__(self, origFile, removedLine):
        reconfigFile = ReconfigFileData(origFile)
        removedLineDic = {}
        with open(removedLine) as f:
            for line in f:
                newLine = line.strip()
                sep = newLine.split(' ')
                if len(sep) == 0:
                    continue
                if len(sep) >= 3:
                    key = sep[0] + '_' + sep[1]
                    removedLineDic[key] = 1
        branch = reconfigFile.GetBranchInfo()
        for i in branch[:]:
            key = str(i[1]) + '_' + str(i[2])
            if key in removedLineDic.keys():
                branch.remove(i)
        dirName = os.path.dirname(origFile)
        reconfigFile.SaveTo(dirName + '/断线后Reconfig.txt')