genericmap.cpp 20.8 KB
Newer Older
Pierre Kraemer's avatar
Pierre Kraemer committed
1 2 3
/*******************************************************************************
* CGoGN: Combinatorial and Geometric modeling with Generic N-dimensional Maps  *
* version 0.1                                                                  *
4
* Copyright (C) 2009-2011, IGG Team, LSIIT, University of Strasbourg           *
Pierre Kraemer's avatar
Pierre Kraemer committed
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
*                                                                              *
* This library is free software; you can redistribute it and/or modify it      *
* under the terms of the GNU Lesser General Public License as published by the *
* Free Software Foundation; either version 2.1 of the License, or (at your     *
* option) any later version.                                                   *
*                                                                              *
* This library is distributed in the hope that it will be useful, but WITHOUT  *
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or        *
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License  *
* for more details.                                                            *
*                                                                              *
* You should have received a copy of the GNU Lesser General Public License     *
* along with this library; if not, write to the Free Software Foundation,      *
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.           *
*                                                                              *
20
* Web site: http://cgogn.u-strasbg.fr/                                         *
Pierre Kraemer's avatar
Pierre Kraemer committed
21 22 23 24 25
* Contact information: cgogn@unistra.fr                                        *
*                                                                              *
*******************************************************************************/

#include "Topology/generic/genericmap.h"
26
#include "Topology/generic/attributeHandler.h"
Pierre Kraemer's avatar
Pierre Kraemer committed
27
#include "Topology/generic/dartmarker.h"
Sylvain Thery's avatar
Sylvain Thery committed
28 29
#include "Topology/generic/traversorCell.h"

Pierre Kraemer's avatar
Pierre Kraemer committed
30 31
#include "Geometry/vector_gen.h"
#include "Geometry/matrix.h"
32
#include "Container/registered.h"
Pierre Kraemer's avatar
Pierre Kraemer committed
33 34 35

