#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <math.h>
#include <iostream>
using namespace std;

class Linear;
class LT;

#include "lt.h"
#include "linear.h"
#include "graph.h"

void randseed();

class Recv
{
  LT *lt;
  Linear *li;
  int K;
  int maxN;
  int *p;
  int lin;
  int ltn;
  int *liprob;
  int *ltprob;
public:
  double liexcess;
  double ltexcess;
  Rcvgraph *gr;
public:
  Recv(int K_, int maxN_);
  ~Recv();
  void newp(int q);
  void sample();
  void rate(int T);
};

main()
{
  randseed();
  int K=30;
  Recv rx(K,3*K);

  rx.gr=new Rcvgraph(K,3*K);
  rx.gr->setmode(3);

  const int n=7; //degree

  for(int t=t;t<3000;t++)
  {

    int a[n];
    int b[n];
    
    a[0]=0;
    a[1]=1+rand()%(K-n+1);

    for(int j=2;j<n;j++)
      a[j]=a[j-1]+1+rand()%(K-(a[j-1]+1)-(n-1-j));

    int k;
    for(k=0;k<n;k++)
      if(a[k]!=k)
        break;
    if(k==n)
      continue;


    b[0]=0;
    b[1]=a[1]+rand()%(K-a[1]-n+2);

    for(int j=2;j<n;j++)
      b[j]=b[j-1]+1+rand()%(K-(b[j-1]+1)-(n-1-j));


    for(k=0;k<n;k++)
      if(b[k]!=k)
        break;
    if(k==n)
      continue;

    for(k=0;k<n;k++)
      if(b[k]!=a[k])
        break;
    if(k==n)
      continue;


    rx.gr->kring_3dN(a,b,n);
    rx.rate(4000);

    printf("%4d,",t);

    for(int j=1;j<n;j++)
      printf("%2d,",a[j]);
    for(int j=1;j<n;j++)
      printf("%2d,",b[j]);

    printf("%f,%f\r\n",rx.ltexcess,rx.liexcess);
  }

  delete rx.gr;
}

void Recv::rate(int T)
{

  for(int i=0;i<=maxN;i++)
  {
    liprob[i]=0;
    ltprob[i]=0;
  }

  for(int t=0;t<T;t++)
  {
    sample();
    for(int i=ltn;i<=maxN;i++)
      ltprob[i]++;
    for(int i=lin;i<=maxN;i++)
      liprob[i]++;
  }

  ltexcess=0;
  liexcess=0;
  for(int i=K;i<=maxN;i++)
  {
    ltexcess+=T-(double)ltprob[i];
    liexcess+=T-(double)liprob[i];
  }
  ltexcess/=T;
  liexcess/=T;

  FILE *f;
  char s[1024];
  sprintf(s,"out-K%d-maxN%d-T%d-mode%d.csv",K,maxN,T,gr->getmode());
  f=fopen(s,"w");
  fprintf(f,",LT %d,linear %d\r\n",gr->getmode(),gr->getmode());
  for(int i=K;i<=maxN;i++)
    fprintf(f,"%3d, %f, %f\r\n",i,
      (double)ltprob[i]/T,
      (double)liprob[i]/T);
  fclose(f);

}

void Recv::sample()
{
  lt->init();
  li->init();
  int ltr=0;
  int lir=0;
  lin=maxN+1;
  ltn=maxN+1;

  if(gr->getmode()==3)
    gr->init();
  for(int i=0; i<maxN && !(ltr==K && lir==K) ;i++)
  {
    int q=rand()%3+1;
    newp(q);

    gr->rx(p);

    if(!(ltr==K))
    {
      ltr=lt->recv(p);
      if(ltr==K)
        ltn=i+1;
    }
    if(!(lir==K))
    {
      lir=li->recv(p);
      if(lir==K)
        lin=i+1;
    }
  }
}

void Recv::newp(int q)
{

  if(q==0)
  {
    for(int i=0;i<K;i++)
      p[i]=rand()%2;
    return;
  }

  for(int i=0;i<K;i++)
    p[i]=0;

  for(int n=0;n<q;n++)
  {
    int i = rand()%(K-n);
    int j = -1;
    for(;i>=0;i--)
    {
      j++;
      for(;p[j];j++)
        ;
    }
    p[j]=1;
  }
}

Recv::Recv(int K_, int maxN_)
{
  K=K_;
  maxN=maxN_;
  lt=new LT(K,maxN);
  li=new Linear(K,maxN);
  lt->init();
  li->init();
  p=new int[K];
  liprob=new int[maxN+1];
  ltprob=new int[maxN+1];

}

Recv::~Recv()
{
  delete lt;
  delete li;
  delete[] p;
  delete[] liprob;
  delete[] ltprob;
}

void randseed()
{
  int ti=time(0);
  printf("srand=%d\r\n",ti);
  srand(ti);
}