support splat files

SplatTiler/SplatTiler.cpp

@@ -48,14 +48,112 @@ struct OctreeNode {
     bool isLeaf() const { return children.empty(); }
 };
 
+// Helper to load raw .splat files (standard 32-byte format)
+bool loadRawSplat(const std::string& filename, spz::GaussianCloud& cloud) {
+    std::ifstream file(filename, std::ios::binary | std::ios::ate);
+    if (!file) return false;
+
+    size_t fileSize = file.tellg();
+    file.seekg(0, std::ios::beg);
+
+    if (fileSize == 0 || fileSize % 32 != 0) {
+        std::cerr << "Error: .splat file size is not a multiple of 32." << std::endl;
+        return false;
+    }
+
+    int32_t numPoints = static_cast<int32_t>(fileSize / 32);
+    cloud.numPoints = numPoints;
+    cloud.shDegree = 0;
+    cloud.antialiased = false;
+
+    cloud.positions.resize(numPoints * 3);
+    cloud.scales.resize(numPoints * 3);
+    cloud.rotations.resize(numPoints * 4);
+    cloud.alphas.resize(numPoints);
+    cloud.colors.resize(numPoints * 3);
+    cloud.sh.clear();
+
+    struct RawSplatPoint {
+        float pos[3];
+        float scale[3];
+        uint8_t color[4];
+        uint8_t rot[4];
+    };
+
+    std::vector<RawSplatPoint> buffer(numPoints);
+    if (!file.read(reinterpret_cast<char*>(buffer.data()), fileSize)) return false;
+
+    const float SH_C0 = 0.28209479177387814f;
+
+    for (int i = 0; i < numPoints; ++i) {
+        const auto& p = buffer[i];
+
+        // Position
+        cloud.positions[i * 3 + 0] = p.pos[0];
+        cloud.positions[i * 3 + 1] = p.pos[1];
+        cloud.positions[i * 3 + 2] = p.pos[2];
+
+        // Scale (Linear -> Log)
+        cloud.scales[i * 3 + 0] = std::log(std::max(1e-6f, p.scale[0]));
+        cloud.scales[i * 3 + 1] = std::log(std::max(1e-6f, p.scale[1]));
+        cloud.scales[i * 3 + 2] = std::log(std::max(1e-6f, p.scale[2]));
+
+        // Color (Uint8 -> SH0)
+        cloud.colors[i * 3 + 0] = (p.color[0] / 255.0f - 0.5f) / SH_C0;
+        cloud.colors[i * 3 + 1] = (p.color[1] / 255.0f - 0.5f) / SH_C0;
+        cloud.colors[i * 3 + 2] = (p.color[2] / 255.0f - 0.5f) / SH_C0;
+
+        // Alpha (Uint8 -> Logit)
+        float a = p.color[3] / 255.0f;
+        if (a < 1.0f/255.0f) a = 1.0f/255.0f;
+        if (a > 254.0f/255.0f) a = 254.0f/255.0f;
+        cloud.alphas[i] = std::log(a / (1.0f - a));
+
+        // Rotation (Uint8 -> Float Quaternion)
+        float r0 = (p.rot[0] - 128.0f) / 128.0f;
+        float r1 = (p.rot[1] - 128.0f) / 128.0f;
+        float r2 = (p.rot[2] - 128.0f) / 128.0f;
+        float r3 = (p.rot[3] - 128.0f) / 128.0f;
+        
+        float len2 = r0*r0 + r1*r1 + r2*r2 + r3*r3;
+        if (len2 > 0.0f) {
+            float invLen = 1.0f / std::sqrt(len2);
+            r0 *= invLen; r1 *= invLen; r2 *= invLen; r3 *= invLen;
+        } else {
+            r0 = 1.0f;
+        }
+
+        cloud.rotations[i * 4 + 0] = r0;
+        cloud.rotations[i * 4 + 1] = r1;
+        cloud.rotations[i * 4 + 2] = r2;
+        cloud.rotations[i * 4 + 3] = r3;
+    }
+
+    return true;
+}
+
 SplatTiler::SplatTiler() {}
 SplatTiler::~SplatTiler() {}
 
