Home | History | Annotate | Download | only in src_util 
      1 # -*- coding: utf-8 -*-
      2 
      3 #-------------------------------------------------------------------------
      4 # drawElements Quality Program utilities
      5 # --------------------------------------
      6 #
      7 # Copyright 2015 The Android Open Source Project
      8 #
      9 # Licensed under the Apache License, Version 2.0 (the "License");
     10 # you may not use this file except in compliance with the License.
     11 # You may obtain a copy of the License at
     12 #
     13 #      http://www.apache.org/licenses/LICENSE-2.0
     14 #
     15 # Unless required by applicable law or agreed to in writing, software
     16 # distributed under the License is distributed on an "AS IS" BASIS,
     17 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     18 # See the License for the specific language governing permissions and
     19 # limitations under the License.
     20 #
     21 #-------------------------------------------------------------------------
     22 
     23 import os
     24 import subprocess
     25 
     26 TEXT_FILE_EXTENSION = [
     27     ".bat",
     28     ".c",
     29     ".cfg",
     30     ".cmake",
     31     ".cpp",
     32     ".css",
     33     ".h",
     34     ".hh",
     35     ".hpp",
     36     ".html",
     37     ".inl",
     38     ".java",
     39     ".js",
     40     ".m",
     41     ".mk",
     42     ".mm",
     43     ".py",
     44     ".rule",
     45     ".sh",
     46     ".test",
     47     ".txt",
     48     ".xml",
     49     ".xsl",
     50     ]
     51 
     52 BINARY_FILE_EXTENSION = [
     53     ".png",
     54     ".pkm",
     55     ".xcf",
     56     ]
     57 
     58 def isTextFile (filePath):
     59     ext = os.path.splitext(filePath)[1]
     60     if ext in TEXT_FILE_EXTENSION:
     61         return True
     62     if ext in BINARY_FILE_EXTENSION:
     63         return False
     64 
     65     # Analyze file contents, zero byte is the marker for a binary file
     66     f = open(filePath, "rb")
     67 
     68     TEST_LIMIT = 1024
     69     nullFound = False
     70     numBytesTested = 0
     71 
     72     byte = f.read(1)
     73     while byte and numBytesTested < TEST_LIMIT:
     74         if byte == "\0":
     75             nullFound = True
     76             break
     77 
     78         byte = f.read(1)
     79         numBytesTested += 1
     80 
     81     f.close()
     82     return not nullFound
     83 
     84 def getProjectPath ():
     85     # File system hierarchy is fixed
     86     scriptDir = os.path.dirname(os.path.abspath(__file__))
     87     projectDir = os.path.normpath(os.path.join(scriptDir, "../.."))
     88     return projectDir
     89 
     90 def git (*args):
     91     process = subprocess.Popen(['git'] + list(args), cwd=getProjectPath(), stdout=subprocess.PIPE)
     92     output = process.communicate()[0]
     93     if process.returncode != 0:
     94         raise Exception("Failed to execute '%s', got %d" % (str(args), process.returncode))
     95     return output
     96 
     97 def getAbsolutePathPathFromProjectRelativePath (projectRelativePath):
     98     return os.path.normpath(os.path.join(getProjectPath(), projectRelativePath))
     99 
    100 def getChangedFiles ():
    101     # Added, Copied, Moved, Renamed
    102     output = git('diff', '--cached', '--name-only', '-z', '--diff-filter=ACMR')
    103     relativePaths = output.split('\0')[:-1] # remove trailing ''
    104     return [getAbsolutePathPathFromProjectRelativePath(path) for path in relativePaths]
    105 
    106 def getAllProjectFiles ():
    107     output = git('ls-files', '--cached', '-z')
    108     relativePaths = output.split('\0')[:-1] # remove trailing ''
    109     return [getAbsolutePathPathFromProjectRelativePath(path) for path in relativePaths]
    110