gallium/radeon: cleanup header inclusion
[mesa.git] / src / gallium / drivers / radeon / radeon_vce_40_2_2.c
index c41b2d03ab860ea4de55a684549756dc93372bd6..6b5da47eec8928967002ef59f343bc19b6aa6d93 100644 (file)
 
 #include "vl/vl_video_buffer.h"
 
-#include "../../winsys/radeon/drm/radeon_winsys.h"
+#include "radeon/drm/radeon_winsys.h"
 #include "radeon_video.h"
 #include "radeon_vce.h"
 
+static const unsigned profiles[7] = { 66, 77, 88, 100, 110, 122, 244 };
+
+static struct rvce_cpb_slot *current_slot(struct rvce_encoder *enc)
+{
+       return LIST_ENTRY(struct rvce_cpb_slot, enc->cpb_slots.prev, list);
+}
+
+static struct rvce_cpb_slot *l0_slot(struct rvce_encoder *enc)
+{
+       return LIST_ENTRY(struct rvce_cpb_slot, enc->cpb_slots.next, list);
+}
+
+static struct rvce_cpb_slot *l1_slot(struct rvce_encoder *enc)
+{
+       return LIST_ENTRY(struct rvce_cpb_slot, enc->cpb_slots.next->next, list);
+}
+
+static void frame_offset(struct rvce_encoder *enc, struct rvce_cpb_slot *slot,
+                        unsigned *luma_offset, unsigned *chroma_offset)
+{
+       unsigned pitch = align(enc->luma->level[0].pitch_bytes, 128);
+       unsigned vpitch = align(enc->luma->npix_y, 16);
+       unsigned fsize = pitch * (vpitch + vpitch / 2);
+
+       *luma_offset = slot->index * fsize;
+       *chroma_offset = *luma_offset + pitch * vpitch;
+}
+
 static void session(struct rvce_encoder *enc)
 {
        RVCE_BEGIN(0x00000001); // session cmd
@@ -78,8 +106,9 @@ static void create(struct rvce_encoder *enc)
 
        RVCE_BEGIN(0x01000001); // create cmd
        RVCE_CS(0x00000000); // encUseCircularBuffer
-       RVCE_CS(0x00000041); // encProfile
-       RVCE_CS(0x0000000a); // encLevel
+       RVCE_CS(profiles[enc->base.profile -
+               PIPE_VIDEO_PROFILE_MPEG4_AVC_BASELINE]); // encProfile
+       RVCE_CS(enc->base.level); // encLevel
        RVCE_CS(0x00000000); // encPicStructRestriction
        RVCE_CS(enc->base.width); // encImageWidth
        RVCE_CS(enc->base.height); // encImageHeight
@@ -154,12 +183,12 @@ static void pic_control(struct rvce_encoder *enc)
        RVCE_CS(0x00000000); // encSPSID
        RVCE_CS(0x00000000); // encPPSID
        RVCE_CS(0x00000040); // encConstraintSetFlags
-       RVCE_CS(0x00000000); // encBPicPattern
+       RVCE_CS(MAX2(enc->base.max_references, 1) - 1); // encBPicPattern
        RVCE_CS(0x00000000); // weightPredModeBPicture
-       RVCE_CS(0x00000001); // encNumberOfReferenceFrames
-       RVCE_CS(0x00000001); // encMaxNumRefFrames
-       RVCE_CS(0x00000000); // encNumDefaultActiveRefL0
-       RVCE_CS(0x00000000); // encNumDefaultActiveRefL1
+       RVCE_CS(MIN2(enc->base.max_references, 2)); // encNumberOfReferenceFrames
+       RVCE_CS(enc->base.max_references + 1); // encMaxNumRefFrames
+       RVCE_CS(0x00000001); // encNumDefaultActiveRefL0
+       RVCE_CS(0x00000001); // encNumDefaultActiveRefL1
        RVCE_CS(0x00000000); // encSliceMode
        RVCE_CS(0x00000000); // encMaxSliceSize
        RVCE_END();
@@ -218,17 +247,6 @@ static void rdo(struct rvce_encoder *enc)
        RVCE_END();
 }
 
