!
! Simulator for terminated convolutional codes (so, far, only rate 1/2)
! BPSK on AWGN Channel
!
! Hybrid decoder - Fano Sequential Decoder and Ordered Statistics Decoder (OSD)a
!
program tccsim

  parameter (N=162,K=50) 
  integer*1 gen(K,N)
  integer*1 gg(64)
  integer*1 mbits(50),mbits2(50)
  integer*4 mettab(-128:127,0:1)

  parameter (NSYM=162)
  parameter (MAXSYM=220)
  character*12 arg
  character*22 msg,msg2
  integer*1 data0(11)
  integer*1 data1(11)
  integer*1 dat(162)
  integer*1 softsym(162)
  integer*1 apmask(162),cw(162)
  real*4    xx0(0:255)
  real      ss(162)

  character*64 g    ! Interleaved polynomial coefficients
  data g/'1101010010001100101001011101100001000000100111100010010010111111'/

  data xx0/                                                      & !Metric table
        1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000,  &
        1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000,  &
        1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000,  &
        1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000,  &
        1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000,  &
        1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000, 1.000,  &
        0.988, 1.000, 0.991, 0.993, 1.000, 0.995, 1.000, 0.991,  &
        1.000, 0.991, 0.992, 0.991, 0.990, 0.990, 0.992, 0.996,  &
        0.990, 0.994, 0.993, 0.991, 0.992, 0.989, 0.991, 0.987,  &
        0.985, 0.989, 0.984, 0.983, 0.979, 0.977, 0.971, 0.975,  &
        0.974, 0.970, 0.970, 0.970, 0.967, 0.962, 0.960, 0.957,  &
        0.956, 0.953, 0.942, 0.946, 0.937, 0.933, 0.929, 0.920,  &
        0.917, 0.911, 0.903, 0.895, 0.884, 0.877, 0.869, 0.858,  &
        0.846, 0.834, 0.821, 0.806, 0.790, 0.775, 0.755, 0.737,  &
        0.713, 0.691, 0.667, 0.640, 0.612, 0.581, 0.548, 0.510,  &
        0.472, 0.425, 0.378, 0.328, 0.274, 0.212, 0.146, 0.075,  &
        0.000,-0.079,-0.163,-0.249,-0.338,-0.425,-0.514,-0.606,  &
       -0.706,-0.796,-0.895,-0.987,-1.084,-1.181,-1.280,-1.376,  &
       -1.473,-1.587,-1.678,-1.790,-1.882,-1.992,-2.096,-2.201,  &
       -2.301,-2.411,-2.531,-2.608,-2.690,-2.829,-2.939,-3.058,  &
       -3.164,-3.212,-3.377,-3.463,-3.550,-3.768,-3.677,-3.975,  &
       -4.062,-4.098,-4.186,-4.261,-4.472,-4.621,-4.623,-4.608,  &
       -4.822,-4.870,-4.652,-4.954,-5.108,-5.377,-5.544,-5.995,  &
       -5.632,-5.826,-6.304,-6.002,-6.559,-6.369,-6.658,-7.016,  &
       -6.184,-7.332,-6.534,-6.152,-6.113,-6.288,-6.426,-6.313,  &
       -9.966,-6.371,-9.966,-7.055,-9.966,-6.629,-6.313,-9.966,  &
       -5.858,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,  &
       -9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,  &
       -9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,  &
       -9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,  &
       -9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,  &
       -9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966,-9.966/

  bias=0.42
  scale=120
!  ndelta=nint(3.4*scale)
  ndelta=100
  ib=150
  slope=2
  do i=0,255
    mettab(i-128,0)=nint(scale*(xx0(i)-bias))
    if(i.gt.ib) mettab(i-128,0)=mettab(ib-128,0)-slope*(i-ib)
    if(i.ge.1) mettab(128-i,1)=mettab(i-128,0)
  enddo
  mettab(-128,1)=mettab(-127,1)
  
