Passing client files to webassembly from the front-end

前端 未结 2 959
轮回少年
轮回少年 2021-02-14 23:39

I\'m looking to pass user-submitted data to a c++ function which I\'ve compiled to wasm. The data is a file which the user submits on the front end via an input tag, like so:

2条回答
  •  南笙
    南笙 (楼主)
    2021-02-15 00:10

    This is a partial answer. It's superior to what I originally did and I feel like it might be closer to what the creators intended. However, I'm still creating more than one copy of the file. Credit to this post for making it click for me.

    This is now my handleFile callback, commented with things I learned.

    handleFile(e){
    
        const file = e.currentTarget.files[0];
        if(!(file instanceof Blob)) return;
        const reader = new FileReader();
        reader.onloadend = evt => {
    
            //evt.target.result is an ArrayBuffer. In js, 
            //you can't do anything with an ArrayBuffer 
            //so we have to ???cast??? it to an Uint8Array
            const uint8_t_arr = new Uint8Array(evt.target.result);
    
            //Right now, we have the file as a unit8array in javascript memory. 
            //As far as I understand, wasm can't directly access javascript memory. 
            //Which is why we need to allocate special wasm memory and then
            //copy the file from javascript memory into wasm memory so our wasm functions 
            //can work on it.
    
            //First we need to allocate the wasm memory. 
            //_malloc returns the address of the new wasm memory as int32.
            //This call is probably similar to 
            //uint8_t * ptr = new uint8_t[sizeof(uint8_t_arr)/sizeof(uint8_t_arr[0])]
            const uint8_t_ptr = window.Module._malloc(uint8_t_arr.length);
    
            //Now that we have a block of memory we can copy the file data into that block
            //This is probably similar to 
            //std::memcpy(uint8_t_ptr, uint8_t_arr, sizeof(uint8_t_arr)/sizeof(uint8_t_arr[0]))
            window.Module.HEAPU8.set(uint8_t_arr, uint8_t_ptr);
    
            //The only thing that's now left to do is pass 
            //the address of the wasm memory we just allocated
            //to our function as well as the size of our memory.
            window.Module.readFile(uint8_t_ptr, uint8_t_arr.length);
    
            //At this point we're forced to wait until wasm is done with the memory. 
            //Your site will now freeze if the memory you're working on is big. 
            //Maybe we can somehow let our wasm function run on a seperate thread and pass a callback?
    
            //Retreiving our (modified) memory is also straight forward. 
            //First we get some javascript memory and then we copy the 
            //relevant chunk of the wasm memory into our javascript object.
            const returnArr = new Uint8Array(uint8_t_arr.length);
            //If returnArr is std::vector, then is probably similar to 
            //returnArr.assign(ptr, ptr + dataSize)
            returnArr.set(window.Module.HEAPU8.subarray(uint8_t_ptr, uint8_t_ptr + uint8_t_arr.length));
    
            //Lastly, according to the docs, we should call ._free here.
            //Do we need to call the gc somehow?
            window.Module._free(uint8_t_ptr);
    
        }
        reader.readAsArrayBuffer(file);
    }
    

    Here is readFile.cpp.

    #include 
    
    //We get out pointer as a plain int from javascript
    void readFile(const int & addr, const size_t & len){
      //We use a reinterpret_cast to turn our plain int into a uint8_t pointer. After
      //which we can play with the data just like we would normally.
      uint8_t * data = reinterpret_cast(addr);
      for(size_t i = 0; i < len; ++i){
        data[i] += 1;
      }
    }
    
    //Using this command to compile
    //  emcc --bind -O3 readFile.cpp -s WASM=1 -s TOTAL_MEMORY=268435456 -o api.js --std=c++11
    //Note that you need to make sure that there's enough memory available to begin with.
    //I got only 16mb without passing the TOTAL_MEMORY setting.
    EMSCRIPTEN_BINDINGS(my_module) {
      emscripten::function("readFile", &readFile);
    }
    

提交回复
热议问题