Add star generation program

This commit is contained in:
Simon Bruder 2022-12-27 16:12:00 +01:00
parent 29d4b0a4a0
commit 002b834449
Signed by: simon
GPG key ID: 8D3C82F9F309F8EC
4 changed files with 72 additions and 0 deletions

1
.envrc Normal file
View file

@ -0,0 +1 @@
use nix

1
.gitignore vendored Normal file
View file

@ -0,0 +1 @@
/genstar/doc

59
genstar/Main.hs Normal file
View file

@ -0,0 +1,59 @@
module Main where
import Control.Monad (join)
-- | One point of a path, consisting of its \(x\) and \(y\) coordinates.
newtype Vertex = Point (Double, Double)
instance Show Vertex where
show (Point (x, y)) = show x ++ " " ++ show y
-- | 'fromAngle' @angle@
-- creats a point on the unit circle at the specified angle
-- (specified in radians).
fromAngle angle = Point (sin angle, cos angle)
-- | 'scale' @s@ @p@ scales the vector to @p@ by the factor @s@.
scale :: Integer -> Vertex -> Vertex
scale s (Point (x, y)) = Point (fromInteger s * x, fromInteger s * y)
-- | 'move' @p1@ @p2@ adds the coordinates of @p1@ and @p2@,
-- effectively moving @p1@ by @p2@ (or vice-versa).
move :: Vertex -> Vertex -> Vertex
move (Point (x1, y1)) (Point (x2, y2)) = Point (x1 + x2, y1 + y2)
-- | Points \(p_0, p_2, \dots, p_{n-1}\)
-- that should be connected,
-- so that \(p_{i \bmod n}\) is connected to \(p_{i+1 \bmod n}\).
newtype PolygonPath = PolygonPath [Vertex]
instance Show PolygonPath where
show (PolygonPath (v : vs)) = "<path d=\"M " ++ show v ++ join (map (\v -> " L " ++ show v) vs) ++ " Z\"/>\n"
show (PolygonPath []) = ""
-- | 'regularPolygonMod' @n@ @rf@ @pos@
-- creates a regular @n@-gon
-- with the center at @pos@
-- that is modified in the sense that it takes @rf@,
-- which is used to set the radius for each individual point.
regularPolygonMod :: Integer -> (Integer -> Integer) -> Vertex -> PolygonPath
regularPolygonMod n rf pos = PolygonPath (map (\step -> move pos $ scale (rf step) $ fromAngle (fromInteger step * 2 * pi / fromInteger n)) [0 .. (n - 1)])
-- | 'regularPolygon' @n@ @r@ @pos@
-- creates a regular polygon
-- with the radius @r@
-- and the center at @pos@.
regularPolygon :: Integer -> Integer -> Vertex -> PolygonPath
regularPolygon n r = regularPolygonMod n (const r)
-- | 'star' @n@ @r1@ @r2@ @pos@
-- creates a star
-- with @n@ spikes,
-- the inner radius @r1@,
-- the outer radius @r2@
-- and the center @pos@.
star :: Integer -> Integer -> Integer -> Vertex -> PolygonPath
star n r1 r2 = regularPolygonMod (n * 2) ((take (fromInteger n * 2) (cycle [r1, r2]) !!) . fromInteger)
main :: IO ()
main = putStr $ join $ map (\n -> show n ++ "-gon: " ++ show (regularPolygon n 100 (Point (100, 100)))) [5, 8, 10, 15, 18, 30]

11
shell.nix Normal file
View file

@ -0,0 +1,11 @@
{ pkgs ? import <nixpkgs> { } }:
pkgs.mkShell {
name = "emi5";
nativeBuildInputs = with pkgs; [
(haskellPackages.ghcWithPackages (pkgs: with pkgs; [
QuickCheck
]))
];
}