1

import os, re, operator

     2

from wxPopen import ProcessLogger

     3

import hashlib

     4

     5

includes_re =  re.compile('\s*#include\s*["<]([^">]*)[">].*')

     6

     7

class toolchain_gcc():

     8

    """

     9

    This abstract class contains GCC specific code.

    10

    It cannot be used as this and should be inherited in a target specific

    11

    class such as target_linux or target_win32

    12

    """

    13

    def __init__(self, PluginsRootInstance):

    14

        self.PluginsRootInstance = PluginsRootInstance

    15

        self.buildpath = None

    16

        self.SetBuildPath(self.PluginsRootInstance._getBuildPath())

    17

    18

    def getBuilderCFLAGS(self):

    19

        """

    20

        Returns list of builder specific CFLAGS

    21

        """

    22

        return [self.PluginsRootInstance.GetTarget().getcontent()["value"].getCFLAGS()]

    23

    24

    def getBuilderLDFLAGS(self):

    25

        """

    26

        Returns list of builder specific LDFLAGS

    27

        """

    28

        return self.PluginsRootInstance.LDFLAGS + \

    29

               [self.PluginsRootInstance.GetTarget().getcontent()["value"].getLDFLAGS()]

    30

    31

    def GetBinaryCode(self):

    32

        try:

    33

            return open(self.exe_path, "rb").read()

    34

        except Exception, e:

    35

            return None

    36

    37

    def _GetMD5FileName(self):

    38

        return os.path.join(self.buildpath, "lastbuildPLC.md5")

    39

    40

    def ResetBinaryCodeMD5(self):

    41

        self.md5key = None

    42

        try:

    43

            os.remove(self._GetMD5FileName())

    44

        except Exception, e:

    45

            pass

    46

    47

    def GetBinaryCodeMD5(self):

    48

        if self.md5key is not None:

    49

            return self.md5key

    50

        else:

    51

            try:

    52

                return open(self._GetMD5FileName(), "r").read()

    53

            except Exception, e:

    54

                return None

    55

    56

    def SetBuildPath(self, buildpath):

    57

        if self.buildpath != buildpath:

    58

            self.buildpath = buildpath

    59

            self.exe = self.PluginsRootInstance.GetProjectName() + self.extension

    60

            self.exe_path = os.path.join(self.buildpath, self.exe)

    61

            self.md5key = None

    62

            self.srcmd5 = {}

    63

    64

    def check_and_update_hash_and_deps(self, bn):

    65

        # Get latest computed hash and deps

    66

        oldhash, deps = self.srcmd5.get(bn,(None,[]))

    67

        # read source

    68

        src = open(os.path.join(self.buildpath, bn)).read()

    69

        # compute new hash

    70

        newhash = hashlib.md5(src).hexdigest()

    71

        # compare

    72

        match = (oldhash == newhash)

    73

        if not match:

    74

            # file have changed

    75

            # update direct dependencies

    76

            deps = []

    77

            for l in src.splitlines():

    78

                res = includes_re.match(l)

    79

                if res is not None:

    80

                    depfn = res.groups()[0]

    81

                    if os.path.exists(os.path.join(self.buildpath, depfn)):

    82

                        #print bn + " depends on "+depfn

    83

                        deps.append(depfn)

    84

            # store that hashand deps

    85

            self.srcmd5[bn] = (newhash, deps)

    86

        # recurse through deps

    87

        # TODO detect cicular deps.

    88

        return reduce(operator.and_, map(self.check_and_update_hash_and_deps, deps), match)

    89

    90

    def build(self):

    91

        # Retrieve toolchain user parameters

    92

        toolchain_params = self.PluginsRootInstance.GetTarget().getcontent()["value"]

    93

        self.compiler = toolchain_params.getCompiler()

    94

        self.linker = toolchain_params.getLinker()

    95

    96

        Builder_CFLAGS = ' '.join(self.getBuilderCFLAGS())

    97

    98

        ######### GENERATE OBJECT FILES ########################################

    99

        obns = []

   100

        objs = []

   101

        relink = False

   102

        for Location, CFilesAndCFLAGS, DoCalls in self.PluginsRootInstance.LocationCFilesAndCFLAGS:

   103

            if Location:

   104

                self.PluginsRootInstance.logger.write(_("Plugin : ") + self.PluginsRootInstance.GetChildByIECLocation(Location).GetCurrentName() + " " + str(Location)+"\n")

   105

            else:

   106

                self.PluginsRootInstance.logger.write(_("PLC :\n"))

   107

   108

            for CFile, CFLAGS in CFilesAndCFLAGS:

   109

                bn = os.path.basename(CFile)

   110

                obn = os.path.splitext(bn)[0]+".o"

   111

                objectfilename = os.path.splitext(CFile)[0]+".o"

   112

   113

                match = self.check_and_update_hash_and_deps(bn)

   114

   115

                if match:

   116

                    self.PluginsRootInstance.logger.write("   [pass]  "+bn+" -> "+obn+"\n")

   117

                else:

   118

                    relink = True

   119

   120

                    self.PluginsRootInstance.logger.write("   [CC]  "+bn+" -> "+obn+"\n")

   121

   122

                    status, result, err_result = ProcessLogger(

   123

                           self.PluginsRootInstance.logger,

   124

                           "\"%s\" -c \"%s\" -o \"%s\" %s %s"%

   125

                               (self.compiler, CFile, objectfilename, Builder_CFLAGS, CFLAGS)

   126

                           ).spin()

   127

   128

                    if status :

   129

                        self.srcmd5.pop(bn)

   130

                        self.PluginsRootInstance.logger.write_error(_("C compilation of %s failed.\n")%bn)

   131

                        return False

   132

   133

                obns.append(obn)

   134

                objs.append(objectfilename)

   135

   136

        ######### GENERATE library FILE ########################################

   137

        # Link all the object files into one binary file

   138

        self.PluginsRootInstance.logger.write(_("Linking :\n"))

   139

        if relink:

   140

            objstring = []

   141

   142

            # Generate list .o files

   143

            listobjstring = '"' + '"  "'.join(objs) + '"'

   144

   145

            ALLldflags = ' '.join(self.getBuilderLDFLAGS())

   146

   147

            self.PluginsRootInstance.logger.write("   [CC]  " + ' '.join(obns)+" -> " + self.exe + "\n")

   148

   149

            status, result, err_result = ProcessLogger(

   150

                   self.PluginsRootInstance.logger,

   151

                   "\"%s\" %s -o \"%s\" %s"%

   152

                       (self.linker,

   153

                        listobjstring,

   154

                        self.exe_path,

   155

                        ALLldflags)

   156

                   ).spin()

   157

   158

            if status :

   159

                return False

   160

   161

        else:

   162

            self.PluginsRootInstance.logger.write("   [pass]  " + ' '.join(obns)+" -> " + self.exe + "\n")

   163

   164

        # Calculate md5 key and get data for the new created PLC

   165

        data=self.GetBinaryCode()

   166

        self.md5key = hashlib.md5(data).hexdigest()

   167

   168

        # Store new PLC filename based on md5 key

   169

        f = open(self._GetMD5FileName(), "w")

   170

        f.write(self.md5key)

   171

        f.close()

   172

   173

        return True

   174