! Get command-line argument(s)
  nargs=iargc()
  if(nargs.ne.3) then
     print*,'Usage: tccsim "message" ntrials ndepth'
     go to 999
  endif
  call getarg(1,msg)                             !Get message from command line
  write(*,1000) msg
1000 format('Message: ',a22)
  call getarg(2,arg) 
  read(arg,*) ntrials 
  call getarg(3,arg) 
  read(arg,*) ndepth 

  nbits=50+31               !User bits=99, constraint length=32
  nbytes=(nbits+7)/8
  limit=20000

  data0=0
!  call wqencode(msg,ntype0,data0)             !Source encoding
data0(1:7)=1
  write(*,1002) data0(1:7),data0(1:6),data0(7)/64
1002 format(/'Source-encoded message, 50 bits:'/'Hex:    ',7z3.2/    &
          'Binary: ',6b9.8,b3.2)

! Demonstrates how to create the generator matrix from a string that contains the interleaved 
! polynomial coefficients
  gen=0
  do j=1,64
    read(g(j:j),"(i1)") gg(j)  ! read polynomial coeffs from string
  enddo
  do i=1,K
    gen(i,2*(i-1)+1:2*(i-1)+64)=gg  ! fill the generator matrix with cyclic shifts of gg
  enddo

! get message bits from data0      
  nbits=0
  do i=1,7
    do ib=7,0,-1
      nbits=nbits+1
      if(nbits .le. 50) then
        mbits(nbits)=0
        if(btest(data0(i),ib)) mbits(nbits)=1
      endif
    enddo
  enddo

! Encode message bits using the generator matrix, generating a 162-bit codeword.
  cw=0
  do i=1,50
    if(mbits(i).eq.1) cw=mod(cw+gen(i,:),2)
  enddo

  write(*,*) 'Codeword from generator matrix: '
  write(*,'(162i1)') cw
 
  call encode232(data0,nbytes,dat,MAXSYM)     !Convolutional encoding
  write(*,*) 'Codeword from encode232: '
  write(*,'(162i1)') dat

!  call inter_mept(dat,1)                      !Interleaving

! Here, we have channel symbols.

!  call inter_mept(dat,-1)                     !Remove interleaving

  call init_random_seed()
  call sgran()

  do isnr=10,-20,-1
    sigma=1/sqrt(2*(10**((isnr/2.0)/10.0)))
    ngood=0
    nbad=0
    do i=1,ntrials 
      do j=1,162
        ss(j)=-(2*dat(j)-1)+sigma*gran()           !Simulate soft symbols
      enddo

      rms=sqrt(sum(ss**2))
      ss=100*ss/rms
      where(ss>127.0) ss=127.0
      where(ss<-127.0) ss=-127.0
      softsym=ss

! Call the sequential (Fano algorithm) decoder
      nbits=50+31
      call fano232(softsym,nbits,mettab,ndelta,limit,data1,ncycles,metric,nerr)
      iflag=0
      nhardmin=0
      dmin=0.0

! If Fano fails, call OSD
      if(nerr.ne.0 .and. ndepth.ge.0) then 
        apmask=0
        cw=0
        call osdwspr(-ss/127,apmask,ndepth,cw,nhardmin,dmin)

! OSD produces a codeword, but code is not systematic
! Use Fano with hard decisions to retrieve the message from the codeword
        cw=-(2*cw-1)*64
        nbits=50+31
        call fano232(cw,nbits,mettab,ndelta,limit,data1,ncycles,metric,nerr)
        iflag=1
      endif

!  call wqdecode(data1,msg2,ntype1)         
!  write(*,*) msg2,iflag,nhardmin,dmin
      if(nerr.eq.0 .and. any(data1.ne.data0)) nbad=nbad+1
      if(nerr.eq.0 .and. all(data1.eq.data0)) ngood=ngood+1
    enddo

  write(*,'(f4.1,i8,i8)') isnr/2.0,ngood,nbad
  enddo

999 end program tccsim

include '../wsprcode/wspr_old_subs.f90'