python 点边界处理

最新推荐文章于 2022-09-29 22:24:27 发布

qiu_zhongya

最新推荐文章于 2022-09-29 22:24:27 发布

阅读量2k

点赞数

本文链接：https://blog.csdn.net/qiu_zhongya/article/details/78266093

版权

#encoding=utf-8

import json
import math
class Point(object):
	def __init__(self, x, z):
		self.x = x
		self.z = z
	def ToString(self):
		return "{ x:"+ str(self.x) + " z:" + str(self.z) + "}"
	def Equal(self, point):
		return self.x == point.x and self.z == point.z

class Line(object):
	def __init__(self, point1, point2):
		self.v0 = point1
		self.v1 = point2
		self.status = False
		if self.v0.x == self.v1.x and self.v0.z != self.v1.z:
			self.direction = 1
		elif self.v0.x != self.v1.x and self.v0.z == self.v1.z:
			self.direction = 2
		else:
			self.direction = 3
	def Equal(self, Line):
		return self.v0.Equal(Line.v1) and self.v1.Equal(Line.v0) or self.v0.Equal(Line.v0) and self.v1.Equal(Line.v1)
	def Length(self):
		return math.sqrt(pow(self.v0.x - self.v1.x, 2) + pow(self.v0.z - self.v1.z, 2))
	def ToString(self):
		return "{ v0 :"+ self.v0.ToString() + ", v1 :" + self.v1.ToString() +"}"
	def GetLengthDir(self):
		return " len :"+ str(self.Length()) + ", direction :" + str(self.direction)

class EdgeHandle(object):
	def __init__(self):
		self.msgId = 0
		self.edges = None
		self.edgeList = None
		self.lineList = []
	def loadJson(self):
		f = open("edges.json", "r")
		self.edges = json.load(f)
		self.edgeList = self.edges["edge"]
		for i in self.edgeList:
			point1 = Point(i["v0"]["x"], i["v0"]["z"])
			point2 = Point(i["v1"]["x"], i["v1"]["z"])
			lineObj = Line(point1, point2)
			if lineObj.direction == 3:
				print "exception line ***********" + lineObj.ToString()
			self.lineList.append(lineObj)
	def ToString(self):
		for i in self.lineList:
			print i.ToString()
	
	def handleEdge(self, ):
		listLen = len(self.lineList)
		join = False
		newlineList = []
		for i in self.lineList:
			for j in self.lineList:
				if i.Equal(j):
					continue
				if i.status or j.status:
					continue
				(canjoin , newline) = self.canjoinLine(i,j)
				if canjoin:
					join = True
					newlineList.append(newline)
					i.status = True
					j.status = True
					#print i.ToString(), j.ToString(), newline.ToString()
		self.lineList =  [i for i in self.lineList if i.status == False]
		self.lineList = self.lineList + newlineList
		return join

	def canjoinLine(self, line1, line2):
		if line1.direction != line2.direction:
			return False, line1
		elif line1.v0.Equal(line2.v0):
			return True, Line(line1.v1 , line2.v1)
		elif line1.v1.Equal(line2.v0):
			return True, Line(line1.v0 , line2.v1)
		elif line1.v0.Equal(line2.v1):
			return True, Line(line1.v1 , line2.v0)
		elif line1.v1.Equal(line2.v1):
			return True, Line(line1.v0 , line2.v0)
		else:
			return False, line1
edgeHandle = EdgeHandle()
edgeHandle.loadJson()
handleTime = 0
while edgeHandle.handleEdge(): 
	handleTime = handleTime +1
	print "******************************************************handleTime: " , handleTime
edgeHandle.ToString()
xlen = 0
zlen = 0
for i in edgeHandle.lineList:
	print i.GetLengthDir()