#include <stdio.h>
#include <stdint.h>
#include <strsafe.h>
#include <windows.h>

#include "ringbuffer.h"
#include "pipe.h"

#ifdef _MSC_VER
  #define inline __inline
#endif

/* Integrate every element of the signal. */
void increment(pipe_t * const p)
{
	while (Pipe_isFilled(p))
	{
		uint32_t item = Pipe_Read(p);
		item++;
		Pipe_Write(p, item);
	}
}

/* Square every element of the signal. */
void square(pipe_t * const p)
{
	while (Pipe_isFilled(p))
	{
		uint32_t item = Pipe_Read(p);
		item = item * item;
		Pipe_Write(p, item);
	}
}

/* Integrate over every element of the signal. */
void integrate(pipe_t * const pipe)
{
  uint32_t state = *((uint32_t*)pipe->state);

  while (Pipe_isFilled(pipe))
	{
    uint32_t item = Pipe_Read(pipe);
		state = state + item;
    Pipe_Write(pipe, state);
	}

  *((uint32_t*)pipe->state) = state;
}

/* Build the sum of all elements of the signal. */
void sum(pipe_t * const pipe)
{
	uint32_t sum = 0;

  while (Pipe_isFilled(pipe))
    sum += Pipe_Read(pipe);
  Pipe_Write(pipe, sum);
}

/* Build the average of all elements of the signal. */
void average(pipe_t * const pipe)
{
	uint32_t sum = 0;
	uint32_t element_counter = 0;
	uint32_t average = 0;

  while (Pipe_isFilled(pipe))
	{
    sum += Pipe_Read(pipe);
		element_counter++;
	}
	average = sum / element_counter;
  Pipe_Write(pipe, average);
}

/* Print the signal. */
void print(pipe_t * const pipe)
{
	printf("\nOutput:\n");
  while (Pipe_isFilled(pipe))
    printf("%d\n", Pipe_Read(pipe));
}

/* Logging function. Set by user. */
void log(pipe_t * const source, pipe_t * const target, uint32_t element)
{
  if (Pipe_isFull(target))
    printf("Error: Pipe %s is full!\n", target->name);

  if (source->state == NULL && target->state == NULL)
    printf("%s -> %d -> %s\n", source->name, element, target->name);
  else if (source->state != NULL && target->state != NULL)
    printf("%s(%d) -> %d -> %s(%d)\n", source->name, *((uint32_t*)source->state), element, target->name, *((uint32_t*)target->state));
  else if (source->state != NULL)
    printf("%s(%d) -> %d -> %s\n", source->name, *((uint32_t*)source->state), element, target->name);
	else
    printf("%s -> %d -> %s(%d)\n", source->name, element, target->name, *((uint32_t*)target->state));
}


#define BUF_SIZE 255

void DisplayMessage(HANDLE hScreen,
  char *ThreadName, int Data, int Count)
{

  TCHAR msgBuf[BUF_SIZE];
  size_t cchStringSize;
  DWORD dwChars;

  // Print message using thread-safe functions.
  StringCchPrintf(msgBuf, BUF_SIZE,
    TEXT("Executing iteration %02d of %s"
    " having data = %02d \n"),
    Count, ThreadName, Data);
  StringCchLength(msgBuf, BUF_SIZE, &cchStringSize);
  WriteConsole(hScreen, msgBuf, cchStringSize,
    &dwChars, NULL);
}

DWORD WINAPI Thread_no_2(LPVOID lpParam)
{

  int     Data = 0;
  int     count = 0;
  HANDLE  hStdout = NULL;

  // Get Handle To screen.
  // Else how will we print?
  if ((hStdout =
    GetStdHandle(STD_OUTPUT_HANDLE))
    == INVALID_HANDLE_VALUE)
    return 1;

  // Cast the parameter to the correct
  // data type passed by callee i.e main() in our case.
  Data = *((int*)lpParam);

  for (count = 0; count <= 4; count++)
  {
    DisplayMessage(hStdout, "Thread_no_1", Data, count);
  }

  return 0;
}

DWORD WINAPI Thread_no_1(LPVOID lpParam)
{

  int     Data = 0;
  int     count = 0;
  HANDLE  hStdout = NULL;

  // Get Handle To screen.
  // Else how will we print?
  if ((hStdout =
    GetStdHandle(STD_OUTPUT_HANDLE))
    == INVALID_HANDLE_VALUE)
    return 1;

  // Cast the parameter to the correct
  // data type passed by callee i.e main() in our case.
  Data = *((int*)lpParam);

  for (count = 0; count <= 4; count++)
  {
    DisplayMessage(hStdout, "Thread_no_2", Data, count);
  }

  return 0;
}

int main(void)
{
	uint32_t counter = 0;

	/* Create pipes and connect them */
	Pipe_Create(increment_pipe, 4, 1, NULL, NULL);
  Pipe_Create(square_pipe, 4, 1, NULL, NULL);
	Pipe_Create(integrate_pipe, 8, 2, &counter, log);
	Pipe_Create(sum_pipe, 8, 1, NULL, log);
	Pipe_Create(average_pipe, 8, 1, NULL, log);
	Pipe_Create(print_pipe, 4, 1, NULL, log);

	Pipe_Connect(&increment_pipe, &integrate_pipe);
	Pipe_Connect(&square_pipe, &integrate_pipe);
	Pipe_Connect(&integrate_pipe, &sum_pipe);
	Pipe_Connect(&integrate_pipe, &average_pipe);
	Pipe_Connect(&sum_pipe, &print_pipe);
	Pipe_Connect(&average_pipe, &print_pipe);

  /* Create Threads */
  int Data_Of_Thread_1 = 1;
  int Data_Of_Thread_2 = 2;
  HANDLE Handle_Of_Thread_1 = 0;
  HANDLE Handle_Of_Thread_2 = 0;

  Handle_Of_Thread_1 = CreateThread(NULL, 0, Thread_no_1, &Data_Of_Thread_1, 0, NULL);
  if (Handle_Of_Thread_1 == NULL)
    ExitProcess(Data_Of_Thread_1);

  Handle_Of_Thread_2 = CreateThread(NULL, 0, Thread_no_2, &Data_Of_Thread_2, 0, NULL);
  if (Handle_Of_Thread_2 == NULL)
    ExitProcess(Data_Of_Thread_2);

  HANDLE Array_Of_Thread_Handles[] = { Handle_Of_Thread_1, Handle_Of_Thread_1 };
  WaitForMultipleObjects(3, Array_Of_Thread_Handles, TRUE, INFINITE);

  CloseHandle(Handle_Of_Thread_1);
  CloseHandle(Handle_Of_Thread_2);

	/* Generate input */
	Pipe_Insert(&increment_pipe, 1);
	Pipe_Insert(&increment_pipe, 3);
	Pipe_Insert(&increment_pipe, 5);

	Pipe_Insert(&square_pipe, 2);
	Pipe_Insert(&square_pipe, 4);
	Pipe_Insert(&square_pipe, 6);

	/* run the functions (each can run in an own thread) */
	increment(&increment_pipe);
	square(&square_pipe);
	integrate(&integrate_pipe);
	sum(&sum_pipe);
	average(&average_pipe);
	print(&print_pipe);

	getchar();

	return 0;
}