public void writeArcs(GL4 gl, int neighborsCountBuffer, int atomCount, int arcsCountBuffer)
throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "arcs.txt")))) {
// read counts
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, neighborsCountBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int neighborCounts[] = new int[atomCount];
data.asIntBuffer().get(neighborCounts);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// write arcs
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, arcsCountBuffer);
IntBuffer arcsCounts = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY).asIntBuffer();
for (int i = 0; i < atomCount; i++) {
int totalArcs = 0;
List<Integer> counts = new ArrayList<>();
for (int j = 0; j < neighborCounts[i]; j++) {
int count = arcsCounts.get(i * Scene.MAX_NEIGHBORS + j);
counts.add(count);
totalArcs += count;
}
writer.append(String.format("%4d (%2d): ", i, totalArcs));
for (int count : counts) {
writer.append(String.format("%3d", count));
}
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void writeSphereIsolated(
GL4 gl, int sphereIsolatedCountsBuffer, int sphereIsolatedVSBuffer, int atomCount)
throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "sphere-isolated.txt")))) {
// read counts
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, sphereIsolatedCountsBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int counts[] = new int[atomCount];
data.asIntBuffer().get(counts);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// write counts and indices
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, sphereIsolatedVSBuffer);
ByteBuffer vsData = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
for (int i = 0; i < counts.length; i++) {
writer.append(String.format("%4d (%2d): ", i, counts[i]));
for (int j = 0; j < counts[i]; j++) {
Vector4f vs =
getVec4(vsData, (i * Scene.MAX_SPHERE_ISOLATED_TORI + j) * Scene.SIZEOF_VEC4);
writer.append(String.format("vs: [%f %f %f %f], ", vs.x, vs.y, vs.z, vs.w));
}
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void writeGrid(String filename, GL4 gl, int gridCountsBuffer, int gridIndicesBuffer)
throws IOException {
try (BufferedWriter writer = new BufferedWriter(new FileWriter(new File(debugDir, filename)))) {
// read counts
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, gridCountsBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int counts[] = new int[Scene.CELL_COUNT];
data.asIntBuffer().get(counts);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// write counts and indices
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, gridIndicesBuffer);
IntBuffer indices = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY).asIntBuffer();
Set<Integer> cellIndices = new TreeSet<>();
for (int i = 0; i < Scene.CELL_COUNT; i++) {
if (counts[i] > 0) {
int x = i / Scene.GRID_SIZE / Scene.GRID_SIZE;
int y = (i / Scene.GRID_SIZE) % Scene.GRID_SIZE;
int z = i % Scene.GRID_SIZE;
writer.append(String.format("[%2d,%2d,%2d] (%2d): ", x, y, z, counts[i]));
cellIndices.clear();
for (int j = 0; j < counts[i]; j++) {
cellIndices.add(indices.get(Scene.MAX_CELL_ATOMS * i + j));
}
for (Integer index : cellIndices) {
writer.append(String.format("%6d", index));
}
writer.newLine();
}
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void writeNeighbors(
GL4 gl, int neighborsBuffer, int neighborCountsBuffer, int sphereCount, String filename)
throws IOException {
try (BufferedWriter writer = new BufferedWriter(new FileWriter(new File(debugDir, filename)))) {
// read counts
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, neighborCountsBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int counts[] = new int[sphereCount];
data.asIntBuffer().get(counts);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// write counts and indices
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, neighborsBuffer);
IntBuffer neighbors = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY).asIntBuffer();
Set<Integer> neighborIndices = new LinkedHashSet<>();
for (int i = 0; i < sphereCount; i++) {
if (counts[i] > 0) {
writer.append(String.format("%4d (%2d): ", i, counts[i]));
neighborIndices.clear();
for (int j = 0; j < counts[i]; j++) {
neighborIndices.add(neighbors.get(Scene.MAX_NEIGHBORS * i + j));
}
for (Integer index : neighborIndices) {
writer.append(String.format("%6d", index));
}
writer.newLine();
}
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void writeSphereCapPlanes(
GL4 gl, int sphereCapCountsBuffer, int sphereCapPlanesBuffer, int atomCount)
throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "sphere-caps.txt")))) {
// read counts
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, sphereCapCountsBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int counts[] = new int[atomCount];
data.asIntBuffer().get(counts);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// write counts and planes
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, sphereCapPlanesBuffer);
ByteBuffer planesData = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
for (int i = 0; i < counts.length; i++) {
writer.append(String.format("%4d (%2d): ", i, counts[i]));
for (int j = 0; j < counts[i]; j++) {
Vector4f plane =
getVec4(planesData, (i * GPUGraph.MAX_SPHERE_POLYGON_COUNT + j) * Scene.SIZEOF_VEC4);
writer.append(
String.format("plane: [%f %f %f %f], ", plane.x, plane.y, plane.z, plane.w));
}
writer.newLine();
}
// unbind buffers
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void setupBuffers(GL4 gl, int[] vbo, int index) {
this.vbo = vbo;
this.index = index;
Vertex3D[] vertices = myShape.getVertices();
int[] indices = myShape.getIndices();
float[] fvalues = new float[indices.length * 3];
float[] tvalues = new float[indices.length * 2];
float[] nvalues = new float[indices.length * 3];
for (int i = 0; i < indices.length; i++) {
fvalues[i * 3] = (float) (vertices[indices[i]]).getX();
fvalues[i * 3 + 1] = (float) (vertices[indices[i]]).getY();
fvalues[i * 3 + 2] = (float) (vertices[indices[i]]).getZ();
tvalues[i * 2] = (float) (vertices[indices[i]]).getS();
tvalues[i * 2 + 1] = (float) (vertices[indices[i]]).getT();
nvalues[i * 3] = (float) (vertices[indices[i]]).getNormalX();
nvalues[i * 3 + 1] = (float) (vertices[indices[i]]).getNormalY();
nvalues[i * 3 + 2] = (float) (vertices[indices[i]]).getNormalZ();
}
gl.glBindBuffer(GL.GL_ARRAY_BUFFER, vbo[index++]);
FloatBuffer vertBuf = FloatBuffer.wrap(fvalues);
gl.glBufferData(GL.GL_ARRAY_BUFFER, vertBuf.limit() * 4, vertBuf, GL.GL_STATIC_DRAW);
gl.glBindBuffer(GL.GL_ARRAY_BUFFER, vbo[index++]);
FloatBuffer texBuf = FloatBuffer.wrap(tvalues);
gl.glBufferData(GL.GL_ARRAY_BUFFER, texBuf.limit() * 4, texBuf, GL.GL_STATIC_DRAW);
gl.glBindBuffer(GL.GL_ARRAY_BUFFER, vbo[index++]);
FloatBuffer norBuf = FloatBuffer.wrap(nvalues);
gl.glBufferData(GL.GL_ARRAY_BUFFER, norBuf.limit() * 4, norBuf, GL.GL_STATIC_DRAW);
}
@Override
public void draw(GLAutoDrawable glAutoDrawable, MatrixStack mvStack, Matrix3D pMat) {
GL4 gl = (GL4) glAutoDrawable.getGL();
mvStack.pushMatrix(); // push translate
translate(dxTranslate, dyTranslate, dzTranslate);
mvStack.multMatrix(getTranslation());
mvStack.pushMatrix(); // push rotate
rotate(dxRotate, dyRotate, dzRotate);
mvStack.multMatrix(getRotation());
mvStack.multMatrix(getScale());
gl.glUniformMatrix4fv(IdentityLocs.getMvLoc(), 1, false, mvStack.peek().getFloatValues(), 0);
gl.glUniformMatrix4fv(IdentityLocs.getProjLoc(), 1, false, pMat.getFloatValues(), 0);
gl.glUniformMatrix4fv(
IdentityLocs.getnlocation(),
1,
false,
(mvStack.peek().inverse().transpose().getFloatValues()),
0);
// bind vertex values
gl.glBindBuffer(GL.GL_ARRAY_BUFFER, vbo[index]);
gl.glVertexAttribPointer(0, 3, GL.GL_FLOAT, false, 0, 0);
gl.glEnableVertexAttribArray(0);
// bind normal values
gl.glBindBuffer(GL.GL_ARRAY_BUFFER, vbo[index + 2]);
gl.glVertexAttribPointer(1, 3, GL.GL_FLOAT, false, 0, 0);
gl.glEnableVertexAttribArray(1);
// bind texture values
gl.glBindBuffer(GL.GL_ARRAY_BUFFER, vbo[index + 1]);
gl.glVertexAttribPointer(2, 2, GL.GL_FLOAT, false, 0, 0);
gl.glEnableVertexAttribArray(2);
gl.glActiveTexture(gl.GL_TEXTURE0);
gl.glBindTexture(gl.GL_TEXTURE_2D, texture);
gl.glEnable(GL_CULL_FACE);
gl.glFrontFace(GL_CCW);
gl.glEnable(GL_DEPTH_TEST);
gl.glDepthFunc(GL_LEQUAL);
// draw arrays
gl.glDrawArrays(GL_TRIANGLES, 0, myShape.getIndices().length);
mvStack.popMatrix(); // pop rotate
mvStack.popMatrix(); // pop translate
}
/**
* @param theSource
* @param theSourceID
* @param theDestinationID
* @param theReadOffset
* @param theWriteOffset
* @param theSize
*/
public void copy(GLBuffer theSource, int theReadOffset, int theWriteOffset, int theSize) {
GL4 gl = GLGraphics.currentGL();
gl.glBindBuffer(GLBufferTarget.COPY_READ.glID, theSource._myID);
gl.glBindBuffer(GLBufferTarget.COPY_WRITE.glID, _myID);
gl.glCopyBufferSubData(
GLBufferTarget.COPY_READ.glID,
GLBufferTarget.COPY_WRITE.glID,
theReadOffset,
theWriteOffset,
theSize);
gl.glBindBuffer(GLBufferTarget.COPY_READ.glID, 0);
gl.glBindBuffer(GLBufferTarget.COPY_WRITE.glID, 0);
}
private boolean initVertexArray(GL4 gl4) {
gl4.glGenVertexArrays(1, vertexArrayName, 0);
gl4.glBindVertexArray(vertexArrayName[0]);
{
gl4.glBindBuffer(GL_ARRAY_BUFFER, bufferName[Buffer.F64.ordinal()]);
gl4.glVertexAttribLPointer(Semantic.Attr.POSITION, 3, GL_DOUBLE, 3 * Double.BYTES, 0);
gl4.glBindBuffer(GL_ARRAY_BUFFER, 0);
gl4.glEnableVertexAttribArray(Semantic.Attr.POSITION);
}
gl4.glBindVertexArray(0);
return checkError(gl4, "initVertexArray");
}
private boolean initVertexBuffer(GL4 gl4) {
gl4.glGenBuffers(Buffer.MAX.ordinal(), bufferName, 0);
gl4.glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, bufferName[Buffer.ELEMENT.ordinal()]);
ShortBuffer elementBuffer = GLBuffers.newDirectShortBuffer(elementData);
gl4.glBufferData(GL_ELEMENT_ARRAY_BUFFER, elementSize, elementBuffer, GL_STATIC_DRAW);
gl4.glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, 0);
gl4.glBindBuffer(GL_ARRAY_BUFFER, bufferName[Buffer.F64.ordinal()]);
DoubleBuffer positionBuffer = GLBuffers.newDirectDoubleBuffer(positionData);
gl4.glBufferData(GL_ARRAY_BUFFER, positionSize, positionBuffer, GL_STATIC_DRAW);
gl4.glBindBuffer(GL_ARRAY_BUFFER, 0);
return checkError(gl4, "initArrayBuffer");
}
public void writeCaps(GL4 gl, int capsArrayBuffer, int count) throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "caps.txt")))) {
// write counts and planes
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, capsArrayBuffer);
ByteBuffer capsArray = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
for (int i = 0; i < count; i++) {
Vector4f position = getVec4(capsArray, i * Scene.SIZEOF_CAP);
Vector4f plane = getVec4(capsArray, i * Scene.SIZEOF_CAP + 16);
int atomIdx = capsArray.getInt(i * Scene.SIZEOF_CAP + 32);
int label = capsArray.getInt(i * Scene.SIZEOF_CAP + 36);
int padding0 = capsArray.getInt(i * Scene.SIZEOF_CAP + 40);
int padding1 = capsArray.getInt(i * Scene.SIZEOF_CAP + 44);
writer.append(String.format("%4d: ", i));
writer.append(
String.format(
"position: [%f %f %f %f], ", position.x, position.y, position.z, position.w));
writer.append(String.format("plane: [%f %f %f %f], ", plane.x, plane.y, plane.z, plane.w));
writer.append(String.format("atomIdx: %d, label: %d", atomIdx, label));
writer.append(String.format(", padding0: %d, padding1: %d", padding0, padding1));
writer.newLine();
}
// unbind buffers
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void writeTori(GL4 gl, int toriArrayBuffer, int count) throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "tori.txt")))) {
// write tori
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, toriArrayBuffer);
ByteBuffer toriArray = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
for (int i = 0; i < count; i++) {
Vector4f position = getVec4(toriArray, i * Scene.SIZEOF_TORUS);
float operation = toriArray.getFloat(i * Scene.SIZEOF_TORUS + 28);
String op = (operation > 0f) ? "AND" : (operation < 0f) ? "OR " : "ISOLATED";
Vector4f plane1 = getVec4(toriArray, i * Scene.SIZEOF_TORUS + 48);
Vector4f plane2 = getVec4(toriArray, i * Scene.SIZEOF_TORUS + 64);
writer.append(String.format("%4d: ", i));
writer.append(
String.format(
"center: [%f %f %f], R: %f, ", position.x, position.y, position.z, position.w));
writer.append(String.format("op: %s, ", op));
writer.append(
String.format("plane1: [%f %f %f %f], ", plane1.x, plane1.y, plane1.z, plane1.w));
writer.append(
String.format("plane2: [%f %f %f %f]", plane2.x, plane2.y, plane2.z, plane2.w));
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public static void checkGridOverflow(
GL4 gl, int gridCountsBuffer, int cellCount, int maxCellAtomCount) {
// read counts
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, gridCountsBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int counts[] = new int[cellCount];
data.asIntBuffer().get(counts);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0);
// check counts
for (int c = 0; c < cellCount; c++) {
if (counts[c] >= maxCellAtomCount) {
System.err.println(
"Warning: possible grid cell overflow. Cell: " + c + ", atom count: " + counts[c]);
}
}
}
public void writeFragments(
GL4 gl, int fragmentsIndexBuffer, int fragmentsBuffer, int width, int height)
throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "fragments.txt")))) {
// write A-Buffer dimensions
writer.append("width: " + width + ", height: " + height);
writer.newLine();
// read fragment indices
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, fragmentsIndexBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int indices[] = new int[Scene.MAX_ABUFFER_PIXELS];
data.asIntBuffer().get(indices);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// write fragments
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, fragmentsBuffer);
int stride = Scene.SIZEOF_FRAGMENT;
ByteBuffer fragments = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
List<Fragment> frags = new ArrayList<>();
for (int y = 0; y < height; y++) {
for (int x = 0; x < width; x++) {
frags.clear();
int index = indices[y * width + x];
if (index == Scene.INVALID_INDEX) {
continue;
}
while (index != Scene.INVALID_INDEX) {
int color = fragments.getInt(index * stride);
float depth = fragments.getFloat(index * stride + 4);
frags.add(new Fragment(color, depth));
index = fragments.getInt(index * stride + 8);
}
Collections.sort(frags);
writer.append(String.format("[%4d,%4d]: ", x, y));
for (Fragment f : frags) {
writer.append(String.format("%08x@%f ", f.color, f.depth));
}
/*if (frags.size() > 0) {
writer.append(String.format("%08x@%f ", frags.get(0).color, frags.get(0).depth)); // DEBUG
}*/
writer.newLine();
}
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public static void checkNeighborsOverflow(
GL4 gl, int neighborCountsBuffer, int sphereCount, int maxNeighborCount) {
// read counts
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, neighborCountsBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int counts[] = new int[sphereCount];
data.asIntBuffer().get(counts);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0);
// check counts
for (int s = 0; s < sphereCount; s++) {
if (counts[s] >= maxNeighborCount) {
System.err.println(
"Warning: possible neighbors overflow. Sphere: "
+ s
+ ", neighbor count: "
+ counts[s]);
}
}
}
public static void checkArcsOverflow(
GL4 gl,
CLArcs arcs,
int neighborCountsBuffer,
int atomCount,
int maxNeighborCount,
int arcsCountBuffer,
int maxArcsCount) {
int maxCount = 0;
// read counts
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, neighborCountsBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
int neighborCounts[] = new int[atomCount];
data.asIntBuffer().get(neighborCounts);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// write arcs
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, arcsCountBuffer);
IntBuffer arcsCounts = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY).asIntBuffer();
for (int i = 0; i < atomCount; i++) {
for (int j = 0; j < neighborCounts[i]; j++) {
int count = arcsCounts.get(i * maxNeighborCount + j);
if (count > maxArcsCount) {
System.err.println("Error: arcs overflow. Arc count: " + count);
}
maxCount = Math.max(maxCount, count);
}
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, 0);
// System.out.println("Max arc count: " + maxCount);
// int threadCount = Utils.getCounter(gl, countersBuffer, 0);
// System.out.println("Arcs threads finished: " + threadCount);
// int hashCount = Utils.getCounter(gl, countersBuffer, 4);
// System.out.println("Arc hashes written: " + hashCount);
int hashErrorCount = arcs.getHashErrorCount();
if (hashErrorCount > 0) {
System.err.println("Error: arc hash overflow. Instance count: " + hashErrorCount);
}
}
private boolean initBuffer(GL4 gl4) {
boolean validated = true;
gl4.glGenBuffers(Buffer.MAX, bufferName);
gl4.glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, bufferName.get(Buffer.ELEMENT));
ShortBuffer elementBuffer = GLBuffers.newDirectShortBuffer(elementData);
gl4.glBufferData(GL_ELEMENT_ARRAY_BUFFER, elementSize, elementBuffer, GL_STATIC_DRAW);
BufferUtils.destroyDirectBuffer(elementBuffer);
gl4.glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, 0);
gl4.glBindBuffer(GL_ARRAY_BUFFER, bufferName.get(Buffer.VERTEX));
FloatBuffer vertexBuffer = GLBuffers.newDirectFloatBuffer(vertexData);
gl4.glBufferData(GL_ARRAY_BUFFER, vertexSize, vertexBuffer, GL_STATIC_DRAW);
BufferUtils.destroyDirectBuffer(vertexBuffer);
gl4.glBindBuffer(GL_ARRAY_BUFFER, 0);
int[] uniformBufferOffset = {0};
gl4.glGetIntegerv(GL_UNIFORM_BUFFER_OFFSET_ALIGNMENT, uniformBufferOffset, 0);
int uniformBlockSize = Math.max(Mat4.SIZE, uniformBufferOffset[0]);
gl4.glBindBuffer(GL_UNIFORM_BUFFER, bufferName.get(Buffer.TRANSFORM));
gl4.glBufferData(GL_UNIFORM_BUFFER, uniformBlockSize, null, GL_DYNAMIC_DRAW);
gl4.glBindBuffer(GL_UNIFORM_BUFFER, 0);
return validated;
}
public void writeDebugi(GL4 gl, int buffer, int count) throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "debug.txt")))) {
// write debug
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, buffer);
IntBuffer debug = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY).asIntBuffer();
for (int i = 0; i < count; i++) {
writer.append(String.format("%4d: %8d", i, debug.get(i)));
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
private boolean initVertexArray(GL4 gl4) {
boolean validated = true;
gl4.glGenVertexArrays(1, vertexArrayName);
gl4.glBindVertexArray(vertexArrayName.get(0));
{
gl4.glBindBuffer(GL_ARRAY_BUFFER, bufferName.get(Buffer.VERTEX));
gl4.glVertexAttribPointer(Semantic.Attr.POSITION, 2, GL_FLOAT, false, 2 * Vec2.SIZE, 0);
gl4.glVertexAttribPointer(
Semantic.Attr.TEXCOORD, 2, GL_FLOAT, false, 2 * Vec2.SIZE, Vec2.SIZE);
gl4.glBindBuffer(GL_ARRAY_BUFFER, 0);
gl4.glEnableVertexAttribArray(Semantic.Attr.POSITION);
gl4.glEnableVertexAttribArray(Semantic.Attr.TEXCOORD);
gl4.glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, bufferName.get(Buffer.ELEMENT));
}
gl4.glBindVertexArray(0);
return validated;
}
public void writeDebug4f(GL4 gl, int buffer, int count) throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "debug.txt")))) {
// write debug
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, buffer);
ByteBuffer debug = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
for (int i = 0; i < count; i++) {
Vector4f v = getVec4(debug, i * 16);
writer.append(String.format("%4d: [%f %f %f %f]", i, v.x, v.y, v.z, v.w));
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void writeArcHashes(GL4 gl, int arcsBuffer, int arcHashesBuffer, int atomCount)
throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "hashes.txt")))) {
// read arcs
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, arcsBuffer);
ByteBuffer data = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
float arcs[] = new float[Scene.MAX_TOTAL_ARCS * 4];
data.asFloatBuffer().get(arcs);
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// write hashes
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, arcHashesBuffer);
int stride = Scene.SIZEOF_HASH;
ByteBuffer hashes = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
for (int i = 0; i < Scene.MAX_TOTAL_ARC_HASHES; i++) {
int key = hashes.getInt(i * stride);
if (key != Scene.INVALID_KEY) {
char primary = ((key & 0xf0000000) == 0) ? 'p' : 's';
int ai = (key & 0x0fffffff) / atomCount;
int aj = (key & 0x0fffffff) % atomCount;
int atomk = hashes.getInt(i * stride + 4);
int index = hashes.getInt(i * stride + 8);
writer.append(String.format("[%4d, %4d] (%c): %4d, %8d", ai, aj, primary, atomk, index));
float x = arcs[4 * index];
float y = arcs[4 * index + 1];
float z = arcs[4 * index + 2];
float k = arcs[4 * index + 3];
writer.append(String.format(" --> (%f %f %f %f)", x, y, z, k));
} else {
writer.append("(EMPTY)");
}
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void writeAtomsVisible(GL4 gl, int atomsVisibleBuffer, int atomCount) throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "atoms.txt")))) {
// write tori
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, atomsVisibleBuffer);
IntBuffer atomsVisible = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY).asIntBuffer();
for (int i = 0; i < atomCount; i++) {
int visible = atomsVisible.get(i);
writer.append(String.format("%4d: ", i));
if (visible == 1) {
writer.append("1");
}
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
public void writeTriangles(GL4 gl, int trianglesArrayBuffer, int count) throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "triangles.txt")))) {
// write triangles
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, trianglesArrayBuffer);
ByteBuffer triangleArray = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
for (int i = 0; i < count; i++) {
Vector4f position = getVec4(triangleArray, i * Scene.SIZEOF_TRIANGLE);
writer.append(String.format("%4d: ", i));
writer.append(
String.format(
"position: [%f %f %f %f]", position.x, position.y, position.z, position.w));
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
@Override
protected boolean render(GL gl) {
GL4 gl4 = (GL4) gl;
FloatUtil.makePerspective(
projection,
0,
true,
(float) Math.PI * 0.25f,
(float) windowSize.x / windowSize.y,
0.1f,
100.0f);
view = view();
FloatUtil.makeIdentity(model);
FloatUtil.multMatrix(projection, view);
FloatUtil.multMatrix(projection, model);
for (int i = 0; i < projection.length; i++) {
mvp[i] = projection[i];
}
gl4.glProgramUniformMatrix4dv(
programName[Program.VERT.ordinal()], uniformMvp, 1, false, mvp, 0);
gl4.glProgramUniform4dv(
programName[Program.FRAG.ordinal()],
uniformDiffuse,
1,
new double[] {1.0f, 0.5f, 0.0f, 1.0f},
0);
gl4.glViewportIndexedfv(0, new float[] {0, 0, windowSize.x, windowSize.y}, 0);
gl4.glClearBufferfv(GL_COLOR, 0, new float[] {0.0f, 0.0f, 0.0f, 0.0f}, 0);
gl4.glBindProgramPipeline(pipelineName[0]);
gl4.glBindVertexArray(vertexArrayName[0]);
gl4.glBindBuffer(GL_ELEMENT_ARRAY_BUFFER, bufferName[Buffer.ELEMENT.ordinal()]);
gl4.glDrawElementsInstancedBaseVertex(GL_TRIANGLES, elementCount, GL_UNSIGNED_SHORT, 0, 1, 0);
return true;
}
public void writePolygons(GL4 gl, int spheresArrayBuffer, int count) throws IOException {
try (BufferedWriter writer =
new BufferedWriter(new FileWriter(new File(debugDir, "polygons.txt")))) {
// write polygons
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, spheresArrayBuffer);
ByteBuffer polygonsArray = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY);
for (int i = 0; i < count; i++) {
int index = polygonsArray.getInt(i * Scene.SIZEOF_POLYGON + 16);
int label = polygonsArray.getInt(i * Scene.SIZEOF_POLYGON + 20);
int circleStart = polygonsArray.getInt(i * Scene.SIZEOF_POLYGON + 24);
int circleLength = polygonsArray.getInt(i * Scene.SIZEOF_POLYGON + 28);
writer.append(String.format("%4d: ", i));
writer.append(String.format("index: %4d, ", index));
writer.append(String.format("label: %4d, ", label));
writer.append(String.format("circle: [%6d, %2d]", circleStart, circleLength));
writer.newLine();
}
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
}
@Override
protected boolean render(GL gl) {
GL4 gl4 = (GL4) gl;
{
gl4.glBindBuffer(GL_UNIFORM_BUFFER, bufferName.get(Buffer.TRANSFORM));
ByteBuffer pointer =
gl4.glMapBufferRange(
GL_UNIFORM_BUFFER, 0, Mat4.SIZE, GL_MAP_WRITE_BIT | GL_MAP_INVALIDATE_BUFFER_BIT);
Mat4 projection =
glm.perspectiveFov_((float) Math.PI * 0.25f, windowSize.x, windowSize.y, 0.1f, 100.0f);
Mat4 model = new Mat4(1.0f);
pointer.asFloatBuffer().put(projection.mul(viewMat4()).mul(model).toFa_());
// Make sure the uniform buffer is uploaded
gl4.glUnmapBuffer(GL_UNIFORM_BUFFER);
}
gl4.glViewportIndexedf(0, 0, 0, windowSize.x, windowSize.y);
gl4.glClearBufferfv(GL_COLOR, 0, new float[] {1.0f, 0.5f, 0.0f, 1.0f}, 0);
gl4.glBindProgramPipeline(pipelineName.get(0));
gl4.glActiveTexture(GL_TEXTURE0);
gl4.glBindTexture(GL_TEXTURE_2D_ARRAY, textureName.get(0));
gl4.glBindVertexArray(vertexArrayName.get(0));
gl4.glBindBufferBase(
GL_UNIFORM_BUFFER, Semantic.Uniform.TRANSFORM0, bufferName.get(Buffer.TRANSFORM));
gl4.glDrawElementsInstancedBaseVertexBaseInstance(
GL_TRIANGLES, elementCount, GL_UNSIGNED_SHORT, 0, 1, 0, 0);
return true;
}
public void connectedComponents(
GL4 gl,
int toriBuffer,
int edgesBuffer,
int torusCount,
int verticesBuffer,
int vertexCount,
int sphereCount) {
boolean[][] adjacency = new boolean[vertexCount][vertexCount];
int[][] edges = new int[torusCount][2];
int[][] spheres = new int[torusCount][2];
// map surface edges buffer
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, edgesBuffer);
IntBuffer edgesData = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY).asIntBuffer();
// fill adjacency matrix
for (int i = 0; i < torusCount; i++) {
int v0 = edgesData.get(4 * i);
int v1 = edgesData.get(4 * i + 1);
if (v0 != -1 && v1 != -1) {
adjacency[v0][v1] = true;
adjacency[v1][v0] = true;
// store edges
edges[i][0] = v0;
edges[i][1] = v1;
// get adjacent spheres
spheres[i][0] = edgesData.get(4 * i + 2);
spheres[i][1] = edgesData.get(4 * i + 3);
}
}
// unmap edges buffer
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
// create CSR representation of the adjacency matrix
int count = 0;
int[] rows = new int[vertexCount + 1];
rows[0] = count;
for (int i = 0; i < vertexCount; i++) {
for (int j = 0; j < vertexCount; j++) {
if (adjacency[i][j]) {
count++;
}
}
rows[i + 1] = count;
}
int index = 0;
int[] columns = new int[count];
for (int i = 0; i < vertexCount; i++) {
for (int j = 0; j < vertexCount; j++) {
if (adjacency[i][j]) {
columns[index] = j;
index++;
}
}
}
Deque<Integer> vertices = new ArrayDeque<>();
List<Integer> sizes = new ArrayList<>();
boolean[] visited = new boolean[vertexCount];
int[] labels = new int[vertexCount];
int label = 1;
int startVertex;
do {
// find start vertex
startVertex = -1;
for (int i = 0; i < vertexCount; i++) {
if (!visited[i]) {
startVertex = i;
break;
}
}
if (startVertex < 0) {
break;
}
// label one connected component using BFS
int size = 0;
vertices.add(startVertex);
while (!vertices.isEmpty()) {
int vertex = vertices.poll();
if (visited[vertex]) {
continue;
}
visited[vertex] = true;
labels[vertex] = label;
int start = rows[vertex];
int end = rows[vertex + 1];
for (int i = start; i < end; i++) {
vertices.add(columns[i]);
}
size++;
}
sizes.add(size);
label++;
} while (startVertex >= 0);
// find largest connceted component
int largest = 0;
for (int i = 1; i < sizes.size(); i++) {
if (sizes.get(i) > sizes.get(largest)) {
largest = i;
}
}
// map surface vertices buffer
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, verticesBuffer);
IntBuffer verticesBuf = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_WRITE_ONLY).asIntBuffer();
// write surface/cavity flags
for (int i = 0; i < labels.length; i++) {
int value = labels[i] == (largest + 1) ? 1 : 0;
verticesBuf.put(i, value);
}
// unmap edges buffer
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
sphereCircles(gl, edges, spheres, sphereCount, toriBuffer, torusCount);
try {
writeCSR(rows, columns);
writeLabels(labels, sizes);
} catch (IOException ex) {
ex.printStackTrace(System.err);
}
}
private void sphereCircles(
GL4 gl, int[][] edges, int[][] spheres, int sphereCount, int toriBuffer, int torusCount) {
List<List<Integer>> circles = new ArrayList<>();
for (int i = 0; i < sphereCount; i++) {
circles.add(new ArrayList<Integer>());
}
for (int i = 0; i < torusCount; i++) {
circles.get(spheres[i][0]).add(i);
circles.get(spheres[i][1]).add(i);
}
for (int i = 0; i < sphereCount; i++) {
List<Integer> circle = circles.get(i);
if (circle.isEmpty()) {
continue;
}
int current = edges[circle.get(0)][1];
for (int j = 0; j < circle.size() - 1; j++) {
int k;
for (k = j + 1; k < circle.size(); k++) {
int v0 = edges[circle.get(k)][0];
int v1 = edges[circle.get(k)][1];
if (current == v0) {
current = v1;
break;
} else if (current == v1) {
current = v0;
break;
}
}
if (k == circle.size()) {
current = edges[circle.get(j + 1)][1];
System.out.println("Another circle found. Sphere: " + i);
}
if (k < circle.size() && k != j + 1) {
int tmp = circle.get(j + 1);
circle.set(j + 1, circle.get(k));
circle.set(k, tmp);
}
}
}
// map surface edges buffer
gl.glBindBuffer(GL_SHADER_STORAGE_BUFFER, toriBuffer);
FloatBuffer tori = gl.glMapBuffer(GL_SHADER_STORAGE_BUFFER, GL_READ_ONLY).asFloatBuffer();
// fill adjacency matrix
/*for (int i = 0; i < torusCount; i++) {
int v0 = edges.get(4 * i);
int v1 = edges.get(4 * i + 1);
if (v0 != -1 && v1 != -1) {
adjacency[v0][v1] = true;
adjacency[v1][v0] = true;
// get adjacent spheres
spheres[i][0] = edges.get(4 * i + 2);
spheres[i][1] = edges.get(4 * i + 3);
}
}
// unmap edges buffer*/
gl.glUnmapBuffer(GL_SHADER_STORAGE_BUFFER);
}
public void unbind() {
GL4 gl = GLGraphics.currentGL();
gl.glBindBuffer(_myTarget.glID, 0);
}
