OSG加载模型时显示读取进度

目录

1. 前言

2. 开发环境说明

3. 功能实现

3.1. 方法1

3.2. 方法2

3.3. 方法3

4. 附加说明


1. 前言

       OSG中加载模型文件到视景器,一般通过osgDB::readXXXX系列开头的函数来加载模型,如:osgDB::readNodeFile、osgDB::readImageFile、osgDB::readObjectFile等系列函数。如果想深入探究read开头的这些函数内部的实现机制,请参考:osgDB::readNodeFile等函数源码剖析 博文。如果能深入理解read开头的这些函数,理解本博文就很容易了。

      OSG中加载模型文件到视景器的典型代码片段类似如下:

int main()
{
    auto pCowNode = osgDB::readNodeFile("cow.osg");
    if(nullptr == pCowNode)
    {
        OSG_WARN << "cow node is nullptr!" << std::endl; // 注意加换行符,否则控制台不会有输出
        return 1;
    }

    sg::ref_ptr<osgViewer::Viewer> pViewer = new osgViewer::Viewer;
    pViewer ->setSceneData(pCowNode );
    osgDB::readNodeFile

    return viewer->run();
}

上面的代码,不会显示进度,当模型文件很大时,只能干等着加载完,期间,也不知道到底加载多少了,即当前时刻到底加载了百分之几呢?如何实现告知用户加载的百分比进度?

2. 开发环境说明

        本次用到的开发环境如下:

  • OpenSceneGraph 3.6.2。
  • Visual studio 2022 64位社区版。
  • Windows 11 家庭中文版。

3. 功能实现

       本博文以读取osg文件来讲解,通过三种方法来说明如何实现读取文件进度。

3.1. 方法1

       本方法直接通过操作读取指定后缀名的文件读写器类实现读取进度。OSG读取模型文件在内部流程是按下述步骤进行:

  1. 获取该文件的后缀名。
  2. osg内部维护了一个后缀名和读写器类对象构成的map。根据步骤1中的后缀名从map中找到读写器类对象。
  3. 利用步骤2中的读写器类对象调用read开头的函数读取模型。

用于读取后缀名为osg的文件的读写器类位于osg源码的如下子目录:

src/osgPlugins/osg/ReaderWriterOSG.cpp

 用于读取后缀名为osg的文件的读写器类以插件形式位于osg解决方案下的如下工程:

该工程下ReaderWriterOSG.cpp文件的OSGReaderWriter就是用来读写后缀名为osg的读写器类,如下:

          OSGReaderWriter最终是通过调用如下的readNode函数实现读取osg文件的:

        virtual ReadResult readNode(std::istream& fin, const Options* options) const
        {
            loadWrappers();

            fin.imbue(std::locale::classic());

            Input fr;
            fr.attach(&fin);
            fr.setOptions(options);
            ...... // 其它代码略
       }

这样,就可以直接构造一个OSGReaderWriter类对象,通过传入流对象读取osg文件,同时将该流对象传入一个线程,在线程中实时获取当前流的文件指针的位置,从而获取到读取进度,代码实现如下:

#include <osgViewer/Viewer>
#include <osgDB/ReaderWriter>
#include <osgDB/ReadFile>
#include <osgDB/FileNameUtils>
#include <osgDB/FileUtils>
//#include <osgDB/Registry>
#include <fstream>
#include<thread>


/* 此处加入读取指定后缀名文件的读写器文件或头文件。本例读取的是osg文件,对应的读写器类为OSGReaderWriter
   最好是换成你本机的相对路径
*/
#include"E:/osg/OpenSceneGraph-OpenSceneGraph-3.6.2/src/osgPlugins/osg/ReaderWriterOSG.cpp"

