Parcourir la source

Add incr merge program

Jérôme BUISINE il y a 4 ans
Parent
commit
c74e450eab
3 fichiers modifiés avec 188 ajouts et 19 suppressions
  1. 4 1
      main/CMakeLists.txt
  2. 182 0
      main/rawls_merge_incr.cpp
  3. 2 18
      reconstruct_png.sh

+ 4 - 1
main/CMakeLists.txt

@@ -1,11 +1,14 @@
 add_executable(rawls_convert rawls_convert.cpp)
 add_executable(rawls_merge rawls_merge.cpp)
 add_executable(rawls_h_flip rawls_h_flip.cpp)
+add_executable(rawls_merge_incr rawls_merge_incr.cpp)
 
 target_link_libraries(rawls_convert LINK_PUBLIC rawls)
 target_link_libraries(rawls_merge LINK_PUBLIC rawls stdc++fs)
 target_link_libraries(rawls_h_flip LINK_PUBLIC rawls stdc++fs)
+target_link_libraries(rawls_merge_incr LINK_PUBLIC rawls stdc++fs)
 
 set_property(TARGET rawls_merge PROPERTY CXX_STANDARD 17)
 set_property(TARGET rawls_convert PROPERTY CXX_STANDARD 17)
-set_property(TARGET rawls_h_flip PROPERTY CXX_STANDARD 17)
+set_property(TARGET rawls_h_flip PROPERTY CXX_STANDARD 17)
+set_property(TARGET rawls_merge_incr PROPERTY CXX_STANDARD 17)

+ 182 - 0
main/rawls_merge_incr.cpp

