src/intel/compiler/brw_ir_allocator.h

   1 /* -*- c++ -*- */
   2 /*
   3  * Copyright © 2010-2014 Intel Corporation
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9  * and/or sell copies of the Software, and to permit persons to whom the
  10  * Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice (including the next
  13  * paragraph) shall be included in all copies or substantial portions of the
  14  * Software.
  15  *
  16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  22  * IN THE SOFTWARE.
  23  */
  24
  25 #ifndef BRW_IR_ALLOCATOR_H
  26 #define BRW_IR_ALLOCATOR_H
  27
  28 #include "main/macros.h"
  29
  30 namespace brw {
  31    /**
  32     * Simple allocator used to keep track of virtual GRFs.
  33     */
  34    class simple_allocator {
  35    public:
  36       simple_allocator() :
  37          sizes(NULL), offsets(NULL), count(0), total_size(0), capacity(0)
  38       {
  39       }
  40
  41       ~simple_allocator()
  42       {
  43          free(offsets);
  44          free(sizes);
  45       }
  46
  47       unsigned
  48       allocate(unsigned size)
  49       {
  50          assert(size > 0);
  51          if (capacity <= count) {
  52             capacity = MAX2(16, capacity * 2);
  53             sizes = (unsigned *)realloc(sizes, capacity * sizeof(unsigned));
  54             offsets = (unsigned *)realloc(offsets, capacity * sizeof(unsigned));
  55          }
  56
  57          sizes[count] = size;
  58          offsets[count] = total_size;
  59          total_size += size;
  60
  61          return count++;
  62       }
  63
  64       /**
  65        * Array of sizes for each allocation.  The allocation unit is up to the
  66        * back-end, but it's expected to be one scalar value in the FS back-end
  67        * and one vec4 in the VEC4 back-end.
  68        */
  69       unsigned *sizes;
  70
  71       /**
  72        * Array of offsets from the start of the VGRF space in allocation
  73        * units.
  74        */
  75       unsigned *offsets;
  76
  77       /** Total number of VGRFs allocated. */
  78       unsigned count;
  79
  80       /** Cumulative size in allocation units. */
  81       unsigned total_size;
  82
  83    private:
  84       unsigned capacity;
  85    };
  86 }
  87
  88 #endif