diffusion-rs-sys 0.1.20

Rust bindings for stable-diffusion.cpp (FFI bindings)
Documentation
// Copyright 2014 Google Inc. All Rights Reserved.
//
// Use of this source code is governed by a BSD-style license
// that can be found in the COPYING file in the root of the source
// tree. An additional intellectual property rights grant can be found
// in the file PATENTS. All contributing project authors may
// be found in the AUTHORS file in the root of the source tree.
// -----------------------------------------------------------------------------
//
// Author: Djordje Pesut (djordje.pesut@imgtec.com)

#include "src/dsp/dsp.h"

#if defined(WEBP_USE_MIPS32)

#include "src/enc/cost_enc.h"

static int GetResidualCost_MIPS32(int ctx0, const VP8Residual* const res) {
  int temp0, temp1;
  int v_reg, ctx_reg;
  int n = res->first;
  // should be prob[VP8EncBands[n]], but it's equivalent for n=0 or 1
  int p0 = res->prob[n][ctx0][0];
  CostArrayPtr const costs = res->costs;
  const uint16_t* t = costs[n][ctx0];
  // bit_cost(1, p0) is already incorporated in t[] tables, but only if ctx != 0
  // (as required by the syntax). For ctx0 == 0, we need to add it here or it'll
  // be missing during the loop.
  int cost = (ctx0 == 0) ? VP8BitCost(1, p0) : 0;
  const int16_t* res_coeffs = res->coeffs;
  const int res_last = res->last;
  const int const_max_level = MAX_VARIABLE_LEVEL;
  const int const_2 = 2;
  const uint16_t** p_costs = &costs[n][0];
  const size_t inc_p_costs = NUM_CTX * sizeof(*p_costs);

  if (res->last < 0) {
    return VP8BitCost(0, p0);
  }

  __asm__ volatile(
      ".set    push                                                     \n\t"
      ".set    noreorder                                                \n\t"
      "subu    %[temp1],      %[res_last],        %[n]                  \n\t"
      "sll     %[temp0],      %[n],               1                     \n\t"
      "blez    %[temp1],      2f                                        \n\t"
      " addu   %[res_coeffs], %[res_coeffs],      %[temp0]              \n\t"
      "1:                                                               \n\t"
      "lh      %[v_reg],      0(%[res_coeffs])                          \n\t"
      "addiu   %[n],          %[n],               1                     \n\t"
      "negu    %[temp0],      %[v_reg]                                  \n\t"
      "slti    %[temp1],      %[v_reg],           0                     \n\t"
      "movn    %[v_reg],      %[temp0],           %[temp1]              \n\t"
      "sltiu   %[temp0],      %[v_reg],           2                     \n\t"
      "move    %[ctx_reg],    %[v_reg]                                  \n\t"
      "movz    %[ctx_reg],    %[const_2],         %[temp0]              \n\t"
      "sll     %[temp1],      %[v_reg],           1                     \n\t"
      "addu    %[temp1],      %[temp1],           %[VP8LevelFixedCosts] \n\t"
      "lhu     %[temp1],      0(%[temp1])                               \n\t"
      "slt     %[temp0],      %[v_reg],           %[const_max_level]    \n\t"
      "movz    %[v_reg],      %[const_max_level], %[temp0]              \n\t"
      "addu    %[cost],       %[cost],            %[temp1]              \n\t"
      "sll     %[v_reg],      %[v_reg],           1                     \n\t"
      "sll     %[ctx_reg],    %[ctx_reg],         2                     \n\t"
      "addu    %[v_reg],      %[v_reg],           %[t]                  \n\t"
      "lhu     %[temp0],      0(%[v_reg])                               \n\t"
      "addu    %[p_costs],    %[p_costs],         %[inc_p_costs]        \n\t"
      "addu    %[t],          %[p_costs],         %[ctx_reg]            \n\t"
      "addu    %[cost],       %[cost],            %[temp0]              \n\t"
      "addiu   %[res_coeffs], %[res_coeffs],      2                     \n\t"
      "bne     %[n],          %[res_last],        1b                    \n\t"
      " lw     %[t],          0(%[t])                                   \n\t"
      "2:                                                               \n\t"
      ".set    pop                                                      \n\t"
      : [cost] "+&r"(cost), [t] "+&r"(t), [n] "+&r"(n), [v_reg] "=&r"(v_reg),
        [ctx_reg] "=&r"(ctx_reg), [p_costs] "+&r"(p_costs),
        [temp0] "=&r"(temp0), [temp1] "=&r"(temp1),
        [res_coeffs] "+&r"(res_coeffs)
      : [const_2] "r"(const_2), [const_max_level] "r"(const_max_level),
        [VP8LevelFixedCosts] "r"(VP8LevelFixedCosts), [res_last] "r"(res_last),
        [inc_p_costs] "r"(inc_p_costs)
      : "memory");

  // Last coefficient is always non-zero
  {
    const int v = abs(res->coeffs[n]);
    assert(v != 0);
    cost += VP8LevelCost(t, v);
    if (n < 15) {
      const int b = VP8EncBands[n + 1];
      const int ctx = (v == 1) ? 1 : 2;
      const int last_p0 = res->prob[b][ctx][0];
      cost += VP8BitCost(0, last_p0);
    }
  }
  return cost;
}

