BVHSerializer.h

#pragma once

#include <vector>

#define MAX_NODE_NUM    100
#define MAX_CHILDBONE   6
namespace ParaEngine
{
    using namespace std;

    class Quaternion;
    class CParaXModel;

    class CBVHSerializer
    {
    public:
        CBVHSerializer();
        ~CBVHSerializer(void);
        
        enum BVH_NODES
        {
            Hips = 0,//

            LeftHip,//lfemur        
            LeftKnee,// ltibia
            LeftAnkle,//    lfoot
            LeftToe,//     ltoe

            RightHip,// rfemur
            RightKnee,//    rtibia
            RightAnkle,//   rfoot
            RightToe,//    rtoe

            Chest,//        upperback
            Chest2,//   thorax
            Chest3,//      spine_bone3
            Chest4,//      spine_bone4
            Chest5,//      spine_bone5
            Chest6,//      spine_bone6
            Chest7,//      spine_bone7

            LeftCollar,//   lshoulderjoint
            LeftShoulder,// lhumerus
            LeftElbow,//    lradius
            LeftWrist,//    lwrist


            LeftFinger0,//    lThumb_0Bone
            LeftFinger01,//   lThumb_1Bone
            LeftFinger02,//   lThumb_2Bone

            LeftFinger1,//     lfinger1_0Bone
            LeftFinger11,//    lfinger1_1Bone
            LeftFinger12,//    lfinger1_2Bone
            //.
            //.
            //.
            LeftFinger4,//     lfinger4_0Bone
            LeftFinger41,//    lfinger4_1Bone
            LeftFinger42,//    lfinger4_2Bone


            RightCollar,//  rshoulderjoint
            RightShoulder,//    rhumerus
            RightElbow,//   rradius
            RightWrist,//   rwrist

            RightFinger0,//    rThumb_0Bone
            RightFinger01,//   rThumb_1Bone
            RightFinger02,//   rThumb_2Bone

            RightFinger1,//     rfinger1_0Bone
            RightFinger11,//    rfinger1_1Bone
            RightFinger12,//    rfinger1_2Bone
            //.
            //.
            //.
            RightFinger4,//     rfinger4_0Bone
            RightFinger41,//    rfinger4_1Bone
            RightFinger42,//    rfinger4_2Bone

            Head,//     head
            Neck,//     neck

            MAX_BVH_NODE,
        };

        struct MotionCaptureNode
        {
            string m_nodeName;
            int m_nBoneIndex;
        };

        struct UniBone{
            int m_nParentIndex;
            vector<int> m_childBones;
            int m_nSubBones;
            int m_nSubLevels;
            Vector3 m_vOffsetToParent;
            float m_fBoneLength;
            Vector3 m_vAbsOffset;
            string m_sMarkerName;
            bool m_bCriticalBone;
        public:
            inline int GetChildCount(){
                return (int)m_childBones.size();
            };
            inline bool IsUnKnownBone(){return m_sMarkerName.empty();};
            inline float GetBoneLength(){ return m_fBoneLength;}
            inline const Vector3& GetOffset(){ return m_vOffsetToParent;}
        };

        struct BoneChildCountCompare
        {   
            BoneChildCountCompare(vector<UniBone>& bones)
            {
                m_pBones = &bones;
            }

            bool operator()(const int& _Left, const int& _Right) const
            {   // apply operator> to operands
                return (((*m_pBones)[_Left].m_nSubLevels > (*m_pBones)[_Right].m_nSubLevels) ||
                    ((((*m_pBones)[_Left].m_nSubLevels == (*m_pBones)[_Right].m_nSubLevels)) && (*m_pBones)[_Left].m_fBoneLength > (*m_pBones)[_Right].m_fBoneLength) );
            }
        private:
            vector<UniBone>* m_pBones;
        };

        struct BoneMarkerCompare
        {   
            BoneMarkerCompare(vector<UniBone>& bones)
            {
                m_pBones = &bones;
            }

            bool operator()(const int& _Left, const int& _Right) const
            {   // apply operator> to operands

                if((*m_pBones)[_Right].m_sMarkerName.empty())
                    return true;
                else if((*m_pBones)[_Left].m_sMarkerName.empty())
                    return true;
                else if((*m_pBones)[_Left].m_sMarkerName[0] == 'L' && (*m_pBones)[_Right].m_sMarkerName[0] == 'R')
                    return true;
                else if((*m_pBones)[_Right].m_sMarkerName[0] == 'C' || (*m_pBones)[_Right].m_sMarkerName[0] == 'N')
                    return true;
                else
                    return false;
            }
        private:
            vector<UniBone>* m_pBones;
        };

        enum CharacterStudioNodeEnum{
            C7=0,CLAV,LANK,LBHD, LBWT, LELB, LFHD, LFIN, LFWT, LKNE, LMT5, LSHO, LTOE, LWRE, LWRI, RANK, RBHD, RBWT, RELB, RFHD, RFIN, RFWT, RKNE, RMT5, RSHO, RTOE, RWRE, RWRI, STRN, T10, 
        };
        enum UniversalNodeEnum{
            UN_Root,
            UN_Root1,
            UN_Root2,
            UN_Spine0,
                UN_Left_Thigh,
                    UN_Left_Calf,
                UN_Right_Thigh,
                    UN_Right_Calf,
                UN_Spine1,UN_Spine2,UN_Spine3,UN_Spine4,
                    UN_Neck,
                        UN_Head,
                        UN_Left_Clavicle,
                            UN_Left_UpperArm,
                        UN_Right_Clavicle,
                            UN_Right_UpperArm,
        };
        
