cmd-midx.py

   1 #!/usr/bin/env python
   2 import sys, math, struct, glob, sha
   3 import options, git
   4 from helpers import *
   5
   6 PAGE_SIZE=4096
   7 SHA_PER_PAGE=PAGE_SIZE/200.
   8
   9
  10 def merge(idxlist, bits, table):
  11     count = 0
  12     for e in git.idxmerge(idxlist):
  13         count += 1
  14         prefix = git.extract_bits(e, bits)
  15         table[prefix] = count
  16         yield e
  17
  18
  19 def do_midx(outdir, outfilename, infilenames):
  20     if not outfilename:
  21         assert(outdir)
  22         sum = sha.sha('\0'.join(infilenames)).hexdigest()
  23         outfilename = '%s/midx-%s.midx' % (outdir, sum)
  24
  25     inp = []
  26     total = 0
  27     for name in infilenames:
  28         ix = git.PackIndex(name)
  29         inp.append(ix)
  30         total += len(ix)
  31
  32     log('Merging %d indexes (%d objects).\n' % (len(infilenames), total))
  33     if total < 1024 and len(infilenames) < 3:
  34         log('%s: not enough objects for a .midx to be useful.\n' % outdir)
  35         return
  36
  37     pages = int(total/SHA_PER_PAGE) or 1
  38     bits = int(math.ceil(math.log(pages, 2)))
  39     entries = 2**bits
  40     log('Table size: %d (%d bits)\n' % (entries*4, bits))
  41
  42     table = [0]*entries
  43
  44     try:
  45         os.unlink(outfilename)
  46     except OSError:
  47         pass
  48     f = open(outfilename + '.tmp', 'w+')
  49     f.write('MIDX\0\0\0\1')
  50     f.write(struct.pack('!I', bits))
  51     assert(f.tell() == 12)
  52     f.write('\0'*8*entries)
  53
  54     for e in merge(inp, bits, table):
  55         f.write(e)
  56
  57     f.write('\0'.join([os.path.basename(p) for p in infilenames]))
  58
  59     f.seek(12)
  60     f.write(struct.pack('!%dQ' % entries, *table))
  61     f.close()
  62     os.rename(outfilename + '.tmp', outfilename)
  63
  64     # this is just for testing
  65     if 0:
  66         p = git.PackMidx(outfilename)
  67         assert(len(p.idxnames) == len(infilenames))
  68         print p.idxnames
  69         assert(len(p) == total)
  70         pi = iter(p)
  71         for i in merge(inp, total, bits, table):
  72             assert(i == pi.next())
  73             assert(p.exists(i))
  74
  75     print outfilename
  76
  77 optspec = """
  78 bup midx [options...] <idxnames...>
  79 --
  80 o,output=  output midx filename (default: auto-generated)
  81 a,auto     automatically create .midx from any unindexed .idx files
  82 f,force    automatically create .midx from *all* .idx files
  83 """
  84 o = options.Options('bup midx', optspec)
  85 (opt, flags, extra) = o.parse(sys.argv[1:])
  86
  87 if extra and (opt.auto or opt.force):
  88     log("bup midx: you can't use -f/-a and also provide filenames\n")
  89     o.usage()
  90
  91 git.check_repo_or_die()
  92
  93 if extra:
  94     do_midx(git.repo('objects/pack'), opt.output, extra)
  95 elif opt.auto or opt.force:
  96     paths = [git.repo('objects/pack')]
  97     paths += glob.glob(git.repo('index-cache/*/.'))
  98     if opt.force:
  99         for path in paths:
 100             do_midx(path, opt.output, glob.glob('%s/*.idx' % path))
 101     elif opt.auto:
 102         for path in paths:
 103             m = git.MultiPackIndex(path)
 104             needed = {}
 105             for pack in m.packs:  # only .idx files without a .midx are open
 106                 if pack.name.endswith('.idx'):
 107                     needed[pack.name] = 1
 108             del m
 109             do_midx(path, opt.output, needed.keys())
 110 else:
 111     log("bup midx: you must use -f or -a or provide input filenames\n")
 112     o.usage()