如何在Common Lisp中将2D字节数组写入二进制文件？

Question

我想对于具有Common Lisp经验的人来说，这是一个简单的问题。对我来说不多，刚开始使用LISP的我。

正如您在下面的下一个片段中看到的，我创建了一个类型为UNSIGNED BYTE的800 x 600数组。

(defun test-binary-save ()
    (let*
        ((width 800)
         (height 600)
         (arr (make-array (list width height) 
                :element-type '(mod 256)
                :initial-element 0)))
        (utilities::save-array-as-pgm "test.pgm" arr)))

而且我的实用程序包中的函数应该以P5 PGM格式写磁盘。

(defun save-array-as-pgm (filename buffer)
    "Writes a byte array as a PGM file (P5) to a file."
    (with-open-file 
        (stream filename 
            :element-type '(unsigned-byte 8)
            :direction :output 
            :if-does-not-exist :create
            :if-exists :supersede)
        (let* 
            ((dimensions (array-dimensions buffer))
             (width (first dimensions))
             (height (second dimensions))
             (header (format nil "P5~A~D ~D~A255~A" 
                #\newline
                width height #\newline
                #\newline)))

            (loop 
                :for char :across header 
                :do (write-byte (char-code char) stream))
            ;(write-sequence buffer stream) <<-- DOES NOT WORK - is not of type SEQUENCE
        ))
    filename)

执行相同功能的等效（和正常工作）C函数看起来像这样。

static
int
save_pgm
( const char* filename
, size_t width
, size_t height
, const uint8_t* pixels
)
{
    if(NULL == filename)
        return 0;
    if(NULL == pixels)
        return 0;

    FILE *out = fopen(filename, "wb");
    if(NULL != out)
    {
        fprintf(out, "P5\n%zu %zu\n255\n", width, height);
        size_t nbytes = width * height;
        fwrite(pixels,1,nbytes,out);
        fclose(out);
        return 1;
    }

    return 0;
}

谁能告诉我如何修复我的save-array-as-pgm函数，最好是一次性编写数组，而不是使用循环和(write-byte (aref buffer y x) stream)？

在我决定在这里问这个问题之前，我在谷歌上搜索了很多，只发现了一些引用一些做奇特二进制文件的软件包的引用-但这是一个简单的案例，我正在寻找一个简单的解决方案。

Answer 1

Common Lisp支持displaced数组：

CL-USER 6 > (let ((array (make-array (list 3 4)
                                     :initial-element 1
                                     :element-type 'bit)))
              (make-array (reduce #'* (array-dimensions array))
                          :element-type 'bit
                          :displaced-to array))
#*111111111111

现在有一个问题，Lisp实现如何能够通过置换后的数组访问该数组。

Answer 2

如果您想在Common Lisp中进行严重的位推送，请使用一维数组。

Answer 3

看起来好像并没有那么困难……一旦我发现可以将2D数组“投射”到1D数组然后简单地使用write-sequence。

[要找到解决方案，我必须检查github上的sbcl源代码以掌握make-array的工作原理并找到-sbcl-特定功能array-storage-vector。

正如我猜想的那样，多维数组使用一维后备数组进行数据存储。

[save-array-as-pgm函数现在看起来像这样：

(defun save-array-as-pgm (filename buffer)
    "Writes a byte array as a PGM file (P5) to a file."
    (with-open-file 
        (stream filename 
            :element-type '(unsigned-byte 8)
            :direction :output 
            :if-does-not-exist :create
            :if-exists :supersede)
        (let* 
            ((dimensions (array-dimensions buffer))
             (width (first dimensions))
             (height (second dimensions))
             (header (format nil "P5~A~D ~D~A255~A" 
                #\newline
                width height #\newline
                #\newline)))

            (loop 
                :for char :across header 
                :do (write-byte (char-code char) stream))
            (write-sequence (sb-c::array-storage-vector buffer) stream)
        ))
    filename)