mirror of
https://github.com/schrodinger/pymol-open-source.git
synced 2026-06-05 20:14:24 +08:00
c96a54561d
New default algorithm for `--use-vtkm=no`. It's not as good as the VTK-m implementation, but should be better than marching tetrahedra and doesn't require any external libraries.
57 lines
1.3 KiB
C++
57 lines
1.3 KiB
C++
/**
|
|
* @file Marching cubes
|
|
*
|
|
* Based on https://github.com/ilastik/marching_cubes
|
|
*
|
|
* License: BSD-3-Clause
|
|
*
|
|
* Copyright 2018, The ilastik development team
|
|
* Copyright 2020, Schrodinger, Inc.
|
|
*/
|
|
|
|
#ifndef MARCHING_CUBES_H
|
|
#define MARCHING_CUBES_H
|
|
|
|
#include <memory>
|
|
|
|
namespace mc
|
|
{
|
|
|
|
struct Point {
|
|
float x, y, z;
|
|
float operator[](size_t pos) const { return (&x)[pos]; }
|
|
float& operator[](size_t pos) { return (&x)[pos]; }
|
|
};
|
|
|
|
class Field
|
|
{
|
|
public:
|
|
virtual ~Field() = default;
|
|
virtual size_t xDim() const = 0;
|
|
virtual size_t yDim() const = 0;
|
|
virtual size_t zDim() const = 0;
|
|
virtual float get(size_t x, size_t y, size_t z) const = 0;
|
|
virtual Point get_point(size_t x, size_t y, size_t z) const = 0;
|
|
Point get_gradient(size_t x, size_t y, size_t z) const;
|
|
};
|
|
|
|
/**
|
|
* Isosurface mesh with faces, vertices and normals
|
|
*/
|
|
struct Mesh {
|
|
size_t vertexCount = 0; //!< the number of vertices
|
|
std::unique_ptr<Point[]> vertices; //!< vertex positions
|
|
std::unique_ptr<Point[]> normals; //!< normal direction of each vertex
|
|
|
|
size_t faceCount = 0; //!< the number of faces
|
|
std::unique_ptr<size_t[]>
|
|
faces; //!< the faces given by 3 vertex indices (length = faceCount * 3)
|
|
};
|
|
|
|
Mesh march(const Field& volume, float isoLevel, bool gradient_normals = true);
|
|
|
|
void calculateNormals(Mesh& mesh);
|
|
|
|
} // namespace mc
|
|
#endif
|