    public:
        void SaveCSM(const string& sFileName);

        bool InitHierachy(CParaXModel& xmesh);

        bool SaveBVH(const string& sFileName, const string& sMarkerNameFile);

    private:
        vector <MotionCaptureNode> m_nodes;

        MotionCaptureNode m_UniversalNodes[MAX_NODE_NUM];

        vector<UniBone> m_bones;
        int m_BVHnodes[MAX_BVH_NODE];
        CParaXModel* m_pXMesh;
        
        int m_nRootBoneIndex;

        int m_nFPS;

        bool m_bRotateY;

        bool m_bEscapeUnknownBones;

        bool m_bEscapeUnmarkedBones;

        bool m_bExportBVHPosition;

        bool m_bSwapChestHip;

        int m_nHipBoneIndex;
        int m_nChestBoneIndex;

        bool BuildInternalBones();

        bool BuildBipedUniversalBones();

        inline int GetBoneIndex(int nIndex){return m_UniversalNodes[(UniversalNodeEnum)nIndex].m_nBoneIndex;};

        int GetBoneCount(){return (int)m_nodes.size();};

        bool DumpUniversalBones();
        bool DumpParaXBones(int nLevel, int nBoneIndex, CParaFile* pFile = NULL);
        bool WriteBVHNode(int nLevel, int nBoneIndex, CParaFile* pFile, bool bEscapeUnknownBones = true);
        bool WriteBVHNodeAnimation(int nLevel, int nBoneIndex, CParaFile* pFile, bool bEscapeUnknownBones = true);

        bool BeginBVHNode(CParaFile* pFile, int nLevel, const string& sMarker, const Vector3& vOffset, bool bExportPos, bool bExportRot);
        bool EndBVHNode(CParaFile* pFile, int nLevel,const string& sMarker);
        bool WriteQuaternion(CParaFile* pFile, const Quaternion& q, bool bRotY = false);
        bool WriteEndSite(CParaFile* pFile, int nLevel, int nBoneIndex);


        bool WriteHips(int nLevel, bool bHierachy, CParaFile* pFile);
        bool WriteLeftHip(int nLevel, bool bHierachy, CParaFile* pFile);
            bool WriteLeftKnee(int nLevel, bool bHierachy, CParaFile* pFile);
                bool WriteLeftAnkle(int nLevel, bool bHierachy, CParaFile* pFile);
                    bool WriteLeftToe(int nLevel, bool bHierachy, CParaFile* pFile);
                
        bool WriteRightHip(int nLevel, bool bHierachy, CParaFile* pFile);
            bool WriteRightKnee(int nLevel, bool bHierachy, CParaFile* pFile);
                bool WriteRightAnkle(int nLevel, bool bHierachy, CParaFile* pFile);
                    bool WriteRightToe(int nLevel, bool bHierachy, CParaFile* pFile);
            

        bool WriteChest(int nLevel, bool bHierachy, CParaFile* pFile);
        bool WriteChest2(int nLevel, bool bHierachy, CParaFile* pFile);
            bool WriteNeck(int nLevel, bool bHierachy, CParaFile* pFile);
                bool WriteHead(int nLevel, bool bHierachy, CParaFile* pFile);
            bool WriteLeftCollar(int nLevel, bool bHierachy, CParaFile* pFile);
                bool WriteLeftShoulder(int nLevel, bool bHierachy, CParaFile* pFile);
                    bool WriteLeftElbow(int nLevel, bool bHierachy, CParaFile* pFile);
                        bool WriteLeftWrist(int nLevel, bool bHierachy, CParaFile* pFile);
                            bool WriteLeftFinger0(int nLevel, bool bHierachy, CParaFile* pFile);
            bool WriteRightCollar(int nLevel, bool bHierachy, CParaFile* pFile);
                bool WriteRightShoulder(int nLevel, bool bHierachy, CParaFile* pFile);
                    bool WriteRightElbow(int nLevel, bool bHierachy, CParaFile* pFile);
                        bool WriteRightWrist(int nLevel, bool bHierachy, CParaFile* pFile);
                            bool WriteRightFinger0(int nLevel, bool bHierachy, CParaFile* pFile);
        


        int ComputeSubBonesCount(int nBoneIndex);

        bool SaveMarkerFile(const string& sFileName);
        bool LoadMarkerFile(const string& sFileName);

        int GetMarkedParentIndex(int nIndex);

        bool UpdateCriticalBone(int nBoneIndex);

        bool SortBVHMarker();
        bool AutoGenerateBVHMarker(int nBoneIndex, bool bGenChest, bool bGenHips);

        bool GenHipMarker(int nBoneIndex, int* ChestBoneIndex);
        bool GenChestMarker(int nBoneIndex);

    };

}