Merged master 8748
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@@ -1,168 +0,0 @@
|
||||
/* ALIST-TO-PCHK.C - Convert a parity check matrix from alist format. */
|
||||
|
||||
/* Copyright (c) 1995-2012 by Radford M. Neal.
|
||||
*
|
||||
* Permission is granted for anyone to copy, use, modify, and distribute
|
||||
* these programs and accompanying documents for any purpose, provided
|
||||
* this copyright notice is retained and prominently displayed, and note
|
||||
* is made of any changes made to these programs. These programs and
|
||||
* documents are distributed without any warranty, express or implied.
|
||||
* As the programs were written for research purposes only, they have not
|
||||
* been tested to the degree that would be advisable in any important
|
||||
* application. All use of these programs is entirely at the user's own
|
||||
* risk.
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <math.h>
|
||||
|
||||
#include "alloc.h"
|
||||
#include "intio.h"
|
||||
#include "open.h"
|
||||
#include "mod2sparse.h"
|
||||
#include "mod2dense.h"
|
||||
#include "mod2convert.h"
|
||||
#include "rcode.h"
|
||||
|
||||
|
||||
void bad_alist_file(void);
|
||||
void usage(void);
|
||||
|
||||
|
||||
/* MAIN PROGRAM. */
|
||||
|
||||
int main
|
||||
( int argc,
|
||||
char **argv
|
||||
)
|
||||
{
|
||||
char *alist_file, *pchk_file;
|
||||
FILE *af, *pf;
|
||||
int mxrw, mxcw;
|
||||
int *rw, *cw;
|
||||
int i, j, k;
|
||||
int tot, trans;
|
||||
int nxt;
|
||||
|
||||
trans = 0;
|
||||
|
||||
for (;;)
|
||||
{
|
||||
if (argc>1 && strcmp(argv[1],"-t")==0)
|
||||
{ trans = 1;
|
||||
argc -= 1;
|
||||
argv += 1;
|
||||
}
|
||||
else
|
||||
{ break;
|
||||
}
|
||||
}
|
||||
|
||||
if (argc!=3)
|
||||
{ usage();
|
||||
}
|
||||
|
||||
pchk_file = argv[2];
|
||||
alist_file = argv[1];
|
||||
|
||||
af = open_file_std(alist_file,"r");
|
||||
|
||||
if (af==NULL)
|
||||
{ fprintf(stderr,"Can't open alist file: %s\n",alist_file);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if (fscanf(af,"%d",&M)!=1 || M<1
|
||||
|| fscanf(af,"%d",&N)!=1 || N<1
|
||||
|| fscanf(af,"%d",&mxrw)!=1 || mxrw<0 || mxrw>N
|
||||
|| fscanf(af,"%d",&mxcw)!=1 || mxcw<0 || mxcw>M)
|
||||
{ bad_alist_file();
|
||||
}
|
||||
|
||||
rw = (int *) chk_alloc (M, sizeof *rw);
|
||||
|
||||
for (i = 0; i<M; i++)
|
||||
{ if (fscanf(af,"%d",&rw[i])!=1 || rw[i]<0 || rw[i]>N)
|
||||
{ bad_alist_file();
|
||||
}
|
||||
}
|
||||
|
||||
cw = (int *) chk_alloc (N, sizeof *cw);
|
||||
|
||||
for (j = 0; j<N; j++)
|
||||
{ if (fscanf(af,"%d",&cw[j])!=1 || cw[j]<0 || cw[j]>M)
|
||||
{ bad_alist_file();
|
||||
}
|
||||
}
|
||||
|
||||
H = mod2sparse_allocate(M,N);
|
||||
|
||||
do { if (fscanf(af,"%d",&nxt)!=1) nxt = -1; } while (nxt==0);
|
||||
|
||||
tot = 0;
|
||||
|
||||
for (i = 0; i<M; i++)
|
||||
{ for (k = 0; k<rw[i]; k++)
|
||||
{ if (nxt<=0 || nxt>N || mod2sparse_find(H,i,nxt-1))
|
||||
{ bad_alist_file();
|
||||
}
|
||||
mod2sparse_insert(H,i,nxt-1);
|
||||
tot += 1;
|
||||
do { if (fscanf(af,"%d",&nxt)!=1) nxt = -1; } while (nxt==0);
|
||||
}
|
||||
}
|
||||
|
||||
for (j = 0; j<N; j++)
|
||||
{ for (k = 0; k<cw[j]; k++)
|
||||
{ if (nxt<=0 || nxt>M || !mod2sparse_find(H,nxt-1,j))
|
||||
{ bad_alist_file();
|
||||
}
|
||||
tot -= 1;
|
||||
do { if (fscanf(af,"%d",&nxt)!=1) nxt = -1; } while (nxt==0);
|
||||
}
|
||||
}
|
||||
|
||||
if (tot!=0 || nxt!=-1 || !feof(af))
|
||||
{ bad_alist_file();
|
||||
}
|
||||
|
||||
if (trans)
|
||||
{ mod2sparse *HT;
|
||||
HT = H;
|
||||
H = mod2sparse_allocate(N,M);
|
||||
mod2sparse_transpose(HT,H);
|
||||
}
|
||||
|
||||
pf = open_file_std(pchk_file,"wb");
|
||||
if (pf==NULL)
|
||||
{ fprintf(stderr,"Can't create parity check file: %s\n",pchk_file);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
intio_write(pf,('P'<<8)+0x80);
|
||||
|
||||
if (ferror(pf) || !mod2sparse_write(pf,H) || fclose(pf)!=0)
|
||||
{ fprintf(stderr,"Error writing to parity check file %s\n",pchk_file);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
/* COMPLAIN THAT ALIST FILE IS BAD. */
|
||||
|
||||
void bad_alist_file()
|
||||
{ fprintf(stderr,"Alist file doesn't have the right format\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
|
||||
/* PRINT USAGE MESSAGE AND EXIT. */
|
||||
|
||||
void usage(void)
|
||||
{ fprintf(stderr,"Usage: alist-to-pchk [ -t ] alist-file pchk-file\n");
|
||||
exit(1);
|
||||
}
|
||||
Reference in New Issue
Block a user