~manxi-david/pyeffect/pyeffect

# Copyright (C) 2011 David Manzanares <manxi.david@gmail.com>

#This file is part of PyEffect.

#

# PyEffect is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# PyEffect 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 General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with PyEffect. If not, see <http://www.gnu.org/licenses/>.

 

#This is the PyEffect file manager, this module contains all the functions that reads, writes  and delete external files

 

#Purpose:

#Read the source (.3) and write the output of the stages in .3p, and .s

 

#Third party libraries:

import os

 

#Own libraries:

import args

import out

import s1

import s3

 

def getSource():

        """Open and read the source code in .3 format"""

        tg = args.target

        if tg[len(tg) - 2:len(tg)] != ".3":

                out.error("Source file <<"+tg+">> must have .3 extension")

        inputFile = open(tg, "r")

        source = inputFile.read()

        return source

 

def putStage1():

        """Write the STAGE1 into the .3p file"""

        outputFileS1 = open(args.target[0:(len(args.target) - 1)] + "3p", "w")

        outputFileS1.write(s1.all)

        outputFileS1.close()

 

def putStage3():

        """Write the STAGE3 into the .3s file"""

        outputFileS1 = open(args.target[0:(len(args.target) - 1)] + "3s", "w")

        outputFileS1.write(s3.lns.getStrVerbose())

        outputFileS1.close()

 

def putStage4():

        """Write the STAGE3 into the .s file"""

        from s4 import s4

        outputFileS3 = open(args.target[0:(len(args.target) - 1)] + "s", "w")

        outputFileS3.write(s4.asm)

        outputFileS3.close()

 

def cleanDir(Dir):

        """Clean the directory, remove all *.s *.3p and *.o files"""

        os.system("rm -f " + Dir + "*.s " + Dir + "*.3p " + Dir + "*.3s " + Dir + "*.o " + Dir + "*.log")

 

def clean():

        """Clean the directory wich contains the args.target if args.keep ==False"""

        if args.keep == False:

                i = len(args.target) - 1

                while args.target[i] != "/" and i > 0:

                        i = i - 1

                if i > 0:

                        cleanDir(args.target[:i + 1])

                else:

                        cleanDir("")