1f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org/*
2f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * Copyright 2010 Marek Olšák <maraeo@gmail.com>
3f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org *
4f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * Permission is hereby granted, free of charge, to any person obtaining a
5f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * copy of this software and associated documentation files (the "Software"),
6f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * to deal in the Software without restriction, including without limitation
7f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * on the rights to use, copy, modify, merge, publish, distribute, sub
8f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * license, and/or sell copies of the Software, and to permit persons to whom
9f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * the Software is furnished to do so, subject to the following conditions:
10f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org *
11f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * The above copyright notice and this permission notice (including the next
12f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * paragraph) shall be included in all copies or substantial portions of the
13f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * Software.
14f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org *
15f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT. IN NO EVENT SHALL
18f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * THE AUTHOR(S) AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM,
19f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR
20f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
21f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org * USE OR OTHER DEALINGS IN THE SOFTWARE. */
22f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
23f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "pipe/p_context.h"
24f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "util/u_index_modify.h"
25f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "util/u_inlines.h"
26f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
27f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org/* Ubyte indices. */
28f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
29f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgvoid util_shorten_ubyte_elts_to_userptr(struct pipe_context *context,
30f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					struct pipe_index_buffer *ib,
31f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					int index_bias,
32f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					unsigned start,
33f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					unsigned count,
34f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					void *out)
35f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org{
36f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_transfer *src_transfer = NULL;
37f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    const unsigned char *in_map;
38f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned short *out_map = out;
39f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned i;
40f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
41f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    if (ib->user_buffer) {
42f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       in_map = ib->user_buffer;
43f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    } else {
44f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       in_map = pipe_buffer_map(context, ib->buffer,
45f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                PIPE_TRANSFER_READ |
46f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                PIPE_TRANSFER_UNSYNCHRONIZED,
47f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                &src_transfer);
48f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    }
49f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    in_map += start;
50f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
51f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    for (i = 0; i < count; i++) {
52f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        *out_map = (unsigned short)(*in_map + index_bias);
53f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        in_map++;
54f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        out_map++;
55f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    }
56f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
57f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    if (src_transfer)
58f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       pipe_buffer_unmap(context, src_transfer);
59f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
60f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
61f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgvoid util_shorten_ubyte_elts(struct pipe_context *context,
62f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			     struct pipe_index_buffer *ib,
63f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			     struct pipe_resource **out_buf,
64f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			     int index_bias,
65f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			     unsigned start,
66f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			     unsigned count)
67f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org{
68f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_resource* new_elts;
69f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned short *out_map;
70f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_transfer *dst_transfer;
71f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
72f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    new_elts = pipe_buffer_create(context->screen,
73f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  PIPE_BIND_INDEX_BUFFER,
74f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  PIPE_USAGE_STATIC,
75f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  2 * count);
76f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
77f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    out_map = pipe_buffer_map(context, new_elts, PIPE_TRANSFER_WRITE,
78f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                              &dst_transfer);
79f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    util_shorten_ubyte_elts_to_userptr(context, ib, index_bias,
80f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                       start, count, out_map);
81f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    pipe_buffer_unmap(context, dst_transfer);
82f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
83f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    pipe_resource_reference(out_buf, NULL);
84f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    *out_buf = new_elts;
85f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
86f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
87f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
88f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org/* Ushort indices. */
89f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
90f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgvoid util_rebuild_ushort_elts_to_userptr(struct pipe_context *context,
91f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					 struct pipe_index_buffer *ib,
92f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					 int index_bias,
93f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					 unsigned start, unsigned count,
94f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org					 void *out)
95f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org{
96f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_transfer *in_transfer = NULL;
97f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    const unsigned short *in_map;
98f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned short *out_map = out;
99f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned i;
100f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
101f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    if (ib->user_buffer) {
102f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       in_map = ib->user_buffer;
103f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    } else {
104f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       in_map = pipe_buffer_map(context, ib->buffer,
105f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                PIPE_TRANSFER_READ |
106f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                PIPE_TRANSFER_UNSYNCHRONIZED,
107f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                &in_transfer);
108f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    }
109f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    in_map += start;
110f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
111f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    for (i = 0; i < count; i++) {
112f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        *out_map = (unsigned short)(*in_map + index_bias);
113f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        in_map++;
114f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        out_map++;
115f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    }
116f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
117f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    if (in_transfer)
118f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       pipe_buffer_unmap(context, in_transfer);
119f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
120f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
121f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgvoid util_rebuild_ushort_elts(struct pipe_context *context,
122f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			      struct pipe_index_buffer *ib,
123f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			      struct pipe_resource **out_buf,
124f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			      int index_bias,
125f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			      unsigned start, unsigned count)
126f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org{
127f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_transfer *out_transfer = NULL;
128f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_resource *new_elts;
129f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned short *out_map;
130f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
131f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    new_elts = pipe_buffer_create(context->screen,
132f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  PIPE_BIND_INDEX_BUFFER,
133f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  PIPE_USAGE_STATIC,
134f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  2 * count);
135f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
136f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    out_map = pipe_buffer_map(context, new_elts,
137f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                              PIPE_TRANSFER_WRITE, &out_transfer);
138f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    util_rebuild_ushort_elts_to_userptr(context, ib, index_bias,
139f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                        start, count, out_map);
140f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    pipe_buffer_unmap(context, out_transfer);
141f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
142f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    pipe_resource_reference(out_buf, NULL);
143f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    *out_buf = new_elts;
144f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
145f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
146f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
147f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org/* Uint indices. */
148f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
149f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgvoid util_rebuild_uint_elts_to_userptr(struct pipe_context *context,
150f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org				       struct pipe_index_buffer *ib,
151f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org				       int index_bias,
152f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org				       unsigned start, unsigned count,
153f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org				       void *out)
154f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org{
155f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_transfer *in_transfer = NULL;
156f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    const unsigned int *in_map;
157f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned int *out_map = out;
158f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned i;
159f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
160f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    if (ib->user_buffer) {
161f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       in_map = ib->user_buffer;
162f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    } else {
163f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       in_map = pipe_buffer_map(context, ib->buffer,
164f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                PIPE_TRANSFER_READ |
165f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                PIPE_TRANSFER_UNSYNCHRONIZED,
166f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                &in_transfer);
167f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    }
168f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    in_map += start;
169f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
170f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    for (i = 0; i < count; i++) {
171f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        *out_map = (unsigned int)(*in_map + index_bias);
172f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        in_map++;
173f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org        out_map++;
174f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    }
175f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
176f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    if (in_transfer)
177f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       pipe_buffer_unmap(context, in_transfer);
178f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
179f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
180f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgvoid util_rebuild_uint_elts(struct pipe_context *context,
181f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			    struct pipe_index_buffer *ib,
182f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			    struct pipe_resource **out_buf,
183f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			    int index_bias,
184f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org			    unsigned start, unsigned count)
185f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org{
186f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_transfer *out_transfer = NULL;
187f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    struct pipe_resource *new_elts;
188f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    unsigned int *out_map;
189f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
190f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    new_elts = pipe_buffer_create(context->screen,
191f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  PIPE_BIND_INDEX_BUFFER,
192f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  PIPE_USAGE_STATIC,
193f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                  2 * count);
194f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
195f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    out_map = pipe_buffer_map(context, new_elts,
196f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                              PIPE_TRANSFER_WRITE, &out_transfer);
197f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    util_rebuild_uint_elts_to_userptr(context, ib, index_bias,
198f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                      start, count, out_map);
199f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    pipe_buffer_unmap(context, out_transfer);
200f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
201f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    pipe_resource_reference(out_buf, NULL);
202f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org    *out_buf = new_elts;
203f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
204