/*

* Copyright (c) 2001-2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: 

* Area-related classes implementation

*

*/

#include "r_areaset.h"

#include "r_global.h"

#include "r_rom.h"

extern TBool gGenDepGraph;

extern string gDepInfoFile;

using namespace std;

////////////////////////////////////////////////////////////////////////

Area::Area(const char* aName, TLinAddr aDestBaseAddr, TUint aMaxSize, Area* aNext)

: iFirstPagedCode(0),

iName(0),

iDestBaseAddr(aDestBaseAddr),

iSrcBaseAddr(0),

iSrcLimitAddr(0),

iMaxSize(aMaxSize),

iIsDefault(strcmp(aName, AreaSet::KDefaultAreaName) == 0),

iFiles(0),

iNextFilePtrPtr(&iFiles),

iNextArea(aNext)	   

{

	size_t len = strlen(aName) + 1;

	iName = new char[len];

	memcpy(iName,aName,len);

}

Area::~Area() {

	ReleaseAllFiles();

	if(iName)

		delete []iName;

}

/**

Increase the size of the area.

The reallocation must not exceed the area maximum size.

@param aSrcLimitAddr New source top address

@param aOverflow Number of overflow bytes if failure.

@return success indication

*/

TBool Area::ExtendSrcLimitAddr(TLinAddr aSrcLimitAddr, TUint& aOverflow) {

	// must have been set before

	assert(iSrcBaseAddr != 0);

	// can only allocate more

	assert(aSrcLimitAddr > iSrcBaseAddr);

	if (aSrcLimitAddr-iSrcBaseAddr > iMaxSize) {

		aOverflow = aSrcLimitAddr-iSrcBaseAddr-iMaxSize;

		return EFalse;

	}

	iSrcLimitAddr = aSrcLimitAddr;

	return ETrue;

}

/**

Add a file at end of the list of files contained in this area.

@param aFile File to add.  Must be allocated on the heap.  Ownership

is transfered from the caller to the callee.

*/

void Area::AddFile(TRomBuilderEntry* aFile) {

	assert(aFile != 0);

	*iNextFilePtrPtr = aFile; 

	iNextFilePtrPtr = &(aFile->iNextInArea);

}

void Area::ReleaseAllFiles() {

	for (TRomBuilderEntry *next = 0, *current = iFiles;

		current != 0;

		current = next) {

		next = current->iNextInArea; 

		delete current;

	}

	iFiles = 0;

	iNextFilePtrPtr = &iFiles;

}

////////////////////////////////////////////////////////////////////////

void FilesInAreaIterator::GoToNext() {

	assert(iCurrentFile!=0);

	iCurrentFile = iCurrentFile->iNextInArea;

}

////////////////////////////////////////////////////////////////////////

const char AreaSet::KDefaultAreaName[] = "DEFAULT AREA";

AreaSet::AreaSet()

: iNonDefaultAreas(0),

iDefaultArea(0),

iAreaCount(0) {

}

AreaSet::~AreaSet() {

	ReleaseAllAreas();

}

inline TBool IsInRange(TLinAddr aAddr, TLinAddr aDestBaseAddr, TLinAddr aEndAddr) {

	return aDestBaseAddr <= aAddr && aAddr <= aEndAddr;

}

/**

Add a new area.

Areas must have unique name, not overlap one another and not overflow

the 32-bit address range.  

@param aOverlappingArea On return ptr to name of overlapping area if

any, 0 otherwise.

@return EAdded if success, an error code otherwise.  

*/