// 线程函数,在该函数中不停地检查文件指针的位置
void readFileProgressFun(std::ifstream& im)
{
    im.seekg(0, std::ifstream::end);
    auto fileSize = im.tellg(); // 获取模型文件的总长度

    im.seekg(0, std::ifstream::beg); // 定位到模型文件的开头

    auto bEnd = false;
    while (!bEnd)
    {
        auto curPos = im.tellg(); // 获取文件指针当前位置
        if (-1 == curPos) // 到达文件尾部了,即文件全部读取完
        {
            /* 因为每读取一次文件,文件指针就前进一段距离(如:blocksize),除了最后一次读取的外,其它
            *  每次文件指针前进的距离应该是一样的。但最后一次读取的字节数可能不够blocksize,此时im.tellg()就会返回-1,表示文件指针已经到尾部了,此时就将当前位置设为文件大小
            */
            curPos = fileSize; 
            bEnd = true;
        }

        std::cout << curPos * 100.0 / fileSize <<"%" << std::endl;
    }
}

int main()
{
    
    std::string fileName = osgDB::findDataFile("cow.osg", nullptr);
    if (!osgDB::fileExists(fileName))
    {
        OSG_WARN << "cow.osg is not exist!" << std::endl; // 加换行符,否则控制台不会显示打印信息,下同
        return osgDB::ReaderWriter::ReadResult::FILE_NOT_FOUND;
    }

    //osgDB::ifstream istream(R"(cow.osg)", std::ios::in | std::ios::binary);
    osgDB::ifstream istream(fileName.c_str(), std::ios::in | std::ios::binary);
    auto bIsGood = istream.good();
    if (!bIsGood) // 检查读取流是否打开成功 
    {
        OSG_WARN << "istream is failed" << std::endl; 
        return 1;
    }

    OSGReaderWriter rw; // 定义一个用于读取osg的文件读写器类对象

    // 开启一个线程,在此线程中,显示读取进度
    std::thread readFileProgressThread(readFileProgressFun, std::ref(istream));
    readFileProgressThread.detach();
    auto pCowNode = rw.readNode(istream, nullptr).takeNode();
    if (nullptr == pCowNode)
    {
        OSG_WARN << "cow node is null" << std::endl;  
        return 1;
    }

    osg::ref_ptr<osgViewer::Viewer> pViewer = new osgViewer::Viewer;
    pViewer->setSceneData(pCowNode);
 
    return pViewer->run();
}

说明:

       上面代码的思路是:直接构造一个读取osg文件的读写器类对象(第60行),然后通过文件名构造一个输入文件流对象,然后将该流对象作为第1个参数传入读写器类对象的readNode函数,且将该流对象传入线程函数,当线程执行时,在一个循环中检查流的文件指针位置,并打印出读取百分比,当文件指针到达文件尾部时,线程循环跳出,文件读取完。

      获取osg文件的路径时,请按照第44行那样调用osgDB::findDataFile函数获取,该函数先从osgDB::Options查找指定的文件,并返回文件在本机文件系统中的绝对路径;如果找不到,再从当前工作目录查找,并返回文件的绝对路径;如果找不到,再从本机环境变量OSG_FILE_PATH中设置的目录查找,并返回文件在本机文件系统中的绝对路径,如果上面的都找不到,则返回空字符串。如果你想像第51行那样,则请写入绝对路径或相对路径,保证能找到该文件。

      上述代码运行结果如下:

3.2. 方法2

       方法1是直接构造一个读取指定后缀名文件的读写器类对象,然后再读取文件的,这要求开发人员对osg的源码很熟。有时我们并不知道指定后缀名的读写器类是哪个类,且每次文件的类型即后缀名变了,如:方法1是读取后缀名为.osg的文件,下次碰到后缀名为.ive的文件时,得改方法1的第60行代码。如果能够根据传入的模型文件的后缀名,让程序为我们自动选择读写器类,那就省事多了。接下来是方法2实现的步骤:

1. 先写个读取文件的callback,让readNodeFile的时候调一下我们的callback.

class ReadFileCB : public osgDB::Registry::ReadFileCallback

