import Control.Monad (forM,forM_,(<=<),unless,replicateM)
import Data.Array.IO (IOUArray,newArray,newArray_,readArray,writeArray,freeze)
import Data.Array.Unboxed (listArray,Array,UArray,(!))
import Data.Int (Int64)
import Data.IORef (IORef,newIORef,readIORef,writeIORef,modifyIORef)
import Data.Ix

fooArray = newArray_ (0,3) :: IO (IOUArray Int Int64)

main = do
	let len = 3
	a <- newArray_ (0, len) :: IO (IOUArray Int Int64)
	putStrLn "print a"
	forM_ [0..len] (print <=< readArray a)
	forM_ [0..len] (\i -> writeArray a i (fromIntegral $ (i+1)*2))
	putStrLn "print updated a"
	forM_ [0..len] $ \i -> do
		v <- readArray a i
		print v
	
	rx <- newIORef (123 :: Int64)
	x <- readIORef rx -- copy value, not reference
	putStr "x "
	print x -- 123
	writeIORef rx 999
	putStr "x "
	print x -- 123
	x <- readIORef rx
	putStr "x "
	print x -- 999

	foo <- fooArray
	putStrLn "print foo"
	print =<< forM [0..len] (readArray foo)
	forM_ [0..len] (\i -> writeArray foo i (fromIntegral $ (i+1)*2))
	putStrLn "print updated foo"
	print =<< forM [0..len] (readArray foo)
	bar <- fooArray
	putStrLn "print bar"
	print =<< forM [0..len] (readArray bar)
	
	(primes, isPrime) <- primeTool (10^8)
	print (take 30 primes)
	print (filter isPrime [10^7..10^7+300])
	
	arr <- genIntArray 10 0
	forM_ [0..9] $ \i -> writeIORef (arr!i) (i*(i+1)`div`2)
	print =<< forM [0..9] (readIORef.(arr!))
	
	table <- genMatrix 5 5 0
	forM_ [0..4] $ \i -> do
		forM_ [0..4] $ \j -> do
			writeIORef (table!i!j) (i*100+j)
	print =<< forM [0..4] (\i -> forM [0..4] (readIORef.(table!i!)))



genArray :: (Integral a,Ix a) => a -> IO b -> IO (Array a b)
genArray n initf = listArray (0,n-1) <$> replicateM (fromIntegral n) initf

genIntArray :: (Integral a, Ix a) => Integral b => a -> b -> IO (Array a (IORef b))
genIntArray n initv = genArray n (newIORef initv)

genMatrix :: (Integral a, Ix a) => a -> a -> b -> IO (Array a (Array a (IORef b)))
genMatrix n m initv = genArray n (genArray m (newIORef initv))



primeTool n = do
    table <- newArray (0,n+6) False :: IO (IOUArray Int64 Bool)
    primes <- newIORef [3,2]
    forM_ [1..((n+5) `div` 6)] $ \i -> do
        let x = 6*i - 1
        r <- readArray table x
        unless r $ do
            modifyIORef primes (x:)
            forM_ [3,5..((n+5) `div` x)] $ \k -> do
                writeArray table (k*x) True
        let y = x + 2
        q <- readArray table y
        unless q $ do
            modifyIORef primes (y:)
            forM_ [3,5..((n+5) `div` y)] $ \k -> do
                writeArray table (k*y) True
        writeArray table (y+2) True
    
    writeArray table 1 True
    table <- freeze table :: IO (UArray Int64 Bool)
    primes <- reverse <$> readIORef primes
    
    let isPrime v | even v = v == 2
                  | v < n  = not (table!v)
                  | otherwise = all ((/=0).mod v) $ takeWhile ((<=v).(^2)) primes

    return (primes, isPrime)

