/* axtSplitByTarget - Split a single axt file into one file per target. */

/* Copyright (C) 2011 The Regents of the University of California 
 * See kent/LICENSE or http://genome.ucsc.edu/license/ for licensing information. */
#include "common.h"
#include "linefile.h"
#include "hash.h"
#include "options.h"
#include "portable.h"
#include "axt.h"


int tStartSize = 0;
double tSS = 0.0;
int chunkSize = 0;
int totalWritten = 0;

void usage()
/* Explain usage and exit. */
{
errAbort(
  "axtSplitByTarget - Split a single axt file into one file per target\n"
  "usage:\n"
  "   axtSplitByTarget in.axt outDir\n"
  "options:\n"
  "   -tStartSize=N - Split into multiple files per target, with each file\n"
  "                   outDir/tName.M.axt containing records whose tStart is\n"
  "                   between M*N and ((M+1)*N)-1, i.e. M = floor(tStart/N).\n"
  "   -chunkSize=N - split input into files of size N (multiple targets per file)\n"
  " Only one of chunkSize and tStartSize can be set.\n"
  );
}

static FILE *getSplitFile(struct hash *outHash, char *outDir, char *tName,
			  int tStart)
{
static char lastName[1024];
static char outName[1024];
static int fileCount = 0;
FILE *f;
if (tStartSize > 0)
    {
    int fNum = (int)floor((double)tStart / tSS);
    safef(outName, sizeof(outName), "%s/%s.%d.axt", outDir, tName, fNum);
    }
else if (chunkSize > 0)
    {
    if ((outName[0] == 0) || ((totalWritten > chunkSize) && !sameString(lastName, tName)))
	{
	if ((f = hashFindVal(outHash, outName)) != NULL)
	    fclose(f);
	safef(outName, sizeof(outName), "%s/chunk%d.axt",outDir,fileCount++);
	totalWritten = 0;
	}
    strcpy(lastName, tName);
    }
else
    {
    safef(outName, sizeof(outName), "%s/%s.axt", outDir, tName);
    }
f = hashFindVal(outHash, outName);
if (f == NULL)
    {
    f = mustOpen(outName, "w");
    hashAdd(outHash, outName, f);
    }
return f;
}

void axtSplitByTarget(char *inName, char *outDir)
/* axtSplitByTarget - Split a single axt file into one file per target. */
{
struct hash *outHash = newHash(8);  /* FILE valued hash */
struct lineFile *lf = lineFileOpen(inName, TRUE);
struct axt *axt;

makeDir(outDir);
while ((axt = axtRead(lf)) != NULL)
    {
    FILE *f = getSplitFile(outHash, outDir, axt->tName, axt->tStart);
    axtWrite(axt, f);
    totalWritten += strlen(axt->tName) + strlen(axt->qName) + 40 + strlen(axt->qSym)+ strlen(axt->tSym);
    axtFree(&axt);
    }
}

int main(int argc, char *argv[])
/* Process command line. */
{
optionHash(&argc, argv);
if (argc != 3)
    usage();
tStartSize = optionInt("tStartSize", 0);
chunkSize = optionInt("chunkSize", 0);
if (chunkSize && tStartSize)
    usage();
tSS = (double)tStartSize;
axtSplitByTarget(argv[1], argv[2]);
return 0;
}