# -*- Mode: Python -*-

# generate a PBM image where an 'on' pixel indicates a prime.

import string

def primes_gen (primes_file):
    f = open (primes_file, 'rb')
    # skip the first two lines
    f.readline()
    f.readline()
    while 1:
        line = f.readline()
        if not line:
            break
        line.strip()
        nums = [int (x) for x in line.split()]
        for num in nums:
            #print num
            yield num

def emit_byte (byte, file):
    if len(byte) != 8:
        # pad byte if necessary
        byte += ['0'] * (8-len(byte))
    as_bin = ''.join (byte)
    as_int = string.atoi (as_bin, 2)
    file.write (chr (as_int))

def emit_pbm (pg, w=2560, h=1600):
    f = open ('primes_%d.pbm' % (w,), 'wb')
    f.write ('P4 %d %d\n' % (w,h))
    byte = []
    next = pg.next()
    n = 0
    for y in xrange (h):
        for x in xrange (w):
            if n == next:
                bit = '1'
                next = pg.next()
            else:
                bit = '0'
            byte.append (bit)
            if len(byte) == 8:
                emit_byte (byte, f)
                byte = []
            n += 1
        # pad end of row if necessary
        if len(byte):
            emit_byte (byte, f)
            byte = []
        #f.write ('\n')

    f.close()

# some prime widths

widths = [
    811,821,823,827,829,839,853,857,859,863,877,881,883,887,907,911,919,929,937,941,
    947,953,967,971,977,983,991,997,1009,1013,1019,1021,1031,1033,1039,1049,1051,1061,1063,1069,
    1087,1091,1093,1097,1103,1109,1117,1123,1129,1151,1153,1163,1171,1181,1187,1193,1201,1213,1217,1223,
    1229,1231,1237,1249,1259,1277,1279,1283,1289,1291,1297,1301,1303,1307,1319,1321,1327,1361,1367,1373,
    1381,1399,1409,1423,1427,1429,1433,1439,1447,1451,1453,1459,1471,1481,1483,1487,1489,1493,1499,1511,
    1523,1531,1543,1549,1553,1559,1567,1571,1579,1583,1597,1601,1607,1609,1613,1619,1621,1627,1637,1657,
    1663,1667,1669,1693,1697,1699,1709,1721,1723,1733,1741,1747,1753,1759,1777,1783,1787,1789,1801,1811,
    1823,1831,1847,1861,1867,1871,1873,1877,1879,1889,1901,1907,1913,1931,1933,1949,1951,1973,1979,1987,
    1993,1997,1999,2003,2011,2017,2027,2029,2039,2053,2063,2069,2081,2083,2087,2089,2099,2111,2113,2129,
    2131,2137,2141,2143,2153,2161,2179,2203,2207,2213,2221,2237,2239,2243,2251,2267,2269,2273,2281,2287,
    2293,2297,2309,2311,2333,2339,2341,2347,2351,2357,2371,2377,2381,2383,2389,2393,2399,2411,2417,2423,
    2437,2441,2447,2459,2467,2473,2477,2503
    ]

widths = range (2300, 2320)

widths = [900]

# 2309 is one of the more interesting ones...
# 2310 is even better (2 * 3 * 5 * 7 * 11)

if __name__ == '__main__':
    import os
    import sys
    for w in widths:
        print w,
        pg = primes_gen (sys.argv[1])
        emit_pbm (pg, w, 1400)
        os.system ('/opt/local/bin/pnmtopng primes_%d.pbm > primes_%d.png' % (w,w))
        os.system ('rm primes_%d.pbm' % (w,))