然后安装这个回调,它是这样被安装调用的:

osgDB::Registry::instance()->setReadFileCallback(new ReadFileCB);

2. 在callback中手动获取读写的插件,然后使用如下代码:

osgDB::ifstream istream(fileName.c_str(), std::ios::in | std::ios::binary);

读取文件,使用如下代码:

rw->readNode(istream);

来读取结点。

3. 起一个线程,实时的获取istream在整个文件中的位置,来求百分比。下面是线程类,先获取了总长度_length,然后在run里看看当前位置,当读到最后的时候线程退出。

整个代码如下: 

#include <osgViewer/Viewer>
#include <osgDB/ReaderWriter>
#include <osgDB/ReadFile>
#include <osgDB/FileNameUtils>
#include <osgDB/FileUtils>
#include <osgDB/Registry>
#include <fstream>


class CRenderingThread : public OpenThreads::Thread
{
public:
    CRenderingThread(osgDB::ifstream* fin) :_fin(fin)
    {
        fin->seekg(0, std::ifstream::end);
        _length = fin->tellg();
        fin->seekg(0, std::ifstream::beg);

    };
    virtual ~CRenderingThread() {};

    virtual void run()
    {
        int pos = _fin->tellg();
        while (pos < _length)
        {
            pos = _fin->tellg();
            if(-1 == pos)
            {
               pos = _length;
            }
            std::cout << 100.0*pos / _length<<"%" << std::endl;
        }
    };

protected:
    osgDB::ifstream* _fin;
    int _length;
};


class ReadFileCB : public osgDB::Registry::ReadFileCallback
{
public:

    virtual osgDB::ReaderWriter::ReadResult readNode(const std::string& file, const osgDB::ReaderWriter::Options* opt)
    {
        //第一步是获取OSG、IVE ReaderWriter
        std::string ext = osgDB::getLowerCaseFileExtension(file);
        osgDB::ReaderWriter* rw = osgDB::Registry::instance()->getReaderWriterForExtension(ext);

        if (!rw)
        {
            return osgDB::ReaderWriter::ReadResult::FILE_NOT_HANDLED;
        }

        std::string fileName = osgDB::findDataFile(file, opt);
        if (fileName.empty()) return osgDB::ReaderWriter::ReadResult::FILE_NOT_FOUND;

        //osgDB::ifstream fin(fileName.c_str());
        osgDB::ifstream istream(fileName.c_str(), std::ios::in | std::ios::binary);
        CRenderingThread crt(&istream);
        crt.startThread();
        auto bIsGood = istream.good();
        if (bIsGood)
        {
            std::cout << "Using default format readerwriter" << std::endl;

            osgDB::ReaderWriter::ReadResult rr = rw->readNode(istream);
            while (crt.isRunning()) {}
            return rr;
        }
        else
        {
            return osgDB::ReaderWriter::ReadResult::ERROR_IN_READING_FILE;
        }
    }
};

int main()
{
    osg::ref_ptr<osgViewer::Viewer> viewer = new osgViewer::Viewer;
    osgDB::Registry::instance()->setReadFileCallback(new ReadFileCB);
    viewer->setSceneData(osgDB::readNodeFile("ceep.ive"));
    return viewer->run();
}

注意:

  • 方法1和方法2开启线程采取的不同方法,前者采用std::thread类,后者采用开源线程库  OpenThreads::Thread。
  • 如果读取的是后缀为.osg的文件,则第52行返回的rw就是OSGReaderWriter类对象。
  • 只要实现了某种后缀名文件的读写器类,并将其以插件的形式挂载到osg内核了,本方法就可可以通过第50行代码获取到该读写器类对象。

3.3. 方法3

      下面的代码使用文件回调ReadFileCallback实现了文件读取进度的控制。其基本思想为:重构标准模板库std::basic_filebuf的uflow函数,在其中记录当前读入的字节数;然后使用std::istream类加载文件流,并直接使用readNode函数流的方式来读取文件。但本方法不一定适用于osg支持的所有文件格式,因为有的格式可能不支持数据流方式读取。

