Exemple #1
0
void 
PassOptix::addScene (const std::string &sceneName)
{
	if (m_SceneVector.end() == std::find (m_SceneVector.begin(), m_SceneVector.end(), sceneName)) {
	
		m_SceneVector.push_back (sceneName);
	
		std::set<std::string> *materialNames = new std::set<std::string>;
		RENDERMANAGER->getScene(sceneName)->getMaterialNames(materialNames);
		
		std::set<std::string>::iterator iter;
		iter = materialNames->begin();
		for ( ; iter != materialNames->end(); ++iter) {
			
			if (m_MaterialMap.count((*iter)) == 0)
				m_MaterialMap[(*iter)] = MaterialID(DEFAULTMATERIALLIBNAME, (*iter));
		}
		delete materialNames;


		IScene *sc = RENDERMANAGER->getScene(sceneName);
		sc->compile();

/*		std::vector<SceneObject *> objs = sc->getAllObjects();
		for (unsigned int i = 0; i < objs.size(); ++i) {

			IRenderable &r = objs[i]->getRenderable();
			VertexData &v = r.getVertexData();
			// just to make sure we get this data compiled
			//r.getVertexData().compile();

			// which attrs should we send to optix?
			unsigned int id = v.getBufferID(0);
			// clear OpenGL Errors so far; otherwise Optix won't work
			int e = glGetError();

			optix::Buffer buffers[VertexData::MaxAttribs];
			try {
				for (unsigned int b = 0; b < VertexData::MaxAttribs; ++b) { 
					if (v.getBufferID(b)) {
						buffers[b] = o_Context->createBufferFromGLBO(RT_BUFFER_INPUT,v.getBufferID(b));
						buffers[b]->setFormat(RT_FORMAT_FLOAT4);
						buffers[b]->setSize(v.getDataOf(0).size());
					}
				}
			}
			catch ( optix::Exception& e ) {
				NAU_THROW("Optix Error: Adding scene %s to pass %s (creating buffers from VBOs) [%s]",
										sc->getName().c_str(), m_Name.c_str(),e.getErrorString().c_str()); 
			}

			std::vector<IMaterialGroup *> mg = r.getMaterialGroups();
			for (unsigned int g = 0; g < mg.size(); ++g) {

				if (mg[g]->getNumberOfPrimitives() > 0) {
					try {
						optix::Geometry geom = o_Context->createGeometry();
						geom->setPrimitiveCount(mg[g]->getNumberOfPrimitives());
						geom->setBoundingBoxProgram(o_BoundingBoxProgram);
						geom->setIntersectionProgram(o_GeometryIntersectionProgram);

						geom["vertex_buffer"]->setBuffer(buffers[0]);
						for (unsigned int b = 1; b < VertexData::MaxAttribs; ++b) {
							if (v.getBufferID(b))
								geom[VertexData::Syntax[b]]->setBuffer(buffers[b]);
						}
						id = mg[g]->getIndexData().getBufferID();
						optix::Buffer indices = o_Context->createBufferFromGLBO(RT_BUFFER_INPUT, id);
						indices->setFormat(RT_FORMAT_UNSIGNED_INT);
						indices->setSize(mg[g]->getIndexData().getIndexSize());

						geom["index_buffer"]->setBuffer(indices);
						o_Material = o_Context->createMaterial();
						o_Material->setClosestHitProgram(0, o_ClosestHitProgram);
						
						o_Material["diffuse"]->set4fv(MATERIALLIBMANAGER->getMaterial(this->m_MaterialMap[mg[g]->getMaterialName()])->getColor().getDiffuse());

						o_GeomInstances.push_back(o_Context->createGeometryInstance());
						o_GeomInstances[o_GeomInstances.size()-1]->setMaterialCount(1);
						o_GeomInstances[o_GeomInstances.size()-1]->setMaterial(0, o_Material);
						o_GeomInstances[o_GeomInstances.size()-1]->setGeometry(geom);
					}
					catch ( optix::Exception& e ) {
						NAU_THROW("Optix Error: Adding scene %s to pass %s (adding material groups) [%s]",
												sc->getName().c_str(), m_Name.c_str(),e.getErrorString().c_str()); 
					}
				}
			}
		}
		try {
			o_GeomGroup->setChildCount(o_GeomInstances.size());
			for (unsigned int i = 0; i < o_GeomInstances.size(); ++i) {
				o_GeomGroup->setChild(i,o_GeomInstances[i]);
			}

			o_GeomGroup->setAcceleration(o_Context->createAcceleration("Bvh","Bvh"));
		}
		catch ( optix::Exception& e ) {
			NAU_THROW("Optix Error: Adding scene %s to pass %s (adding instances to geometry group) [%s]",
												sc->getName().c_str(), m_Name.c_str(),e.getErrorString().c_str()); 
		}

		//o_Context["geometry"]->set(o_GeomGroup);






*/		

	}
}
Exemple #2
0
void
PassOptix::optixInit() {

	glGetError();

	// Can these be removed due to texture count?
	o_TexLib.addTexture(1);
	o_Context["tex0"]->set(o_TexLib.getTexture(1));

	std::set<std::string> *materialNames = new std::set<std::string>;
	
	std::vector<std::string>::iterator scenesIter;
	scenesIter = m_SceneVector.begin();
	for ( ; scenesIter != m_SceneVector.end(); ++scenesIter) {

		IScene *aScene = RENDERMANAGER->getScene (*scenesIter);
		aScene->compile();

		// Adding Materials to optix lib
		aScene->getMaterialNames(materialNames);
		std::set<std::string>::iterator matIter;
		matIter = materialNames->begin();
		for( ; matIter != materialNames->end(); ++matIter) {
		
			o_MatLib.addMaterial(m_MaterialMap[*matIter]);
		}

		materialNames->clear();


		std::vector<SceneObject *> objs = aScene->getAllObjects();
		std::vector<nau::scene::SceneObject*>::iterator objsIter;

		objsIter = objs.begin();
		for ( ; objsIter != objs.end(); ++objsIter) {
			o_GeomLib.addSceneObject(*objsIter, m_MaterialMap);
//			o_GeomLib.addSceneObject((*objsIter)->getId(), m_MaterialMap);
		}
	}
	o_GeomLib.buildGeometryGroup();
	o_Context["top_object"]->set(o_GeomLib.getGeometryGroup());

	o_MatLib.applyMissPrograms();


	std::map<std::string, std::string>::iterator iter;
	iter = o_InputBuffers.begin();
	for ( ; iter != o_InputBuffers.end(); ++iter) {
		try {
			unsigned int id = RESOURCEMANAGER->getTexture(iter->second)->getPropi(ITexture::ID);
			if (RESOURCEMANAGER->getTexture(iter->second)->getPrope(ITexture::DIMENSION) == GL_TEXTURE_2D) {
				optix::TextureSampler rtWorldSpaceTexture = o_Context->createTextureSamplerFromGLImage(id, RT_TARGET_GL_TEXTURE_2D);
				rtWorldSpaceTexture->setWrapMode(0, RT_WRAP_CLAMP_TO_EDGE);
				rtWorldSpaceTexture->setWrapMode(1, RT_WRAP_CLAMP_TO_EDGE);
				rtWorldSpaceTexture->setIndexingMode(RT_TEXTURE_INDEX_ARRAY_INDEX);
				rtWorldSpaceTexture->setReadMode(RT_TEXTURE_READ_ELEMENT_TYPE);
				rtWorldSpaceTexture->setMaxAnisotropy(1.0f);
				rtWorldSpaceTexture->setFilteringModes(RT_FILTER_NEAREST, RT_FILTER_NEAREST, RT_FILTER_NONE);
				o_Context[iter->first]->setTextureSampler(rtWorldSpaceTexture);
			}
		}
		catch(optix::Exception& e) {
			NAU_THROW("Optix Error: Input Buffer preparation in pass %s [%s]", m_Name.c_str(), e.getErrorString().c_str());
		}
	}

	std::map<std::string, databuffer>::iterator iter2;
	iter2 = o_OutputDataBuffer.begin();
	ITexture *texID;

	for ( ; iter2 != o_OutputDataBuffer.end() ; ++iter2) {

		texID = RESOURCEMANAGER->getTexture(iter2->second.texName);
//		int format = texID->getPrope(ITexture::FORMAT);
		int tex = texID->getPropi(ITexture::ID);

		unsigned int pbo;
		glGenBuffers(1, &pbo);
		glBindBuffer(GL_PIXEL_UNPACK_BUFFER, pbo);
		// need to allow different types
		glBufferData(GL_PIXEL_UNPACK_BUFFER, texID->getPropi(ITexture::WIDTH)*texID->getPropi(ITexture::HEIGHT)*texID->getPropi(ITexture::ELEMENT_SIZE), 0, GL_STREAM_READ);
		glBindBuffer(GL_PIXEL_UNPACK_BUFFER, 0);

		optix::Buffer ob = o_Context->createBufferFromGLBO(RT_BUFFER_OUTPUT, pbo);
		ob->setSize(texID->getPropi(ITexture::WIDTH), texID->getPropi(ITexture::HEIGHT));
		// same here (types)
		ob->setFormat(getOptixFormat(texID));

		o_Context[iter2->first]->setBuffer(ob);

		o_OutputDataBuffer[iter2->first].pbo = pbo;
		//
	}

#if (TEST == 2)
	try {
		unsigned int id = RESOURCEMANAGER->getTexture("Deferred Render Targets::pos")->getId();
		optix::ITextureSampler rtWorldSpaceTexture = o_Context->createTextureSamplerFromGLImage(id, RT_TARGET_GL_TEXTURE_2D);
		rtWorldSpaceTexture->setWrapMode(0, RT_WRAP_CLAMP_TO_EDGE);
		rtWorldSpaceTexture->setWrapMode(1, RT_WRAP_CLAMP_TO_EDGE);
		rtWorldSpaceTexture->setIndexingMode(RT_TEXTURE_INDEX_ARRAY_INDEX);
		rtWorldSpaceTexture->setReadMode(RT_TEXTURE_READ_ELEMENT_TYPE);
		rtWorldSpaceTexture->setMaxAnisotropy(1.0f);
		rtWorldSpaceTexture->setFilteringModes(RT_FILTER_NEAREST, RT_FILTER_NEAREST, RT_FILTER_NONE);
		o_Context["pos_buffer"]->setTextureSampler(rtWorldSpaceTexture);
	}
	catch(optix::Exception& e) {
		NAU_THROW("Optix Error: Input Buffer preparation in pass %s [%s]", m_Name.c_str(), e.getErrorString().c_str());
	}
#endif
	o_OptixIsPrepared = true;
}