namespace CGoGN
{
36 37

std::map<std::string, RegisteredBaseAttribute*>* GenericMap::m_attributes_registry_map = NULL ;
38

39
GenericMap::GenericMap() : m_nbThreads(1)
Pierre Kraemer's avatar
Pierre Kraemer committed
40
{
41
	if(m_attributes_registry_map == NULL)
42
		m_attributes_registry_map = new std::map<std::string, RegisteredBaseAttribute*> ;
43

Pierre Kraemer's avatar
Pierre Kraemer committed
44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
	// register all known types
	registerAttribute<Dart>("Dart");
	registerAttribute<Mark>("Mark");

	registerAttribute<long>("long");
	registerAttribute<int>("int");
	registerAttribute<short>("short");
	registerAttribute<char>("char");

	registerAttribute<unsigned long>("unsigned long");
	registerAttribute<unsigned int>("unsigned int");
	registerAttribute<unsigned short>("unsigned short");
	registerAttribute<unsigned char>("unsigned char");

	registerAttribute<Geom::Vec2f>(Geom::Vec2f::CGoGNnameOfType());
	registerAttribute<Geom::Vec3f>(Geom::Vec3f::CGoGNnameOfType());
	registerAttribute<Geom::Vec4f>(Geom::Vec4f::CGoGNnameOfType());

	registerAttribute<Geom::Vec2d>(Geom::Vec2d::CGoGNnameOfType());
	registerAttribute<Geom::Vec3d>(Geom::Vec3d::CGoGNnameOfType());
	registerAttribute<Geom::Vec4d>(Geom::Vec4d::CGoGNnameOfType());

	registerAttribute<Geom::Matrix33f>(Geom::Matrix33f::CGoGNnameOfType());
	registerAttribute<Geom::Matrix44f>(Geom::Matrix44f::CGoGNnameOfType());

	registerAttribute<Geom::Matrix33d>(Geom::Matrix33d::CGoGNnameOfType());
	registerAttribute<Geom::Matrix44d>(Geom::Matrix44d::CGoGNnameOfType());

72
	for(unsigned int i = 0; i < NB_ORBITS; ++i)
Pierre Kraemer's avatar
Pierre Kraemer committed
73
	{
74
		m_attribs[i].setOrbit(i) ;
75
		m_attribs[i].setRegistry(m_attributes_registry_map) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
76
		m_embeddings[i] = NULL ;
77 78 79
		for(unsigned int j = 0; j < NB_THREAD; ++j)
		{
			m_marksets[i][j].clear() ;
Pierre Kraemer's avatar
Pierre Kraemer committed
80
			m_markTables[i][j] = NULL ;
81
		}
Pierre Kraemer's avatar
Pierre Kraemer committed
82
	}
83 84 85

	// get & lock marker for boundary
	m_boundaryMarker =  m_marksets[DART][0].getNewMark();
Pierre Kraemer's avatar
Pierre Kraemer committed
86 87 88 89
}

GenericMap::~GenericMap()
{
90 91 92
	// release marker for boundary
	m_marksets[DART][0].releaseMark(m_boundaryMarker);

Pierre Kraemer's avatar
Pierre Kraemer committed
93 94 95 96 97
	for(unsigned int i = 0; i < NB_ORBITS; ++i)
	{
		if(isOrbitEmbedded(i))
			m_attribs[i].clear(true) ;
	}
98 99 100 101 102

	for(std::multimap<AttributeMultiVectorGen*, AttributeHandlerGen*>::iterator it = attributeHandlers.begin(); it != attributeHandlers.end(); ++it)
		(*it).second->setInvalid() ;
	attributeHandlers.clear() ;

103
	if(m_attributes_registry_map)
104
	{
105
		delete m_attributes_registry_map;
106 107
		m_attributes_registry_map = NULL;
	}
Pierre Kraemer's avatar
Pierre Kraemer committed
108 109
}

110 111
void GenericMap::clear(bool removeAttrib)
{
Sylvain Thery's avatar
Sylvain Thery committed
112
	if (removeAttrib)
113
	{
Sylvain Thery's avatar
Sylvain Thery committed
114
		for(unsigned int i = 0; i < NB_ORBITS; ++i)
115
		{
Sylvain Thery's avatar
Sylvain Thery committed
116
			m_attribs[i].clear(true) ;
117 118
			m_embeddings[i] = NULL ;

Sylvain Thery's avatar
Sylvain Thery committed
119 120 121 122 123 124
			for(unsigned int j = 0; j < NB_THREAD; ++j)
			{
				if (i != DART)		// keep Marker reservation for DartMarker
					m_marksets[i][j].clear() ;
				m_markTables[i][j] = NULL ;
			}
125
		}
126 127
		for(std::multimap<AttributeMultiVectorGen*, AttributeHandlerGen*>::iterator it = attributeHandlers.begin(); it != attributeHandlers.end(); ++it)
			(*it).second->setInvalid() ;
128
		attributeHandlers.clear() ;
129
	}
Sylvain Thery's avatar
Sylvain Thery committed
130 131 132 133 134 135 136
	else
	{
		for(unsigned int i = 0; i < NB_ORBITS; ++i)
		{
			m_attribs[i].clear(false) ;
		}
	}
137 138
}

Pierre Kraemer's avatar
Pierre Kraemer committed
139 140 141 142 143 144 145
/****************************************
 *         EMBEDDING MANAGEMENT         *
 ****************************************/

void GenericMap::setDartEmbedding(unsigned int orbit, Dart d, unsigned int emb)
{
	assert(isOrbitEmbedded(orbit) || !"Invalid parameter: orbit not embedded");
Pierre Kraemer's avatar
Pierre Kraemer committed
146
	unsigned int old = getEmbedding(orbit, d);
Pierre Kraemer's avatar
Pierre Kraemer committed
147 148 149 150 151 152

	// if same emb nothing to do
	if (old == emb)
		return;
	// if different then unref the old emb
	if (old != EMBNULL)
153 154
	{
		if(m_attribs[orbit].unrefLine(old))
155 156
		{
			for (unsigned int t = 0; t < m_nbThreads; ++t)
Pierre Kraemer's avatar
Pierre Kraemer committed
157
				m_markTables[orbit][t]->operator[](old).clear();
158
		}
159
	}
Pierre Kraemer's avatar
Pierre Kraemer committed
160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179
	// ref the new emb
	if (emb != EMBNULL)
		m_attribs[orbit].refLine(emb);
	// affect the embedding to the dart
	(*m_embeddings[orbit])[d.index] = emb ;
}

/****************************************
 *        ATTRIBUTES MANAGEMENT         *
 ****************************************/

void GenericMap::initOrbitEmbedding(unsigned int orbit, bool realloc)
{
	assert(isOrbitEmbedded(orbit) || !"Invalid parameter: orbit not embedded") ;
	DartMarker mark(*this) ;
	for(Dart d = begin(); d != end(); next(d))
	{
		if(!mark.isMarked(d))
		{
			mark.markOrbit(orbit, d) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
180
			if(realloc || getEmbedding(orbit, d) == EMBNULL)
Pierre Kraemer's avatar
Pierre Kraemer committed
181 182 183 184 185 186 187 188 189 190 191
				embedNewCell(orbit, d) ;
		}
	}
}

template <typename R>
bool GenericMap::registerAttribute(const std::string &nameType)
{
	RegisteredBaseAttribute* ra = new RegisteredAttribute<R>;
	if (ra == NULL)
	{
192
		CGoGNerr << "Erreur enregistrement attribut" << CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
193 194 195 196 197
		return false;
	}

	ra->setTypeName(nameType);

198
	m_attributes_registry_map->insert(std::pair<std::string, RegisteredBaseAttribute*>(nameType,ra));
Pierre Kraemer's avatar
Pierre Kraemer committed
199 200 201
	return true;
}

Sylvain Thery's avatar
Sylvain Thery committed
202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
/****************************************
 *   EMBEDDING ATTRIBUTES MANAGEMENT    *
 ****************************************/

void GenericMap::addEmbedding(unsigned int orbit)
{
	assert(!isOrbitEmbedded(orbit) || !"Invalid parameter: orbit already embedded") ;

	std::ostringstream oss;
	oss << "EMB_" << orbit;

	AttributeContainer& dartCont = m_attribs[DART] ;
	AttributeMultiVector<unsigned int>* amv = dartCont.addAttribute<unsigned int>(oss.str()) ;
	m_embeddings[orbit] = amv ;

	// set new embedding to EMBNULL for all the darts of the map
	for(unsigned int i = dartCont.begin(); i < dartCont.end(); dartCont.next(i))
		amv->operator[](i) = EMBNULL ;

	AttributeContainer& cellCont = m_attribs[orbit];
	for (unsigned int t = 0; t < m_nbThreads; ++t)
	{
		std::stringstream ss ;
		ss << "Mark_"<< t ;
		AttributeMultiVector<Mark>* amvMark = cellCont.addAttribute<Mark>(ss.str()) ;
		for(unsigned int i = cellCont.begin(); i < cellCont.end(); cellCont.next(i))
			amvMark->operator[](i).clear() ;
		m_markTables[orbit][t] = amvMark ;
	}
}


234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254
/****************************************
 *          THREAD MANAGEMENT           *
 ****************************************/

void GenericMap::addThreadMarker(unsigned int nb)
{
	unsigned int th ;

	for (unsigned int j = 0; j < nb; ++j)
	{
		th = m_nbThreads ;
		m_nbThreads++ ;

		for (unsigned int i = 0; i < NB_ORBITS; ++i)
		{
			if (isOrbitEmbedded(i))
			{
				std::stringstream ss ;
				ss << "Mark_"<< th ;
				AttributeContainer& cellCont = m_attribs[i] ;
				AttributeMultiVector<Mark>* amvMark = cellCont.addAttribute<Mark>(ss.str()) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
255
				m_markTables[i][th] = amvMark ;
256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279
			}
		}
	}
}

unsigned int GenericMap::getNbThreadMarkers()
{
	return m_nbThreads;
}

void GenericMap::removeThreadMarker(unsigned int nb)
{
	unsigned int th = 0;
	while ((m_nbThreads > 1) && (nb > 0))
	{
		th = --m_nbThreads ;
		--nb;
		for (unsigned int i = 0; i < NB_ORBITS; ++i)
		{
			if (isOrbitEmbedded(i))
			{
				std::stringstream ss ;
				ss << "Mark_"<< th ;
				AttributeContainer& cellCont = m_attribs[i] ;
280
				cellCont.removeAttribute<Mark>(ss.str()) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
281
				m_markTables[i][th] = NULL ;
282 283 284 285
			}
		}
	}
}
Pierre Kraemer's avatar
Pierre Kraemer committed
286 287 288 289 290 291 292 293

/****************************************
 *             SAVE & LOAD              *
 ****************************************/

void GenericMap::update_m_emb_afterLoad()
{
	// get container of dart orbit
294
	AttributeContainer& cont = m_attribs[DART] ;
Pierre Kraemer's avatar
Pierre Kraemer committed
295 296 297

	// get the list of attributes
	std::vector<std::string> listeNames;
Pierre Kraemer's avatar
Pierre Kraemer committed
298
	cont.getAttributesNames(listeNames);
Pierre Kraemer's avatar
Pierre Kraemer committed
299

Pierre Kraemer's avatar
Pierre Kraemer committed
300
	// check if there are EMB_X attributes
Pierre Kraemer's avatar
Pierre Kraemer committed
301 302 303 304 305 306 307
	for (unsigned int i = 0;  i < listeNames.size(); ++i)
	{
		std::string sub = listeNames[i].substr(0,listeNames[i].size()-1);
		if (sub == "EMB_")
		{
			unsigned int orb = listeNames[i][4]-'0'; // easy atoi computation for one char;

308 309
			AttributeMultiVector<unsigned int>* amv = cont.getDataVector<unsigned int>(i);
			m_embeddings[orb] = amv ;
310
			CGoGNout << "Ajoute m_emb[" << orb << "] : " << i << CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
		}
	}
}

bool GenericMap::saveMapXml(const std::string& filename, bool compress)
{
	xmlTextWriterPtr writer = xmlNewTextWriterFilename(filename.c_str(), 0);
	xmlTextWriterStartDocument(writer,NULL,NULL,NULL);

	// Entete
	int rc = xmlTextWriterStartElement(writer, BAD_CAST "CGoGN_Map");
	rc = xmlTextWriterWriteAttribute(writer,  BAD_CAST "type", BAD_CAST this->mapTypeName().c_str());
	rc = xmlTextWriterWriteFormatAttribute(writer,  BAD_CAST "nb_max_orbits","%u",NB_ORBITS);

	// save m_attribs
	for (unsigned int i=0; i<NB_ORBITS; ++i)
	{
		m_attribs[i].saveXml(writer,i);
	}

	// save m_orbMarker
	rc = xmlTextWriterStartElement(writer, BAD_CAST "Orbit_MarkerSet");
	for (unsigned int i=0; i<NB_ORBITS; ++i)
	{
		int rc = xmlTextWriterStartElement(writer, BAD_CAST "MarkerSet");
		rc = xmlTextWriterWriteFormatAttribute(writer,  BAD_CAST "orbit", "%u", i);
Pierre Kraemer's avatar
Pierre Kraemer committed
337 338 339
		rc = xmlTextWriterWriteFormatAttribute(writer,  BAD_CAST "val", "%u", m_marksets[i][0].getMarkVal());
//		rc = xmlTextWriterWriteAttribute(writer,  BAD_CAST "bin", BAD_CAST m_marksets[i][0].getMarkerAsBinaryString().c_str());
//		m_marksets[i] ;
Pierre Kraemer's avatar
Pierre Kraemer committed
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367
		rc = xmlTextWriterEndElement(writer);
	}
	rc = xmlTextWriterEndElement(writer);

	// save m_BoundaryMarkerVal
//	rc = xmlTextWriterStartElement(writer, BAD_CAST "Boundary_Marker");
//	rc = xmlTextWriterWriteFormatAttribute(writer,  BAD_CAST "val", "%u", m_BoundaryMarkerVal.getMarkVal());
//	rc = xmlTextWriterEndElement(writer);

	rc = xmlTextWriterEndElement(writer); // of map

	xmlTextWriterEndDocument(writer);
	xmlFreeTextWriter(writer);

	return true;
}


bool GenericMap::loadMapXml(const std::string& filename, bool compress)
{

	xmlDocPtr doc = xmlReadFile(filename.c_str(), NULL, 0);
	xmlNodePtr map_node = xmlDocGetRootElement(doc);

	// Check if it is a CGoGN_Map file
//	if (strcmp((char*)(map_node->name),(char*)"CGoGN_Map")!=0)
	if (!chechXmlNode(map_node,"CGoGN_Map"))
	{
368
		CGoGNerr << "Wrong xml format: Root node != CGoGN_Map"<< CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
369 370 371 372 373
		return false;
	}

	// check the file type
	xmlChar *prop = xmlGetProp(map_node, BAD_CAST "type");
374
	CGoGNout << "Loading "<< prop <<" xml file"<< CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
375 376 377 378 379 380

	// check the nb max orbits
	prop = xmlGetProp(map_node, BAD_CAST "nb_max_orbits");
	unsigned int nbo = atoi((char*)prop);
	if (nbo != NB_ORBITS)
	{
381
		CGoGNerr << "Wrong nb max orbits in xml map"<< CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
382 383 384 385 386 387 388 389 390 391 392 393
		return false;
	}

	/***********************************************
	*	 			 load attributs
	************************************************/
	for (xmlNode* cur_node = map_node->children; cur_node; cur_node = cur_node->next)
	{
		// for each attribute
//		if (strcmp((char*)(cur_node->name),(char*)"Attributes_Container")==0)
		if (chechXmlNode(cur_node, "Attributes_Container"))
		{
394
			CGoGNout << "LOAD ATTRIBUT"<< CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
395
			// get the orbit id
396
			unsigned int id = AttributeContainer::getIdXmlNode(cur_node);
Pierre Kraemer's avatar
Pierre Kraemer committed
397 398 399 400
			// and load container
			unsigned int nba = m_attribs[id].getNbAttributes();


401
			CGoGNout << "attribut "<<id<<" size="<< m_attribs[id].size()<< "  nbatt="<< nba<< CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
402 403 404 405 406 407 408 409 410

			m_attribs[id].loadXml(cur_node);
		}
	}

	/***********************************************
	*   creation of the m_embeddings pointers table
	************************************************/
//	// get attribute names of dart orbit
411
//	AttributeContainer& contDart = m_attribs[DART] ;
Pierre Kraemer's avatar
Pierre Kraemer committed
412 413 414 415 416 417
//	std::vector< std::string > tableNames;
//	contDart.getAttributesStrings(tableNames);
//
//	// find orbit frome name and store pointer in right place
//	for (unsigned int i = 0; i< tableNames.size(); ++i)
//	{
418
////		CGoGNout << i <<" : "<< tableNames[i]<<CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
419 420 421 422 423
//
//		std::string& name = tableNames[i];
//		std::string is_an_emb = name.substr(0,4);
//		if (is_an_emb == "EMB_")
//		{
424
//			AttributeMultiVector<unsigned int>& amv = contDart.getDataVector<unsigned int>(i) ;
Pierre Kraemer's avatar
Pierre Kraemer committed
425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454
//
//			std::string orbitname = name.substr(4, name.size()-4);
//			std::istringstream iss(orbitname);
//			unsigned int orbit;
//			iss >> orbit;
//
//			m_embeddings[orbit] = &amv ;
//		}
//	}
//
//
	update_m_emb_afterLoad();

	/***********************************************
	*	 load Orbit_MarkerSet & BoundaryMarker
	************************************************/
	xmlNode* cur_node = map_node->children;
	bool read1=false;
	bool read2=false;
	while (!(read1 || read2) && cur_node)// scan nodes to find the one with right name
	{
//		if (strcmp((char*)(cur_node->name),(char*)"Orbit_MarkerSet") == 0)
		if (chechXmlNode(cur_node, "Orbit_MarkerSet"))
		{
			for (xmlNode* mark_node = cur_node->children; mark_node; mark_node = mark_node->next)
			{
				xmlChar* prop = xmlGetProp(mark_node, BAD_CAST "orbit");
				unsigned int orb = atoi((char*)prop);
				prop = xmlGetProp(mark_node, BAD_CAST "val");
				unsigned int val = atoi((char*)prop);
Pierre Kraemer's avatar
Pierre Kraemer committed
455
				m_marksets[orb][0].setMarkVal(val);
Pierre Kraemer's avatar
Pierre Kraemer committed
456 457 458 459 460 461
			}
			read1 =true;
		}
		else
		{
//			if (strcmp((char*)(cur_node->name),(char*)"Boundary_Marker") == 0)
462
			CGoGNout << "Orbit_MarkerSet"<<CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
463 464 465 466 467 468 469 470 471 472 473 474 475 476
//			if (chechXmlNode(cur_node, "Boundary_Marker"))
//			{
//				xmlChar* prop = xmlGetProp(cur_node, BAD_CAST "val");
//				unsigned int val = atoi((char*)prop);
//				m_BoundaryMarkerVal.setMarkVal(val);
//				read2 =true;
//			}
		}
		// next node
		cur_node = cur_node->next;
	}

	if (!(read1 && read2))
	{
477
		CGoGNerr <<"Error reading Marker in xml node"<<CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
478 479 480 481 482 483 484 485 486 487 488 489 490 491 492
		return false;
	}

	xmlFreeDoc(doc);

	update_m_emb_afterLoad();

	return true ;
}

bool GenericMap::saveMapBin(const std::string& filename)
{
	CGoGNostream fs(filename.c_str(), std::ios::out|std::ios::binary);
	if (!fs)
	{
493
		CGoGNerr << "Unable to open file for writing: " << filename << CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522
		return false;
	}

	// Entete
	char* buff = new char[256];
	for (int i=0; i< 256; ++i)
		buff[i] = char(255);
	const char* cgogn = "CGoGN_Map";
	memcpy(buff,cgogn,10);
	std::string mt = mapTypeName();
	const char* mtc = mt.c_str();
	memcpy(buff+32,mtc,mt.size()+1);
	unsigned int *buffi = reinterpret_cast<unsigned int*>(buff+64);
	*buffi = NB_ORBITS;
	fs.write(reinterpret_cast<const char*>(buff), 256);

	delete buff;

	// save m_attribs
	for (unsigned int i=0; i<NB_ORBITS; ++i)
	{
		m_attribs[i].saveBin(fs,i);
	}


	std::vector<unsigned int> buffer;
	buffer.reserve(NB_ORBITS+1);
	for (unsigned int i=0; i<NB_ORBITS; ++i)
	{
Pierre Kraemer's avatar
Pierre Kraemer committed
523
		buffer.push_back(m_marksets[i][0].getMarkVal());
Pierre Kraemer's avatar
Pierre Kraemer committed
524 525 526 527 528 529 530 531 532 533 534 535 536
	}
//	buffer.push_back(m_BoundaryMarkerVal.getMarkVal());

	fs.write(reinterpret_cast<const char*>(&(buffer[0])), buffer.size()*sizeof(unsigned int));

	return true;
}

bool GenericMap::loadMapBin(const std::string& filename)
{
	CGoGNistream fs(filename.c_str(), std::ios::in|std::ios::binary);
	if (!fs)
	{
537
		CGoGNerr << "Unable to open file for loading" << CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
538 539 540 541 542 543 544 545 546 547 548
		return false;
	}

	// entete
	char* buff = new char[256];
	fs.read(reinterpret_cast<char*>(buff), 256);

	std::string buff_str(buff);
	// Check file type
	if (buff_str != "CGoGN_Map")
	{
549
		CGoGNerr<< "Wrong binary file format"<< CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
550 551 552 553 554
		return false;
	}

	// Check map type
	buff_str = std::string(buff+32);
555
	CGoGNout << "Map type file = "<< buff_str<< CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
556 557 558 559 560 561 562 563

	std::string localType = this->mapTypeName();
	localType = localType.substr(0,localType.size()-1);

	std::string fileType = buff_str.substr(0,buff_str.size()-1);

	if (fileType != localType)
	{
564
		CGoGNerr << "Not possible to load "<< fileType<< " into "<< localType << " object"<<CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
565 566 567 568 569 570 571 572
		return false;
	}

	// Check max nb orbit
	unsigned int *ptr_nbo = reinterpret_cast<unsigned int*>(buff+64);
	unsigned int nbo = *ptr_nbo;
	if (nbo != NB_ORBITS)
	{
573
		CGoGNerr << "Wrond max orbit number in file" << CGoGNendl;
Pierre Kraemer's avatar
Pierre Kraemer committed
574 575 576 577 578 579
		return  false;
	}

	// load attrib container
	for (unsigned int i=0; i<NB_ORBITS; ++i)
	{
580
		unsigned int id = AttributeContainer::loadBinId(fs);
Pierre Kraemer's avatar
Pierre Kraemer committed
581 582 583 584 585 586 587 588 589
		m_attribs[id].loadBin(fs);
	}

	// load MarkerSet & boundary
	std::vector<unsigned int> buffer;
	buffer.resize(NB_ORBITS);
	fs.read(reinterpret_cast<char*>(&(buffer[0])), (NB_ORBITS+1)*sizeof(unsigned int));
	for (unsigned int i=0; i<NB_ORBITS; ++i)
	{
Pierre Kraemer's avatar
Pierre Kraemer committed
590
		m_marksets[i][0].setMarkVal(buffer[i]);
Pierre Kraemer's avatar
Pierre Kraemer committed
591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613
	}
//	m_BoundaryMarkerVal.setMarkVal(buffer[NB_ORBITS]);

	update_m_emb_afterLoad();

	return true;
}

/****************************************
 *           DARTS TRAVERSALS           *
 ****************************************/

bool GenericMap::foreach_dart(FunctorType& f, const FunctorSelect& good)
{
	for (Dart d = begin(); d != end(); next(d))
	{
		if(good(d))
			if (f(d))
				return true;
	}
	return false;
}

Sylvain Thery's avatar
Sylvain Thery committed
614
bool GenericMap::foreach_dart_of_orbit(unsigned int orbit, Dart d, FunctorType& f, unsigned int thread)
Pierre Kraemer's avatar
Pierre Kraemer committed
615 616 617
{
	switch(orbit)
	{
618 619 620 621 622 623
		case DART: return f(d);
		case VERTEX: return foreach_dart_of_vertex(d, f, thread);
		case EDGE: return foreach_dart_of_edge(d, f, thread);
		case ORIENTED_FACE: return foreach_dart_of_oriented_face(d, f, thread);
		case FACE: return foreach_dart_of_face(d, f, thread);
		case VOLUME: return foreach_dart_of_volume(d, f, thread);
Sylvain Thery's avatar
Sylvain Thery committed
624
//		case -1: return foreach_dart_of_cc(d,f,thread);
Pierre Kraemer's avatar
Pierre Kraemer committed
625 626 627 628 629
		default: assert(!"Cells of this dimension are not handled");
	}
	return false;
}

Sylvain Thery's avatar
Sylvain Thery committed
630
bool GenericMap::foreach_orbit(unsigned int orbit, FunctorType& fonct, const FunctorSelect& good, unsigned int thread)
Pierre Kraemer's avatar
Pierre Kraemer committed
631
{
Pierre Kraemer's avatar
Pierre Kraemer committed
632
	TraversorCell<GenericMap> trav(*this, orbit, good, thread);
633
	bool found = false;
Pierre Kraemer's avatar
Pierre Kraemer committed
634

Sylvain Thery's avatar
Sylvain Thery committed
635
	for (Dart d = trav.begin(); !found && d != trav.end(); d = trav.next())
Pierre Kraemer's avatar
Pierre Kraemer committed
636
	{
637
		if ((fonct)(d))
Pierre Kraemer's avatar
Pierre Kraemer committed
638
			found = true;
Pierre Kraemer's avatar
Pierre Kraemer committed
639 640 641 642 643 644 645 646 647 648 649
	}
	return found;
}

unsigned int GenericMap::getNbOrbits(unsigned int orbit, const FunctorSelect& good)
{
	FunctorCount fcount;
	foreach_orbit(orbit, fcount, good);
	return fcount.getNb();
}

Sylvain Thery's avatar
Sylvain Thery committed
650 651 652
void GenericMap::viewAttributesTables()
{
	std::cout << "======================="<< std::endl;
653
	for (unsigned int i = 0; i < NB_ORBITS; ++i)
Sylvain Thery's avatar
Sylvain Thery committed
654
	{
655
		std::cout << "ATTRIBUTE_CONTAINER " << i << std::endl;
Sylvain Thery's avatar
Sylvain Thery committed
656 657 658 659 660
		AttributeContainer& cont = m_attribs[i] ;

		// get the list of attributes
		std::vector<std::string> listeNames;
		cont.getAttributesNames(listeNames);
661 662 663
		for (std::vector<std::string>::iterator it = listeNames.begin(); it != listeNames.end(); ++it)
			std::cout << "    " << *it << std::endl;
		std::cout << "-------------------------" << std::endl;
Sylvain Thery's avatar
Sylvain Thery committed
664
	}
665 666
	std::cout << "m_embeddings: " << std::hex;
	for (unsigned int i = 0; i < NB_ORBITS; ++i)
Sylvain Thery's avatar
Sylvain Thery committed
667
		std::cout << (long)(m_embeddings[i]) << " / ";
668
	std::cout << std::endl << "-------------------------" << std::endl;
Sylvain Thery's avatar
Sylvain Thery committed
669 670

	std::cout << "m_markTables: ";
671
	for (unsigned int i = 0; i < NB_ORBITS; ++i)
Sylvain Thery's avatar
Sylvain Thery committed
672
		std::cout << (long)(m_markTables[i][0]) << " / ";
673
	std::cout << std::endl << "-------------------------" << std::endl << std::dec;
Sylvain Thery's avatar
Sylvain Thery committed
674 675
}

676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709
void GenericMap::boundaryMark(Dart d)
{
	m_markTables[DART][0]->operator[](d.index).setMark(m_boundaryMarker);
}

void GenericMap::boundaryUnmark(Dart d)
{
	m_markTables[DART][0]->operator[](d.index).unsetMark(m_boundaryMarker);
}

bool GenericMap::isBoundaryMarked(Dart d)
{
	return m_markTables[DART][0]->operator[](d.index).testMark(m_boundaryMarker);
}

void GenericMap::boundaryMarkOrbit(unsigned int orbit, Dart d)
{
	FunctorMark<GenericMap> fm(*this, m_boundaryMarker, m_markTables[DART][0]) ;
	foreach_dart_of_orbit(orbit, d, fm, 0) ;
}

void GenericMap::boundaryUnmarkOrbit(unsigned int orbit, Dart d)
{
	FunctorUnmark<GenericMap> fm(*this, m_boundaryMarker, m_markTables[DART][0]) ;
	foreach_dart_of_orbit(orbit, d, fm, 0) ;
}

void GenericMap::boundaryUnmarkAll()
{
	AttributeContainer& cont = getAttributeContainer(DART) ;
	for (unsigned int i = cont.begin(); i != cont.end(); cont.next(i))
		m_markTables[DART][0]->operator[](i).unsetMark(m_boundaryMarker);
}

Pierre Kraemer's avatar
Pierre Kraemer committed
710
} // namespace CGoGN