jtrace.scenes.FourSpheres.java Source code

Introduction

Here is the source code for jtrace.scenes.FourSpheres.java

Source

/*
 * Copyright (C) 2012 Tim Vaughan <tgvaughan@gmail.com>
 *
 * 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/>.
 */
package jtrace.scenes;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import javax.imageio.ImageIO;
import jtrace.Camera;
import jtrace.Colour;
import jtrace.LightSource;
import jtrace.Scene;
import jtrace.object.Plane;
import jtrace.object.Sphere;
import jtrace.texture.AmbientFinish;
import jtrace.texture.CheckeredPigment;
import jtrace.texture.DiffuseFinish;
import jtrace.texture.FlatTexture;
import jtrace.texture.SolidPigment;
import jtrace.texture.SpecularFinish;
import org.apache.commons.math3.geometry.euclidean.threed.Vector3D;

/**
 * Slightly more complex test scene.
 *
 * @author Tim Vaughan <tgvaughan@gmail.com>
 */
public class FourSpheres {

    public static void main(String[] args) throws IOException {

        Camera camera = new Camera(new Vector3D(-1, 2, 5), new Vector3D(0, 0, 0), Vector3D.PLUS_J, 1.0, 1.6);

        Scene scene = new Scene();
        scene.setCamera(camera);

        scene.addLightSource(new LightSource(new Vector3D(-3, 3, 3), 4));

        FlatTexture glossRed = (new FlatTexture(new SolidPigment(new Colour(1, 0, 0))))
                .addFinish(new DiffuseFinish(1.0)).addFinish(new AmbientFinish(0.1))
                .addFinish(new SpecularFinish(1.0, 100));

        FlatTexture glossGreen = (new FlatTexture(new SolidPigment(new Colour(0, 1, 0))))
                .addFinish(new DiffuseFinish(1.0)).addFinish(new AmbientFinish(0.1))
                .addFinish(new SpecularFinish(1.0, 100));

        FlatTexture glossBlue = (new FlatTexture(new SolidPigment(new Colour(0, 0, 1))))
                .addFinish(new DiffuseFinish(1.0)).addFinish(new AmbientFinish(0.1))
                .addFinish(new SpecularFinish(1.0, 100));

        FlatTexture glossYellow = (new FlatTexture(new SolidPigment(new Colour(1, 1, 0))))
                .addFinish(new DiffuseFinish(1.0)).addFinish(new AmbientFinish(0.1))
                .addFinish(new SpecularFinish(1.0, 100));

        Sphere redSphere = new Sphere(new Vector3D(-1, 0, 0), 0.4);
        redSphere.addTexture(glossRed);
        scene.addObject(redSphere);

        Sphere greenSphere = new Sphere(new Vector3D(0, 0, 1), 0.4);
        greenSphere.addTexture(glossGreen);
        scene.addObject(greenSphere);

        Sphere blueSphere = new Sphere(new Vector3D(0, 0, -1), 0.4);
        blueSphere.addTexture(glossBlue);
        scene.addObject(blueSphere);

        Sphere yellowSphere = new Sphere(new Vector3D(1, 0, 0), 0.4);
        yellowSphere.addTexture(glossYellow);
        scene.addObject(yellowSphere);

        FlatTexture floorTexture = new FlatTexture(
                new CheckeredPigment(new Colour(.5, .5, .5), new Colour(1, 1, 1), 1));
        floorTexture.addFinish(new DiffuseFinish(1.0));
        floorTexture.addFinish(new AmbientFinish(0.05));

        Plane plane = new Plane(new Vector3D(0, -0.4, 0), Vector3D.PLUS_J, Vector3D.PLUS_K);
        plane.addTexture(floorTexture);
        scene.addObject(plane);

        BufferedImage image = scene.render(1440, 900, 10);
        ImageIO.write(image, "PNG", new File("out.png"));
    }
}