AreaSet::TAddResult AreaSet::AddArea(const char* aNewName,

									 TLinAddr aNewDestBaseAddr,

									 TUint aNewMaxSize,

									 const char*& aOverlappingArea) {

	assert(aNewName != 0 && strlen(aNewName) > 0);

	assert(aNewMaxSize > 0);

	aOverlappingArea = 0;

	//

	// Checking new area validity

	//

	if (aNewDestBaseAddr+aNewMaxSize <= aNewDestBaseAddr)

		return EOverflow;

	TLinAddr newEndAddr = aNewDestBaseAddr+aNewMaxSize-1;

	// iterate non default areas first, then the default one if any

	Area* area=iNonDefaultAreas; 

	while (area != 0) {

		if (strcmp(area->Name(), aNewName) == 0)

			return EDuplicateName;

		TLinAddr curDestBaseAddr = area->DestBaseAddr();

		TLinAddr curEndAddr = area->DestBaseAddr()+area->MaxSize()-1;

		if (IsInRange(newEndAddr, curDestBaseAddr, curEndAddr) ||

			IsInRange(aNewDestBaseAddr, curDestBaseAddr, curEndAddr) ||

			IsInRange(curDestBaseAddr, aNewDestBaseAddr, newEndAddr)) {

			aOverlappingArea = area->Name();

			return EOverlap;

		}

		if (area->iNextArea == 0 && area != iDefaultArea)

			area = iDefaultArea;

		else

			area = area->iNextArea;

	}

	//

	// Adding new area

	//

	if (strcmp(KDefaultAreaName, aNewName) == 0)

		iDefaultArea = new Area(aNewName, aNewDestBaseAddr, aNewMaxSize);

	else

		iNonDefaultAreas = new Area(aNewName, aNewDestBaseAddr, aNewMaxSize, iNonDefaultAreas);

	++iAreaCount;

	return EAdded;

}

/**

Remove every area added to the set.

As a side-effect every file added to the areas is deleted.

*/

void AreaSet::ReleaseAllAreas() {

	for (Area *next = 0, *current = iNonDefaultAreas; current != 0; current = next) {

		next = current->iNextArea;

		delete current;

	}

	iNonDefaultAreas = 0;

	if(iDefaultArea){

		delete iDefaultArea;

		iDefaultArea = 0;

	}

}

/**

Find an area from its name.

@return A pointer to the area or 0 if the name is unknown.  The

returned pointer becomes invalid when "this" is destructed.

*/

Area* AreaSet::FindByName(const char* aName) const {

	assert(aName != 0 && strlen(aName) > 0);

	if (iDefaultArea && strcmp(iDefaultArea->Name(), aName) == 0)

		return iDefaultArea;

	for (Area* area=iNonDefaultAreas; area != 0; area = area->iNextArea) {

		if (strcmp(area->Name(), aName) == 0)

			return area;

	}

	return 0;

}

////////////////////////////////////////////////////////////////////////

void NonDefaultAreasIterator::GoToNext() {

	assert(iCurrentArea!=0);

	iCurrentArea = iCurrentArea->iNextArea;

}

TInt Area::SortFilesForPagedRom() {

	Print(ELog,"Sorting files to paged/unpaged.\n");

	TRomBuilderEntry* extention[2] = {0,0};

	TRomBuilderEntry* unpaged[2] = {0,0};

	TRomBuilderEntry* normal[2] = {0,0};

	TRomBuilderEntry* current = iFiles;

	while(current) {

		TRomBuilderEntry** list;

		if((current->iRomImageFlags & (KRomImageFlagPrimary|KRomImageFlagVariant|KRomImageFlagExtension|KRomImageFlagDevice)) ||

			current->HCRDataFile())

			list = extention;

		else if(current->iRomImageFlags&(KRomImageFlagCodeUnpaged))

			list = unpaged;

		else if(current->iResource && (current->iOverrideFlags&KOverrideCodeUnpaged) && gPagedRom)

			list = unpaged;

		else

			list = normal;

		if(list!=normal) {

			Print(ELog, "Unpaged file %s\n",current->iRomNode->BareName());

		}

		if(!list[0])

			list[0] = current;

		else

			list[1]->iNextInArea = current;

		list[1] = current;

		current = current->iNext;

	}

	if(extention[1]) {

		if(unpaged[0]) {

			extention[1]->iNextInArea = unpaged[0];

			unpaged[1]->iNextInArea = normal[0];

		}

		else

			extention[1]->iNextInArea = normal[0];

		if (normal[1])

			normal[1]->iNextInArea = 0;

		iFiles = extention[0];

	}

	else{

		Print(EError,"No primary files.\n");

		return KErrGeneral;

	}

	iFirstPagedCode = normal[0];

	Print(ELog,"\n");

	if(gGenDepGraph)

		WriteDependenceGraph();

	return KErrNone;

}

