def GenKparentSeq(Octree,K): LevelNum =len(Octree) nodeNum = Octree[-1].node[-1].nodeid Seq = np.ones((nodeNum,K),'int')*255 LevelOctant = np.zeros((nodeNum,K,2),'int') # Level and Octant Pos = np.zeros((nodeNum,K,3),'int'); #padding 0 ChildID = [[] for _ in range(nodeNum)] Seq[0,K-1] = Octree[0].node[0].oct LevelOctant[0,K-1,0] = 1 LevelOctant[0,K-1,1] = 1 Pos[0,K-1,:] = Octree[0].node[0].pos Octree[0].node[0].parent = 1 # set to 1 n= 0 for L in range(0,LevelNum): for node in Octree[L].node: Seq[n,K-1] = node.oct Seq[n,0:K-1] = Seq[node.parent-1,1:K] LevelOctant[n,K-1,:] = [L+1,node.octant] LevelOctant[n,0:K-1] = LevelOctant[node.parent-1,1:K,:] Pos[n,K-1] = node.pos Pos[n,0:K-1,:] = Pos[node.parent-1,1:K,:] if (L==LevelNum-1): pass n+=1 assert n==nodeNum DataStruct = {'Seq':Seq,'Level':LevelOctant,'ChildID':ChildID,'Pos':Pos} return DataStruct

这是一个生成K父节点序列的函数GenKparentSeq。

参数Octree是一个八叉树的节点列表，K是指定的父节点序列的长度。

首先，计算Octree的层数LevelNum，并获取最后一个节点的节点ID。

然后，创建Seq数组，大小为(nodeNum, K)，填充为255。Seq数组用于存储父节点序列。

接下来，创建LevelOctant数组，大小为(nodeNum, K, 2)，填充为0。LevelOctant数组用于存储每个节点的层级和八叉树的八个子节点。

然后，创建Pos数组，大小为(nodeNum, K, 3)，填充为0。Pos数组用于存储每个节点的位置信息。

接下来，创建ChildID列表，长度为nodeNum，用于存储每个节点的子节点ID。

然后，设置根节点的Seq、LevelOctant、Pos的初始值。根节点的父节点设置为1。

接下来，使用两层循环遍历Octree中的每个节点。在循环中，将每个节点的信息复制到Seq、LevelOctant、Pos中的相应位置。同时更新父节点的信息。

最后，使用assert语句检查生成的节点数量是否与预期的nodeNum相等。

将Seq、LevelOctant、ChildID和Pos组合成一个字典DataStruct，并返回该字典作为结果。

该函数返回一个包含生成的父节点序列和相关数据的字典。字典中包含Seq、LevelOctant、ChildID和Pos四个键，分别对应父节点序列、层级和八叉树的子节点、子节点ID和节点位置信息。