整理paperArtist代码

Author: MrZ626

assets/ai/paperArtist.lua

@@ -33,80 +33,48 @@ local simulateDrop=require'assets.ai.util'.simulateDrop
 local lockPiece=require'assets.ai.util'.lockPiece
 local clearLine=require'assets.ai.util'.clearLine
 local getBoundCount=paperArtist.getBoundCount
-local directions={'0','R','F','L'}
 
 ---@param F Mat<boolean> Field (read)
 ---@param shape Mat<boolean>
----@return number x, number y, string dir
-function paperArtist.findPosition(F,shape)
-    local best={
-        score=-1e99,
-        x=4,y=4,dir='0',
-    }
-    local w=F[1] and #F[1] or 10
-    for d=1,4 do
-        for cx=1,w-#shape[1]+1 do
-            local F1=TABLE.copy(F,1)
-            local cy,colH=simulateDrop(F1,shape,cx)
-
-            lockPiece(F1,shape,cx,cy)
-            local clear=clearLine(F1,cy,cy+#shape-1)
-            local score
-            if #F1==0 then
-                -- Nobody can refuse AC
-                score=1e99
-            else
-                local HBC,VBC=getBoundCount(F1)
-                score=clear*2-HBC*3-VBC*2-cy
-            end
-
-            local minH=math.min(unpack(colH))
-            for i=1,#colH do
-                score=score-(colH[i]-minH)*1.26
-            end
-
-            if score>best.score then
-                best.x,best.y,best.dir=cx,cy,directions[d]
-                best.score=score
-            end
-        end
-        if d<4 then shape=TABLE.rotate(shape,'R') end
+---@param X number
+function paperArtist.getSimScore(F,shape,X)
+    F=TABLE.copy(F,1)
+    local cy,colH=simulateDrop(F,shape,X)
+    lockPiece(F,shape,X,cy)
+    local linesClear=clearLine(F,cy,cy+#shape-1)
+    local score
+    if #F==0 then
+        -- Nobody can refuse AC
+        score=1e99
+    else
+        local HBC,VBC=getBoundCount(F)
+        local minH=math.min(unpack(colH))
+        local totalH=MATH.sum(colH)
+        local shadowCreated=totalH-minH*#colH
+        score=2*linesClear-cy-shadowCreated*1.26-HBC*3-VBC*2
     end
-    return best.x,best.y,best.dir
+    return score,cy
 end
 
+local rotate=TABLE.rotate
+local getSimScore=paperArtist.getSimScore
+local directions={'0','R','F','L'}
+
 local floor=math.floor
 local ins,rem=table.insert,table.remove
 
 ---@param F Mat<boolean> Field (read)
 ---@param shape Mat<boolean>
 ---@return {x:number, y:number, dir:string}[]
-function paperArtist.findBestPositions(F,shape,counts)
+function paperArtist.search(F,shape,dirCount,bestCount)
     local posList={{
         score=-1e99,
         x=4,y=4,dir='0',
     }}
     local w=F[1] and #F[1] or 10
-    for d=1,4 do
+    for d=1,dirCount do
         for cx=1,w-#shape[1]+1 do
-            local F1=TABLE.copy(F,1)
-            local cy,colH=simulateDrop(F1,shape,cx)
-
-            lockPiece(F1,shape,cx,cy)
-            local clear=clearLine(F1,cy,cy+#shape-1)
-            local score
-            if #F1==0 then
-                -- Nobody can refuse AC
-                score=1e99
-            else
-                local HBC,VBC=getBoundCount(F1)
-                score=clear*2-HBC*3-VBC*2-cy
-            end
-
-            local minH=math.min(unpack(colH))
-            for i=1,#colH do
-                score=score-(colH[i]-minH)*1.26
-            end
+            local score,cy=getSimScore(F,shape,cx)
 
             local pos={
                 score=score,
@@ -122,13 +90,21 @@ function paperArtist.findBestPositions(F,shape,counts)
                 end
             end
             ins(posList,i,pos)
-            if #posList>counts then
+            if #posList>bestCount then
                 rem(posList)
             end
         end
-        if d<4 then shape=TABLE.rotate(shape,'R') end
+        if d<4 then shape=rotate(shape,'R') end
     end
     return posList
 end
 
+---@param F Mat<boolean> Field (read)
+---@param shape Mat<boolean>
+---@return number x, number y, string dir
+function paperArtist.getBestPosition(F,shape)
+    local best=paperArtist.search(F,shape,4,1)[1]
+    return best.x,best.y,best.dir
+end
+
 return paperArtist

assets/ai/util.lua

@@ -5,7 +5,7 @@ local Util={}
 ---@param field Mat<boolean> Field (read)
 ---@param shape Mat<boolean>
 ---@param X number
----@return number Y, number[] delta
+---@return number Y, number[] colH the height of the piece if only check collision with this column
 function Util.simulateDrop(field,shape,X)
     local w=#shape[1]
     local shapeBottom,fieldTop={},{}
@@ -18,11 +18,11 @@ function Util.simulateDrop(field,shape,X)
         while y>0 and not field[y][X+x-1] do y=y-1 end
         fieldTop[x]=y
     end
-    local delta={}
+    local colH={}
     for i=1,w do
-        delta[i]=fieldTop[i]-shapeBottom[i]
+        colH[i]=fieldTop[i]-shapeBottom[i]
     end
-    return math.max(unpack(delta))+1,delta
+    return math.max(unpack(colH))+1,colH
 end
 
 ---@param field Mat<boolean> Field (write)