@@ -0,0 +1,182 @@
+#include <iostream>
+#include <string.h>
+#include <memory>
+
+#include "lodepng.h"
+#include "rawls.h"
+
+#include <algorithm>
+#include <filesystem>
+
+void writeProgress(float progress){
+    int barWidth = 200;
+
+    std::cout << "[";
+    int pos = barWidth * progress;
+    for (int i = 0; i < barWidth; ++i) {
+        if (i < pos) std::cout << "=";
+        else if (i == pos) std::cout << ">";
+        else std::cout << " ";
+    }
+    std::cout << "] " << int(progress * 100.0) << " %\r";
+    std::cout.flush();
+}
+
+/*
+ * Save current step images from current buffer
+ */
+bool saveCurrentImage(int width, int height, int nbChanels, float* buffer, std::string outfileName){
+    
+    // create outfile 
+    if (rawls::HasExtension(outfileName, ".ppm")){
+        rawls::saveAsPPM(width, height, nbChanels, buffer, outfileName);
+    } 
+    else if (rawls::HasExtension(outfileName, ".png")){
+        rawls::saveAsPNG(width, height, nbChanels, buffer, outfileName);
+    } 
+
+    // TODO : add this option
+    /*else if (rawls::HasExtension(outfileName, ".rawls") || rawls::HasExtension(outfileName, ".rawls_20")){
+        // need to get comments from an image
+        std::string comments = rawls::getCommentsRAWLS(imagesPath.at(0));
+
+        // Here no gamma conversion is done, only mean of samples
+        rawls::saveAsRAWLS(width, height, nbChanels, comments, buffer, outfileName);
+    }*/
+    else{
+        std::cout << "Unexpected output extension image" << std::endl;
+        return false;
+    }
+
+    return true;
+}
+
+/*
+ * Incremental merge of `rawls` images
+ */
+int main(int argc, char *argv[]){
+
+    std::string folderName;
+    std::string outputFolder;
+    std::string prefixImageName;
+    std::string imageExtension;
+
+    unsigned step = 10;
+    unsigned maxSamples = 0;
+    bool random;
+
+    for (int i = 1; i < argc; ++i) {
+        if (!strcmp(argv[i], "--folder") || !strcmp(argv[i], "-folder")) {
+            folderName = argv[++i];
+        } else if (!strcmp(argv[i], "--step") || !strcmp(argv[i], "-step")) {
+            step = atoi(argv[++i]);
+        }else if (!strcmp(argv[i], "--random") || !strcmp(argv[i], "-random")) {
+            random = bool(atoi(argv[++i]));
+        }else if (!strcmp(argv[i], "--output") || !strcmp(argv[i], "-output")) {
+            outputFolder = argv[++i];
+        }else if (!strcmp(argv[i], "--prefix") || !strcmp(argv[i], "-prefix")) {
+            prefixImageName = argv[++i];
+        }else if (!strcmp(argv[i], "--max") || !strcmp(argv[i], "-max")) {
+            maxSamples = atoi(argv[++i]);
+        }else if (!strcmp(argv[i], "--extension") || !strcmp(argv[i], "-extension")) {
+            imageExtension = argv[++i];
+        }
+    }
+
+    std::vector<std::string> imagesPath;
+
+    for (const auto & entry : std::filesystem::directory_iterator(folderName)){
+        std::string imageName = entry.path().string();
+        if (rawls::HasExtension(imageName, ".rawls") || rawls::HasExtension(imageName, ".rawls_20")){
+            imagesPath.push_back(imageName);
+        }
+    }
+
+    // sort or shuffle the images path
+    if (!random){
+        std::sort(imagesPath.begin(), imagesPath.end(), std::less<std::string>());
+    }else{
+        std::random_shuffle(imagesPath.begin(), imagesPath.end());
+    }
+
+
+    unsigned width, height, nbChanels;
+    float* outputStepBuffer;
+    float* outputBuffer;
+
+    if (imagesPath.size() > 0){
+
+        std::tuple<unsigned, unsigned, unsigned> dimensions = rawls::getDimensionsRAWLS(imagesPath.at(0));
+
+        width = std::get<0>(dimensions);
+        height = std::get<1>(dimensions);
+        nbChanels = std::get<2>(dimensions);
+        outputBuffer = new float[width * height * nbChanels];
+        outputStepBuffer = new float[width * height * nbChanels];
+
+        // init values of buffer
+        for (int i = 0; i < height * width * nbChanels; i++){
+            outputBuffer[i] = 0;
+            outputStepBuffer[i] = 0;
+        }
+    }
+    else
+    {
+        std::cout << "Folder is empty..." << std::endl;
+        return 1;
+    }
+
+    // just for indication
+    float progress = 0.0;
+    unsigned bufferSize = width * height * nbChanels;
+
+    for (unsigned i = 1; i < maxSamples; i++){
+
+        // read into folder all `.rawls` file and merge pixels values
+        float* buffer = rawls::getPixelsRAWLS(imagesPath.at(i));
+
+        for(unsigned y = 0; y < height; y++){
+
+            for(unsigned x = 0; x < width; x++) {
+
+                for(unsigned j = 0; j < nbChanels; j++){
+                    
+                    float value = buffer[nbChanels * width * y + nbChanels * x + j];
+                    outputBuffer[nbChanels * width * y + nbChanels * x + j] = outputBuffer[nbChanels * width * y + nbChanels * x + j] + value;
+                }
+            }
+        }
+
+        // save a new 
+        if (i % step == 0){
+             // mean all samples values by number of samples used
+            for (int j = 0; j < height * width * nbChanels; j++){
+                outputStepBuffer[j] = outputBuffer[j] / i;
+            }
+
+            std::string suffix = std::to_string(i);
+
+            while(suffix.length() < 5){
+                suffix = "0" + suffix;
+            }
+
+            // TODO : build outfileName
+            std::string outfileName = outputFolder + "/" + prefixImageName + "_" + suffix + "." + imageExtension;
+
+            saveCurrentImage(width, height, nbChanels, outputStepBuffer, outfileName);
+            writeProgress(progress);
+        }
+
+        // update and write progress information
+        progress += (1 / (float)maxSamples);
+
+        delete buffer;
+    }
+    
+    writeProgress(1.);
+    std::cout << std::endl;
+
+    // delete the outputbuffer used
+    delete outputBuffer;
+    delete outputStepBuffer;
+}

+ 2 - 18
reconstruct_png.sh

@@ -44,23 +44,7 @@ do
     
     mkdir -p $output_scene_path_fixed
 
-    samples=$step
-
-    while [ $samples -le ${nb_elements} ]
-    do
-      suffix=$samples
-
-      while [ ${#suffix} -le 5 ]
-      do
-        suffix="0"${suffix}
-      done
-
-      # need to create `outfile` name
-      outfile="${folder}_${suffix}.png"
-      echo "$folder: [merge of ${samples} samples] into ${outfile}"
-
-      ./${build_folder}/main/rawls_merge --folder ${folder_path} --samples ${samples} --random 0 --outfile ${output_scene_path_fixed}/${outfile}
-      samples=$(( $samples + $step ))
-    done
+    ./${build_folder}/main/rawls_merge_incr --folder ${folder_path} --output $output_scene_path_fixed --step ${step} --random 0 --prefix ${folder} --max ${nb_elements} --extension ".png"
+      
   fi
 done