#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream.h>
#include <fstream.h>

#include "e:\opt\common\timings.h"

int main(int argc, char *argv[])
{
    const int BUFSIZE = 16384;
    const int BUFCOUNT = 256;
    const int INPUTLINESIZE = 1024;
    char InputLine[INPUTLINESIZE];
    char *Buffer[BUFCOUNT];
    int Displacement[BUFCOUNT];
    int TotalDisplacement[BUFCOUNT];
    char CurrentChar;
    char* InputFileName;
    char* OutputFileName;
    ifstream InputFile;
    ofstream OutputFile;
    int PassCount;
    int CurrentLength;
    int NewLength;
    int LineLength;
    char* OriginalInputFileName;
    char* OriginalOutputFileName;
    int TotalKeys = 0;
    int TotalData = 0;
    bool StatisticsDisplayed = false;
    int TotalWrites = 0;
    int i;

    if (argc < 4)
        {
        printf("Usage: zensort passcount infile outfile\n");
        exit(1);
        }
    else
        {
        PassCount = atoi(argv[1]);
        OriginalInputFileName = argv[2];
        OriginalOutputFileName = argv[3];
        }

char temp[100];
start_timing();

    for (i = 0; i < BUFCOUNT; i ++)
         {
         Buffer[i] = new char[BUFSIZE];
         }

    for (int Pass = PassCount - 1; Pass >= 0; Pass --)
        {

        if ((PassCount - Pass) % 2 == 1)
            {
            InputFileName = OriginalInputFileName;
            OutputFileName = OriginalOutputFileName;
            }
        else
            {
            InputFileName = OriginalOutputFileName;
            OutputFileName = OriginalInputFileName;
            }

        InputFile.open(InputFileName,ios::in|ios::binary);
        OutputFile.open(OutputFileName,ios::out|ios::binary);

        for (i = 0; i < BUFCOUNT; i ++)
            {
            memset(Buffer[i],0,BUFSIZE);
            Displacement[i] = 0;
            TotalDisplacement[i] = 0;
            }

        while (true)
            {
            InputFile.getline(InputLine,INPUTLINESIZE);
            if (!InputFile)
                break;
            TotalKeys ++;
            LineLength = strlen(InputLine);
            TotalData += LineLength + 1;
            if (Pass >= LineLength)
                CurrentChar = 0;
            else
                CurrentChar = InputLine[Pass];
            Displacement[CurrentChar] += LineLength + 1;
            }
        InputFile.close();

        for (i = 1; i < BUFCOUNT; i ++)
            {
            TotalDisplacement[i] = TotalDisplacement[i-1] + Displacement[i-1];
            }

if ((Pass == PassCount - 1) && StatisticsDisplayed == false)
  {
  printf("Total keys: %d\n", TotalKeys);
  printf("Total data: %d\n", TotalData);
  StatisticsDisplayed = true;
  }

sprintf(temp,"Finished counting on pass %d",PassCount-Pass);
timing(temp);

        InputFile.open(InputFileName,ios::in|ios::binary);

        while (true)
            {
            InputFile.getline(InputLine,INPUTLINESIZE);
            if (!InputFile)
                break;
            strcat(InputLine,"\n");
            LineLength = strlen(InputLine);
            if (Pass >= LineLength-1)
                CurrentChar = 0;
            else
                CurrentChar = InputLine[Pass];
            CurrentLength = strlen(Buffer[CurrentChar]);
            NewLength =  CurrentLength + LineLength;
            if (NewLength >= BUFSIZE)
                {
                OutputFile.seekp(TotalDisplacement[CurrentChar]);
                TotalDisplacement[CurrentChar] += CurrentLength;
                OutputFile.write(Buffer[CurrentChar],CurrentLength);
                TotalWrites ++;
                memset(Buffer[CurrentChar],0,BUFSIZE);
                }
            strcat(Buffer[CurrentChar],InputLine);
            }

        for (i = 0; i < BUFCOUNT; i ++)
            {
            CurrentLength = strlen(Buffer[i]);
            if (CurrentLength > 0)
                {
                OutputFile.seekp(TotalDisplacement[i]);
                OutputFile.write(Buffer[i],CurrentLength);
                TotalWrites ++;
                }
            }
        InputFile.close();
        OutputFile.close();

sprintf(temp,"Finished distributing on pass %d",PassCount-Pass);
timing(temp);
    }

    printf("Total writes: %d\n", TotalWrites);
    
    for (i = 0; i < BUFCOUNT; i ++)
        {
        delete [] Buffer[i];
        }

    end_timing();

    return 0;
}