/*
 * MMI optimized DSP utils
 * Copyright (c) 2000, 2001 Fabrice Bellard.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 * MMI optimization by Leon van Stuivenberg <leonvs@iae.nl>
 * clear_blocks_mmi() by BroadQ
 */

#include "../dsputil.h"
#include "mmi.h"

void ff_mmi_idct_put(uint8_t *dest, int line_size, DCTELEM *block);
void ff_mmi_idct_add(uint8_t *dest, int line_size, DCTELEM *block);


static void clear_blocks_mmi(DCTELEM * blocks)
{
        asm volatile(
        ".set noreorder    \n"
        "addiu $9, %0, 768 \n"
        "nop               \n"
        "1:                \n"
        "sq $0, 0(%0)      \n"
        "move $8, %0       \n"
        "addi %0, %0, 64   \n"
        "sq $0, 16($8)     \n"
        "slt $10, %0, $9   \n"
        "sq $0, 32($8)     \n"
        "bnez $10, 1b      \n"
        "sq $0, 48($8)     \n"
        ".set reorder      \n"
        : "+r" (blocks) ::  "$8", "$9", "memory" );
}


static void get_pixels_mmi(DCTELEM *block, const uint8_t *pixels, int line_size)
{
    int i;
    for(i=0;i<8;i++) {
        asm volatile(
        ".set   push            \n\t"
        ".set   mips3           \n\t"
        "ld     $8, 0(%1)       \n\t"
        "add    %1, %1, %2      \n\t"
        "pextlb $8, $0, $8      \n\t"
        "sq     $8, 0(%0)       \n\t"
        ".set   pop             \n\t"
        :: "r" (block), "r" (pixels), "r" (line_size) : "$8", "memory" );
        block += 8;
    }
}


static void put_pixels8_mmi(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    int i;
    for(i=0; i<h; i++) {
        asm volatile(
        ".set   push            \n\t"
        ".set   mips3           \n\t"
        "ldr    $8, 0(%1)       \n\t"
        "ldl    $8, 7(%1)       \n\t"
        "add    %1, %1, %2      \n\t"
        "sd     $8, 0(%0)       \n\t"
        "add    %0, %0, %2      \n\t"
        ".set   pop             \n\t"
        :: "r" (block), "r" (pixels), "r" (line_size) : "$8", "memory" );
    }
}


static void put_pixels16_mmi(uint8_t *block, const uint8_t *pixels, int line_size, int h)
{
    int i;
    for(i=0; i<(h>>2); i++) {
        asm volatile (
        ".set   push            \n\t"
        ".set   mips3           \n\t"
#define PUTPIX16 \
        "ldr    $8, 0(%1)       \n\t" \
        "ldl    $8, 7(%1)       \n\t" \
        "ldr    $9, 8(%1)       \n\t" \
        "ldl    $9, 15(%1)      \n\t" \
        "add    %1, %1, %2      \n\t" \
        "pcpyld $8, $9, $8      \n\t" \
        "sq     $8, 0(%0)       \n\t" \
        "add    %0, %0, %2      \n\t"
        PUTPIX16
        PUTPIX16
        PUTPIX16
        PUTPIX16
        ".set   pop             \n\t"
        :: "r" (block), "r" (pixels), "r" (line_size) : "$8", "$9", "memory" );
    }
}


void dsputil_init_mmi(DSPContext* c, AVCodecContext *avctx)
{
    const int idct_algo= avctx->idct_algo;

    c->clear_blocks = clear_blocks_mmi;

    c->put_pixels_tab[1][0] = put_pixels8_mmi;
    c->put_no_rnd_pixels_tab[1][0] = put_pixels8_mmi;

    c->put_pixels_tab[0][0] = put_pixels16_mmi;
    c->put_no_rnd_pixels_tab[0][0] = put_pixels16_mmi;

    c->get_pixels = get_pixels_mmi;
       
    if(idct_algo==FF_IDCT_AUTO || idct_algo==FF_IDCT_PS2){
        c->idct_put= ff_mmi_idct_put;
        c->idct_add= ff_mmi_idct_add;
        c->idct_permutation_type= FF_LIBMPEG2_IDCT_PERM;
    }
}