java读取revit文件缩略图

package test;

import org.apache.poi.poifs.filesystem.POIFSDocument;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.apache.poi.poifs.storage.BlockWritable;

import java.io.ByteArrayOutputStream;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Iterator;

/**
 * Created with IntelliJ IDEA.
 * User: wzhoufeng [e-mail]
 * Date: 20-6-23
 * Time: 上午8:14
 * To change this template use File | Settings | File Templates.
 */
public class RvtFileHeaderImage {

    public static void main(String[] args){
        try {
            POIFSFileSystem poifsFileSystem = new POIFSFileSystem(new FileInputStream("E:\\11.rvt"));
            Iterator<Object> iterator = poifsFileSystem.getViewableIterator();
            while (iterator.hasNext()){
                Object o = iterator.next();
                if(o instanceof org.apache.poi.poifs.filesystem.DocumentNode){
                    org.apache.poi.poifs.filesystem.DocumentNode documentNode = (org.apache.poi.poifs.filesystem.DocumentNode)o;

                    if(documentNode.getName().equals("RevitPreview4.0")){

                        Iterator<Object> objectIterator = documentNode.getViewableIterator();
                        POIFSDocument document =    null;
                        while (objectIterator.hasNext()){
                            Object ob = objectIterator.next();
                            if(ob instanceof POIFSDocument){
                                document = (POIFSDocument)ob;
                            }
                        }

                        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
                        //document.writeBlocks(byteArrayOutputStream);
                        BlockWritable[] blockWritables = document.getSmallBlocks();
                        for (BlockWritable writable : blockWritables){
                            writable.writeBlocks(byteArrayOutputStream);
                        }
                        byte[] arr = byteArrayOutputStream.toByteArray();
                        byteArrayOutputStream.close();

                        int start = getFileHeader(arr);

                        FileOutputStream fileOutputStream = new FileOutputStream("E:\\out.png");
                        byte[] newArr = new byte[arr.length-start];
                        for(int is=start;is<arr.length;is++){
                            newArr[is-start] = arr[is];
                        }
                        fileOutputStream.write(newArr);
                        fileOutputStream.close();
                        break;
                    }
                }

            }

        } catch (IOException e) {
            e.printStackTrace();
        }


    }

    static int getFileHeader(byte[] rawData){
        for (int i = 0; i < rawData.length; i++)
        {

            byte bt = rawData[i];
            if(bt==0x50){                                  //P
                //89504E47
                if(rawData[i+1]==0x4E && rawData[i+2]==0x47){  //NG
                    return i - 1;
                }
            }

//            String x  = Character.toString((char)bt);
//            if(x.equals("P")){
//                System.out.println(x);
//
//                if(Character.toString((char)rawData[i+1]).equals("N") && Character.toString((char)rawData[i+2]).equals("G")){
//                    return i - 1;
//                }
//            }
        }
        return 0;
    }

}

 使用java语言获取revit文件中的缩略图，代码依赖POI jar