这是一个实现,summarize :: [ComplexCampus] -> ComplexCampus
它使用带有 Uniplate 的 Lenses(如您所提到的)来汇总 ComplexCampus 列表和单个 ComplexCampus。
{-# Language TemplateHaskell,DeriveDataTypeable #-}
import Control.Lens
import Data.Data.Lens
import Data.Typeable
import Data.Data
import Data.List(transpose,genericLength)
data Location = Location { _x :: Double, _y :: Double } deriving(Show,Typeable,Data)
data CampusLocation = CampusLocation { _firstBuilding :: Location, _firsecondBuilding :: Location }deriving(Show,Typeable,Data)
data ComplexCampus = ComplexCampus { _buildings :: [Location] } deriving(Show,Typeable,Data)
makeLenses ''Location
makeLenses ''CampusLocation
makeLenses ''ComplexCampus
l1 = Location 1 10
l2 = Location 2 20
l3 = Location 3 30
c1 = CampusLocation l1 l2
c2 = CampusLocation l2 l3
c3 = CampusLocation l1 l3
campusLocs = [c1,c2,c3]
c1' = ComplexCampus [l1, l2]
c2' = ComplexCampus [l2, l3]
c3' = ComplexCampus [l1, l3]
campusLocs' = [c1',c2',c3']
average l = (sum l) / (genericLength l)
-- returns average location for a list of locations
averageLoc locs = Location {
_x = average $ locs ^.. biplate . x,
_y = average $ locs ^.. biplate . y
}
summarize :: [ComplexCampus] -> ComplexCampus
summarize ccs = ComplexCampus $ ccs ^.. biplate . buildings ^.. folding transpose . to averageLoc
在这里使用 biplate 可能有点矫枉过正,但无论averageLoc
我们biplate
在位置列表上使用来获取所有x
字段和所有y
字段。如果您想将 a 汇总ComplexCampus
为一个Location
,我们可以使用它来从顶层biplate
提取所有x
值和所有值。y
ComplexBuilding
例如:
campusLocs' ^.. biplate . x
给我们所有的 x 值并campusLocs' ^.. biplate . y
给我们所有的 y 值
同样,要获取所有位置,我们可以这样做:
(campusLocs' ^.. biplate) ::[Location]
或者,如果我们想要每个 Double:
(campusLocs' ^.. biplate) ::[Double]