#include <GooVector.h>
#include <PDFDoc.h>
#include "parseargs.h"
#include "config.h"
#include <poppler-config.h>

static GBool printVersion = gFalse;
static GBool printHelp = gFalse;

static const ArgDesc argDesc[] = {
  {"-v",       argFlag,     &printVersion,  0,
   "print copyright and version info"},
  {"-h",       argFlag,     &printHelp,     0,
   "print usage information"},
  {"-help",    argFlag,     &printHelp,     0,
   "print usage information"},
  {"--help",   argFlag,     &printHelp,     0,
   "print usage information"},
  {"-?",       argFlag,     &printHelp,     0,
   "print usage information"},
  {NULL}
};

///////////////////////////////////////////////////////////////////////////
int main(int argc, char *argv[])
///////////////////////////////////////////////////////////////////////////
// Merge PDF files given by arguments 1 to argc-2 and write the result
// to the file specified by argument argc-1.
///////////////////////////////////////////////////////////////////////////
{
	int objectsCount = 0;
	Guint numOffset = 0;
	GooVector<Object> pages;
	GooVector<Guint> offsets;
	XRef *yRef;
	FILE *f;
	OutStream *outStr;
	int i;
	int j, rootNum;	
	GooVector<PDFDoc *> docs;
	double version = 0;
	char *fileName = argv[argc-1];
	int exitCode;
	
	exitCode = 99;
	if (argc <= 3 || printVersion || printHelp) {
		fprintf(stderr, "pdfmerge version %s\n", PACKAGE_VERSION);
		fprintf(stderr, "%s\n", popplerCopyright);
		fprintf(stderr, "%s\n", xpdfCopyright);
		if (!printVersion) {
		  printUsage("pdfmerge", "<PDF-sourcefile-1>..<PDF-sourcefile-n> <PDF-destfile>", argDesc);
		}
		if (printVersion || printHelp)
		  exitCode = 0;
		goto err0;
	}
	exitCode = 0;

	for (i = 1; i < argc-1; i++) {
		GooString *gfileName = new GooString(argv[i]);
		PDFDoc *doc = new PDFDoc(gfileName, NULL, NULL, NULL);
		if (doc->isOk()) {
			docs.push_back(doc);
			if ((doc->getPDFMajorVersion () + doc->getPDFMinorVersion() / 10.0) > version)
				version = doc->getPDFMajorVersion () + doc->getPDFMinorVersion() / 10.0;
			continue;
		}
		break;
	}

	if (!(f = fopen(fileName, "wb"))) {
		error(-1, "Couldn't open file '%s'", fileName);
		return -1;
	}
	outStr = new FileOutStream(f,0);
	
	yRef = new XRef();
	yRef->add(0, 65535, 0, gFalse);
	PDFDoc::writeHeader(outStr, version);
	
	for (i = 0; i < (int) docs.size(); i++) {
		for (j=1; j <= docs[i]->getNumPages(); j++) {
		   PDFRectangle *cropBox = NULL;
//		   docs[i]->countRef = new XRef();
		   if (docs[i]->getCatalog()->getPage(j)->isCropped())
			   cropBox = docs[i]->getCatalog()->getPage(j)->getCropBox();
		   docs[i]->ReplacePageDict(j, 
			   docs[i]->getCatalog()->getPage(j)->getRotate(),
			   docs[i]->getCatalog()->getPage(j)->getMediaBox(),
			   cropBox, NULL);
			Ref *refPage = docs[i]->getCatalog()->getPageRef(j);
			Object page;
			docs[i]->getXRef()->fetch(refPage->num, refPage->gen, &page);
			pages.push_back(page);
			offsets.push_back(numOffset);
			Dict *pageDict = page.getDict();
			docs[i]->MarkPageObjects(pageDict, yRef, numOffset);
		}
		objectsCount += docs[i]->WritePageObjects(outStr, yRef, numOffset);
		numOffset = yRef->getNumObjects() + 1;
	}

	rootNum = yRef->getNumObjects() + 1;
	yRef->add(rootNum,0,outStr->getPos(),gTrue);
	outStr->printf("%d 0 obj\n", rootNum);
	outStr->printf("<< /Type /Catalog /Pages %d 0 R", rootNum + 1); 
	outStr->printf(">>\nendobj\n");
	objectsCount++;
		
	yRef->add(rootNum + 1,0,outStr->getPos(),gTrue);
	outStr->printf("%d 0 obj\n", rootNum + 1);
	outStr->printf("<< /Type /Pages /Kids [");
	for (j = 0; j < (int) pages.size(); j++)
		outStr->printf(" %d 0 R", rootNum + j + 2);
	outStr->printf(" ] /Count %d >>\nendobj\n", pages.size());
	objectsCount++;

	for (i = 0; i < (int) pages.size(); i++) {
		yRef->add(rootNum + i + 2,0,outStr->getPos(),gTrue);
		outStr->printf("%d 0 obj\n", rootNum  + i + 2);
		outStr->printf("<< ");
		Dict *pageDict = pages[i].getDict();
		for (j = 0; j < pageDict->getLength(); j++) {
			if (j > 0) outStr->printf(" ");
			const char *key = pageDict->getKey(j);
			Object value; pageDict->getValNF(j, &value);
			if (strcmp(key, "Parent") == 0) {
				outStr->printf("/Parent %d 0 R", rootNum + 1);
			} else {
				outStr->printf("/%s ", key);
				PDFDoc::writeObject(&value, NULL, outStr, yRef, offsets[i]);
			}
			value.free();
		}
		outStr->printf(" >>\nendobj\n");
		objectsCount++;
	}
	Guint uxrefOffset = outStr->getPos();
	yRef->writeToFile(outStr, gFalse /* do not write unnecessary entries */);

	Ref ref;
	ref.num = rootNum;
	ref.gen = 0;
	PDFDoc::writeTrailer(uxrefOffset, objectsCount, outStr, (GBool) gFalse, 0, &ref, yRef, fileName, outStr->getPos());

	outStr->close();
	fclose(f);
	delete yRef;
	for (j = 0; j < (int) pages.size(); j++) {
		pages[j].free();
	}
	for (i = 0; i < (int) docs.size(); i++) {
		delete docs[i];
	}
 err0:

  return exitCode;
}