#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;
  Rcvgraph *gr;
public:
  double liperf;
  double ltperf;
public:
  Recv(int K_, int maxN_);
  ~Recv();
  void newp(int q);
  void sample();
  void rate(int T, int s1, int s2, int s3);
};


/*
main()
{
  randseed();
  Recv rx(18,54);
  rx.rate(15000,3,6,13);
}
*/

main()
{
  randseed();
  int Nn=18;
  Recv rx(18,54);

  for(int s1=2;s1<=3 && s1<Nn;s1++)
    for(int s2=1;s2<Nn;s2++)
      for(int s3=s2+1;s3<Nn;s3++)
        if(!(s2==1 && s3==s1))
        {
          rx.rate(10000,s1,s2,s3);
          printf("%d,%d,%d,%f,%f\r\n",
            s1,s2,s3,rx.ltperf,rx.liperf);
        }
}


void Recv::rate(int T, int s1, int s2, int s3)
{
  gr=new Rcvgraph(K,0,2*K);
  gr->kring_sides(s1,s2,s3);

  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]++;
  }

  delete gr;


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

  FILE *f;
  char s[1024];
  sprintf(s,"out-K%d-maxN%d-T=%d.csv",K,maxN,T);
  f=fopen(s,"w");
  fprintf(f,",LT,linear\r\n");
  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;

  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);
}