ldr2pov.py

#!/usr/bin/env python

"""
ldr2pov.py - An LDraw to POV-Ray convertor tool.

Copyright (C) 2009 David Boddie <david@boddie.org.uk>

This file is part of the ldraw Python package.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
"""
import argparse
import sys

from ldraw.tools import vector_position, get_model
from ldraw.writers.povray import POVRayWriter

SKY_SPHERE_FORMAT_STRING = """sky_sphere {
  pigment
  {
    rgb <%s>
  }
}
"""

CAMERA_FORMAT_STRING = """camera {
  location <%s>
  look_at <%s>
}
"""

LIGHT_FORMAT_STRING = """light_source {
  <%1.3f, %1.3f, %1.3f>, rgb <0.6, 0.6, 0.6>
}

light_source {
  <%1.3f, %1.3f, %1.3f>, rgb <0.6, 0.6, 0.6>
}

light_source {
  <%1.3f, %1.3f, %1.3f>, rgb <0.6, 0.6, 0.6>
}

light_source {
  <%1.3f, %1.3f, %1.3f>, rgb <0.6, 0.6, 0.6>
}

"""


def main():
    """ ldr2pov main function """

    description = """Converts the LDraw file to a POV-Ray file.
    
The camera position is a single x,y,z argument where each coordinate
should be specified as a floating point number.
The look at position is a single x,y,z argument where each coordinate
should be specified as a floating point number.
The optional sky colour is a single red,green,blue argument where
each component should be specified as a floating point number between
0.0 and 1.0 inclusive.

"""
    parser = argparse.ArgumentParser(description=description)
    parser.add_argument('ldraw_file')
    parser.add_argument('pov_file')
    parser.add_argument('camera_position', type=vector_position)
    parser.add_argument('look_at_position', type=vector_position, default=vector_position("0,0,0"))
    parser.add_argument('--sky')

    args = parser.parse_args()

    ldr2pov(args.ldraw_file, args.pov_file,
            args.camera_position, args.look_at_position, args.sky)


def ldr2pov(ldraw_path, pov_path,
            camera_position, look_at_position, sky):
    """ actual ldr2pov implementation """
    model, parts = get_model(ldraw_path)

    with open(pov_path, "w") as pov_file:
        pov_file.write('#include "include/colors.inc"\n\n')
        writer = POVRayWriter(parts, pov_file)
        writer.write(model)

        dd = camera_position.repr
        dd = dd.split(',')
        tt1 = float(dd[0])
        tt2 = float(dd[1])
        tt3 = float(dd[2])



        if not writer.lights:
            lights = (tt1, tt2, tt3,
                      tt1, tt2, tt3,
                      tt1, tt2, tt3,
                      tt1, tt2, tt3)
            pov_file.write(LIGHT_FORMAT_STRING % lights)

        pov_file.write(
            CAMERA_FORMAT_STRING % (camera_position.repr,
                                    look_at_position.repr)
        )

        if sky:
            pov_file.write(SKY_SPHERE_FORMAT_STRING % sky)

    for message, part in writer.warnings:
        sys.stderr.write((message + "\n") % part)

#
# if __name__ == "__main__":
#     main()