-bool SplatTiler::load(const std::string& plyFilename) {
+bool SplatTiler::load(const std::string& filename) {
+    std::string ext = fs::path(filename).extension().string();
+    std::transform(ext.begin(), ext.end(), ext.begin(), ::tolower);
+
     spz::UnpackOptions opts;
     // Load as RDF (Raw) to avoid implicit conversion. We will handle orientation via the matrix.
     opts.to = spz::CoordinateSystem::RDF;
-    m_cloud = spz::loadSplatFromPly(plyFilename, opts);
+
+    if (ext == ".ply") {
+        m_cloud = spz::loadSplatFromPly(filename, opts);
+    } else if (ext == ".spz") {
+        m_cloud = spz::loadSpz(filename, opts);
+    } else if (ext == ".splat") {
+        return loadRawSplat(filename, m_cloud);
+    } else {
+        std::cerr << "Error: Unsupported file extension '" << ext << "'. Supported: .ply, .spz, .splat" << std::endl;
+        return false;
+    }
+
     return m_cloud.numPoints > 0;
 }
 

SplatTiler/SplatTiler.h

@@ -31,8 +31,8 @@ public:
     SplatTiler();
     ~SplatTiler();
 
-    // Load a PLY file into memory
-    bool load(const std::string& plyFilename);
+    // Load a PLY/SPZ/SPLAT file into memory
+    bool load(const std::string& filename);
 
     // Run the tiling and export process
     void process(const TileConfig& config);

SplatTiler/main.cpp

@@ -6,7 +6,7 @@
 #include "SplatTiler.h"
 
 void printUsage(const char* progName) {
-    std::cout << "Usage: " << progName << " [options] <input_ply>\n";
+    std::cout << "Usage: " << progName << " [options] <input_file>\n";
     std::cout << "Options:\n";
     std::cout << "  -o, --output <dir>       Output directory (default: output_tiles)\n";
     std::cout << "  -m, --max-points <num>   Max points per tile (default: 65536)\n";
@@ -25,7 +25,7 @@ int main(int argc, char* argv[]) {
     }
 
     tile3dgs::TileConfig config;
-    std::string inputPly;
+    std::string inputFile;
 
     for (int i = 1; i < argc; ++i) {
         std::string arg = argv[i];
@@ -123,25 +123,25 @@ int main(int argc, char* argv[]) {
             printUsage(argv[0]);
             return 1;
         } else {
-            inputPly = arg;
+            inputFile = arg;
         }
     }
 
-    if (inputPly.empty()) {
-        std::cerr << "Error: No input PLY file specified.\n";
+    if (inputFile.empty()) {
+        std::cerr << "Error: No input file specified.\n";
         printUsage(argv[0]);
         return 1;
     }
 
-    if (!std::filesystem::exists(inputPly)) {
-        std::cerr << "Error: Input file does not exist: " << inputPly << "\n";
+    if (!std::filesystem::exists(inputFile)) {
+        std::cerr << "Error: Input file does not exist: " << inputFile << "\n";
         return 1;
     }
 
-    std::cout << "Loading " << inputPly << "...\n";
+    std::cout << "Loading " << inputFile << "...\n";
     tile3dgs::SplatTiler tiler;
-    if (!tiler.load(inputPly)) {
-        std::cerr << "Failed to load PLY file (or file is empty).\n";
+    if (!tiler.load(inputFile)) {
+        std::cerr << "Failed to load file (or file is empty).\n";
         return 1;
     }
 

client/src/components/MeasurePanel.jsx

@@ -22,9 +22,10 @@ import {
 import DownloadIcon from "@mui/icons-material/Download";
 import DeleteIcon from "@mui/icons-material/Delete";
 import EditIcon from "@mui/icons-material/Edit";
+import { getDefaultBucket } from "../utils/awsConfig";
 
 const DEFAULT_CUSTOM_MODEL = {
-  bucket: import.meta.env.VITE_S3_BUCKET || "ddy-first-bucket",
+  bucket: getDefaultBucket(),
   prefix: "data/3dtiles",
   tileset: "custom/tileset.json",
   offsetHeight: 0,

client/src/utils/awsConfig.js

@@ -14,10 +14,10 @@ import { S3Client, GetObjectCommand } from "@aws-sdk/client-s3";
 import { fromCognitoIdentityPool } from "@aws-sdk/credential-provider-cognito-identity";
 import { CognitoIdentityClient } from "@aws-sdk/client-cognito-identity";
 
-export const AWS_REGION = "us-east-2";
+export const AWS_REGION = import.meta.env.VITE_AWS_REGION || "us-east-2";
 // Cognito pool ID is safe to embed in client code (not a secret)
-export const COGNITO_IDENTITY_POOL_ID = "us-east-2:1387102d-5a87-4742-8a69-1baa70f37984";
-export const DEFAULT_BUCKET = "ddy-first-bucket";
+export const COGNITO_IDENTITY_POOL_ID = import.meta.env.VITE_AWS_COGNITO_IDENTITY_POOL_ID || "us-east-2:1387102d-5a87-4742-8a69-1baa70f37984";
+export const DEFAULT_BUCKET = import.meta.env.VITE_S3_BUCKET || "ddy-first-bucket";
 
 /**
  * Initialize S3 client with Cognito credentials
@@ -155,6 +155,5 @@ export function getAwsRegion() {
  * Validate AWS configuration
  */
 export function validateAwsConfig() {
-  // Configuration is hardcoded, always valid
-  return true;
+  return !!(AWS_REGION && COGNITO_IDENTITY_POOL_ID && DEFAULT_BUCKET);
 }