【OSG学习笔记】Day 45: osg::Camera::DrawCallback （抓取图片）

osg::Camera::DrawCallback 抓取图片在OpenSceneGraphOSG三维渲染开发中相机Camera是场景渲染的核心入口控制着三维场景到二维屏幕的投影、绘制流程等关键逻辑。而osg::Camera::DrawCallback作为OSG提供的相机绘制阶段回调机制是实现帧缓存读取、屏幕截图、渲染后处理、自定义绘制等高级功能的核心技术。本文将结合OSG屏幕截图实战代码深度解析osg::Camera::DrawCallback的继承关系、工作原理、使用方法帮你彻底掌握这一OSG核心组件。基础认知1. 核心定义osg::Camera::DrawCallback是OSG中专门用于监听相机渲染流程的抽象回调类它允许开发者在相机渲染的前置阶段PreDraw和后置阶段PostDraw插入自定义逻辑无需修改OSG底层渲染源码。简单来说它是相机渲染的「钩子函数」相机渲染到指定阶段时会自动执行我们重写的回调逻辑。2. 核心应用场景结合本文截图代码该回调最常用的场景包括屏幕截图/帧缓存读取PostDraw阶段渲染完成后读取像素渲染后处理滤镜、灰度、模糊等效果自定义绘制叠加UI、标记、辅助图形渲染状态监控/性能统计类的继承关系核心图谱osg::Camera::DrawCallback并非孤立存在它遵循OSG的引用计数继承架构完整继承关系如下osg::Referenced OSG所有智能指针管理类的基类 ↓ 继承 osg::Camera::DrawCallback 抽象回调基类 ↓ 继承 自定义回调类如本文的 CaptureDrawCallback关键说明osg::ReferencedOSG所有需要自动内存管理的类的顶级基类提供引用计数功能配合osg::ref_ptr智能指针使用避免内存泄漏。✅ 本文中CaptureDrawCallback无需手动释放由OSG自动管理生命周期。osg::Camera::DrawCallback纯虚抽象类核心接口virtualvoidoperator()(constosg::Cameracamera)const0;这是必须重写的纯虚函数自定义逻辑全部写在该函数内。自定义子类CaptureDrawCallback继承DrawCallback重写operator()实现具体功能本文中为读取帧缓存、截图数据采集。相机渲染流程与回调执行时机OSG相机的渲染分为三个核心阶段回调的执行时机直接决定功能是否生效回调类型执行时机适用场景本文使用情况PreDrawCallback相机开始渲染之前渲染前修改状态、清空缓存不适用DrawCallback相机渲染过程中中间层自定义绘制不适用PostDrawCallback相机渲染完成后读取像素、截图、后处理✅ 核心使用关键结论屏幕截图必须使用 PostDrawCallback因为只有渲染完成后GPU帧缓存才有完整的场景像素数据此时调用readPixels才能拿到有效画面否则会出现黑屏、残缺等问题。结合截图代码DrawCallback 实战解析全部代码#includeosgViewer/Viewer#includeosgViewer/GraphicsWindow#includeosg/Node#includeosg/Geode#includeosg/Group#includeosg/Camera#includeosg/Image#includeosg/BufferObject#includeosgDB/ReadFile#includeosgDB/WriteFile#includeosgUtil/Optimizer#includeiostream// 定义全局变量用于在回调和事件处理器之间传递图像数据osg::ref_ptrosg::Imageimage_cnew osg::Image();/*********************************************************************** * 相机绘制回调在绘制完成后读取帧缓存 * 继承自 osg::Camera::DrawCallback在相机绘制流程中被调用 ***********************************************************************/structCaptureDrawCallback:public osg::Camera::DrawCallback{public:// 构造函数接收外部传入的osg::Image对象CaptureDrawCallback(osg::ref_ptrosg::Imageimage){_imageimage;}~CaptureDrawCallback(){}// 重载operator()绘制回调的核心执行函数virtualvoidoperator()(constosg::Cameracamera)const{// 1. 获取窗口系统接口跨平台抽象层适配不同操作系统的窗口APIosg::ref_ptrosg::GraphicsContext::WindowingSystemInterfacewsiosg::GraphicsContext::getWindowingSystemInterface();unsignedintwidth,height;// 2. 获取主屏幕分辨率也可以根据窗口句柄获取窗口大小wsi-getScreenResolution(osg::GraphicsContext::ScreenIdentifier(0),width,height);// 3. 为Image分配内存宽、高、像素格式、数据类型_image-allocateImage(width,height,1,GL_RGB,GL_UNSIGNED_BYTE);// 4. 读取帧缓存数据从(0,0)开始读取整个窗口的RGB像素数据_image-readPixels(0,0,width,height,GL_RGB,GL_UNSIGNED_BYTE);}// 成员变量存储要写入数据的Image对象osg::ref_ptrosg::Image_image;};/*********************************************************************** * 事件处理器处理键盘输入触发截图操作 * 继承自 osgGA::GUIEventHandler用于捕获用户交互事件 ***********************************************************************/class ImageHandler:public osgGA::GUIEventHandler{public:ImageHandler(){}~ImageHandler(){}// 重载handle()函数事件处理的核心逻辑virtual boolhandle(constosgGA::GUIEventAdapterea,osgGA::GUIActionAdapteraa){// 1. 将ActionAdapter转换为Viewer指针获取Viewer实例osg::ref_ptrosgViewer::Viewerviewerdynamic_castosgViewer::Viewer*(aa);if(viewernullptr)returnfalse;// 静态变量记录截图序号实现连续截图ScreenShot000.bmp, 001.bmp...staticint_screenCaptureSequence0;// 2. 判断事件类型键盘按下事件switch(ea.getEventType()){caseosgGA::GUIEventAdapter::KEYDOWN:{// 按下c或C键时触发截图if(ea.getKey()c||ea.getKey()C){charfilename[128];// 格式化文件名生成带序号的BMP文件sprintf(filename,ScreenShot%04d.bmp,_screenCaptureSequence);// 写入文件使用osgDB工具类将Image数据保存为BMP格式osgDB::writeImageFile(*(image_c.get()),filename);}break;}default:// 其他事件不处理返回false表示事件未被消费returnfalse;}// 返回true表示事件已被处理returntrue;}};/*********************************************************************** * 主函数程序入口 ***********************************************************************/intmain(){// 1. 创建Viewer实例OSG渲染主循环控制器osg::ref_ptrosgViewer::Viewerviewernew osgViewer::Viewer();// 2. 创建场景根节点osg::ref_ptrosg::Grouprootnew osg::Group();// 3. 读取外部模型示例为cow.osg可替换为其他OSG支持的模型格式osg::ref_ptrosg::NodenodeosgDB::readNodeFile(cow.osg);if(nodenullptr){std::cerrError: 无法读取模型文件 cow.osg请确保文件在程序目录下std::endl;return-1;}root-addChild(node.get());// 4. 场景优化使用OSG内置优化器提升渲染性能osgUtil::Optimizer optimizer;optimizer.optimize(root.get());// 5. 设置相机绘制回调在绘制完成后读取帧缓存viewer-getCamera()-setPostDrawCallback(newCaptureDrawCallback(image_c.get()));// 6. 设置场景数据将根节点传入Viewerviewer-setSceneData(root.get());// 7. 添加事件处理器注册截图按键事件viewer-addEventHandler(newImageHandler());// 8. 初始化窗口并启动渲染主循环viewer-realize();viewer-run();return0;}1. 自定义回调类定义继承与重写// 1. 继承 osg::Camera::DrawCallbackstructCaptureDrawCallback:publicosg::Camera::DrawCallback{public:// 构造函数接收图像对象用于存储截图数据CaptureDrawCallback(osg::ref_ptrosg::Imageimage){_imageimage;}// 2. 重写纯虚函数 operator() → 核心逻辑入口virtualvoidoperator()(constosg::Cameracamera)const{// 获取屏幕/窗口尺寸unsignedintwidth,height;autowsiosg::GraphicsContext::getWindowingSystemInterface();wsi-getScreenResolution(osg::GraphicsContext::ScreenIdentifier(0),width,height);// 分配图像内存_image-allocateImage(width,height,1,GL_RGB,GL_UNSIGNED_BYTE);// 3. 核心功能读取GPU帧缓存像素数据截图核心_image-readPixels(0,0,width,height,GL_RGB,GL_UNSIGNED_BYTE);}// 成员变量存储像素数据osg::ref_ptrosg::Image_image;};代码关键点严格继承osg::Camera::DrawCallback遵循OSG继承规范重写operator()纯虚函数这是回调的唯一入口函数内实现帧缓存读取将GPU数据同步到CPU的osg::Image中。2. 回调注册到相机在main函数中将自定义回调绑定到相机的PostDraw阶段// 注册后置绘制回调 → 渲染完成后自动执行viewer-getCamera()-setPostDrawCallback(newCaptureDrawCallback(image_c.get()));这一步是回调生效的关键告诉OSG相机渲染完成后执行CaptureDrawCallback的逻辑。3. 配合事件处理器完成截图DrawCallback负责采集数据键盘事件处理器负责保存数据// 按下c键将Image数据写入文件osgDB::writeImageFile(*(image_c.get()),filename);使用核心规范必须继承重写自定义类必须公有继承osg::Camera::DrawCallback且必须实现operator()纯虚函数。const 约束不可忽略operator()函数是const修饰的函数内不能修改类的非mutable成员变量。执行时机选对截图/后处理 →setPostDrawCallback渲染前初始化 →setPreDrawCallback内存自动管理基于osg::Referenced回调对象用new创建即可无需手动delete。总结osg::Camera::DrawCallback是OSG相机渲染的核心扩展接口它通过继承抽象类回调钩子的设计让开发者可以灵活介入相机渲染流程。结合本文截图代码我们可以总结其核心价值继承关系osg::Referenced→osg::Camera::DrawCallback→ 自定义回调核心作用在相机渲染的指定阶段执行自定义逻辑实战用法PostDraw回调读取帧缓存 事件处理器保存图像实现稳定截图设计优势无侵入式扩展不修改OSG源码满足个性化渲染需求。