--- /dev/null
+/* copyright (c) 1996, 1997, 1998, 1999 William R. Pearson and the
+ U. of Virginia */
+
+/* $Name: fa_34_26_5 $ - $Id: work_thr.c,v 1.23 2007/04/26 18:33:20 wrp Exp $ */
+
+/* work_thr.c - threaded worker */
+
+/* modified 21-Oct-1998 to work with reverse complement for DNA */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <time.h>
+#include <sys/types.h>
+#include <signal.h>
+
+#include "defs.h" /* various constants */
+#include "mw.h" /* defines beststr */
+#include "structs.h"
+#include "param.h" /* pstruct, thr_str, buf_head, rstruct */
+
+/***************************************/
+/* thread global variable declarations */
+/***************************************/
+
+#define XTERNAL
+#include "thr.h"
+#undef XTERNAL
+
+void alloc_pam (int, int, struct pstruct *);
+int **alloc_pam2p(int, int);
+void revcomp(unsigned char *seq, int n, int *c_nt);
+#ifdef WIN32
+void pthread_exit(void *);
+#else
+void THR_EXIT(void *);
+#endif
+
+/* functions getting/sending buffers to threads (thr_sub.c) */
+extern void wait_thr(void);
+extern int get_wbuf(struct buf_head **cur_buf, int max_work_buf);
+extern void put_wbuf(struct buf_head *cur_buf, int max_work_buf);
+
+/* dropxx.c functions */
+extern void init_work (unsigned char *aa0, int n0,
+ struct pstruct *ppst, void **f_arg);
+
+extern void do_work (unsigned char *aa0, int n0, unsigned char *aa1, int n1,
+ int frame,
+ struct pstruct *ppst, void *f_str, int qr_flg,
+ struct rstruct *rst);
+
+extern void close_work (unsigned char *, int, struct pstruct *, void **);
+
+extern void irand(int);
+extern int shuffle(unsigned char *, unsigned char *, int);
+extern int wshuffle(unsigned char *, unsigned char *, int, int, int *);
+extern void qshuffle(unsigned char *aa0, int n0, int nm0);
+extern void free_pam2p(int **);
+
+void
+work_thread (struct thr_str *work_info)
+{
+ struct buf_head *cur_buf;
+ struct buf_str *cur_buf_p;
+ struct buf_str *p_rbuf;
+ unsigned char *aa1s;
+ int cur_cnt;
+ int my_worker;
+ int i, j, npam, n0, nm0;
+ int ix_score, debug_lib, zsflag, zs_win, do_shuffle, ieven=0;
+ int frame;
+
+ struct rstruct rrst;
+ struct pstruct my_pst, *my_ppst;
+ unsigned char *aa0[6], *aa0s;
+ void *f_str[6], *qf_str;
+
+ my_worker = work_info->worker;
+
+ wait_thr(); /* wait for start_thread predicate to drop to 0 */
+
+ /* do init_work */
+
+ /* let each thread have its own copy of the query */
+ n0 = work_info->n0;
+ nm0 = work_info->nm0;
+
+ if ((aa0[0]=(unsigned char *)calloc((size_t)n0+2,sizeof(unsigned char)))
+ ==NULL) {
+ fprintf(stderr," cannot allocate aa00[%d] for worker %d\n",
+ n0, my_worker);
+ exit(1);
+ }
+ *aa0[0]='\0';
+ aa0[0]++;
+ memcpy(aa0[0],work_info->aa0,n0+1);
+
+ /* make certain that all but 0 have their own copy of pst */
+ if (my_worker) {
+ my_ppst = &my_pst;
+ memcpy(my_ppst,work_info->ppst,sizeof(struct pstruct));
+
+ alloc_pam(MAXSQ, MAXSQ, my_ppst);
+
+ npam = (my_pst.ext_sq_set) ? my_pst.nsqx : my_pst.nsq;
+
+ for (i=0; i<=npam; i++) {
+ for (j=0; j<=npam; j++) {
+ my_pst.pam2[0][i][j] = work_info->ppst->pam2[0][i][j];
+ my_pst.pam2[1][i][j] = work_info->ppst->pam2[1][i][j];
+ }
+ }
+
+ if (work_info->ppst->pam_pssm && work_info->ppst->pam2p[0]) {
+ my_ppst->pam2p[0] = alloc_pam2p(n0,npam);
+ my_ppst->pam2p[1] = alloc_pam2p(n0,npam);
+ for (i=0; i<n0; i++) {
+ for (j=0; j <= npam; j++) {
+ my_pst.pam2p[0][i][j] = work_info->ppst->pam2p[0][i][j];
+ my_pst.pam2p[1][i][j] = work_info->ppst->pam2p[1][i][j];
+ }
+ }
+ }
+ }
+ else my_ppst=work_info->ppst;
+
+ /* note that aa[5,4,3,2] are never used, but are provided so that frame
+ can range from 0 .. 5; likewise for f_str[5..2] */
+
+ aa0[5] = aa0[4] = aa0[3] = aa0[2] = aa0[1] = aa0[0];
+ init_work (aa0[0], n0, my_ppst, &f_str[0]);
+
+ f_str[5] = f_str[4] = f_str[3] = f_str[2] = f_str[1] = f_str[0];
+
+ if (work_info->qframe == 2) {
+ if ((aa0[1]=(unsigned char *)calloc((size_t)n0+2,sizeof(unsigned char)))==NULL) {
+ fprintf(stderr," cannot allocate aa01[%d] for worker %d\n",
+ n0, my_worker);
+ }
+ *aa0[1]='\0';
+ aa0[1]++;
+ memcpy(aa0[1],work_info->aa0,n0+1);
+ revcomp(aa0[1],n0,my_ppst->c_nt);
+ init_work (aa0[1], n0, my_ppst, &f_str[1]);
+ }
+
+ if (work_info->qshuffle) {
+ if ((aa0s=(unsigned char *)calloc(n0+2,sizeof(char)))==NULL) {
+ fprintf(stderr,"cannot allocate aa0s[%d]\n",n0+2);
+ exit(1);
+ }
+ *aa0s='\0';
+ aa0s++;
+ memcpy(aa0s,aa0[0],n0);
+ qshuffle(aa0s,n0,nm0);
+ init_work (aa0s, n0, my_ppst, &qf_str);
+ }
+
+ ix_score = my_ppst->score_ix;
+ debug_lib = my_ppst->debug_lib;
+ zsflag = my_ppst->zsflag;
+ zs_win = my_ppst->zs_win;
+
+ if (zsflag >= 10) {
+ if((aa1s=calloc(work_info->max_tot+1,sizeof(char))) == NULL) {
+ fprintf(stderr,"unable to allocate shuffled library sequence\n");
+ }
+ else {
+ *aa1s=0;
+ aa1s++;
+ do_shuffle =1;
+ irand(0);
+ }
+ }
+ else {do_shuffle = 0;}
+
+ /* main work loop */
+ while (get_wbuf(&cur_buf,work_info->max_work_buf)) {
+
+ cur_cnt = cur_buf->buf_cnt;
+ if (cur_cnt == -1) break;
+ cur_buf_p = cur_buf->buf;
+
+ while (cur_cnt--) { /* count down the number of sequences */
+ p_rbuf = cur_buf_p++; /* step through each sequence */
+ p_rbuf->rst.score[0] = p_rbuf->rst.score[1] = p_rbuf->rst.score[2] = 0;
+ frame = p_rbuf->frame;
+
+#ifdef DEBUG
+ if (debug_lib) {
+ if (frame >= 2) fprintf(stderr,"* frame: %d\n",frame);
+ for (i=0; i<p_rbuf->n1; i++)
+ if (p_rbuf->aa1b[i]>=my_ppst->nsqx) {
+ fprintf(stderr,
+ "%s residue[%d/%d] %d range (%d)\n",
+ p_rbuf->libstr,i,p_rbuf->n1,p_rbuf->aa1b[i],my_ppst->nsqx);
+ p_rbuf->aa1b[i]=0;
+ p_rbuf->n1=i-1;
+ break;
+ }
+ }
+#endif
+
+ do_work (aa0[frame], n0, p_rbuf->aa1b, p_rbuf->n1, frame,
+ my_ppst, f_str[frame], 0, &p_rbuf->rst);
+
+ if (work_info->qshuffle) {
+ do_work(aa0s,n0,p_rbuf->aa1b, p_rbuf->n1, frame,
+ my_ppst, qf_str, 1, &rrst);
+ p_rbuf->qr_score = rrst.score[ix_score];
+ p_rbuf->qr_escore = rrst.escore;
+ }
+
+ if (do_shuffle) {
+ if (zs_win > 0) wshuffle(p_rbuf->aa1b,aa1s,p_rbuf->n1,zs_win,&ieven);
+ else shuffle(p_rbuf->aa1b,aa1s,p_rbuf->n1);
+
+ do_work (aa0[frame], n0, aa1s, p_rbuf->n1, frame,
+ my_ppst, f_str[frame], 0, &rrst);
+ p_rbuf->r_score = rrst.score[ix_score];
+ p_rbuf->r_escore = rrst.escore;
+ }
+ }
+
+ cur_buf->have_results = 1;
+
+ put_wbuf(cur_buf,work_info->max_work_buf);
+
+ } /* end main while */
+
+ close_work(aa0[0], n0, my_ppst, &f_str[0]);
+ free(aa0[0]-1);
+ if (work_info->qframe == 2) {
+ close_work(aa0[1], n0, my_ppst, &f_str[1]);
+ free(aa0[1]-1);
+ }
+
+ if (do_shuffle) free(aa1s-1);
+
+ if (my_worker) {
+ free(my_pst.pam2[1][0]);
+ free(my_pst.pam2[0][0]);
+ free(my_pst.pam2[1]);
+ free(my_pst.pam2[0]);
+
+ if (my_pst.pam_pssm) {
+ free_pam2p(my_pst.pam2p[0]);
+ free_pam2p(my_pst.pam2p[1]);
+ }
+ }
+
+#ifdef WIN32
+ pthread_exit(&work_info->status);
+#else
+ THR_EXIT(&work_info->status);
+#endif
+
+} /* end work_thread */
+