#include "config.h"
#include <stdint.h>
#include <stdlib.h>
#include "parasail.h"
#include "parasail/memory.h"
#include "parasail/internal_neon.h"
#ifdef PARASAIL_TABLE
static inline void arr_store_si128(
int *array,
simde__m128i vH,
int32_t t,
int32_t seglen,
int32_t d,
int32_t dlen)
{
array[1LL*(0*seglen+t)*dlen + d] = (int64_t)simde_mm_extract_epi64(vH, 0);
array[1LL*(1*seglen+t)*dlen + d] = (int64_t)simde_mm_extract_epi64(vH, 1);
}
#endif
#ifdef PARASAIL_ROWCOL
static inline void arr_store_col(
int *col,
simde__m128i vH,
int32_t t,
int32_t seglen)
{
col[0*seglen+t] = (int64_t)simde_mm_extract_epi64(vH, 0);
col[1*seglen+t] = (int64_t)simde_mm_extract_epi64(vH, 1);
}
#endif
#ifdef PARASAIL_TABLE
#define FNAME parasail_nw_stats_table_scan_neon_128_64
#define PNAME parasail_nw_stats_table_scan_profile_neon_128_64
#else
#ifdef PARASAIL_ROWCOL
#define FNAME parasail_nw_stats_rowcol_scan_neon_128_64
#define PNAME parasail_nw_stats_rowcol_scan_profile_neon_128_64
#else
#define FNAME parasail_nw_stats_scan_neon_128_64
#define PNAME parasail_nw_stats_scan_profile_neon_128_64
#endif
#endif
parasail_result_t* FNAME(
const char * const restrict s1, const int s1Len,
const char * const restrict s2, const int s2Len,
const int open, const int gap, const parasail_matrix_t *matrix)
{
parasail_profile_t *profile = NULL;
parasail_result_t *result = NULL;
PARASAIL_CHECK_NULL(s2);
PARASAIL_CHECK_GT0(s2Len);
PARASAIL_CHECK_GE0(open);
PARASAIL_CHECK_GE0(gap);
PARASAIL_CHECK_NULL(matrix);
if (matrix->type == PARASAIL_MATRIX_TYPE_PSSM) {
PARASAIL_CHECK_NULL_PSSM_STATS(s1);
}
else {
PARASAIL_CHECK_NULL(s1);
PARASAIL_CHECK_GT0(s1Len);
}
profile = parasail_profile_create_stats_neon_128_64(s1, s1Len, matrix);
if (!profile) return NULL;
result = PNAME(profile, s2, s2Len, open, gap);
parasail_profile_free(profile);
return result;
}
parasail_result_t* PNAME(
const parasail_profile_t * const restrict profile,
const char * const restrict s2, const int s2Len,
const int open, const int gap)
{
int32_t i = 0;
int32_t j = 0;
int32_t k = 0;
int32_t s1Len = 0;
int32_t end_query = 0;
int32_t end_ref = 0;
const parasail_matrix_t *matrix = NULL;
int32_t segWidth = 0;
int32_t segLen = 0;
int32_t offset = 0;
int32_t position = 0;
simde__m128i* restrict pvP = NULL;
simde__m128i* restrict pvPm = NULL;
simde__m128i* restrict pvPs = NULL;
simde__m128i* restrict pvE = NULL;
simde__m128i* restrict pvEM = NULL;
simde__m128i* restrict pvES = NULL;
simde__m128i* restrict pvEL = NULL;
simde__m128i* restrict pvH = NULL;
simde__m128i* restrict pvHM = NULL;
simde__m128i* restrict pvHS = NULL;
simde__m128i* restrict pvHL = NULL;
simde__m128i* restrict pvHMax = NULL;
simde__m128i* restrict pvHMMax = NULL;
simde__m128i* restrict pvHSMax = NULL;
simde__m128i* restrict pvHLMax = NULL;
simde__m128i* restrict pvGapper = NULL;
simde__m128i* restrict pvGapperL = NULL;
int64_t* restrict boundary = NULL;
simde__m128i vGapO;
simde__m128i vGapE;
int64_t NEG_LIMIT = 0;
int64_t POS_LIMIT = 0;
simde__m128i vZero;
simde__m128i vOne;
int64_t score = 0;
int64_t matches = 0;
int64_t similar = 0;
int64_t length = 0;
simde__m128i vNegLimit;
simde__m128i vPosLimit;
simde__m128i vSaturationCheckMin;
simde__m128i vSaturationCheckMax;
simde__m128i vNegInfFront;
simde__m128i vSegLenXgap;
simde__m128i vSegLen;
parasail_result_t *result = NULL;
PARASAIL_CHECK_NULL(profile);
PARASAIL_CHECK_NULL(profile->profile64.score);
PARASAIL_CHECK_NULL(profile->matrix);
PARASAIL_CHECK_GT0(profile->s1Len);
PARASAIL_CHECK_NULL(s2);
PARASAIL_CHECK_GT0(s2Len);
PARASAIL_CHECK_GE0(open);
PARASAIL_CHECK_GE0(gap);
i = 0;
j = 0;
k = 0;
s1Len = profile->s1Len;
end_query = s1Len-1;
end_ref = s2Len-1;
matrix = profile->matrix;
segWidth = 2;
segLen = (s1Len + segWidth - 1) / segWidth;
offset = (s1Len - 1) % segLen;
position = (segWidth - 1) - (s1Len - 1) / segLen;
pvP = (simde__m128i*)profile->profile64.score;
pvPm = (simde__m128i*)profile->profile64.matches;
pvPs = (simde__m128i*)profile->profile64.similar;
vGapO = simde_mm_set1_epi64x(open);
vGapE = simde_mm_set1_epi64x(gap);
NEG_LIMIT = (-open < matrix->min ? INT64_MIN + open : INT64_MIN - matrix->min) + 1;
POS_LIMIT = INT64_MAX - matrix->max - 1;
vZero = simde_mm_setzero_si128();
vOne = simde_mm_set1_epi64x(1);
score = NEG_LIMIT;
matches = 0;
similar = 0;
length = 0;
vNegLimit = simde_mm_set1_epi64x(NEG_LIMIT);
vPosLimit = simde_mm_set1_epi64x(POS_LIMIT);
vSaturationCheckMin = vPosLimit;
vSaturationCheckMax = vNegLimit;
vNegInfFront = vZero;
vSegLen = simde_mm_slli_si128(simde_mm_set1_epi64x(segLen), 8);
vNegInfFront = simde_mm_insert_epi64(vNegInfFront, NEG_LIMIT, 0);
vSegLenXgap = simde_mm_add_epi64(vNegInfFront,
simde_mm_slli_si128(simde_mm_set1_epi64x(-segLen*gap), 8));
#ifdef PARASAIL_TABLE
result = parasail_result_new_table3(segLen*segWidth, s2Len);
#else
#ifdef PARASAIL_ROWCOL
result = parasail_result_new_rowcol3(segLen*segWidth, s2Len);
#else
result = parasail_result_new_stats();
#endif
#endif
if (!result) return NULL;
result->flag |= PARASAIL_FLAG_NW | PARASAIL_FLAG_SCAN
| PARASAIL_FLAG_STATS
| PARASAIL_FLAG_BITS_64 | PARASAIL_FLAG_LANES_2;
#ifdef PARASAIL_TABLE
result->flag |= PARASAIL_FLAG_TABLE;
#endif
#ifdef PARASAIL_ROWCOL
result->flag |= PARASAIL_FLAG_ROWCOL;
#endif
pvE = parasail_memalign_simde__m128i(16, segLen);
pvEM = parasail_memalign_simde__m128i(16, segLen);
pvES = parasail_memalign_simde__m128i(16, segLen);
pvEL = parasail_memalign_simde__m128i(16, segLen);
pvH = parasail_memalign_simde__m128i(16, segLen);
pvHM = parasail_memalign_simde__m128i(16, segLen);
pvHS = parasail_memalign_simde__m128i(16, segLen);
pvHL = parasail_memalign_simde__m128i(16, segLen);
pvHMax = parasail_memalign_simde__m128i(16, segLen);
pvHMMax = parasail_memalign_simde__m128i(16, segLen);
pvHSMax = parasail_memalign_simde__m128i(16, segLen);
pvHLMax = parasail_memalign_simde__m128i(16, segLen);
pvGapper = parasail_memalign_simde__m128i(16, segLen);
pvGapperL = parasail_memalign_simde__m128i(16, segLen);
boundary = parasail_memalign_int64_t(16, s2Len+1);
if (!pvE) return NULL;
if (!pvEM) return NULL;
if (!pvES) return NULL;
if (!pvEL) return NULL;
if (!pvH) return NULL;
if (!pvHM) return NULL;
if (!pvHS) return NULL;
if (!pvHL) return NULL;
if (!pvHMax) return NULL;
if (!pvHMMax) return NULL;
if (!pvHSMax) return NULL;
if (!pvHLMax) return NULL;
if (!pvGapper) return NULL;
if (!pvGapperL) return NULL;
if (!boundary) return NULL;
parasail_memset_simde__m128i(pvHM, vZero, segLen);
parasail_memset_simde__m128i(pvHS, vZero, segLen);
parasail_memset_simde__m128i(pvHL, vZero, segLen);
parasail_memset_simde__m128i(pvE, vNegLimit, segLen);
parasail_memset_simde__m128i(pvEM, vZero, segLen);
parasail_memset_simde__m128i(pvES, vZero, segLen);
parasail_memset_simde__m128i(pvEL, vZero, segLen);
{
simde__m128i vGapper = simde_mm_sub_epi64(vZero,vGapO);
simde__m128i vGapperL = vOne;
for (i=segLen-1; i>=0; --i) {
simde_mm_store_si128(pvGapper+i, vGapper);
simde_mm_store_si128(pvGapperL+i, vGapperL);
vGapper = simde_mm_sub_epi64(vGapper, vGapE);
vGapperL = simde_mm_add_epi64(vGapperL, vOne);
}
}
{
int32_t index = 0;
for (i=0; i<segLen; ++i) {
int32_t segNum = 0;
simde__m128i h;
for (segNum=0; segNum<segWidth; ++segNum) {
int64_t tmp = -open-gap*(segNum*segLen+i);
h.i64[segNum] = tmp < INT64_MIN ? INT64_MIN : tmp;
}
simde_mm_store_si128(&pvH[index], h);
++index;
}
}
{
boundary[0] = 0;
for (i=1; i<=s2Len; ++i) {
int64_t tmp = -open-gap*(i-1);
boundary[i] = tmp < INT64_MIN ? INT64_MIN : tmp;
}
}
for (j=0; j<s2Len; ++j) {
simde__m128i vE;
simde__m128i vE_ext;
simde__m128i vE_opn;
simde__m128i vEM;
simde__m128i vES;
simde__m128i vEL;
simde__m128i vHt;
simde__m128i vHtM;
simde__m128i vHtS;
simde__m128i vHtL;
simde__m128i vF;
simde__m128i vF_ext;
simde__m128i vF_opn;
simde__m128i vFM;
simde__m128i vFS;
simde__m128i vFL;
simde__m128i vH;
simde__m128i vHM;
simde__m128i vHS;
simde__m128i vHL;
simde__m128i vHp;
simde__m128i vHpM;
simde__m128i vHpS;
simde__m128i vHpL;
simde__m128i *pvW;
simde__m128i vW;
simde__m128i *pvWM;
simde__m128i vWM;
simde__m128i *pvWS;
simde__m128i vWS;
simde__m128i case1;
simde__m128i case2;
simde__m128i vGapper;
simde__m128i vGapperL;
vHp = simde_mm_load_si128(pvH+(segLen-1));
vHpM = simde_mm_load_si128(pvHM+(segLen-1));
vHpS = simde_mm_load_si128(pvHS+(segLen-1));
vHpL = simde_mm_load_si128(pvHL+(segLen-1));
vHp = simde_mm_slli_si128(vHp, 8);
vHpM = simde_mm_slli_si128(vHpM, 8);
vHpS = simde_mm_slli_si128(vHpS, 8);
vHpL = simde_mm_slli_si128(vHpL, 8);
vHp = simde_mm_insert_epi64(vHp, boundary[j], 0);
pvW = pvP + matrix->mapper[(unsigned char)s2[j]]*segLen;
pvWM= pvPm+ matrix->mapper[(unsigned char)s2[j]]*segLen;
pvWS= pvPs+ matrix->mapper[(unsigned char)s2[j]]*segLen;
vHt = simde_mm_sub_epi64(vNegLimit, pvGapper[0]);
vF = vNegLimit;
vFM = vZero;
vFS = vZero;
vFL = vZero;
for (i=0; i<segLen; ++i) {
vH = simde_mm_load_si128(pvH+i);
vHM= simde_mm_load_si128(pvHM+i);
vHS= simde_mm_load_si128(pvHS+i);
vHL= simde_mm_load_si128(pvHL+i);
vE = simde_mm_load_si128(pvE+i);
vEM= simde_mm_load_si128(pvEM+i);
vES= simde_mm_load_si128(pvES+i);
vEL= simde_mm_load_si128(pvEL+i);
vW = simde_mm_load_si128(pvW+i);
vWM = simde_mm_load_si128(pvWM+i);
vWS = simde_mm_load_si128(pvWS+i);
vGapper = simde_mm_load_si128(pvGapper+i);
vGapperL = simde_mm_load_si128(pvGapperL+i);
vE_opn = simde_mm_sub_epi64(vH, vGapO);
vE_ext = simde_mm_sub_epi64(vE, vGapE);
case1 = simde_mm_cmpgt_epi64(vE_opn, vE_ext);
vE = simde_mm_max_epi64(vE_opn, vE_ext);
vEM = simde_mm_blendv_epi8(vEM, vHM, case1);
vES = simde_mm_blendv_epi8(vES, vHS, case1);
vEL = simde_mm_blendv_epi8(vEL, vHL, case1);
vEL = simde_mm_add_epi64(vEL, vOne);
vGapper = simde_mm_add_epi64(vHt, vGapper);
case1 = simde_mm_or_si128(
simde_mm_cmpgt_epi64(vF, vGapper),
simde_mm_cmpeq_epi64(vF, vGapper));
vF = simde_mm_max_epi64(vF, vGapper);
vFM = simde_mm_blendv_epi8(vHtM, vFM, case1);
vFS = simde_mm_blendv_epi8(vHtS, vFS, case1);
vFL = simde_mm_blendv_epi8(
simde_mm_add_epi64(vHtL, vGapperL),
vFL, case1);
vHp = simde_mm_add_epi64(vHp, vW);
vHpM = simde_mm_add_epi64(vHpM, vWM);
vHpS = simde_mm_add_epi64(vHpS, vWS);
vHpL = simde_mm_add_epi64(vHpL, vOne);
case1 = simde_mm_cmpgt_epi64(vE, vHp);
vHt = simde_mm_max_epi64(vE, vHp);
vHtM = simde_mm_blendv_epi8(vHpM, vEM, case1);
vHtS = simde_mm_blendv_epi8(vHpS, vES, case1);
vHtL = simde_mm_blendv_epi8(vHpL, vEL, case1);
simde_mm_store_si128(pvE+i, vE);
simde_mm_store_si128(pvEM+i, vEM);
simde_mm_store_si128(pvES+i, vES);
simde_mm_store_si128(pvEL+i, vEL);
simde_mm_store_si128(pvH+i, vHp);
simde_mm_store_si128(pvHM+i, vHpM);
simde_mm_store_si128(pvHS+i, vHpS);
simde_mm_store_si128(pvHL+i, vHpL);
vHp = vH;
vHpM = vHM;
vHpS = vHS;
vHpL = vHL;
}
vHt = simde_mm_slli_si128(vHt, 8);
vHtM = simde_mm_slli_si128(vHtM, 8);
vHtS = simde_mm_slli_si128(vHtS, 8);
vHtL = simde_mm_slli_si128(vHtL, 8);
vHt = simde_mm_insert_epi64(vHt, boundary[j+1], 0);
vGapper = simde_mm_load_si128(pvGapper);
vGapperL = simde_mm_load_si128(pvGapperL);
vGapper = simde_mm_add_epi64(vHt, vGapper);
case1 = simde_mm_or_si128(
simde_mm_cmpgt_epi64(vGapper, vF),
simde_mm_cmpeq_epi64(vGapper, vF));
vF = simde_mm_max_epi64(vF, vGapper);
vFM = simde_mm_blendv_epi8(vFM, vHtM, case1);
vFS = simde_mm_blendv_epi8(vFS, vHtS, case1);
vFL = simde_mm_blendv_epi8(
vFL,
simde_mm_add_epi64(vHtL, vGapperL),
case1);
for (i=0; i<segWidth-2; ++i) {
simde__m128i vFt = simde_mm_slli_si128(vF, 8);
simde__m128i vFtM = simde_mm_slli_si128(vFM, 8);
simde__m128i vFtS = simde_mm_slli_si128(vFS, 8);
simde__m128i vFtL = simde_mm_slli_si128(vFL, 8);
vFt = simde_mm_add_epi64(vFt, vSegLenXgap);
case1 = simde_mm_or_si128(
simde_mm_cmpgt_epi64(vFt, vF),
simde_mm_cmpeq_epi64(vFt, vF));
vF = simde_mm_max_epi64(vF, vFt);
vFM = simde_mm_blendv_epi8(vFM, vFtM, case1);
vFS = simde_mm_blendv_epi8(vFS, vFtS, case1);
vFL = simde_mm_blendv_epi8(
vFL,
simde_mm_add_epi64(vFtL, vSegLen),
case1);
}
vF = simde_mm_slli_si128(vF, 8);
vFM = simde_mm_slli_si128(vFM, 8);
vFS = simde_mm_slli_si128(vFS, 8);
vFL = simde_mm_slli_si128(vFL, 8);
vF = simde_mm_add_epi64(vF, vNegInfFront);
case1 = simde_mm_cmpgt_epi64(vF, vHt);
vH = simde_mm_max_epi64(vF, vHt);
vHM = simde_mm_blendv_epi8(vHtM, vFM, case1);
vHS = simde_mm_blendv_epi8(vHtS, vFS, case1);
vHL = simde_mm_blendv_epi8(vHtL, vFL, case1);
for (i=0; i<segLen; ++i) {
vHp = simde_mm_load_si128(pvH+i);
vHpM = simde_mm_load_si128(pvHM+i);
vHpS = simde_mm_load_si128(pvHS+i);
vHpL = simde_mm_load_si128(pvHL+i);
vE = simde_mm_load_si128(pvE+i);
vEM = simde_mm_load_si128(pvEM+i);
vES = simde_mm_load_si128(pvES+i);
vEL = simde_mm_load_si128(pvEL+i);
vF_opn = simde_mm_sub_epi64(vH, vGapO);
vF_ext = simde_mm_sub_epi64(vF, vGapE);
vF = simde_mm_max_epi64(vF_opn, vF_ext);
case1 = simde_mm_cmpgt_epi64(vF_opn, vF_ext);
vFM = simde_mm_blendv_epi8(vFM, vHM, case1);
vFS = simde_mm_blendv_epi8(vFS, vHS, case1);
vFL = simde_mm_blendv_epi8(vFL, vHL, case1);
vFL = simde_mm_add_epi64(vFL, vOne);
vH = simde_mm_max_epi64(vHp, vE);
vH = simde_mm_max_epi64(vH, vF);
case1 = simde_mm_cmpeq_epi64(vH, vHp);
case2 = simde_mm_cmpeq_epi64(vH, vF);
vHM = simde_mm_blendv_epi8(
simde_mm_blendv_epi8(vEM, vFM, case2),
vHpM, case1);
vHS = simde_mm_blendv_epi8(
simde_mm_blendv_epi8(vES, vFS, case2),
vHpS, case1);
vHL = simde_mm_blendv_epi8(
simde_mm_blendv_epi8(vEL, vFL, case2),
vHpL, case1);
simde_mm_store_si128(pvH+i, vH);
simde_mm_store_si128(pvHM+i, vHM);
simde_mm_store_si128(pvHS+i, vHS);
simde_mm_store_si128(pvHL+i, vHL);
vSaturationCheckMin = simde_mm_min_epi64(vSaturationCheckMin, vH);
vSaturationCheckMax = simde_mm_max_epi64(vSaturationCheckMax, vH);
vSaturationCheckMax = simde_mm_max_epi64(vSaturationCheckMax, vHM);
vSaturationCheckMax = simde_mm_max_epi64(vSaturationCheckMax, vHS);
vSaturationCheckMax = simde_mm_max_epi64(vSaturationCheckMax, vHL);
#ifdef PARASAIL_TABLE
arr_store_si128(result->stats->tables->score_table, vH, i, segLen, j, s2Len);
arr_store_si128(result->stats->tables->matches_table, vHM, i, segLen, j, s2Len);
arr_store_si128(result->stats->tables->similar_table, vHS, i, segLen, j, s2Len);
arr_store_si128(result->stats->tables->length_table, vHL, i, segLen, j, s2Len);
#endif
}
#ifdef PARASAIL_ROWCOL
{
vH = simde_mm_load_si128(pvH + offset);
vHM = simde_mm_load_si128(pvHM + offset);
vHS = simde_mm_load_si128(pvHS + offset);
vHL = simde_mm_load_si128(pvHL + offset);
for (k=0; k<position; ++k) {
vH = simde_mm_slli_si128(vH, 8);
vHM = simde_mm_slli_si128(vHM, 8);
vHS = simde_mm_slli_si128(vHS, 8);
vHL = simde_mm_slli_si128(vHL, 8);
}
result->stats->rowcols->score_row[j] = (int64_t) simde_mm_extract_epi64 (vH, 1);
result->stats->rowcols->matches_row[j] = (int64_t) simde_mm_extract_epi64 (vHM, 1);
result->stats->rowcols->similar_row[j] = (int64_t) simde_mm_extract_epi64 (vHS, 1);
result->stats->rowcols->length_row[j] = (int64_t) simde_mm_extract_epi64 (vHL, 1);
}
#endif
}
#ifdef PARASAIL_ROWCOL
for (i=0; i<segLen; ++i) {
simde__m128i vH = simde_mm_load_si128(pvH+i);
simde__m128i vHM = simde_mm_load_si128(pvHM+i);
simde__m128i vHS = simde_mm_load_si128(pvHS+i);
simde__m128i vHL = simde_mm_load_si128(pvHL+i);
arr_store_col(result->stats->rowcols->score_col, vH, i, segLen);
arr_store_col(result->stats->rowcols->matches_col, vHM, i, segLen);
arr_store_col(result->stats->rowcols->similar_col, vHS, i, segLen);
arr_store_col(result->stats->rowcols->length_col, vHL, i, segLen);
}
#endif
{
simde__m128i vH = simde_mm_load_si128(pvH + offset);
simde__m128i vHM = simde_mm_load_si128(pvHM + offset);
simde__m128i vHS = simde_mm_load_si128(pvHS + offset);
simde__m128i vHL = simde_mm_load_si128(pvHL + offset);
for (k=0; k<position; ++k) {
vH = simde_mm_slli_si128(vH, 8);
vHM = simde_mm_slli_si128(vHM, 8);
vHS = simde_mm_slli_si128(vHS, 8);
vHL = simde_mm_slli_si128(vHL, 8);
}
score = (int64_t) simde_mm_extract_epi64 (vH, 1);
matches = (int64_t) simde_mm_extract_epi64 (vHM, 1);
similar = (int64_t) simde_mm_extract_epi64 (vHS, 1);
length = (int64_t) simde_mm_extract_epi64 (vHL, 1);
}
if (simde_mm_movemask_epi8(simde_mm_or_si128(
simde_mm_cmplt_epi64(vSaturationCheckMin, vNegLimit),
simde_mm_cmpgt_epi64(vSaturationCheckMax, vPosLimit)))) {
result->flag |= PARASAIL_FLAG_SATURATED;
score = 0;
matches = 0;
similar = 0;
length = 0;
end_query = 0;
end_ref = 0;
}
result->score = score;
result->end_query = end_query;
result->end_ref = end_ref;
result->stats->matches = matches;
result->stats->similar = similar;
result->stats->length = length;
parasail_free(boundary);
parasail_free(pvGapperL);
parasail_free(pvGapper);
parasail_free(pvHLMax);
parasail_free(pvHSMax);
parasail_free(pvHMMax);
parasail_free(pvHMax);
parasail_free(pvHL);
parasail_free(pvHS);
parasail_free(pvHM);
parasail_free(pvH);
parasail_free(pvEL);
parasail_free(pvES);
parasail_free(pvEM);
parasail_free(pvE);
return result;
}