J3DShape / J3DShapeDraw: More OK

3 files changed, 15 insertions, 7 deletions

diff --git a/include/JSystem/J3DGraphBase/J3DShape.h b/include/JSystem/J3DGraphBase/J3DShape.h
index 5bac7d8ddf..866dbe34e5 100644
--- a/include/JSystem/J3DGraphBase/J3DShape.h
+++ b/include/JSystem/J3DGraphBase/J3DShape.h
@@ -90,6 +90,9 @@ public:
     inline J3DMaterial* getMaterial() const { return mMaterial; }
     inline u32 getPipeline() const { return (mFlags >> 2) & 0x07; }
     inline u32 getTexMtxLoadType() const { return mFlags & 0xF000; }
+    inline u32 getMtxGroupNum() const { return mMtxGroupNum; }
+    inline J3DShapeDraw* getShapeDraw(u32 idx) const { return mShapeDraw[idx]; }
+    inline J3DShapeMtx* getShapeMtx(u32 idx) const { return mShapeMtx[idx]; }
     static void resetVcdVatCache() { sOldVcdVatCmd = NULL; }
 
     static void* sOldVcdVatCmd;
diff --git a/include/JSystem/J3DGraphBase/J3DShapeDraw.h b/include/JSystem/J3DGraphBase/J3DShapeDraw.h
index 8caf151850..8ad82a0c3d 100644
--- a/include/JSystem/J3DGraphBase/J3DShapeDraw.h
+++ b/include/JSystem/J3DGraphBase/J3DShapeDraw.h
@@ -5,13 +5,16 @@
 
 class J3DShapeDraw {
 public:
-    /* 80314924 */ void countVertex(u32);
+    /* 80314924 */ u32 countVertex(u32);
     /* 80314974 */ void addTexMtxIndexInDL(u32, u32, u32);
     /* 80314ABC */ J3DShapeDraw(u8 const*, u32);
     /* 80314AD4 */ void draw() const;
 
     /* 80314B00 */ virtual ~J3DShapeDraw();
 
+    u8* getDisplayList() const { return (u8*)mDisplayList; }
+    u32 getDisplayListSize() const { return mDisplayListSize; }
+
 private:
     /* 0x04 */ u32 mDisplayListSize;
     /* 0x08 */ void* mDisplayList;
diff --git a/include/JSystem/J3DGraphBase/J3DVertex.h b/include/JSystem/J3DGraphBase/J3DVertex.h
index 360e4a0cea..567a41ebe0 100644
--- a/include/JSystem/J3DGraphBase/J3DVertex.h
+++ b/include/JSystem/J3DGraphBase/J3DVertex.h
@@ -21,17 +21,19 @@ public:
     J3DVertexData();
 
     void* getVtxPosArray() const { return mVtxPosArray; }
-
     void* getVtxNrmArray() const { return mVtxNrmArray; }
-
-    _GXColor* getVtxColorArray(u8 idx) const { return mVtxColorArray[idx]; }
-
+    GXColor* getVtxColorArray(u8 idx) const { return mVtxColorArray[idx]; }
+    void* getVtxTexCoordArray(u8 idx) const { return mVtxTexCoordArray[idx]; }
+    void* getVtxNBTArray() const { return mVtxNBTArray; }
     u32 getNrmNum() const { return mNrmNum; }
-
     u32 getVtxNum() const { return mVtxNum; }
-
     GXVtxAttrFmtList* getVtxAttrFmtList() { return mVtxAttrFmtList; }
 
+    void setVtxPosFrac(u8 frac) { mVtxPosFrac = frac; }
+    void setVtxPosType(GXCompType type) { mVtxPosType = type; }
+    void setVtxNrmFrac(u8 frac) { mVtxNrmFrac = frac; }
+    void setVtxNrmType(GXCompType type) { mVtxNrmType = type; }
+
 private:
     /* 0x00 */ u32 mVtxNum;
     /* 0x04 */ u32 mNrmNum;