package {
          import flash.display.Sprite;
          import flash.display.StageAlign;
          import flash.display.StageScaleMode;
          import flash.events.Event;
          import flash.events.KeyboardEvent;
          import flash.ui.Keyboard;
          
          import org.wiiflash.Wiimote;
          import org.wiiflash.events.ButtonEvent;
          import org.wiiflash.events.WiimoteEvent;
          
          [SWF(backgroundColor=0xffffff)]
          public class @ax-actionscript-wii-Trees2 extends Sprite
          {
                  private var trees:Array;
                  private var numTrees:uint = 200;
                  private var fl:Number = 250;
                  private var vpX:Number = stage.stageWidth / 2;
                  private var vpY:Number = stage.stageHeight / 2;
                  private var floor:Number = 50;
                  private var ax:Number = 0;
                  private var ay:Number = 0;
                  private var az:Number = 0;
                  private var vx:Number = 0;
                  private var vy:Number = 0;
                  private var vz:Number = 0;
                  private var gravity:Number = 0.3;
                  private var friction:Number = 0.98;
                  
                  private var cursorX:Number = stage.stageWidth / 2;
                  private var cursorY:Number = stage.stageHeight / 2;
                  private var myWiimote:Wiimote;
                  
                  public function @ax-actionscript-wii-Trees2()
                  {
                          init();
                  }
                  
                  private function init():void
                  {
                          stage.displayState = "fullScreen";
                          
                          myWiimote = new Wiimote();
                          myWiimote.connect();
                          
                          stage.align = StageAlign.TOP_LEFT;
                          //stage.scaleMode = StageScaleMode.NO_SCALE;
                          
                          trees = new Array();
                          for(var i:uint = 0; i < numTrees; i++)
                          {
                                  var tree:actionscript_wii_Tree = new actionscript_wii_Tree();
                                  trees.push(tree);
                                  tree.xpos = Math.random() * 2000 - 1000;
                                  tree.ypos = floor;
                                  tree.zpos = Math.random() * 10000;
                                  addChild(tree);
                          }
                          addEventListener(Event.ENTER_FRAME, onEnterFrame);
                          stage.addEventListener(KeyboardEvent.KEY_DOWN, onKeyDown);
                          stage.addEventListener(KeyboardEvent.KEY_UP, onKeyUp);
                          
                          myWiimote.addEventListener(WiimoteEvent.UPDATE, onUpdated);
                          myWiimote.addEventListener(ButtonEvent.A_PRESS, onAPressed)
                          myWiimote.addEventListener(ButtonEvent.A_RELEASE, onAReleased);;
                          myWiimote.addEventListener(ButtonEvent.B_PRESS, onBPressed);
                          myWiimote.addEventListener(ButtonEvent.B_RELEASE, onBReleased);
                          
                  }
                  
                  private function onEnterFrame(event:Event):void
                  {
                          vx += ax;
                          vy += ay;
                          vz += az;
                          vy -= gravity;
                          for(var i:uint = 0; i < numTrees; i++)
                          {
                                  var tree:actionscript_wii_Tree = trees[i];
                                  move(tree);
                          }
                          vx *= friction;
                          vy *= friction;
                          vz *= friction;
                          sortZ();
                  }
                  
                  private function onKeyDown(event:KeyboardEvent):void
                  {
                          switch(event.keyCode)
                          {
                                  case Keyboard.UP :
                                  az = -1;
                                  break;
                                  
                                  case Keyboard.DOWN :
                                  az = 1;
                                  break;
                                  
                                  case Keyboard.LEFT :
                                  ax = 1;
                                  break;
                                  
                                  case Keyboard.RIGHT :
                                  ax = -1;
                                  break;
                                  
                                  case Keyboard.SPACE :
                                  ay = 1;
                                  break;
                                  
                                  default :
                                  break;
                          }
                  }
                  
                  private function onKeyUp(event:KeyboardEvent):void
                  {
                          switch(event.keyCode)
                          {
                                  case Keyboard.UP :
                                  case Keyboard.DOWN :
                                  az = 0;
                                  break;
                                  
                                  case Keyboard.LEFT :
                                  case Keyboard.RIGHT :
                                  ax = 0;
                                  break;
                                  
                                  case Keyboard.SPACE :
                                  ay = 0;
                                  break;
                                  
                                  default :
                                  break;
                          }
                  }
                  
                  private function move(tree:actionscript_wii_Tree):void
                  {
                          tree.xpos += vx;
                          tree.ypos += vy;
                          tree.zpos += vz;
                          
                          if(tree.ypos < floor)
                          {
                                  tree.ypos = floor;
                          }
                          
                          if(tree.zpos < -fl)
                          {
                                  tree.zpos += 10000;
                          }
                          if(tree.zpos > 10000 - fl)
                          {
                                  tree.zpos -= 10000;
                          }
                          
                          var scale:Number = fl / (fl + tree.zpos);
                          tree.scaleX = tree.scaleY = scale;
                          tree.x = vpX + tree.xpos * scale;
                          tree.y = vpY + tree.ypos * scale;
                          tree.alpha = scale;
                  }
                  
                  private function sortZ():void
                  {
                          trees.sortOn("zpos", Array.DESCENDING | Array.NUMERIC);
                          for(var i:uint = 0; i < numTrees; i++)
                          {
                                  var tree:actionscript_wii_Tree = trees[i];
                                  setChildIndex(tree, i);
                          }
                  }
                  
                  private function onUpdated ( pEvt:WiimoteEvent ):void 
                  {
                          /*if (myWiimote.ir.p1==true) {// && myWiimote.ir.p2==true) {
                                   //statusText.text = "Status:" + ((1-myWiimote.ir.x1) * stage.stageWidth) + (myWiimote.ir.y1 * stage.stageHeight);
                                   cursorX = (1-myWiimote.ir.x1) * stage.stageWidth;
                                  cursorY = myWiimote.ir.y1 * stage.stageHeight;
                                  ax = (2*myWiimote.ir.x1) - 1;
                                  ay = 1-myWiimote.ir.y1;
                                  
                                  //force feedback
                                  if (cursorX<10 || cursorX>stage.stageWidth-10) {
                                          myWiimote.rumbleTimeout = 300;        
                                  }
                                  if (cursorY<10 || cursorY>stage.stageHeight-10) {
                                          myWiimote.rumbleTimeout = 300;        
                                  }
                          }*/
                          
                          //motion sensor
                          ax = -myWiimote.sensorX;
                          ay = myWiimote.sensorY;
                          az = -myWiimote.sensorY - 0.5;
                  }
                  
                  private function onAPressed ( pEvt:ButtonEvent ):void
                  {        
                          az = 1;
                  }
                  
                  private function onAReleased ( pEvt:ButtonEvent ):void
                  {        
                          az = 0;
                  }
                  
                  private function onBPressed ( pEvt:ButtonEvent ):void
                  {        
                          az = -1;
                  }
                  
                  private function onBReleased ( pEvt:ButtonEvent ):void
                  {        
                          az = 0;
                  }
          }
  }