{-# LANGUAGE FlexibleContexts #-} {-# LANGUAGE MultiParamTypeClasses #-} ----------------------------------------------------------------------------- -- | -- Module : Diagrams.TwoD.Factorization -- Copyright : (c) 2012 Brent Yorgey -- License : BSD-style (see LICENSE) -- Maintainer : byorgey@cis.upenn.edu -- -- Factorization diagrams, as seen at -- -- and -- -- and on the cover of Hacker Monthly -- (): visually represent the -- prime factorization of n by drawing n dots recursively grouped -- according to the factors. -- -- <> -- ----------------------------------------------------------------------------- module Diagrams.TwoD.Factorization where import Data.Char (digitToInt) import Data.List.Split (chunksOf) import Diagrams.Prelude import Math.NumberTheory.Primes.Factorisation (factorise) -- | @primeLayout@ takes a positive integer p (the idea is for it to -- be prime, though it doesn't really matter) and a diagram, and lays -- out p rotated copies of the diagram in a circular pattern. -- -- There is a special case for @p = 2@: if the given diagram is taller -- than it is wide, then the two copies will be placed beside each -- other; if wider then tall, they will be placed one above the -- other. -- -- The regular @p@-gon connecting the centers of the laid-out -- diagrams is also filled in with vertical bars of color -- representing the number @p@. In particular, there is one color -- for each decimal digit (the provided list should have length 10 -- and represents the digits 0-9), and the colors, read left to -- right, give the decimal expansion of @p@. -- -- > import Diagrams.TwoD.Factorization -- > plExample -- > = pad 1.1 . centerXY -- > . hcat' (with & sep .~ 0.5) -- > . map (sized (Width 1)) -- > $ [ primeLayout defaultColors 5 (circle 1 # fc black) -- > , primeLayout defaultColors 103 (square 1 # fc green # lw 0) -- > , primeLayout (repeat white) 13 (circle 1 # lc orange) -- > ] -- -- <> primeLayout :: (Backend b R2, Renderable (Path R2) b) => [Colour Double] -> Integer -> Diagram b R2 -> Diagram b R2 primeLayout _ 2 d | width d >= height d = (d === strutY (height d / 3) === d # reflectY) # centerY | otherwise = (d ||| strutX (width d / 3) ||| d) # centerX primeLayout colors p d = (mconcat $ map (\n -> d # translateY r # rotateBy (fromIntegral n/fromIntegral p)) [0..p-1] ) <> colorBars colors p poly where poly = polygon (with & polyType .~ PolyRegular (fromIntegral p) r & polyOrient .~ OrientH ) w = max (width d) (height d) r = w * c / sin (tau / (2 * fromIntegral p)) c = 0.75 -- | Draw vertical bars of color inside a polygon which represent the -- decimal expansion of @p@, using the provided list of colors to -- represent the digits 0-9. -- -- > import Diagrams.TwoD.Factorization -- > colorBarsEx = colorBars defaultColors 3526 (square 1) -- -- <> colorBars :: (Renderable (Path R2) b) => [Colour Double] -> Integer -> Path R2 -> Diagram b R2 colorBars colors p poly | p <= 11 = stroke poly # fc (colors!!(fromIntegral p `mod` 10)) # lw 0 colorBars colors p poly = bars # clipBy poly where barColors = map ((colors!!) . digitToInt) (show p) barW = width poly / fromIntegral (length barColors) barH = height poly bars = (hcat $ map (\c -> rect barW barH # fc c # lc c) barColors) # centerX -- | A default set of digit colors, based very loosely on the color -- code for resistors (), -- lightened up a bit by blending with white. -- -- <> defaultColors :: [Colour Double] defaultColors = map (blend 0.1 white) [black,red,orange,yellow,green,blue,gray,purple,white,brown] -- > import Diagrams.TwoD.Factorization -- > showDefaultColors = hcat $ zipWith showColor defaultColors [0..] -- > where -- > showColor c d = text (show d) <> square 1 # fc c # lw 0 -- | Create a centered factorization diagram from the given list of -- factors (intended to be primes, but again, any positive integers -- will do; note how the below example uses 6), by recursively -- folding according to 'primeLayout', with the 'defaultColors' and -- a base case of a black circle. -- -- > import Diagrams.TwoD.Factorization -- > factorDiagram'Ex = factorDiagram' [2,5,6] -- -- <> factorDiagram' :: (Backend b R2, Renderable (Path R2) b) => [Integer] -> Diagram b R2 factorDiagram' = centerXY . foldr (primeLayout defaultColors) (circle 1 # fc black) -- | Create a default factorization diagram for the given integer, by -- factoring it and calling 'factorDiagram'' on its prime -- factorization (with the factors ordered from smallest to -- biggest). -- -- > import Diagrams.TwoD.Factorization -- > factorDiagramEx = factorDiagram 700 -- -- <> factorDiagram :: (Backend b R2, Renderable (Path R2) b) => Integer -> Diagram b R2 factorDiagram = factorDiagram' . concatMap (uncurry $ flip replicate) . factorise -- | Place a diagram inside a square with the given side length, -- centering and scaling it to fit with a bit of padding. -- -- > import Diagrams.TwoD.Factorization -- > ensquareEx = ensquare 1 (circle 25) ||| ensquare 1 (factorDiagram 30) -- -- <> ensquare :: (Backend b R2, Renderable (Path R2) b) => Double -> Diagram b R2 -> Diagram b R2 ensquare n d = d # centerXY # sized (Dims (0.8*n) (0.8*n)) <> square n -- | @fdGrid n@ creates a grid of factorization diagrams, given a list -- of lists of integers: the inner lists represent L-R rows, which -- are laid out from top to bottom. -- -- > import Diagrams.TwoD.Factorization -- > fdGridEx = fdGrid [[7,6,5],[4,19,200],[1,10,50]] -- -- <> fdGrid :: (Renderable (Path R2) b, Backend b R2) => [[Integer]] -> Diagram b R2 fdGrid = vcat . map hcat . (map . map) (ensquare 1 . factorDiagram) -- | @fdGridList n@ creates a grid containing the factorization -- diagrams of all the numbers from @1@ to @n^2@, ordered left to -- right, top to bottom (like the grid seen on the cover of Hacker -- Monthly, ). -- -- > import Diagrams.TwoD.Factorization -- > grid100 = fdGridList 10 -- > grid100Big = grid100 -- -- <> fdGridList :: (Renderable (Path R2) b, Backend b R2) => Integer -> Diagram b R2 fdGridList n = fdGrid . chunksOf (fromIntegral n) $ [1..n*n] -- | @fdTable n@ creates a \"multiplication table\" of factorization -- diagrams, with the diagrams for @1@ to @n@ along both the top row -- and left column, and the diagram for @m*n@ in row @m@ and column -- @n@. -- -- > import Diagrams.TwoD.Factorization -- > fdMultTableEx = fdMultTable 13 -- -- <> fdMultTable :: (Renderable (Path R2) b, Backend b R2) => Integer -> Diagram b R2 fdMultTable n = fdGrid $ [ [r*c | c <- [1 .. n]] | r <- [1 .. n] ]