/*
* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU Lesser General Public License for more details.
* 
* You should have received a copy of the GNU Lesser General Public License
* along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef ELFSECTIONMANAGER_H_
#define ELFSECTIONMANAGER_H_

#include <vector>

#include "filefragment.h"
//
// Copyright (c) 2008 Symbian Software Ltd. All rights reserved.
//

#include "elfstringtable.h"
#include "outputfile.h"
#include "elfheader.h"
#include "e32romimage.h"
#include "elfsection.h"
#include "defs.h" // String


class ElfSectionHeaderStringTable : public ElfStringTable {
public:
	size_t AddName(String aName) {
		return AddString(aName);
	}

};

class ElfSectionManager : public FileFragmentOwner {
public:
	ElfSectionManager(Elf32Header & aElf32Header, E32RomImage & aRomImage, OutputFile & aOutputFile) :
		iElf32Header(aElf32Header), iRomImage(aRomImage), 
		iOutputFile(aOutputFile), iOffset(0), iData(NULL),
		iSectionStringTableSectionAdded(false)
		{}
	
	// The FileFragmentOwner protocol
	virtual void GetFileFragmentData(FileFragmentData & aFileFragmentData );
	virtual size_t Size();
	virtual void DeleteFileFragmentData();
	virtual void AddData(OutputFile & aOutputFile);
	
	void AddData();
	void AddSection(ElfSection & aSection);
	
	int NumSections() { return iSections.size(); }
	
private:
	// Don't want one of these to be copied
	ElfSectionManager(const ElfSectionManager & aElfSectionManager);
	
	ElfSectionManager & operator=(const ElfSectionManager & aElfSectionManager);
	
private:
	size_t SectionHeaderSize();
	void AddSectionStringTable();
	void AddSectionTable();
	void EnsureSectionStringTableSectionAdded();
	
private:
	typedef std::vector<ElfSection> SectionList;
	Elf32Header & iElf32Header;
	E32RomImage & iRomImage;
	OutputFile & iOutputFile;
	SectionList iSections;
	size_t iOffset;
	ElfSectionHeaderStringTable iStringTable;
	Elf32_Shdr * iData;
	bool iSectionStringTableSectionAdded;
	
};

#endif /*ELFSECTIONMANAGER_H_*/