static void SetResidualCoeffs_MIPS32(const int16_t* WEBP_RESTRICT const coeffs,
                                     VP8Residual* WEBP_RESTRICT const res) {
  const int16_t* p_coeffs = (int16_t*)coeffs;
  int temp0, temp1, temp2, n, n1;
  assert(res->first == 0 || coeffs[0] == 0);

  __asm__ volatile(
      ".set     push                                      \n\t"
      ".set     noreorder                                 \n\t"
      "addiu    %[p_coeffs],   %[p_coeffs],    28         \n\t"
      "li       %[n],          15                         \n\t"
      "li       %[temp2],      -1                         \n\t"
      "0:                                                   \n\t"
      "ulw      %[temp0],      0(%[p_coeffs])             \n\t"
      "beqz     %[temp0],      1f                         \n\t"
#if defined(WORDS_BIGENDIAN)
      " sll     %[temp1],      %[temp0],       16         \n\t"
#else
      " srl     %[temp1],      %[temp0],       16         \n\t"
#endif
      "addiu    %[n1],         %[n],           -1         \n\t"
      "movz     %[temp0],      %[n1],          %[temp1]   \n\t"
      "movn     %[temp0],      %[n],           %[temp1]   \n\t"
      "j        2f                                        \n\t"
      " addiu   %[temp2],      %[temp0],       0          \n\t"
      "1:                                                   \n\t"
      "addiu    %[n],          %[n],           -2         \n\t"
      "bgtz     %[n],          0b                         \n\t"
      " addiu   %[p_coeffs],   %[p_coeffs],    -4         \n\t"
      "2:                                                   \n\t"
      ".set     pop                                       \n\t"
      : [p_coeffs] "+&r"(p_coeffs), [temp0] "=&r"(temp0), [temp1] "=&r"(temp1),
        [temp2] "=&r"(temp2), [n] "=&r"(n), [n1] "=&r"(n1)
      :
      : "memory");
  res->last = temp2;
  res->coeffs = coeffs;
}

//------------------------------------------------------------------------------
// Entry point

extern void VP8EncDspCostInitMIPS32(void);

WEBP_TSAN_IGNORE_FUNCTION void VP8EncDspCostInitMIPS32(void) {
  VP8GetResidualCost = GetResidualCost_MIPS32;
  VP8SetResidualCoeffs = SetResidualCoeffs_MIPS32;
}

#else  // !WEBP_USE_MIPS32

WEBP_DSP_INIT_STUB(VP8EncDspCostInitMIPS32)

#endif  // WEBP_USE_MIPS32