void Area::WriteDependenceGraph() {

	TDepInfoList::iterator infoIt;

	TStringList::iterator strIt;

	TDepInfoList myDepInfoList;

	TRomBuilderEntry* e = iFirstPagedCode;

	char buffer[255];

	TInt count = 0;

	TStringList nameList;

	while(e) {

		DepInfo tmpDepInfo;

		TRomNode* rn = e->iRomNode;

		TInt ll = rn->FullNameLength();

		char* mm = new char[ll+1];

		rn->GetFullName(mm);

		sprintf(buffer, "f%d", count);

		tmpDepInfo.portName = buffer;

		tmpDepInfo.index = count;

		myDepInfoList[mm] = tmpDepInfo;

		nameList.push_back(mm);

		delete []mm;

		e = e->iNextInArea;

		count++;

	}

	e = iFirstPagedCode;

	count = 0;

	while(e) {

		TRomNode* rn = e->iRomNode;

		TRomFile* rf = rn->iRomFile;

		TInt j;

		TStringList depFiles;

		for(j=0; j < rf->iNumDeps; ++j) {

			TRomFile* f=rf->iDeps[j];

			TRomBuilderEntry* start = iFiles;

			while(start && start->iRomNode->iRomFile != f)

				start = start->iNextInArea;

			if(start && (start->iRomNode->iRomFile == f)) {

				TRomNode* target = start->iRomNode;

				TInt l = target->FullNameLength();

				char* fname = new char[l+1];

				target->GetFullName(fname);

				if(myDepInfoList.find(fname) != myDepInfoList.end()) {

					depFiles.push_back(fname);

					myDepInfoList[fname].beenDepended = ETrue;

				}

				delete []fname;

			}

		}

		if(depFiles.size() > 0) {

			myDepInfoList[nameList[count]].depFilesList=depFiles;

			myDepInfoList[nameList[count]].dependOthers = ETrue;

		}

		count++;

		e=e->iNextInArea;

	}

	ofstream os;

	string filename(gDepInfoFile.c_str(), gDepInfoFile.length() - 3);

	filename = filename + "dot";

	os.open(filename.c_str());

	os << "digraph ROM {\n";

	os << "rankdir = LR;\n";

	os << "fontsize = 10;\n";

	os << "fontname = \"Courier New\";\n";

	os << "label = \"ROM DEPENDENCE GRAPH DOT FILE\";\n";

	os << "node[shape = plaintext];\n";

	os << "dependence[label=<<FONT FACE=\"Courier new\" POINT-SIZE=\"10pt\">\n";

	os << "<TABLE BORDER=\"0\" CELLBORDER=\"1\" CELLSPACING=\"0\">\n";

	//for(infoIt = myDepInfoList.begin(); infoIt != myDepInfoList.end(); infoIt++)

	for(strIt = nameList.begin(); strIt != nameList.end(); strIt++) {

		string tmp = *strIt;

		string::iterator charIt;

		for(charIt=tmp.begin(); charIt != tmp.end(); charIt++) {

			if(*charIt == '\\')

				*charIt = '/';

		}

		if(myDepInfoList[*strIt].beenDepended && myDepInfoList[*strIt].dependOthers) {

			os << "\t<TR><TD PORT=\"" << myDepInfoList[*strIt].portName << "\" ALIGN=\"LEFT\" BGCOLOR=\"yellow\">\n";

			os << "\t<FONT COLOR=\"red\">" << tmp << "</FONT>\n";

			os << "\t</TD></TR>\n";

		}

		else if(myDepInfoList[*strIt].beenDepended) {

			os << "\t<TR><TD PORT=\"" << myDepInfoList[*strIt].portName << "\" ALIGN=\"LEFT\" BGCOLOR=\"gray\">\n";

			os << "\t<FONT COLOR=\"red\">" << tmp << "</FONT>\n";

			os << "\t</TD></TR>\n";

		}

		else if(myDepInfoList[*strIt].dependOthers) {

			os << "\t<TR><TD PORT=\"" << myDepInfoList[*strIt].portName << "\" ALIGN=\"LEFT\" BGCOLOR=\"cyan\">\n";

			os << "\t<FONT COLOR=\"blue\">" << tmp << "</FONT>\n";

			os << "\t</TD></TR>\n";

		}

		else {

			os << "\t<TR><TD PORT=\"" << myDepInfoList[*strIt].portName << "\" ALIGN=\"LEFT\">";

			os << tmp;

			os << "</TD></TR>\n";

		}

	}

	os << "</TABLE>\n";

	os << "</FONT>>]\n";

	TBool lastEdge = ETrue;

	TBool first = ETrue;

	for(infoIt = myDepInfoList.begin(); infoIt != myDepInfoList.end(); infoIt++) {

		if(!infoIt->second.dependOthers) {

			continue;

		}

		for(strIt = infoIt->second.depFilesList.begin(); strIt != infoIt->second.depFilesList.end(); strIt++) {

			TBool tmpEdge = ETrue;

			if(infoIt->second.index < myDepInfoList[*strIt].index)

			{	

				tmpEdge = EFalse;

			}

			if(first) {

				lastEdge = tmpEdge;

				first = EFalse;

				if(lastEdge) {

					os << "edge[color=forestgreen];\n";

				}

				else {

					os << "edge[color=red];\n";

				}

			}

			else {

				if(lastEdge != tmpEdge) {

					lastEdge = tmpEdge;

					if(lastEdge) {

						os << "edge[color=forestgreen];\n";

					}

					else {

						os << "edge[color=red];\n";

					}

				}

			}

			os << "dependence: " << infoIt->second.portName << " -> dependence: " << myDepInfoList[*strIt].portName << ";\n";

		}

	}

	os << "}\n";

	os.close();

	filename = filename.substr(0, filename.size()-4);

	filename = filename + ".backwarddep.dot";

	os.open(filename.c_str());

	os << "digraph ROM {\n";

	os << "rankdir = LR;\n";

	os << "fontsize = 10;\n";

	os << "fontname = \"Courier New\";\n";

	os << "label = \"ROM FORWARD DEPENDENCE GRAPH DOT FILE\";\n";

	os << "node[shape = plaintext];\n";

	os << "dependence[label=<<FONT FACE=\"Courier new\" POINT-SIZE=\"10pt\">\n";

	os << "<TABLE BORDER=\"0\" CELLBORDER=\"1\" CELLSPACING=\"0\">\n";

	//for(infoIt = myDepInfoList.begin(); infoIt != myDepInfoList.end(); infoIt++)

	for(strIt = nameList.begin(); strIt != nameList.end(); strIt++) {

		string tmp = *strIt;

		string::iterator charIt;

		for(charIt=tmp.begin(); charIt != tmp.end(); charIt++) {

			if(*charIt == '\\')

				*charIt = '/';

		}

		if(myDepInfoList[*strIt].beenDepended && myDepInfoList[*strIt].dependOthers) {

			os << "\t<TR><TD PORT=\"" << myDepInfoList[*strIt].portName << "\" ALIGN=\"LEFT\" BGCOLOR=\"yellow\">\n";

			os << "\t<FONT COLOR=\"red\">" << tmp << "</FONT>\n";

			os << "\t</TD></TR>\n";

		}

		else if(myDepInfoList[*strIt].beenDepended) {

			os << "\t<TR><TD PORT=\"" << myDepInfoList[*strIt].portName << "\" ALIGN=\"LEFT\" BGCOLOR=\"gray\">\n";

			os << "\t<FONT COLOR=\"red\">" << tmp << "</FONT>\n";

			os << "\t</TD></TR>\n";

		}

		else if(myDepInfoList[*strIt].dependOthers) {

			os << "\t<TR><TD PORT=\"" << myDepInfoList[*strIt].portName << "\" ALIGN=\"LEFT\" BGCOLOR=\"cyan\">\n";

			os << "\t<FONT COLOR=\"blue\">" << tmp << "</FONT>\n";

			os << "\t</TD></TR>\n";

		}

		else {

			os << "\t<TR><TD PORT=\"" << myDepInfoList[*strIt].portName << "\" ALIGN=\"LEFT\">";

			os << tmp;

			os << "</TD></TR>\n";

		}

	}

	os << "</TABLE>\n";

	os << "</FONT>>]\n";

	os << "edge[color=red];\n";

	for(infoIt = myDepInfoList.begin(); infoIt != myDepInfoList.end(); infoIt++) {

		if(!infoIt->second.dependOthers) {

			continue;

		}

		for(strIt = infoIt->second.depFilesList.begin(); strIt != infoIt->second.depFilesList.end(); strIt++) {

			if(infoIt->second.index < myDepInfoList[*strIt].index)