-static void frame_offset(struct rvce_encoder *enc, unsigned frame_num,
-                        unsigned *luma_offset, unsigned *chroma_offset)
-{
-       unsigned pitch = align(enc->luma->level[0].pitch_bytes, 128);
-       unsigned vpitch = align(enc->luma->npix_y, 16);
-       unsigned fsize = pitch * (vpitch + vpitch / 2);
-
-       *luma_offset = (frame_num % RVCE_NUM_CPB_FRAMES) * fsize;
-       *chroma_offset = *luma_offset + pitch * vpitch;
-}
-
 static void encode(struct rvce_encoder *enc)
 {
        int i;
@@ -265,15 +283,25 @@ static void encode(struct rvce_encoder *enc)
        RVCE_CS(0x00000000); // encInputPic(Addr|Array)Mode
        RVCE_CS(0x00000000); // encInputPicTileConfig
        RVCE_CS(enc->pic.picture_type); // encPicType
-       RVCE_CS(enc->pic.picture_type == 3); // encIdrFlag
+       RVCE_CS(enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_IDR); // encIdrFlag
        RVCE_CS(0x00000000); // encIdrPicId
        RVCE_CS(0x00000000); // encMGSKeyPic
-       RVCE_CS(0x00000001); // encReferenceFlag
+       RVCE_CS(!enc->pic.not_referenced); // encReferenceFlag
        RVCE_CS(0x00000000); // encTemporalLayerIndex
        RVCE_CS(0x00000000); // num_ref_idx_active_override_flag
        RVCE_CS(0x00000000); // num_ref_idx_l0_active_minus1
        RVCE_CS(0x00000000); // num_ref_idx_l1_active_minus1
-       for (i = 0; i < 4; ++i) {
+
+       i = enc->pic.frame_num - enc->pic.ref_idx_l0;
+       if (i > 1 && enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_P) {
+               RVCE_CS(0x00000001); // encRefListModificationOp
+               RVCE_CS(i - 1);      // encRefListModificationNum
+       } else {
+               RVCE_CS(0x00000000); // encRefListModificationOp
+               RVCE_CS(0x00000000); // encRefListModificationNum
+       }
+
+       for (i = 0; i < 3; ++i) {
                RVCE_CS(0x00000000); // encRefListModificationOp
                RVCE_CS(0x00000000); // encRefListModificationNum
        }
@@ -281,39 +309,56 @@ static void encode(struct rvce_encoder *enc)
                RVCE_CS(0x00000000); // encDecodedPictureMarkingOp
                RVCE_CS(0x00000000); // encDecodedPictureMarkingNum
                RVCE_CS(0x00000000); // encDecodedPictureMarkingIdx
-       }
-       for (i = 0; i < 4; ++i) {
                RVCE_CS(0x00000000); // encDecodedRefBasePictureMarkingOp
                RVCE_CS(0x00000000); // encDecodedRefBasePictureMarkingNum
        }
 
+       // encReferencePictureL0[0]
        RVCE_CS(0x00000000); // pictureStructure
-
-       if (enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_IDR) { 
+       if(enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_P ||
+          enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_B) {
+               struct rvce_cpb_slot *l0 = l0_slot(enc);
+               frame_offset(enc, l0, &luma_offset, &chroma_offset);
+               RVCE_CS(l0->picture_type); // encPicType
+               RVCE_CS(l0->frame_num); // frameNumber
+               RVCE_CS(l0->pic_order_cnt); // pictureOrderCount
+               RVCE_CS(luma_offset); // lumaOffset
+               RVCE_CS(chroma_offset); // chromaOffset
+       } else {
                RVCE_CS(0x00000000); // encPicType
                RVCE_CS(0x00000000); // frameNumber
                RVCE_CS(0x00000000); // pictureOrderCount
                RVCE_CS(0xffffffff); // lumaOffset
                RVCE_CS(0xffffffff); // chromaOffset
        }
-       else if(enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_P) {
-               frame_offset(enc, enc->pic.frame_num - 1, &luma_offset, &chroma_offset);
-               RVCE_CS(0x00000000); // encPicType
-               RVCE_CS(enc->pic.frame_num - 1); // frameNumber
-               RVCE_CS(enc->pic.frame_num - 1); // pictureOrderCount
+
+       // encReferencePictureL0[1]
+       RVCE_CS(0x00000000); // pictureStructure
+       RVCE_CS(0x00000000); // encPicType
+       RVCE_CS(0x00000000); // frameNumber
+       RVCE_CS(0x00000000); // pictureOrderCount
+       RVCE_CS(0xffffffff); // lumaOffset
+       RVCE_CS(0xffffffff); // chromaOffset
+
+       // encReferencePictureL1[0]
+       RVCE_CS(0x00000000); // pictureStructure
+       if(enc->pic.picture_type == PIPE_H264_ENC_PICTURE_TYPE_B) {
+               struct rvce_cpb_slot *l1 = l1_slot(enc);
+               frame_offset(enc, l1, &luma_offset, &chroma_offset);
+               RVCE_CS(l1->picture_type); // encPicType
+               RVCE_CS(l1->frame_num); // frameNumber
+               RVCE_CS(l1->pic_order_cnt); // pictureOrderCount
                RVCE_CS(luma_offset); // lumaOffset
                RVCE_CS(chroma_offset); // chromaOffset
-       }
-       for (i = 0; i < 2; ++i) {
-               RVCE_CS(0x00000000); // pictureStructure
+       } else {
                RVCE_CS(0x00000000); // encPicType
                RVCE_CS(0x00000000); // frameNumber
                RVCE_CS(0x00000000); // pictureOrderCount
                RVCE_CS(0xffffffff); // lumaOffset
                RVCE_CS(0xffffffff); // chromaOffset
        }
-       
-       frame_offset(enc, enc->pic.frame_num, &luma_offset, &chroma_offset);
+
+       frame_offset(enc, current_slot(enc), &luma_offset, &chroma_offset);
        RVCE_CS(luma_offset); // encReconstructedLumaOffset
        RVCE_CS(chroma_offset); // encReconstructedChromaOffset
        RVCE_CS(0x00000000); // encColocBufferOffset
@@ -322,8 +367,8 @@ static void encode(struct rvce_encoder *enc)
        RVCE_CS(0x00000000); // encReferenceRefBasePictureLumaOffset
        RVCE_CS(0x00000000); // encReferenceRefBasePictureChromaOffset
        RVCE_CS(0x00000000); // pictureCount
-       RVCE_CS(0x00000000); // frameNumber
-       RVCE_CS(0x00000000); // pictureOrderCount
+       RVCE_CS(enc->pic.frame_num); // frameNumber
+       RVCE_CS(enc->pic.pic_order_cnt); // pictureOrderCount
        RVCE_CS(0x00000000); // numIPicRemainInRCGOP
        RVCE_CS(0x00000000); // numPPicRemainInRCGOP
        RVCE_CS(0x00000000); // numBPicRemainInRCGOP