codecombat/app/lib/surface/WebGLSprite.coffee

SpriteBuilder = require 'lib/sprites/SpriteBuilder'

module.exports = class WebGLSprite extends createjs.SpriteContainer
  childMovieClips: null
  
  constructor: (@spriteSheet, @thangType, @spriteSheetPrefix, @resolutionFactor=SPRITE_RESOLUTION_FACTOR) ->
    @initialize(@spriteSheet)
    if @thangType.get('renderStrategy') isnt 'container'
      @singleChildSprite = new createjs.Sprite(@spriteSheet)
      @addChild(@singleChildSprite)
    @addEventListener 'tick', @handleTick
      
  handleTick: (e) =>
    if @lastTimeStamp
      @tick(e.timeStamp - @lastTimeStamp)
    @lastTimeStamp = e.timeStamp
    
  destroy: ->
    @handleTick = undefined
    @removeAllEventListeners()
      
  play: ->
    @singleChildSprite?.play()
    @paused = false
    
  stop: ->
    @singleChildSprite?.stop()
    @paused = true
  
  gotoAndPlay: (actionName) -> @goto(actionName, false)
  gotoAndStop: (actionName) -> @goto(actionName, true)
  
  goto: (actionName, @paused=true) ->
    @currentAnimation = actionName
    @baseMovieClip = @framerate = null
    
    action = @thangType.getActions()[actionName]
    randomStart = actionName.startsWith('move')
    reg = action.positions?.registration or @thangType.get('positions')?.registration or {x:0, y:0}

    if action.animation
      @framerate = (action.framerate ? 20) * (action.speed ? 1)
      
      if @singleChildSprite
        scale = @resolutionFactor * (action.scale ? @thangType.get('scale') ? 1)
        @regX = -reg.x * scale
        @regY = -reg.y * scale
        func = if @paused then 'gotoAndStop' else 'gotoAndPlay'
        animationName = @spriteSheetPrefix + actionName
        if not (animationName in @spriteSheet.getAnimations())
          @singleChildSprite.gotoAndStop(0)
          return
        @singleChildSprite[func](animationName)
        @singleChildSprite.framerate = action.framerate or 20
        
        if randomStart and frames = @spriteSheet.getAnimation(animationName)?.frames
          @singleChildSprite.currentAnimationFrame = Math.floor(Math.random() * frames.length)

      else
        @regX = -reg.x
        @regY = -reg.y
        @childMovieClips = []
        @baseMovieClip = @buildMovieClip(action.animation)
        if not @baseMovieClip
          @children = []
          return
        @children = @baseMovieClip.children
        @frames = action.frames
        @frames = (parseInt(f) for f in @frames.split(',')) if @frames
        @animLength = if @frames then @frames.length else @baseMovieClip.frameBounds.length
        @currentFrame = if randomStart then Math.floor(Math.random() * @animLength) else 0 
        @baseMovieClip.gotoAndStop(@currentFrame)
        @loop = action.loops isnt false
        @goesTo = action.goesTo
        
    if action.container
      if @singleChildSprite
        scale = @resolutionFactor * (action.scale ? @thangType.get('scale') ? 1)
        @regX = -reg.x * scale
        @regY = -reg.y * scale
        animationName = @spriteSheetPrefix + actionName
        if not (animationName in @spriteSheet.getAnimations())
          @singleChildSprite.gotoAndStop(0)
          return
        @singleChildSprite.gotoAndStop(animationName)

      else
        @regX = -reg.x
        @regY = -reg.y
        @childMovieClips = []
        containerName = @spriteSheetPrefix + action.container
        if not (containerName in @spriteSheet.getAnimations())
          @children = []
          return
        sprite = new createjs.Sprite(@spriteSheet)
        sprite.gotoAndStop(containerName)
        sprite.scaleX = sprite.scaleY = 1 / @resolutionFactor
        @children = [sprite]
      
    return
  
  buildMovieClip: (animationName, mode, startPosition, loops) ->
    raw = @thangType.get('raw')
    animData = raw.animations[animationName]
    movieClip = new createjs.MovieClip()

    locals = {}
    _.extend locals, containers = @buildMovieClipContainers(animData.containers)
    return false if not containers
    _.extend locals, animations = @buildMovieClipAnimations(animData.animations)
    return false if not animations
    
    toSkip = {}
    toSkip[shape.bn] = true for shape in animData.shapes
    toSkip[graphic.bn] = true for graphic in animData.graphics

    anim = new createjs.MovieClip()
    anim.initialize(mode ? createjs.MovieClip.INDEPENDENT, startPosition ? 0, loops ? true)

    for tweenData in animData.tweens
      stopped = false
      tween = createjs.Tween
      for func in tweenData
        args = $.extend(true, [], (func.a))
        if @dereferenceArgs(args, locals, toSkip) is false
          console.debug 'Did not dereference args:', args
          stopped = true
          break
        tween = tween[func.n](args...)
      continue if stopped
      anim.timeline.addTween(tween)

    anim.nominalBounds = new createjs.Rectangle(animData.bounds...)
    if animData.frameBounds
      anim.frameBounds = (new createjs.Rectangle(bounds...) for bounds in animData.frameBounds)
    return anim

  buildMovieClipContainers: (localContainers) ->
    map = {}
    for localContainer in localContainers
      outerContainer = new createjs.SpriteContainer(@spriteSheet)
      innerContainer = new createjs.Sprite(@spriteSheet)
      innerContainer.scaleX = innerContainer.scaleY = 1 / @resolutionFactor
      animationName = @spriteSheetPrefix + localContainer.gn
      return false if not (animationName in @spriteSheet.getAnimations())
      innerContainer.gotoAndStop(animationName)
      outerContainer.addChild(innerContainer)
      outerContainer.setTransform(localContainer.t...)
      outerContainer._off = localContainer.o if localContainer.o?
      outerContainer.alpha = localContainer.al if localContainer.al?
      map[localContainer.bn] = outerContainer
    return map

  buildMovieClipAnimations: (localAnimations) ->
    map = {}
    for localAnimation in localAnimations
      animation = @buildMovieClip(localAnimation.gn, localAnimation.a...)
      return false if not animation
      animation.setTransform(localAnimation.t...)
      map[localAnimation.bn] = animation
      @childMovieClips.push(animation)
    return map

  dereferenceArgs: (args, locals, toSkip) ->
    for key, val of args
      if locals[val]
        args[key] = locals[val]
      else if val is null
        args[key] = {}
      else if _.isString(val) and val.indexOf('createjs.') is 0
        args[key] = eval(val) # TODO: Security risk
      else if _.isObject(val) or _.isArray(val)
        res = @dereferenceArgs(val, locals, toSkip)
        return res if res is false
      else if _.isString(val) and toSkip[val]
        return false
    return args

  tick: (delta) ->
    return unless @baseMovieClip and @framerate and not @paused
    newFrame = @currentFrame + @framerate * delta / 1000
    
    if newFrame > @animLength
      if @goesTo
        @gotoAndPlay(@goesTo)
        return
      else if not @loop
        @paused = true
        newFrame = @animLength - 1
        @dispatchEvent('animationend')
      else
        newFrame = newFrame % @animLength
      
    if @frames
      prevFrame = Math.floor(newFrame)
      nextFrame = Math.ceil(newFrame)
      if prevFrame is nextFrame
        @baseMovieClip.gotoAndStop(@frames[newFrame])
      else if nextFrame is @frames.length
        @baseMovieClip.gotoAndStop(@frames[prevFrame])
      else
        # interpolate between frames
        pct = newFrame % 1
        newFrameIndex = @frames[prevFrame] + (pct * (@frames[nextFrame] - @frames[prevFrame]))
        @baseMovieClip.gotoAndStop(newFrameIndex)
    else
      @baseMovieClip.gotoAndStop(newFrame)
      
    @currentFrame = newFrame

    # So, originally I thought I'd have to swap in MovieClips for parallel 
    # SpriteContainers between each frame, but turns out that's not the case.
    # The WebGL rendering system treats the MovieClip like a SpriteContainer,
    # which makes things simpler for me...
    
    # For some reason, though, gotoAndStop doesn't seem to advance the children
    # so I gotta do that manually.
    movieClip.gotoAndStop(newFrame) for movieClip in @childMovieClips
  
  getBounds: ->
    @baseMovieClip.getBounds()