#include <osgViewer/Viewer>
#include <osgDB/ReaderWriter>
#include <osgDB/ReadFile>
#include <osgDB/FileNameUtils>
#include <osgDB/FileUtils>
#include <osgDB/Registry>
#include <iostream>
 
template<typename Elem, typename Tr = std::char_traits<Elem>>
class ProgressingStreamBuf : public std::basic_filebuf<Elem, Tr>
{
public:
    typedef std::basic_filebuf<Elem, Tr> BaseType;
    ProgressingStreamBuf(const std::string& filename)
        :BaseType(), _count(0), _readSize(0)
    {
        if (open(filename.c_str(), std::ios_base::in | std::ios_base::binary))
        {
            pubseekoff(0, std::ios_base::beg, std::ios_base::in);
 
        }
 
    }
protected:
 
    virtual int_type uflow()
    {
        int_type value = BaseType::uflow();
        _count++;
        _readSize += egptr() - gptr();
 
        if (0 == (_count % 10))
        {
            std::cout << _readSize;
        }
        else
        {
            std::cout << ".";
        }
 
        return value;
     }
 
    int _count;
    int _readSize;
};
 
 
class  ProgressingReadCallback : public osgDB::Registry::ReadFileCallback
{
public:
    typedef ProgressingStreamBuf<char> ProgressingStringBuf;
    typedef osgDB::ReaderWriter::ReadResult ReadResult;
    ProgressingReadCallback() {}
 
    virtual osgDB::ReaderWriter::ReadResult readNode(const std::string& file, const osgDB::Options* option)
    {
        std::string ext = osgDB::getLowerCaseFileExtension(file);
        osgDB::ReaderWriter::ReadResult rr;
        osgDB::ReaderWriter* rw = osgDB::Registry::instance()->getReaderWriterForExtension(ext);
 
        if (rw)
        {
            std::string fileName = osgDB::findDataFile(file, option);
            if (fileName.empty()) return ReadResult::ReadResult::FILE_NOT_FOUND;
 
            ProgressingStringBuf* ps = new ProgressingStringBuf(fileName);
            if (!ps->is_open())
            {
                return ReadResult::ERROR_IN_READING_FILE;
            }
 
            std::istream s(ps);
            rr = rw->readNode(s, option);
            delete ps;
        }
        else
        {
            rr = osgDB::Registry::instance()->readNodeImplementation(file, option);
        }
 
        return rr;
    }
};
 
int main(int argc, char**argv)
{
    osg::ArgumentParser arguments(&argc, argv);
    osgDB::Registry::instance()->setReadFileCallback(new ProgressingReadCallback);
    
    osg::Node* model = osgDB::readNodeFiles(arguments);
    if (!model) model = osgDB::readNodeFile("cow.osg");
 
    osgViewer::Viewer viewer;
    viewer.setSceneData(model);
    return viewer.run();
}

运行结果如下:

关于std::basic_filebuf的用法,请参考:std::basic_filebuf 

4. 附加说明

       不论是哪种方法,指定后缀名的文件读写器类插件应存在,否则对于第1种方法无法构造读写器类对象,即第1种方法的第60行没法构造出一个具体的读写器对象;对于第2种方法的第52行的if语句将会为true,从而导致返回

osgDB::ReaderWriter::ReadResult::FILE_NOT_HANDLED

;对于第3种方法,则会执行第79行的else语句。

    现有osg读写插件,如果不支持特定的文件类型,就得自己写个读写器的插件,并将该插件挂载到osg内核上。关于怎么实现自定义的读写器插件,请参考:osg实现自定义插件读取自定义格式的模型文件到场景

       方法2转载自第19节 实例-显示模型读取进度

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值