ylpy

#coding=utf8
import arcpy
import os
import math

#获得各种比例尺的代号
def getScaleDM(idx):
    return chr(idx + 65)

#有经纬度和图幅序号获得图幅号，
#idx=0，为1:100万
#idx=1,为1:50万
#idx=2,为1:25万
#idx=3,为1:10万
#idx=4,为1:5万
#idx=5,为1:2.5万
#idx=6,为1:1万
#idx=7,为1:5千
#idx=8,为1:2千
#idx=9,为1:1千
#idx=10,为1:500
def getnewtf(ll, bb, idx):
    ScaleDM = getScaleDM(idx)
    a = int(math.floor(bb / 4) + 1);
    #b = int(ll / 6) + 31;
    b = int(ll //6) + 31;
    #cc = int(4.0 / dw);
    cc = int(4.0 // dw);
    #cc1 = int((bb - 4 * (a - 1)) / dw); #(int)((bb - 4 * (a - 1)) / dw)
    cc1 = int((bb - 4 * (a - 1)) // dw); #(int)((bb - 4 * (a - 1)) / dw)
    c = cc - cc1;
    if (c == 0):
        c = cc
        a = a + 1
    #d = (int)((ll - 6 * (b - 31)) / dj) + 1
    d = (int)((ll - 6 * (b - 31)) // dj) + 1
    #arcpy.AddMessage("dddd="+str(d))
    #arcpy.AddMessage("aaa="+str(a))
    #arcpy.AddMessage("bbbb="+str(b))
    resultstr = chr(a + 64) + str(b)
    #arcpy.AddMessage("resultstr="+resultstr)
    if (idx > 0):
        resultstr = resultstr + ScaleDM;
        mystr = str(c).zfill(3)
        resultstr = resultstr + mystr + str(d).zfill(3);

    return resultstr;
#获得一个坐标的地理坐标系
def getGCSName(sr):
     if sr.type != "Geographic":
         if hasattr(sr, 'GCS'):
             return sr.GCS
         srname=sr.name

         sr1 = arcpy.SpatialReference()
         if srname.startswith("Xian_1980"):
              sr1.factoryCode=4610 #80
         elif srname.startswith("Beijing_1954") or srname.find("Lambert")>0:
              sr1.factoryCode=4214 #54
         elif  srname.startswith("CGCS2000"):
             sr1.factoryCode=4490 #2000
         else:
             sr1.factoryCode=4326 #1984
         sr1.create()

         return sr1
     else:
       return sr


#把其他类型转字符串
def ConverttoStr(v):
    if type(v)==str:
        return v
    else:
        return str(v)
#字段是否存在
def FieldExists(TableName,FieldName):
    desc = arcpy.Describe(TableName)
    for field in desc.fields:
        if field.Name.upper() ==FieldName:
            return True
            break
    return False
#清除选择对象
def clearselect(in_data):
    arcpy.SelectLayerByAttribute_management(in_data, "CLEAR_SELECTION")
    arcpy.RefreshActiveView()
#获得记录数
def getCount(inFeature):
    result = arcpy.GetCount_management(inFeature)
    count= int(result.getOutput(0))
    return count
#初始化进度条
def initProgress(hint,num):
    arcpy.SetProgressor("step", u"正在"+hint,0,num,1)
#进度条
def step():
    arcpy.SetProgressorLabel(u"正在进行....")
    arcpy.SetProgressorPosition()
#释放进度条
def freeProgress():
    arcpy.ResetProgressor()
#获得字段长度
def getlength(mystr):
    lenTxt = len(mystr)
    lenTxt_utf8 = len(mystr.decode('utf-8'))
    size = int((lenTxt_utf8 - lenTxt)/2 + lenTxt)
    #arcpy.AddMessage("'{0}'最后长度{1},lenTxt={2},lenTxt_utf8={3}".format(mystr,size,lenTxt,lenTxt_utf8))
#获得指定条件的记录数
def getwhereCount(inFeature,wherestr):
    my_temp="my_temp" #临时数据

    arcpy.MakeFeatureLayer_management(inFeature,my_temp,wherestr)
    return getCount(my_temp)
#获得一个表的FID
def getOIDField(jfb_Select):
    desc = arcpy.Describe(jfb_Select)
    OIDField=desc.OIDFieldName
    return OIDField
#删除一个要素中图形相同的，可以删除相同的点、线、面
def delsame(jfb_Select):
    desc = arcpy.Describe(jfb_Select)
    shapeName = desc.ShapeFieldName
    arcpy.DeleteIdentical_management(jfb_Select, [shapeName])
#获得点的距离
def pointDistance(pt1,pt2):
    return math.sqrt((pt1.X-pt2.X)*(pt1.X-pt2.X)+(pt1.Y-pt2.Y)*(pt1.Y-pt2.Y))
#添加一个数据到指定mxd
def AddLayer(mxd,inFeature):
    df=arcpy.mapping.ListDataFrames(mxd)[0]
    addLayer = arcpy.mapping.Layer(inFeature)
    arcpy.mapping.AddLayer(df, addLayer,"AUTO_ARRANGE") #AUTO_ARRANGE，"BOTTOM"，TOP
#删除一个表的所有字段
def delAllField(inFeature):
    fieldList = arcpy.ListFields(inFeature)
    for field in fieldList:
          if not (field.type=="Geometry" or field.type=="OID" or not field.editable):
              arcpy.DeleteField_management(inFeature,field.name)
#获得一个要素类合并后的图形，使用融合工具
def getLayerGeomtry(inFeature):

    num=getCount(inFeature)

    mytemp=inFeature
    if num>1:#融合在
        mytemp="in_memory/mytemp999aa"
        arcpy.Dissolve_management(inFeature, mytemp,"", "", "MULTI_PART")

    desc = arcpy.Describe(mytemp)
    shapefieldname = desc.ShapeFieldName

    rows = arcpy.SearchCursor(mytemp)

    for row in rows:
        pgeometry=row.getValue(shapefieldname)

    if row:
        del row
    if rows:
        del rows
    return pgeometry
#获得一个要素类合并后的图形，自己图形合并
def getLayerGeomtryPro(inFeature):
    desc = arcpy.Describe(inFeature)
    shapefieldname = desc.ShapeFieldName
    rows = arcpy.SearchCursor(inFeature)
    i=0
    for row in rows:
        if i<1:
            pgeometry=row.getValue(shapefieldname)
        else:
            pgeometry=pgeometry.union(row.getValue(shapefieldname))

        i=i+1
    if row:
        del row
    if rows:
        del rows

    return pgeometry
#更新一个表所有图形
def  updateGeometry(inFeature,pgeometry2):
    desc = arcpy.Describe(inFeature)
    shapefieldname = desc.ShapeFieldName
    rows = arcpy.UpdateCursor(inFeature)

    for row in rows:
        row.setValue(shapefieldname,pgeometry2)
        rows.updateRow(row)

    if row:
        del row
    if rows:
        del rows
#度转度分秒
def dtodmsint(num):
    '''
    bef0: 小数点前面的值
    aft012: 转换后小数点后面第一二位数
    aft034: 转换后小数点后面第三四伴数
    '''
    num = eval(num)

    bef0, aft0 = int(num), num-int(num)

    aft012, aft034_t = int(aft0*60), aft0*60-int(aft0*60)

    aft034 = int(round(aft034_t*60))

    if aft034 < 10:
        aft034 = '0'+str(aft034)

    elif aft034 == 60:
        aft034='00'
        aft012 += 1
        if aft012==60:
            bef0=bef0+1
            aft012=0

    elif aft034 > 60:
        print "error:%s"%aft034

    if aft012<10:aft012 = '0' + str(aft012)


    return "%s°%s′%s″"%(bef0, aft012, aft034)
#判断一个字符串是否为数字
def is_number(str):
    try:
        # 因为使用float有一个例外是'NaN'
        if str=='NaN':
            return False
        float(str)
        return True
    except ValueError:
        return False
#度分秒转度
def dmstod(dms):
    #arcpy.AddMessage("================================="+dms+"==============================")
    try:
        p = dms.find('°')
        if p<0:
            return str(dms)
        #arcpy.AddMessage("p="+str(p))
        d=string.atof(dms[0:p].strip())
        #arcpy.AddMessage("d="+str(d))
        p1=dms.find('′')
        #arcpy.AddMessage("p1="+str(p1))
        if p1<0:
             p1=dms.find("'")

        f=0 #
        if p1>0:
            f=string.atof(dms[p+1:p1].strip())
        else:
            p1=p
        #arcpy.AddMessage("f="+str(f))

        p2=dms.find('″')
        if p2<0:
             p2=dms.find('"')
        #arcpy.AddMessage("p2="+str(p2))

        s=0 #
        if p2>0:
            s=string.atof(dms[p1+1:p2].strip())
        #arcpy.AddMessage("s="+str(s))

        return d+f/60+s/3600
    except ValueError:
        return None
#获得一个点的投影坐标系
def getpjxy(gx,gy,sr,GCSsr):
    point = arcpy.Point()
    point.X = gx
    point.Y = gy

    pointGeometry = arcpy.PointGeometry(point,GCSsr)
    newpointgeo=  pointGeometry.projectAs(sr)
    p1=newpointgeo.firstPoint
    return p1.X,p1.Y
#获得一个点的经纬度坐标系
def getGCSxy(gx,gy,sr,GCSsr):
    point = arcpy.Point()
    point.X = gx
    point.Y = gy

    pointGeometry = arcpy.PointGeometry(point,sr)
    newpointgeo=  pointGeometry.projectAs(GCSsr)
    p1=newpointgeo.firstPoint
    return p1.X,p1.Y
#数字转字符保留几位小数  by gisoracle
def floattostr(num,xsnum):
    if xsnum==0:
        return str(int(num))
    nd=round(num,xsnum)
    nstr=str(nd)
    idx=nstr.index('.')
    print idx
    p=len(nstr)-idx-1
    print p
    n=xsnum-p
    print n
    s=""
    for i in range(n):
        s=s+"0"

    return nstr+s


#度转度分秒（秒保留小数位secondNUM)
def dtodmsbydouble(num,secondNUM):
    '''
    bef0: 小数点前面的值
    aft012: 转换后小数点后面第一二位数
    aft034: 转换后小数点后面第三四伴数
    '''
    if secondNUM<1:
        return dtodmsint(num)
    if not isinstance(num,float):
        num = eval(num)

    bef0, aft0 = int(num), num-int(num)

    aft012=int(aft0*60)
    aft034_t = aft0*60-int(aft0*60)

    aft034 =aft034_t*60
    #arcpy.AddMessage("aft034:"+str(aft034))
    aft034=round(aft034,secondNUM)

    if aft034 < 10:
        aft034 = '0'+str(aft034)

    elif aft034 == 60:
        aft034='00'
        aft012 += 1
        if aft012==60:
            bef0=bef0+1
            aft012=0

    elif aft034 > 60:
        print "error:%s"%aft034

    aft034=floattostr(float(aft034),secondNUM)

    if aft012<10:aft012 = '0' + str(aft012)
    return "%s°%s′%s″"%(bef0, aft012, aft034)

#获得字段类型是否为文本
def getFieldTypeIsText(TableName,FieldName):
    desc = arcpy.Describe(TableName)
    FieldName=FieldName.upper()
    for field in desc.fields:
        if field.Name.upper() ==FieldName:
            return field.type=="String"

    return False
#获得指定图层
def getLayer(layername):
    layername=layername.upper()
    mxd = arcpy.mapping.MapDocument("CURRENT")
    try:
        for lyr in arcpy.mapping.ListLayers(mxd):
            if lyr.name.upper()==layername:
                return lyr
        return None
    finally:
        del mxd
#从列表中获得元素的索引idx
def listindex(mylist,aa):
    num=len(mylist)
    for i in range(num):
        if mylist[i]==aa:
            return i
    return -1
#两点线
def gettwoline(pt1,pt2):
    parray=arcpy.Array()
    parray.add(pt1)
    parray.add(pt2)
    polyline = arcpy.Polyline(parray,sr)
    parray.removeAll()
    return polyline
#有坐标生成点
def createPoint(x,y):
    point = arcpy.Point() #create an empty Point object
    point.X = float(x)
    point.Y = float(y)
    return point
    #pointGeometry = arcpy.PointGeometry(point,sr)
    #return pointGeometry
#有最小的xy和最大的XY生成面，有矩形范围生成面
def createPolygon(xx1,yy1,xx2,yy2):

    p1=createPoint(xx1,yy2)
    p2=createPoint(xx2,yy2)
    p3=createPoint(xx2,yy1)
    p4=createPoint(xx1,yy1)

    parray=arcpy.Array()
    parray.add(p1)
    parray.add(p2)
    parray.add(p3)
    parray.add(p4)
    parray.add(p1)
    polygon = arcpy.Polygon(parray,sr)
    parray.removeAll()
    return polygon
#根据一个点创建矩形:
def getRectbyPoint(pnt,width,hight):
     x=pnt.X
     y=pnt.Y
     xx1=x-width/2
     xx2=x+width/2

     yy1=y-hight/2
     yy2=y+hight/2
     pgeo= createPolygon(xx1,yy1,xx2,yy2)
     return pgeo
#字符串转列表
def strtoList(text):
    text = text.decode('utf-8')
    pList=[]
    for i in text:
        pList.append(i)
    return pList
#获得一个数据的工作空间
def getWorkspace(inFeature):
    desc = arcpy.Describe(inFeature)
    path=desc.path.lower()
    if path.endswith(".mdb") or path.endswith(".gdb"):
        return path
    if os.path.isdir(path):#shp
        return path
    outPath, outFC = os.path.split(path)
    return outPath
#根据的一个点，距离和角度，获得新点
def getonePoint(pnt,dis,angle):
    x=pnt.X+dis*math.cos(angle)
    y=pnt.Y+dis*math.sin(angle)
    pt1 =  arcpy.Point(x,y)
    return pt1
#根据TableName，增加所有字段到outFeature
def AddAllField(TableName,outFeature):
    desc = arcpy.Describe(TableName)
    plist = []
    for field in desc.fields:
        if field.type!="Geometry" and field.type!="OID":
            plist.append(field.Name)
            arcpy.AddField_management(outFeature,field.Name ,field.type,field.precision,field.scale,field.length)
    return plist
#旋转某个点，rotateX,rotateY旋转X，y，rotateAngle是旋转角度
def rotateonePoint(pnt,rotateX,rotateY,rotateAngle):
    x=(pnt.X-rotateX)*math.cos(rotateAngle)-(pnt.Y-rotateY)*math.sin(rotateAngle)+rotateX  # (x - rx0)*cos(a) - (y - ry0)*sin(a) + rx0 ;
    y=(pnt.X-rotateX)*math.sin(rotateAngle)-(pnt.Y-rotateY)*math.cos(rotateAngle)+rotateY #(x - rx0)*sin(a) + (y - ry0)*cos(a) + ry0
    pt1 =  arcpy.Point(x,y)
    return pt1
#旋转一个点对象
def MovePoint(pt):
    pnt= pt.getPart(0)
    return rotateonePoint(pnt)
#旋转多点
def MoveMPoint(partgeometry):
    array = arcpy.Array()
    num=partgeometry.partCount

    for i in range(num):

        pt=partgeometry.getPart(i)
        pt=rotateonePoint(pt)
        array.add(pt)
    return array
#旋转很多点
def MoveManyPoint(partgeometry):
    array = arcpy.Array()
    num=partgeometry.count


    for i in range(num):

        pt=partgeometry[i]
        pt=rotateonePoint(pt)
        array.add(pt)
    return array
#把内多边形，分解出来
def splitNgeometry(mgeometry):
    num=mgeometry.count
    Sumarray = arcpy.Array()
    parray = arcpy.Array()
    for i in range(num):
        pt=mgeometry[i]
        if pt:
            parray.add(pt)
        else:#内多边形
            Sumarray.add(parray)
            parray.removeAll()
    Sumarray.add(parray)
    return Sumarray
#旋转一个对象
def Moveobj(geometry):
    part_count = geometry.partCount  #有几部分
    #arcpy.AddMessage("FID:"+str(FID)+",part_count:"+str(part_count))
    Sumarray = arcpy.Array()
    for i in range(part_count):
        partgeometry=geometry.getPart(i)
        SpliArray=splitNgeometry(partgeometry)
        N=SpliArray.count
        #arcpy.AddMessage("NNNNN=====:"+str(N))
        for j in range(N):
            Splitgeometry=SpliArray[j]

            array=MoveManyPoint(Splitgeometry)

            try:
                Sumarray.add(array)
            except Exception as err:
                arcpy.AddError("错误=============j:"+str(j)+","+err.message)
    return Sumarray
#获得FID对应的图形
def getFID(FID,pFeatures):
    FIDField=getOIDField(pFeatures)
    wherestr="{0}={1}".format(FIDField,FID)
    #arcpy.AddMessage(u"===="+wherestr)
    with arcpy.da.SearchCursor(pFeatures, "SHAPE@",wherestr) as cursor:
        try:
            for row in cursor:
                #arcpy.AddMessage(u"————————"+wherestr)
                return row[0]
        finally:
            del cursor
#获得对应FID的图形
def updateFID(FID,pFeatures,geo):
    FIDField=getOIDField(pFeatures)
    wherestr="{0}={1}".format(FIDField,FID)
    with arcpy.da.UpdateCursor(pFeatures, "SHAPE@",wherestr) as cursor:
        try:
            for row in cursor:
                row[0]=geo
                cursor.updateRow(row)
        finally:
            del cursor
#删除对应FID
def delFID(FID,pFeatures):
    FIDField=getOIDField(pFeatures)
    wherestr="{0}={1}".format(FIDField,FID)
    with arcpy.da.UpdateCursor(pFeatures, "SHAPE@",wherestr) as cursor:
        try:
            for row in cursor:
                cursor.deleteRow()
        finally:
            del cursor
#更新某个字段的值
def updateFieldName(tablename,FieldName,Value):
    rows = arcpy.UpdateCursor(tablename)
    for row in rows:
        row.setValue(FieldName,Value)
        rows.updateRow(row)
    if row:
        del row
    del rows
#获得一个图层所有的节点数
def getAllPointCount(fc):
    g = arcpy.Geometry()
    geometryList = arcpy.CopyFeatures_management(fc, g)
    n = 0
    for geometry in geometryList:
        n += geometry.pointCount
    return n