#!/usr/bin/env python

header        = 'De Bruijn torus (16, 32; 3, 3) with 1 as panels and 0 as holes by CMG Lee'
(r_row,r_col) = (300,600)
rot_col       = 270
f             = 0.01
datass        = [[int(char) for char in line] for line in '''
00000010111001011100101110010111
00010011111101001101101010000110
00000010111001011100101110010111
11101100000010110010010101111001
00000010111001011100101110010111
00010011111101001101101010000110
00110001110101101111100010100100
10001010011011010100001100011111
11111101000110100011010001101000
11101100000010110010010101111001
10101000010011110110000100111101
01000110101000011000111111010011
11111101000110100011010001101000
11101100000010110010010101111001
01100100100000111010110111110001
11011111001110000001011001001010
'''.strip().split()]
(n_row,n_col) = (len(datass),len(datass[0]))

import math, struct
def tabbify(cellss, separator='|'):
 cellpadss = [list(rows) + [''] * (len(max(cellss, key=len)) - len(rows)) for rows in cellss]
 fmts = ['%%%ds' % (max([len(str(cell)) for cell in cols])) for cols in zip(*cellpadss)]
 return '\n'.join([separator.join(fmts) % tuple(rows) for rows in cellpadss])
def hex_rgb(colour): ## convert [#]RGB to #RRGGBB and [#]RRGGBB to #RRGGBB
 return '#%s' % (colour if len(colour) > 4 else ''.join([c * 2 for c in colour])).lstrip('#')
def viscam_colour(colour):
 colour_hex      = hex_rgb(colour)
 colour_top5bits = [int(colour_hex[i:i+2], 16) >> 3 for i in range(1,7,2)]
 return (1 << 15) + (colour_top5bits[0] << 10) + (colour_top5bits[1] << 5) + colour_top5bits[2]

## Find vertices
xyzsss = []
for  i_row in range(n_row):
 xyzss = []
 for i_col in range(n_col):
  theta_row = 2 * math.pi * i_row / n_row
  theta_col = 2 * math.pi * i_col / n_col
  r_row_sin = r_row * math.sin(theta_row)
  x = (r_col + r_row_sin) * math.cos(theta_col + math.radians(rot_col))
  y = (r_col + r_row_sin) * math.sin(theta_col + math.radians(rot_col))
  z = r_row * math.cos(theta_row)
  xyzss.append([x,y,z])
 xyzsss.append(xyzss)

## Find facets
facetss = []
for  i_row in range(n_row):
 for i_col in range(n_col):
  xyz_currents      = xyzsss[(i_row    ) % n_row][(i_col    ) % n_col]
  xyz_next_rows     = xyzsss[(i_row + 1) % n_row][(i_col    ) % n_col]
  xyz_next_cols     = xyzsss[(i_row    ) % n_row][(i_col + 1) % n_col]
  xyz_next_row_cols = xyzsss[(i_row + 1) % n_row][(i_col + 1) % n_col]
  xyz_inters        = [f * xyz_currents[i] + (1 - f) * xyz_next_row_cols[i] for i in range(3)]
  if datass[i_row][i_col] == 0: xyz_currents      = xyz_inters
  # else:                         xyz_next_row_cols = xyz_inters
  facetss.append([None,None,None] + xyz_currents + xyz_next_cols + xyz_next_row_cols)
  facetss.append([None,None,None] + xyz_currents + xyz_next_row_cols + xyz_next_rows)

## Calculate normals for facets with unknown normals
for facets in facetss:
 if (facets[0] is None or facets[1] is None or facets[2] is None):
  us      = [facets[i_xyz + 9] - facets[i_xyz + 6] for i_xyz in range(3)]
  vs      = [facets[i_xyz + 6] - facets[i_xyz + 3] for i_xyz in range(3)]
  normals = [us[1]*vs[2] - us[2]*vs[1], us[2]*vs[0] - us[0]*vs[2], us[0]*vs[1] - us[1]*vs[0]]
  normal_length = sum([component * component for component in normals]) ** 0.5
  facets[:3]    = [-round(component / normal_length, 10) for component in normals]
# print(tabbify([['N%s'  % (xyz   )                   for xyz in list('xyz')] +
#                ['%s%d' % (xyz, n) for n in range(3) for xyz in list('XYZ')] + ['RGB']] + facetss))

## Compile STL
outss = ([[('STL\n\n%-73s\n\n' % (header[:73])).encode('utf-8'), struct.pack('<L',len(facetss))]] +
         [[struct.pack('<f',float(value)) for value in facets[:12]] +
          [struct.pack('<H',0 if (len(facets) <= 12) else
                            viscam_colour(facets[12]))] for facets in facetss])
out   = b''.join([bytes(out) for outs in outss for out in outs])
# out += ('\n\n## Python script to generate STL\n\n%s\n' % (open(__file__).read())).encode('utf-8')
print("# bytes:%d\t# facets:%d\ttitle:\"%-73s\"" % (len(out), len(facetss), header[:73]))
with open(__file__[:__file__.rfind('.')] + '.stl', 'wb') as f_out: f_out.write(out)