from Ffs import Ffs

import Section

import os

import subprocess

from GenFdsGlobalVariable import GenFdsGlobalVariable

from CommonDataClass.FdfClass import VerSectionClassObject

class VerSection (VerSectionClassObject):

## The constructor

#

# @param self The object pointer

#

def __init__(self):

VerSectionClassObject.__init__(self)

## GenSection() method

#

# Generate version section

#

# @param self The object pointer

# @param OutputPath Where to place output file

# @param ModuleName Which module this section belongs to

# @param SecNum Index of section

# @param KeyStringList Filter for inputs of section generation

# @param FfsInf FfsInfStatement object that contains this section data

# @param Dict dictionary contains macro and its value

# @retval tuple (Generated file name, section alignment)

#

def GenSection(self,OutputPath, ModuleName, SecNum, KeyStringList, FfsInf = None, Dict = {}):

#

# Prepare the parameter of GenSection

#

if FfsInf != None:

self.Alignment = FfsInf.__ExtendMacro__(self.Alignment)

self.BuildNum = FfsInf.__ExtendMacro__(self.BuildNum)

self.StringData = FfsInf.__ExtendMacro__(self.StringData)

self.FileName = FfsInf.__ExtendMacro__(self.FileName)

OutputFile = os.path.join(OutputPath,

ModuleName + 'SEC' + SecNum + Ffs.SectionSuffix.get('VERSION'))

OutputFile = os.path.normpath(OutputFile)

# Get String Data

StringData = ''

if self.StringData != None:

StringData = self.StringData

elif self.FileName != None:

FileNameStr = GenFdsGlobalVariable.ReplaceWorkspaceMacro(self.FileName)

FileNameStr = GenFdsGlobalVariable.MacroExtend(FileNameStr, Dict)

FileObj = open(FileNameStr, 'r')

StringData = FileObj.read()

StringData = '"' + StringData + '"'

FileObj.close()

else:

StringData = ''

GenFdsGlobalVariable.GenerateSection(OutputFile, None, 'EFI_SECTION_VERSION',

Ui=StringData, Ver=self.BuildNum)

OutputFileList = []

OutputFileList.append(OutputFile)

return OutputFileList, self.Alignment