Tools/bgen/bgen/bgenStackBuffer.py

   1 """Buffers allocated on the stack."""
   2
   3
   4 from bgenBuffer import FixedInputBufferType, FixedOutputBufferType
   5
   6
   7 class StackOutputBufferType(FixedOutputBufferType):
   8
   9         """Fixed output buffer allocated on the stack -- passed as (buffer, size).
  10
  11         Instantiate with the buffer size as parameter.
  12         """
  13
  14         def passOutput(self, name):
  15                 return "%s__out__, %s" % (name, self.size)
  16
  17
  18 class VarStackOutputBufferType(StackOutputBufferType):
  19
  20         """Output buffer allocated on the stack -- passed as (buffer, &size).
  21
  22         Instantiate with the buffer size as parameter.
  23         """
  24
  25         def declareSize(self, name):
  26                 Output("int %s__len__ = %s;", name, self.size)
  27
  28         def passOutput(self, name):
  29                 return "%s__out__, &%s__len__" % (name, name)
  30
  31         def mkvalueArgs(self, name):
  32                 return "%s__out__, (int)%s__len__" % (name, name)
  33
  34
  35 class VarVarStackOutputBufferType(VarStackOutputBufferType):
  36
  37         """Output buffer allocated on the stack -- passed as (buffer, size, &size).
  38
  39         Instantiate with the buffer size as parameter.
  40         """
  41
  42         def passOutput(self, name):
  43                 return "%s__out__, %s__len__, &%s__len__" % (name, name, name)
  44
  45
  46 class ReturnVarStackOutputBufferType(VarStackOutputBufferType):
  47
  48         """Output buffer allocated on the stack -- passed as (buffer, size) -> size.
  49
  50         Instantiate with the buffer size as parameter.
  51         The function's return value is the size.
  52         (XXX Should have a way to suppress returning it separately, too.)
  53         """
  54
  55         def passOutput(self, name):
  56                 return "%s__out__, %s__len__" % (name, name)
  57
  58         def mkvalueArgs(self, name):
  59                 return "%s__out__, (int)_rv" % name