/*---------------------------------------------------------------------------*
  Project:  NintendoWare
  File:     gfx_TransformAnimAdder.cpp
  
  Copyright (C)2009-2011 Nintendo/HAL Laboratory, Inc.  All rights reserved.
  
  These coded instructions, statements, and computer programs contain proprietary
  information of Nintendo and/or its licensed developers and are protected by
  national and international copyright laws. They may not be disclosed to third
  parties or copied or duplicated in any form, in whole or in part, without the
  prior written consent of Nintendo.
  
  The content herein is highly confidential and should be handled accordingly.
  
  $Revision: 31311 $
 *---------------------------------------------------------------------------*/

#include "precompiled.h"

#include <nw/anim/anim_AnimBlend.h>
#include <nw/gfx/gfx_AnimGroup.h>
#include <nw/gfx/gfx_CalculatedTransform.h>
#include <nw/gfx/gfx_TransformAnimAdder.h>
#include <nw/gfx/gfx_TransformAnimBlendOp.h>
#include <nw/gfx/gfx_TransformAnimEvaluator.h>

namespace nw
{
namespace gfx
{

NW_UT_RUNTIME_TYPEINFO_DEFINITION(TransformAnimAdder       , AnimAdder);

//----------------------------------------------------------
const anim::AnimResult*
TransformAnimAdder::GetResult(
    void* target,
    int memberIdx
) const
{
    //----------------------------------------------------------
    // ブレンドオペレーションを取得
    const anim::AnimBlendOp* blendOp = m_AnimGroup->GetBlendOperation(memberIdx);

    //----------------------------------------------------------
    // 補間は操作空間があればそこで行われるべきなので
    // target の CONVERTED フラグを退避してからオンに
    CalculatedTransform* transform =
        reinterpret_cast<CalculatedTransform*>(target);
    const bit32 flagsBak = transform->GetFlags();
    const bool convertedBak = transform->IsEnabledFlags(CalculatedTransform::FLAG_CONVERTED_FOR_BLEND);
    transform->EnableFlags(CalculatedTransform::FLAG_CONVERTED_FOR_BLEND);

    //----------------------------------------------------------
    // target の IGNORE フラグをオンにして未書き込み状態とする
    transform->EnableFlags(CalculatedTransform::FLAG_IS_IGNORE_ALL);

    //----------------------------------------------------------
    // すべての子アニメーションについてループ
    CalculatedTransform workResult;
    bool written = false;
    bool firstRotateFlag = true;
    math::MTX34 firstRotateMtx;
    //for (int animIdx = 0; animIdx < m_AnimObjects.Size(); ++animIdx)
    for (int animIdx = m_AnimObjects.Size() - 1; animIdx >= 0; --animIdx)
    {
        const AnimObject* animObj = m_AnimObjects[animIdx];
        if (animObj == NULL)
        {
            continue;
        }

        const float childWeight = m_Weights[animIdx];
        float srcWeights[3] = { childWeight, childWeight, childWeight };
        TransformAnimEvaluator::DisableSRTWeightsIfNeeded(srcWeights, animObj);

        if (!TransformAnimEvaluator::CheckWeightsNearlyZero(srcWeights))
        {
            // 退避した CONVERTED フラグを work に反映
            // 子が CONVERTED を扱う Blender であれば CONVERTED なままの結果が返ってくることを期待
            // 子が Evaluator の場合はフラグがオフで返ってくるが blendOp 内で変換されるので問題ない
            workResult.EnableFlags(CalculatedTransform::FLAG_CONVERTED_FOR_BLEND, convertedBak);
            
            // 評価
            const anim::AnimResult* childResult =
                animObj->GetResult(&workResult, memberIdx);
            if (childResult != NULL)
            {
                written = true;
                const bool evaluatorFlag =
                    ut::DynamicCast<const TransformAnimEvaluator*>(animObj) != NULL;

                // ブレンド結果の回転行列の行がゼロベクトルになった場合のために
                // 最初のアニメーションの回転行列を保存しておく
                if (evaluatorFlag && firstRotateFlag &&
                    !workResult.IsEnabledFlags(CalculatedTransform::FLAG_IS_IGNORE_ROTATE))
                {
                    firstRotateFlag = false;
                    firstRotateMtx = workResult.TransformMatrix();
                }

                // ブレンド
                if (!blendOp->Blend(reinterpret_cast<anim::AnimResult*>(transform),
                    NULL, childResult, srcWeights))
                {
                    break;
                }
            }
        }
    }

    //----------------------------------------------------------
    // target の CONVERTED フラグを復元
    if (!convertedBak)
    {
        transform->DisableFlags(CalculatedTransform::FLAG_CONVERTED_FOR_BLEND);   
    }

    if (!written) // 有効な子アニメーションなし
    {
        transform->RestoreFlags(CalculatedTransform::FLAG_IS_IGNORE_ALL, flagsBak);
        return NULL;
    }

    //----------------------------------------------------------
    // ブレンド後の処理があれば実行
    if (!convertedBak && blendOp->HasPostBlend())
    {
        if (!blendOp->PostBlend(reinterpret_cast<anim::AnimResult*>(transform), NULL))
        {
            // ブレンド結果の回転行列の行がゼロベクトルなら
            // 最初のアニメーションの回転行列を採用します。
            // 結果が品質上問題となる場合は
            // 元データを微妙にずらして回避する必要があります。
            transform->AdjustZeroRotateMatrix(!firstRotateFlag, firstRotateMtx);
        }
    }
    transform->UpdateScaleFlags();
    transform->UpdateRotateFlagsStrictly();
    transform->UpdateTranslateFlags();
    transform->UpdateCompositeFlags();

    // target を AnimResult にキャストして返す（親のブレンダで使用）
    return reinterpret_cast<anim::AnimResult*>(target);
}

} // namespace gfx
} // namespace nw