aW1wb3J0IENvbnRyb2wuTW9uYWQgKGZvck0sZm9yTV8sKDw9PCksdW5sZXNzLHJlcGxpY2F0ZU0pCmltcG9ydCBEYXRhLkFycmF5LklPIChJT1VBcnJheSxuZXdBcnJheSxuZXdBcnJheV8scmVhZEFycmF5LHdyaXRlQXJyYXksZnJlZXplKQppbXBvcnQgRGF0YS5BcnJheS5VbmJveGVkIChsaXN0QXJyYXksQXJyYXksVUFycmF5LCghKSkKaW1wb3J0IERhdGEuSW50IChJbnQ2NCkKaW1wb3J0IERhdGEuSU9SZWYgKElPUmVmLG5ld0lPUmVmLHJlYWRJT1JlZix3cml0ZUlPUmVmLG1vZGlmeUlPUmVmKQppbXBvcnQgRGF0YS5JeAoKZm9vQXJyYXkgPSBuZXdBcnJheV8gKDAsMykgOjogSU8gKElPVUFycmF5IEludCBJbnQ2NCkKCm1haW4gPSBkbwoJbGV0IGxlbiA9IDMKCWEgPC0gbmV3QXJyYXlfICgwLCBsZW4pIDo6IElPIChJT1VBcnJheSBJbnQgSW50NjQpCglwdXRTdHJMbiAicHJpbnQgYSIKCWZvck1fIFswLi5sZW5dIChwcmludCA8PTwgcmVhZEFycmF5IGEpCglmb3JNXyBbMC4ubGVuXSAoXGkgLT4gd3JpdGVBcnJheSBhIGkgKGZyb21JbnRlZ3JhbCAkIChpKzEpKjIpKQoJcHV0U3RyTG4gInByaW50IHVwZGF0ZWQgYSIKCWZvck1fIFswLi5sZW5dICQgXGkgLT4gZG8KCQl2IDwtIHJlYWRBcnJheSBhIGkKCQlwcmludCB2CgkKCXJ4IDwtIG5ld0lPUmVmICgxMjMgOjogSW50NjQpCgl4IDwtIHJlYWRJT1JlZiByeCAtLSBjb3B5IHZhbHVlLCBub3QgcmVmZXJlbmNlCglwdXRTdHIgInggIgoJcHJpbnQgeCAtLSAxMjMKCXdyaXRlSU9SZWYgcnggOTk5CglwdXRTdHIgInggIgoJcHJpbnQgeCAtLSAxMjMKCXggPC0gcmVhZElPUmVmIHJ4CglwdXRTdHIgInggIgoJcHJpbnQgeCAtLSA5OTkKCglmb28gPC0gZm9vQXJyYXkKCXB1dFN0ckxuICJwcmludCBmb28iCglwcmludCA9PDwgZm9yTSBbMC4ubGVuXSAocmVhZEFycmF5IGZvbykKCWZvck1fIFswLi5sZW5dIChcaSAtPiB3cml0ZUFycmF5IGZvbyBpIChmcm9tSW50ZWdyYWwgJCAoaSsxKSoyKSkKCXB1dFN0ckxuICJwcmludCB1cGRhdGVkIGZvbyIKCXByaW50ID08PCBmb3JNIFswLi5sZW5dIChyZWFkQXJyYXkgZm9vKQoJYmFyIDwtIGZvb0FycmF5CglwdXRTdHJMbiAicHJpbnQgYmFyIgoJcHJpbnQgPTw8IGZvck0gWzAuLmxlbl0gKHJlYWRBcnJheSBiYXIpCgkKCShwcmltZXMsIGlzUHJpbWUpIDwtIHByaW1lVG9vbCAoMTBeOCkKCXByaW50ICh0YWtlIDMwIHByaW1lcykKCXByaW50IChmaWx0ZXIgaXNQcmltZSBbMTBeNy4uMTBeNyszMDBdKQoJCglhcnIgPC0gZ2VuSW50QXJyYXkgMTAgMAoJZm9yTV8gWzAuLjldICQgXGkgLT4gd3JpdGVJT1JlZiAoYXJyIWkpIChpKihpKzEpYGRpdmAyKQoJcHJpbnQgPTw8IGZvck0gWzAuLjldIChyZWFkSU9SZWYuKGFyciEpKQoJCgl0YWJsZSA8LSBnZW5NYXRyaXggNSA1IDAKCWZvck1fIFswLi40XSAkIFxpIC0+IGRvCgkJZm9yTV8gWzAuLjRdICQgXGogLT4gZG8KCQkJd3JpdGVJT1JlZiAodGFibGUhaSFqKSAoaSoxMDAraikKCXByaW50ID08PCBmb3JNIFswLi40XSAoXGkgLT4gZm9yTSBbMC4uNF0gKHJlYWRJT1JlZi4odGFibGUhaSEpKSkKCgoKZ2VuQXJyYXkgOjogKEludGVncmFsIGEsSXggYSkgPT4gYSAtPiBJTyBiIC0+IElPIChBcnJheSBhIGIpCmdlbkFycmF5IG4gaW5pdGYgPSBsaXN0QXJyYXkgKDAsbi0xKSA8JD4gcmVwbGljYXRlTSAoZnJvbUludGVncmFsIG4pIGluaXRmCgpnZW5JbnRBcnJheSA6OiAoSW50ZWdyYWwgYSwgSXggYSkgPT4gSW50ZWdyYWwgYiA9PiBhIC0+IGIgLT4gSU8gKEFycmF5IGEgKElPUmVmIGIpKQpnZW5JbnRBcnJheSBuIGluaXR2ID0gZ2VuQXJyYXkgbiAobmV3SU9SZWYgaW5pdHYpCgpnZW5NYXRyaXggOjogKEludGVncmFsIGEsIEl4IGEpID0+IGEgLT4gYSAtPiBiIC0+IElPIChBcnJheSBhIChBcnJheSBhIChJT1JlZiBiKSkpCmdlbk1hdHJpeCBuIG0gaW5pdHYgPSBnZW5BcnJheSBuIChnZW5BcnJheSBtIChuZXdJT1JlZiBpbml0dikpCgoKCnByaW1lVG9vbCBuID0gZG8KICAgIHRhYmxlIDwtIG5ld0FycmF5ICgwLG4rNikgRmFsc2UgOjogSU8gKElPVUFycmF5IEludDY0IEJvb2wpCiAgICBwcmltZXMgPC0gbmV3SU9SZWYgWzMsMl0KICAgIGZvck1fIFsxLi4oKG4rNSkgYGRpdmAgNildICQgXGkgLT4gZG8KICAgICAgICBsZXQgeCA9IDYqaSAtIDEKICAgICAgICByIDwtIHJlYWRBcnJheSB0YWJsZSB4CiAgICAgICAgdW5sZXNzIHIgJCBkbwogICAgICAgICAgICBtb2RpZnlJT1JlZiBwcmltZXMgKHg6KQogICAgICAgICAgICBmb3JNXyBbMyw1Li4oKG4rNSkgYGRpdmAgeCldICQgXGsgLT4gZG8KICAgICAgICAgICAgICAgIHdyaXRlQXJyYXkgdGFibGUgKGsqeCkgVHJ1ZQogICAgICAgIGxldCB5ID0geCArIDIKICAgICAgICBxIDwtIHJlYWRBcnJheSB0YWJsZSB5CiAgICAgICAgdW5sZXNzIHEgJCBkbwogICAgICAgICAgICBtb2RpZnlJT1JlZiBwcmltZXMgKHk6KQogICAgICAgICAgICBmb3JNXyBbMyw1Li4oKG4rNSkgYGRpdmAgeSldICQgXGsgLT4gZG8KICAgICAgICAgICAgICAgIHdyaXRlQXJyYXkgdGFibGUgKGsqeSkgVHJ1ZQogICAgICAgIHdyaXRlQXJyYXkgdGFibGUgKHkrMikgVHJ1ZQogICAgCiAgICB3cml0ZUFycmF5IHRhYmxlIDEgVHJ1ZQogICAgdGFibGUgPC0gZnJlZXplIHRhYmxlIDo6IElPIChVQXJyYXkgSW50NjQgQm9vbCkKICAgIHByaW1lcyA8LSByZXZlcnNlIDwkPiByZWFkSU9SZWYgcHJpbWVzCiAgICAKICAgIGxldCBpc1ByaW1lIHYgfCBldmVuIHYgPSB2ID09IDIKICAgICAgICAgICAgICAgICAgfCB2IDwgbiAgPSBub3QgKHRhYmxlIXYpCiAgICAgICAgICAgICAgICAgIHwgb3RoZXJ3aXNlID0gYWxsICgoLz0wKS5tb2QgdikgJCB0YWtlV2hpbGUgKCg8PXYpLiheMikpIHByaW1lcwoKICAgIHJldHVybiAocHJpbWVzLCBpc1ByaW1lKQoK