import viz

import vizact

import vizinfo

import viztask

import numpy

import numpy as np

import random

from random import *

from numpy import random

from numpy import array

import vizshape

import vizfx

import vizinfo

import oculus

import time

import projector

import statistics

from statistics import mean

#Oculus setup code

hmd = oculus.Rift()

viz.link(hmd.getSensor(), viz.MainView)

viz.MainView

viz.setMultiSample(8)

viz.go()

#Get RGB value for image and divide number by 250 to get the proportion of that color

viz.clearcolor(.5,.5,.5)

# Setup navigation node and link to main view

navigationNode = viz.addGroup()

viewLink = viz.link(navigationNode, viz.MainView)

viewLink.preMultLinkable(hmd.getSensor())

#Oculus will automatically sense the height of the participant and adjust the view

profile = hmd.getProfile()

global height

height = profile.eyeHeight - .13

navigationNode.setPosition([0,height,0])

# Setup heading reset key

KEYS = { 'reset': 'r'}

vizact.onkeydown(KEYS['reset'], hmd.getSensor().reset)

#Load textures

plaster = viz.addTexture('CeilingLarge.jpg')

wallpaper = viz.addTexture('WallpaperLarge.jpg')

wallpaperlong = viz.addTexture('WallpaperLong.jpg')

carpet = viz.addTexture('CarpetLarge3.jpg')

wood = viz.addTexture('Door.jpg')

mask_img = viz.addTexture('Masktiny.jpg')

#Create skylight

viz.MainView.getHeadLight().disable()

sky_light = viz.addLight(euler=(0,90,0)) #euler = yaw, pitch, roll

sky_light.position(2,0,-1,0)

sky_light.color(viz.WHITE)

sky_light.intensity(2)

#Create backlight

back_light = viz.addLight(euler=(0,215,0)) #euler = yaw, pitch, roll

back_light.position(-2,0,4,0)

back_light.color(viz.WHITE)

back_light.intensity(.75)

#Create frontlight?

front_light = viz.addLight(euler=(0,45,0)) #euler = yaw, pitch, roll

front_light.position(-2,0,4,0)

front_light.color(viz.WHITE)

front_light.intensity(1.5)

#Creating all parts of scene

#Creating all parts of scene

global ground

ground = vizshape.addPlane(size=[10,45]) #set constant size of 40 m wide, 45 m deep. Extra 5m in depth allows for 5m to extend behind participants' view

ground.setPosition(0,0,17.5) #set constant position to begin at viewpoint and set z dimension so 5m is behind participant view and 40m extends in front

ground.texture(carpet)

ground.disable(viz.RENDERING)

global farwall

farwall = vizshape.addPlane(size=[10,2.69], axis = vizshape.AXIS_Z, cullFace=False) #set default size to 2.69m tall by 40 m wide, aligned with Z axis

farwall.setPosition(0,1.345,40)

farwall.texture(wallpaper)

farwall.disable(viz.RENDERING)

global leftwall

leftwall = vizshape.addPlane(size=[45,2.69], axis = vizshape.AXIS_X, cullFace=False) #set constant size of 40 m deep by 2.69 m high, aligned with X axis

leftwall.setPosition(-5,1.345,17.5)

leftwall.texture(wallpaperlong)

leftwall.disable(viz.RENDERING)

global rightwall

rightwall = vizshape.addPlane(size=[45,2.69], axis = vizshape.AXIS_X, cullFace=False) #set constant size of 40 m deep by 2.69 m high, aligned with X axis

rightwall.setPosition(5,1.345,17.5)

rightwall.texture(wallpaperlong)

rightwall.disable(viz.RENDERING)

global ceiling

ceiling = vizshape.addPlane(size=[10,45], cullFace=False) #set constant size of 40 m wide by 40 m deep

ceiling.setPosition(0,2.69,17.5) #set constant position of 2.69 m high, with 40 m in front of participant and 5m behind (matching ground)

ceiling.texture(plaster)

ceiling.disable(viz.RENDERING)

global backwall

backwall = vizshape.addPlane(size=[10,2.69], axis = vizshape.AXIS_Z, cullFace=False) #set default size to 2.69m tall by 40 m wide, aligned with Z axis

backwall.setPosition(0,1.345,-5)

backwall.texture(wallpaper)

backwall.disable(viz.RENDERING)

global mask

mask = vizshape.addPlane(size=[10,10], axis = vizshape.AXIS_Z, cullFace=False)

mask.setPosition(0,0,2)

mask.texture(mask_img)

mask.disable(viz.RENDERING)

#Defining all constant parts of scene. Positions of walls moved to create different size rooms

def room ():

ceiling

#Enable environment

def en_env ():

ceiling.enable(viz.RENDERING)

#Disable environment

def dis_env ():

ceiling.disable(viz.RENDERING)

###########################################################################################################################################

## To execute thresholding:

## - Duration will begin at approx 20 ms. After each trial, at the fixation screen, press the up arrow to increase the duration by 10 ms

## - Continue until participant can reliably report the color of the target (no distance)

## - Press the Enter key (main keyboard, not keypad) to go to the next trial

## - Duration will begin at 120 ms. After each trial, press the down arrow key to decrease the duration by 10 ms

## - Continue until participant cannot reliably report the color of the target, then press enter

## - There will be 5 increasing and 5 decreasing trials, alternating one after the other

## - At the end of the trials, the program will calculate the average threshold duration

## - Take this number, round up to the nearest 5 ms and add 10 ms to use as short duration for experiment

###########################################################################################################################################

#Calibrate headset

def calibrate ():

lever.remove()

times = []

targets = [4.8,6.2,7.6,9,10.4,11.8,13.2,14.6,17.4,20.2,23,25.8,30]

def trial_up ():

print times

def trial_down ():

print times

def KeyEvents(key):

print 'stop'

def task ():

end_text.color(0,0,0)

viztask.schedule(task)