Exemplo n.º 1
0
def main():

    if len(sys.argv) < 2:
        print("Usage: python 7_panorama_viewer.py image_file.ext")
        print("ext: jpg, png, tif and bmp files are accepted.")
        exit(0)

    fname = str(sys.argv[1])
    # just in case you have white spaces in the path...
    for s in sys.argv[2:]:
        fname += " " + str(s)
    print("Reading:", fname)

    # create and configure, show the window later
    rt = TkOptiX()

    rt.set_background_mode("TextureEnvironment")

    rt.set_background(fname)

    rt.start()

    rt.set_camera_glock(True)

    print("done")
Exemplo n.º 2
0
def main():

    # make some data first:
    b = 4
    n = 200
    k = 9.1
    l = 1.1
    m = 7.1

    r0 = 0.06
    q = 11

    x = np.sin(np.linspace(0, 2 * b * k * np.pi, b * n))
    y = np.cos(np.linspace(0, 2 * b * l * np.pi, b * n))
    z = np.sin(np.linspace(0, 2 * b * m * np.pi, b * n))
    pos = np.stack((x, y, z)).T

    r = r0 * 0.5 * (1 +
                    np.sin(np.linspace(0, 2 * b * q * np.pi, b * n))) + 0.002

    # create and configure, show the window later
    optix = TkOptiX()

    optix.set_param(min_accumulation_step=2, max_accumulation_frames=64)
    optix.setup_material("plastic", m_plastic)

    optix.setup_camera("dof_cam",
                       cam_type="DoF",
                       eye=[-4, 3, 4],
                       target=[0, 0, 0],
                       up=[0.21, 0.86, -0.46],
                       focal_scale=0.75)

    # some exposure and gamma adjutments
    optix.set_float("tonemap_exposure", 0.9)
    optix.set_float("tonemap_gamma", 1.4)
    optix.add_postproc("Gamma")

    ################################################################################
    # Try one of the background / lighting settings:

    # ------------------------------------------------------------------------------
    # 1. Ray tracer is initialized in `AmbientLight` mode. There is a constant
    #    background color and an omnidirectional light with the color independent
    #    from the background:

    #optix.setup_light("l1", color=4*np.array([0.99, 0.95, 0.9]), radius=3)
    #optix.set_background([0, 0.02, 0.1]) # ambient and background colors can be
    #optix.set_ambient(0.4)               # RGB array or a gray level

    # ------------------------------------------------------------------------------
    # 2. `Default` (although not set by default initialization) mode uses
    #    background color to paint the background and as the ambient light,
    #    so a brighter one is better looking here:

    #optix.set_background_mode("Default")
    #optix.setup_light("l1", color=np.array([0.99, 0.95, 0.9]), radius=3) # just a little light from the side
    #optix.set_background(0.94)

    # ------------------------------------------------------------------------------
    # 3. Environment map. Background texture is projected on the sphere with
    #    infinite radius, and it is also the source of the ambient light.

    # make a small RGB texture with a vertical gradient
    a = np.linspace(0.94, 0, 10)
    b = np.zeros((10, 2, 3))
    for i in range(10):
        b[i, 0] = np.full(3, a[i])
        b[i, 1] = np.full(3, a[i])

    # extend to RGBA (RGB is fine to use with `set_background()`, but is converted to
    # RGBA internally anyway)
    bg = np.zeros((10, 2, 4), dtype=np.float32)
    bg[:, :, :-1] = b

    optix.set_background_mode("TextureEnvironment")
    optix.setup_light("l1", color=np.array([0.99, 0.95, 0.9]),
                      radius=3)  # just a little light from the side
    optix.set_background(bg)

    # ------------------------------------------------------------------------------
    # 4. Fixed background texture. Background is just a wallpaper, you need to setup
    #    ambient light and/or other lights.

    # make a small RGB texture with a vertical gradient
    #a = np.linspace(0.94, 0, 10)
    #b = np.zeros((10, 2, 3))
    #for i in range(10):
    #    b[i,0]=np.full(3, a[i])
    #    b[i,1]=np.full(3, a[i])

    #optix.set_background_mode("TextureFixed")
    #optix.setup_light("l1", color=4*np.array([0.99, 0.95, 0.9]), radius=3)
    #optix.set_background(b)
    #optix.set_ambient(0.4)

    # ------------------------------------------------------------------------------
    # Note: background mode, lights, colors and textures can be all updated also
    # after the window is open, while the ray tracing running.
    ################################################################################

    # create a plot of parametric curve calculated above, and open the window
    optix.set_data("plot",
                   pos=pos,
                   r=r,
                   c=0.94,
                   geom="BezierChain",
                   mat="plastic")
    #optix.set_data("plot", pos=pos, r=r, c=0.94, geom="BSplineQuad", mat="plastic")
    #optix.set_data("plot", pos=pos, r=r, c=0.94, geom="BSplineCubic", mat="plastic")
    #optix.set_data("plot", pos=pos, r=r, c=0.94, geom="SegmentChain", mat="plastic")

    optix.start()

    print(optix.get_background_mode())

    print("done")
# Fullpoints[:,1] = points[1,:]

plot = TkOptiX()
plot.setup_material("ice", m_metallic)
plot.set_ambient(0.7)
plot.set_background(0.4)

# plot.load_mesh_obj("./mesh/prism", "hexa")

fname = "back.jpg"

plot.set_float("tonemap_exposure", 0.9)
plot.set_float("tonemap_gamma", 1.2)
plot.add_postproc("Gamma")

plot.set_background_mode("TextureEnvironment")
# plot.setup_light("l1", color=1*np.array([0.99, 0.95, 0.9]), radius=3)
plot.set_background(fname)
plot.set_ambient(0.1)

# for i in range(1,6):
# points = np.load(input_path+str(i)+".npz","r+")["arr_0"]
points = generate_snowflake().T

Fullpoints = np.zeros([points.shape[1], 3])

Fullpoints[:, 0] = points[0, :]  #+10*i
Fullpoints[:, 1] = points[1, :]
# Fullpoints[:,2] = 2*i

plot.set_data("flake" + str(1),