]> git.cworth.org Git - vogl/blob - src/voglcore/lzma_Alloc.cpp
- Fixing polygon stipple saving/restoring so it doesn't stomp memory after the polygo...
[vogl] / src / voglcore / lzma_Alloc.cpp
1 /**************************************************************************
2  *
3  * Copyright 2013-2014 RAD Game Tools and Valve Software
4  * Copyright 2010-2014 Rich Geldreich and Tenacious Software LLC
5  * All Rights Reserved.
6  *
7  * Permission is hereby granted, free of charge, to any person obtaining a copy
8  * of this software and associated documentation files (the "Software"), to deal
9  * in the Software without restriction, including without limitation the rights
10  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11  * copies of the Software, and to permit persons to whom the Software is
12  * furnished to do so, subject to the following conditions:
13  *
14  * The above copyright notice and this permission notice shall be included in
15  * all copies or substantial portions of the Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23  * THE SOFTWARE.
24  *
25  **************************************************************************/
26
27 /* Alloc.c -- Memory allocation functions
28 2008-09-24
29 Igor Pavlov
30 Public domain */
31 #include "vogl_core.h"
32 #ifdef _WIN32
33 #include <windows.h>
34 #endif
35 #include <stdlib.h>
36
37 #include "lzma_Alloc.h"
38
39 namespace vogl
40 {
41
42 /* #define _SZ_ALLOC_DEBUG */
43
44 /* use _SZ_ALLOC_DEBUG to debug alloc/free operations */
45 #ifdef _SZ_ALLOC_DEBUG
46 #include <stdio.h>
47     int g_allocCount = 0;
48     int g_allocCountMid = 0;
49     int g_allocCountBig = 0;
50 #endif
51
52     void *MyAlloc(size_t size)
53     {
54         if (size == 0)
55             return 0;
56 #ifdef _SZ_ALLOC_DEBUG
57         {
58             void *p = vogl_malloc(size);
59             fprintf(stderr, "\nAlloc %10d bytes, count = %10d,  addr = %8X", size, g_allocCount++, (unsigned)p);
60             return p;
61         }
62 #else
63         return vogl_malloc(size);
64 #endif
65     }
66
67     void MyFree(void *address)
68     {
69 #ifdef _SZ_ALLOC_DEBUG
70         if (address != 0)
71             fprintf(stderr, "\nFree; count = %10d,  addr = %8X", --g_allocCount, (unsigned)address);
72 #endif
73         vogl_free(address);
74     }
75
76 #ifdef _WIN32
77
78     void *MidAlloc(size_t size)
79     {
80         if (size == 0)
81             return 0;
82 #ifdef _SZ_ALLOC_DEBUG
83         fprintf(stderr, "\nAlloc_Mid %10d bytes;  count = %10d", size, g_allocCountMid++);
84 #endif
85         return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);
86     }
87
88     void MidFree(void *address)
89     {
90 #ifdef _SZ_ALLOC_DEBUG
91         if (address != 0)
92             fprintf(stderr, "\nFree_Mid; count = %10d", --g_allocCountMid);
93 #endif
94         if (address == 0)
95             return;
96         VirtualFree(address, 0, MEM_RELEASE);
97     }
98
99 #ifndef MEM_LARGE_PAGES
100 #undef _7ZIP_LARGE_PAGES
101 #endif
102
103 #ifdef _7ZIP_LARGE_PAGES
104     SIZE_T g_LargePageSize = 0;
105     typedef SIZE_T(WINAPI *GetLargePageMinimumP)();
106 #endif
107
108     void SetLargePageSize()
109     {
110 #ifdef _7ZIP_LARGE_PAGES
111         SIZE_T size = 0;
112         GetLargePageMinimumP largePageMinimum = (GetLargePageMinimumP)
113             GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "GetLargePageMinimum");
114         if (largePageMinimum == 0)
115             return;
116         size = largePageMinimum();
117         if (size == 0 || (size & (size - 1)) != 0)
118             return;
119         g_LargePageSize = size;
120 #endif
121     }
122
123     void *BigAlloc(size_t size)
124     {
125         if (size == 0)
126             return 0;
127 #ifdef _SZ_ALLOC_DEBUG
128         fprintf(stderr, "\nAlloc_Big %10d bytes;  count = %10d", size, g_allocCountBig++);
129 #endif
130
131 #ifdef _7ZIP_LARGE_PAGES
132         if (g_LargePageSize != 0 && g_LargePageSize <= (1 << 30) && size >= (1 << 18))
133         {
134             void *res = VirtualAlloc(0, (size + g_LargePageSize - 1) & (~(g_LargePageSize - 1)),
135                                      MEM_COMMIT | MEM_LARGE_PAGES, PAGE_READWRITE);
136             if (res != 0)
137                 return res;
138         }
139 #endif
140         return VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);
141     }
142
143     void BigFree(void *address)
144     {
145 #ifdef _SZ_ALLOC_DEBUG
146         if (address != 0)
147             fprintf(stderr, "\nFree_Big; count = %10d", --g_allocCountBig);
148 #endif
149
150         if (address == 0)
151             return;
152         VirtualFree(address, 0, MEM_RELEASE);
153     }
154
155 #endif
156 }