stuff and even more stuff

dataset/primgen.py

@@ -0,0 +1,136 @@
+import subprocess as sp
+import subprocess
+import glob, os
+import shutil
+import pointcloud as pc
+
+modelType = "off"
+samplingRate = 0.003
+maxDistance = 0.003
+maxAngleDistance = 0.01
+errorSigma = 0
+k = 16
+pointCloudSize = 2048
+cutOutProb = 0
+maxIterations = 10
+
+toolPath = "C:/Projekte/csg_playground_build/Release/primgen.exe"
+
+def getExtension(modType):
+    if("off" in modType):
+        return "off"
+    elif("obj" in modType):
+        return "obj"
+    elif("csg" in modType):
+        return ".json"
+    else:
+        return ""
+
+
+def run(modelPath, outputFolder):
+
+    executable = "{} {} \"{}\" \"{}\" {} {} {} {} {} {} {} {}".format(
+        toolPath, modelType, modelPath, outputFolder,
+        samplingRate, maxDistance, maxAngleDistance, errorSigma,
+        k, pointCloudSize, cutOutProb, maxIterations)
+
+    print("Call generator with " + executable)
+
+    p = sp.Popen(executable, stdout=subprocess.PIPE)
+    for line in p.stdout:
+        print(line)
+
+    # Wait until process terminates (without using p.wait())
+    # while p.poll() is None:
+        # Process hasn't exited yet, let's wait some
+    #    time.sleep(0.5)
+
+    # Get return code from process
+    return_code = p.returncode
+
+    if return_code == -1:
+        print("Unable to generate primtitives for model {}".format(modelPath));
+    else:
+        print("Done. Exist Code: " + str(return_code))
+
+
+def extract_clusters(outputFolder, cluster_method="none", **kwargs):
+
+    if "none" in cluster_method:
+        return
+
+    all_clusters = []
+
+    for subdir, dirs, _ in os.walk(outputFolder):
+        for dir in dirs:
+            path = os.path.join(subdir, dir)
+            for _,_,files in os.walk(path):
+                for file in files:
+                    file = os.path.join(path, file.lower())
+                    print(file)
+                    if file.endswith("_pc.xyz"):
+                        pointcloud = pc.read_pointcloud(file);
+
+                        clusters = []
+                        if "per_primitive" in cluster_method:
+                            clusters = pc.cluster_per_column(pointcloud, column=7) # primitive id column
+                        elif "cube" in cluster_method:
+                            clusters = pc.cluster_cubes(pointcloud,
+                                                        kwargs.get("cluster_dims", [1,1,1]))
+                        elif "dbscan" in cluster_method:
+                            clusters = pc.cluster_dbscan(pointcloud,
+                                                         selected_indices=kwargs.get("selected_indices", [0,1,2,3,4,5]),
+                                                         eps=kwargs.get("eps", 0.1),
+                                                         min_samples=kwargs.get("min_samples", None)
+                                                         )
+
+                        for idx, cluster in enumerate(clusters):
+                            pos = file.rfind("pc.xyz")
+                            new_file = file[:pos] + str(idx) + "_pc.xyz"
+                            pc.write_pointcloud(new_file, cluster)
+
+                        # all_clusters.extend(clusters)
+
+    return all_clusters
+
+def runForFolder(modelFolder, outputFolder):
+
+    os.chdir(modelFolder)
+    modelFiles = glob.glob("*." + getExtension(modelType))
+    print("Working on {} model files with extension {} from folder {}.".format(len(modelFiles),getExtension(modelType), modelFolder))
+
+    folderIdx = 0
+    for modelFile in modelFiles:
+
+        try:
+            subfolderPath = outputFolder + str(folderIdx) + "/"
+            folderIdx += 1
+
+            print("Check if output sub folder exists...")
+            if os.path.exists(subfolderPath) and os.path.isdir(subfolderPath):
+                shutil.rmtree(subfolderPath)
+                print("Yes => Existing sub folder was deleted.")
+
+            os.mkdir(subfolderPath)
+            print("Successfully created the directory %s " % subfolderPath)
+
+            run(modelFolder + modelFile, subfolderPath)
+        except OSError as err:
+            print("Creation of the directory %s failed." % str(err))
+
+
+
+
+if __name__ == "__main__":
+
+    outputFolder = "D:/output_0/"
+    modelFolder = "C:/Users/friedrich/PycharmProjects/data/models/"
+
+    # clusters = extract_clusters("C:/Projekte/csg_playground_build/testOFF", "cube",
+    #                            cluster_dims=[2,2,2])
+    #                            #eps=0.1, min_samples=0.01, selected_indices=[7])
+
+    # pc.draw_clusters(clusters)
+
+    runForFolder(modelFolder, outputFolder)
+