| | 1232 | |
| | 1233 | /// \ingroup lemon_io |
| | 1234 | /// |
| | 1235 | /// \brief LGF reader for undirected graphs |
| | 1236 | /// |
| | 1237 | /// This utility reads an \ref lgf-format "LGF" file. |
| | 1238 | template <typename _Graph> |
| | 1239 | class GraphReader { |
| | 1240 | public: |
| | 1241 | |
| | 1242 | typedef _Graph Graph; |
| | 1243 | TEMPLATE_GRAPH_TYPEDEFS(Graph); |
| | 1244 | |
| | 1245 | private: |
| | 1246 | |
| | 1247 | |
| | 1248 | std::istream* _is; |
| | 1249 | bool local_is; |
| | 1250 | |
| | 1251 | Graph& _graph; |
| | 1252 | |
| | 1253 | std::string _nodes_caption; |
| | 1254 | std::string _edges_caption; |
| | 1255 | std::string _attributes_caption; |
| | 1256 | |
| | 1257 | typedef std::map<std::string, Node> NodeIndex; |
| | 1258 | NodeIndex _node_index; |
| | 1259 | typedef std::map<std::string, Edge> EdgeIndex; |
| | 1260 | EdgeIndex _edge_index; |
| | 1261 | |
| | 1262 | typedef std::vector<std::pair<std::string, |
| | 1263 | _reader_bits::MapStorageBase<Node>*> > NodeMaps; |
| | 1264 | NodeMaps _node_maps; |
| | 1265 | |
| | 1266 | typedef std::vector<std::pair<std::string, |
| | 1267 | _reader_bits::MapStorageBase<Edge>*> > EdgeMaps; |
| | 1268 | EdgeMaps _edge_maps; |
| | 1269 | |
| | 1270 | typedef std::multimap<std::string, _reader_bits::ValueStorageBase*> |
| | 1271 | Attributes; |
| | 1272 | Attributes _attributes; |
| | 1273 | |
| | 1274 | typedef std::map<std::string, _reader_bits::Section*> Sections; |
| | 1275 | Sections _sections; |
| | 1276 | |
| | 1277 | bool _use_nodes; |
| | 1278 | bool _use_edges; |
| | 1279 | |
| | 1280 | int line_num; |
| | 1281 | std::istringstream line; |
| | 1282 | |
| | 1283 | public: |
| | 1284 | |
| | 1285 | /// \brief Constructor |
| | 1286 | /// |
| | 1287 | /// Construct a undirected graph reader, which reads from the given |
| | 1288 | /// input stream. |
| | 1289 | GraphReader(std::istream& is, Graph& graph) |
| | 1290 | : _is(&is), local_is(false), _graph(graph), |
| | 1291 | _use_nodes(false), _use_edges(false) {} |
| | 1292 | |
| | 1293 | /// \brief Constructor |
| | 1294 | /// |
| | 1295 | /// Construct a undirected graph reader, which reads from the given |
| | 1296 | /// file. |
| | 1297 | GraphReader(const std::string& fn, Graph& graph) |
| | 1298 | : _is(new std::ifstream(fn.c_str())), local_is(true), _graph(graph), |
| | 1299 | _use_nodes(false), _use_edges(false) {} |
| | 1300 | |
| | 1301 | /// \brief Constructor |
| | 1302 | /// |
| | 1303 | /// Construct a undirected graph reader, which reads from the given |
| | 1304 | /// file. |
| | 1305 | GraphReader(const char* fn, Graph& graph) |
| | 1306 | : _is(new std::ifstream(fn)), local_is(true), _graph(graph), |
| | 1307 | _use_nodes(false), _use_edges(false) {} |
| | 1308 | |
| | 1309 | /// \brief Copy constructor |
| | 1310 | /// |
| | 1311 | /// The copy constructor transfers all data from the other reader, |
| | 1312 | /// therefore the copied reader will not be usable more. |
| | 1313 | GraphReader(GraphReader& other) |
| | 1314 | : _is(other._is), local_is(other.local_is), _graph(other._graph), |
| | 1315 | _use_nodes(other._use_nodes), _use_edges(other._use_edges) { |
| | 1316 | |
| | 1317 | other._is = 0; |
| | 1318 | other.local_is = false; |
| | 1319 | |
| | 1320 | _node_index.swap(other._node_index); |
| | 1321 | _edge_index.swap(other._edge_index); |
| | 1322 | |
| | 1323 | _node_maps.swap(other._node_maps); |
| | 1324 | _edge_maps.swap(other._edge_maps); |
| | 1325 | _attributes.swap(other._attributes); |
| | 1326 | |
| | 1327 | _nodes_caption = other._nodes_caption; |
| | 1328 | _edges_caption = other._edges_caption; |
| | 1329 | _attributes_caption = other._attributes_caption; |
| | 1330 | |
| | 1331 | _sections.swap(other._sections); |
| | 1332 | } |
| | 1333 | |
| | 1334 | /// \brief Destructor |
| | 1335 | ~GraphReader() { |
| | 1336 | for (typename NodeMaps::iterator it = _node_maps.begin(); |
| | 1337 | it != _node_maps.end(); ++it) { |
| | 1338 | delete it->second; |
| | 1339 | } |
| | 1340 | |
| | 1341 | for (typename EdgeMaps::iterator it = _edge_maps.begin(); |
| | 1342 | it != _edge_maps.end(); ++it) { |
| | 1343 | delete it->second; |
| | 1344 | } |
| | 1345 | |
| | 1346 | for (typename Attributes::iterator it = _attributes.begin(); |
| | 1347 | it != _attributes.end(); ++it) { |
| | 1348 | delete it->second; |
| | 1349 | } |
| | 1350 | |
| | 1351 | for (typename Sections::iterator it = _sections.begin(); |
| | 1352 | it != _sections.end(); ++it) { |
| | 1353 | delete it->second; |
| | 1354 | } |
| | 1355 | |
| | 1356 | if (local_is) { |
| | 1357 | delete _is; |
| | 1358 | } |
| | 1359 | |
| | 1360 | } |
| | 1361 | |
| | 1362 | private: |
| | 1363 | |
| | 1364 | GraphReader& operator=(const GraphReader&); |
| | 1365 | |
| | 1366 | public: |
| | 1367 | |
| | 1368 | /// \name Reading rules |
| | 1369 | /// @{ |
| | 1370 | |
| | 1371 | /// \brief Node map reading rule |
| | 1372 | /// |
| | 1373 | /// Add a node map reading rule to the reader. |
| | 1374 | template <typename Map> |
| | 1375 | GraphReader& nodeMap(const std::string& caption, Map& map) { |
| | 1376 | checkConcept<concepts::WriteMap<Node, typename Map::Value>, Map>(); |
| | 1377 | _reader_bits::MapStorageBase<Node>* storage = |
| | 1378 | new _reader_bits::MapStorage<Node, Map>(map); |
| | 1379 | _node_maps.push_back(std::make_pair(caption, storage)); |
| | 1380 | return *this; |
| | 1381 | } |
| | 1382 | |
| | 1383 | /// \brief Node map reading rule |
| | 1384 | /// |
| | 1385 | /// Add a node map reading rule with specialized converter to the |
| | 1386 | /// reader. |
| | 1387 | template <typename Map, typename Converter> |
| | 1388 | GraphReader& nodeMap(const std::string& caption, Map& map, |
| | 1389 | const Converter& converter = Converter()) { |
| | 1390 | checkConcept<concepts::WriteMap<Node, typename Map::Value>, Map>(); |
| | 1391 | _reader_bits::MapStorageBase<Node>* storage = |
| | 1392 | new _reader_bits::MapStorage<Node, Map, Converter>(map, converter); |
| | 1393 | _node_maps.push_back(std::make_pair(caption, storage)); |
| | 1394 | return *this; |
| | 1395 | } |
| | 1396 | |
| | 1397 | /// \brief Edge map reading rule |
| | 1398 | /// |
| | 1399 | /// Add an edge map reading rule to the reader. |
| | 1400 | template <typename Map> |
| | 1401 | GraphReader& edgeMap(const std::string& caption, Map& map) { |
| | 1402 | checkConcept<concepts::WriteMap<Edge, typename Map::Value>, Map>(); |
| | 1403 | _reader_bits::MapStorageBase<Edge>* storage = |
| | 1404 | new _reader_bits::MapStorage<Edge, Map>(map); |
| | 1405 | _edge_maps.push_back(std::make_pair(caption, storage)); |
| | 1406 | return *this; |
| | 1407 | } |
| | 1408 | |
| | 1409 | /// \brief Edge map reading rule |
| | 1410 | /// |
| | 1411 | /// Add an edge map reading rule with specialized converter to the |
| | 1412 | /// reader. |
| | 1413 | template <typename Map, typename Converter> |
| | 1414 | GraphReader& edgeMap(const std::string& caption, Map& map, |
| | 1415 | const Converter& converter = Converter()) { |
| | 1416 | checkConcept<concepts::WriteMap<Edge, typename Map::Value>, Map>(); |
| | 1417 | _reader_bits::MapStorageBase<Edge>* storage = |
| | 1418 | new _reader_bits::MapStorage<Edge, Map, Converter>(map, converter); |
| | 1419 | _edge_maps.push_back(std::make_pair(caption, storage)); |
| | 1420 | return *this; |
| | 1421 | } |
| | 1422 | |
| | 1423 | /// \brief Arc map reading rule |
| | 1424 | /// |
| | 1425 | /// Add an arc map reading rule to the reader. |
| | 1426 | template <typename Map> |
| | 1427 | GraphReader& arcMap(const std::string& caption, Map& map) { |
| | 1428 | checkConcept<concepts::WriteMap<Arc, typename Map::Value>, Map>(); |
| | 1429 | _reader_bits::MapStorageBase<Edge>* forward_storage = |
| | 1430 | new _reader_bits::GraphArcMapStorage<Graph, true, Map>(_graph, map); |
| | 1431 | _edge_maps.push_back(std::make_pair('+' + caption, forward_storage)); |
| | 1432 | _reader_bits::MapStorageBase<Edge>* backward_storage = |
| | 1433 | new _reader_bits::GraphArcMapStorage<Graph, false, Map>(_graph, map); |
| | 1434 | _edge_maps.push_back(std::make_pair('-' + caption, backward_storage)); |
| | 1435 | return *this; |
| | 1436 | } |
| | 1437 | |
| | 1438 | /// \brief Arc map reading rule |
| | 1439 | /// |
| | 1440 | /// Add an arc map reading rule with specialized converter to the |
| | 1441 | /// reader. |
| | 1442 | template <typename Map, typename Converter> |
| | 1443 | GraphReader& arcMap(const std::string& caption, Map& map, |
| | 1444 | const Converter& converter = Converter()) { |
| | 1445 | checkConcept<concepts::WriteMap<Arc, typename Map::Value>, Map>(); |
| | 1446 | _reader_bits::MapStorageBase<Edge>* forward_storage = |
| | 1447 | new _reader_bits::GraphArcMapStorage<Graph, true, Map, Converter> |
| | 1448 | (_graph, map, converter); |
| | 1449 | _edge_maps.push_back(std::make_pair('+' + caption, forward_storage)); |
| | 1450 | _reader_bits::MapStorageBase<Edge>* backward_storage = |
| | 1451 | new _reader_bits::GraphArcMapStorage<Graph, false, Map, Converter> |
| | 1452 | (_graph, map, converter); |
| | 1453 | _edge_maps.push_back(std::make_pair('-' + caption, backward_storage)); |
| | 1454 | return *this; |
| | 1455 | } |
| | 1456 | |
| | 1457 | /// \brief Attribute reading rule |
| | 1458 | /// |
| | 1459 | /// Add an attribute reading rule to the reader. |
| | 1460 | template <typename Value> |
| | 1461 | GraphReader& attribute(const std::string& caption, Value& value) { |
| | 1462 | _reader_bits::ValueStorageBase* storage = |
| | 1463 | new _reader_bits::ValueStorage<Value>(value); |
| | 1464 | _attributes.insert(std::make_pair(caption, storage)); |
| | 1465 | return *this; |
| | 1466 | } |
| | 1467 | |
| | 1468 | /// \brief Attribute reading rule |
| | 1469 | /// |
| | 1470 | /// Add an attribute reading rule with specialized converter to the |
| | 1471 | /// reader. |
| | 1472 | template <typename Value, typename Converter> |
| | 1473 | GraphReader& attribute(const std::string& caption, Value& value, |
| | 1474 | const Converter& converter = Converter()) { |
| | 1475 | _reader_bits::ValueStorageBase* storage = |
| | 1476 | new _reader_bits::ValueStorage<Value, Converter>(value, converter); |
| | 1477 | _attributes.insert(std::make_pair(caption, storage)); |
| | 1478 | return *this; |
| | 1479 | } |
| | 1480 | |
| | 1481 | /// \brief Node reading rule |
| | 1482 | /// |
| | 1483 | /// Add a node reading rule to reader. |
| | 1484 | GraphReader& node(const std::string& caption, Node& node) { |
| | 1485 | typedef _reader_bits::MapLookUpConverter<Node> Converter; |
| | 1486 | Converter converter(_node_index); |
| | 1487 | _reader_bits::ValueStorageBase* storage = |
| | 1488 | new _reader_bits::ValueStorage<Node, Converter>(node, converter); |
| | 1489 | _attributes.insert(std::make_pair(caption, storage)); |
| | 1490 | return *this; |
| | 1491 | } |
| | 1492 | |
| | 1493 | /// \brief Edge reading rule |
| | 1494 | /// |
| | 1495 | /// Add an edge reading rule to reader. |
| | 1496 | GraphReader& edge(const std::string& caption, Edge& edge) { |
| | 1497 | typedef _reader_bits::MapLookUpConverter<Edge> Converter; |
| | 1498 | Converter converter(_edge_index); |
| | 1499 | _reader_bits::ValueStorageBase* storage = |
| | 1500 | new _reader_bits::ValueStorage<Edge, Converter>(edge, converter); |
| | 1501 | _attributes.insert(std::make_pair(caption, storage)); |
| | 1502 | return *this; |
| | 1503 | } |
| | 1504 | |
| | 1505 | /// \brief Arc reading rule |
| | 1506 | /// |
| | 1507 | /// Add an arc reading rule to reader. |
| | 1508 | GraphReader& arc(const std::string& caption, Arc& arc) { |
| | 1509 | typedef _reader_bits::GraphArcLookUpConverter<Graph> Converter; |
| | 1510 | Converter converter(_graph, _edge_index); |
| | 1511 | _reader_bits::ValueStorageBase* storage = |
| | 1512 | new _reader_bits::ValueStorage<Arc, Converter>(arc, converter); |
| | 1513 | _attributes.insert(std::make_pair(caption, storage)); |
| | 1514 | return *this; |
| | 1515 | } |
| | 1516 | |
| | 1517 | /// @} |
| | 1518 | |
| | 1519 | /// \name Select section by name |
| | 1520 | /// @{ |
| | 1521 | |
| | 1522 | /// \brief Set \c \@nodes section to be read |
| | 1523 | /// |
| | 1524 | /// Set \c \@nodes section to be read |
| | 1525 | GraphReader& nodes(const std::string& caption) { |
| | 1526 | _nodes_caption = caption; |
| | 1527 | return *this; |
| | 1528 | } |
| | 1529 | |
| | 1530 | /// \brief Set \c \@edges section to be read |
| | 1531 | /// |
| | 1532 | /// Set \c \@edges section to be read |
| | 1533 | GraphReader& edges(const std::string& caption) { |
| | 1534 | _edges_caption = caption; |
| | 1535 | return *this; |
| | 1536 | } |
| | 1537 | |
| | 1538 | /// \brief Set \c \@attributes section to be read |
| | 1539 | /// |
| | 1540 | /// Set \c \@attributes section to be read |
| | 1541 | GraphReader& attributes(const std::string& caption) { |
| | 1542 | _attributes_caption = caption; |
| | 1543 | return *this; |
| | 1544 | } |
| | 1545 | |
| | 1546 | /// @} |
| | 1547 | |
| | 1548 | /// \name Section readers |
| | 1549 | /// @{ |
| | 1550 | |
| | 1551 | /// \brief Add a section processor with line oriented reading |
| | 1552 | /// |
| | 1553 | /// In the \e LGF file extra sections can be placed, which contain |
| | 1554 | /// any data in arbitrary format. These sections can be read with |
| | 1555 | /// this function line by line. The first parameter is the type |
| | 1556 | /// descriptor of the section, the second is a functor, which |
| | 1557 | /// takes just one \c std::string parameter. At the reading |
| | 1558 | /// process, each line of the section will be given to the functor |
| | 1559 | /// object. However, the empty lines and the comment lines are |
| | 1560 | /// filtered out, and the leading whitespaces are stipped from |
| | 1561 | /// each processed string. |
| | 1562 | /// |
| | 1563 | /// For example let's see a section, which contain several |
| | 1564 | /// integers, which should be inserted into a vector. |
| | 1565 | ///\code |
| | 1566 | /// @numbers |
| | 1567 | /// 12 45 23 |
| | 1568 | /// 4 |
| | 1569 | /// 23 6 |
| | 1570 | ///\endcode |
| | 1571 | /// |
| | 1572 | /// The functor is implemented as an struct: |
| | 1573 | ///\code |
| | 1574 | /// struct NumberSection { |
| | 1575 | /// std::vector<int>& _data; |
| | 1576 | /// NumberSection(std::vector<int>& data) : _data(data) {} |
| | 1577 | /// void operator()(const std::string& line) { |
| | 1578 | /// std::istringstream ls(line); |
| | 1579 | /// int value; |
| | 1580 | /// while (ls >> value) _data.push_back(value); |
| | 1581 | /// } |
| | 1582 | /// }; |
| | 1583 | /// |
| | 1584 | /// // ... |
| | 1585 | /// |
| | 1586 | /// reader.sectionLines("numbers", NumberSection(vec)); |
| | 1587 | ///\endcode |
| | 1588 | template <typename Functor> |
| | 1589 | GraphReader& sectionLines(const std::string& type, Functor functor) { |
| | 1590 | LEMON_ASSERT(!type.empty(), "Type is not empty."); |
| | 1591 | LEMON_ASSERT(_sections.find(type) == _sections.end(), |
| | 1592 | "Multiple reading of section."); |
| | 1593 | LEMON_ASSERT(type != "nodes" && type != "arcs" && type != "edges" && |
| | 1594 | type != "attributes", "Multiple reading of section."); |
| | 1595 | _sections.insert(std::make_pair(type, |
| | 1596 | new _reader_bits::LineSection<Functor>(functor))); |
| | 1597 | return *this; |
| | 1598 | } |
| | 1599 | |
| | 1600 | |
| | 1601 | /// \brief Add a section processor with stream oriented reading |
| | 1602 | /// |
| | 1603 | /// In the \e LGF file extra sections can be placed, which contain |
| | 1604 | /// any data in arbitrary format. These sections can be read |
| | 1605 | /// directly with this function. The first parameter is the type |
| | 1606 | /// of the section, the second is a functor, which takes an \c |
| | 1607 | /// std::istream& and an int& parameter, the latter regard to the |
| | 1608 | /// line number of stream. The functor can read the input while |
| | 1609 | /// the section go on, and the line number should be modified |
| | 1610 | /// accordingly. |
| | 1611 | template <typename Functor> |
| | 1612 | GraphReader& sectionStream(const std::string& type, Functor functor) { |
| | 1613 | LEMON_ASSERT(!type.empty(), "Type is not empty."); |
| | 1614 | LEMON_ASSERT(_sections.find(type) == _sections.end(), |
| | 1615 | "Multiple reading of section."); |
| | 1616 | LEMON_ASSERT(type != "nodes" && type != "arcs" && type != "edges" && |
| | 1617 | type != "attributes", "Multiple reading of section."); |
| | 1618 | _sections.insert(std::make_pair(type, |
| | 1619 | new _reader_bits::StreamSection<Functor>(functor))); |
| | 1620 | return *this; |
| | 1621 | } |
| | 1622 | |
| | 1623 | /// @} |
| | 1624 | |
| | 1625 | /// \name Using previously constructed node or edge set |
| | 1626 | /// @{ |
| | 1627 | |
| | 1628 | /// \brief Use previously constructed node set |
| | 1629 | /// |
| | 1630 | /// Use previously constructed node set, and specify the node |
| | 1631 | /// label map. |
| | 1632 | template <typename Map> |
| | 1633 | GraphReader& useNodes(const Map& map) { |
| | 1634 | checkConcept<concepts::ReadMap<Node, typename Map::Value>, Map>(); |
| | 1635 | LEMON_ASSERT(!_use_nodes, "Multiple usage of useNodes() member"); |
| | 1636 | _use_nodes = true; |
| | 1637 | _writer_bits::DefaultConverter<typename Map::Value> converter; |
| | 1638 | for (NodeIt n(_graph); n != INVALID; ++n) { |
| | 1639 | _node_index.insert(std::make_pair(converter(map[n]), n)); |
| | 1640 | } |
| | 1641 | return *this; |
| | 1642 | } |
| | 1643 | |
| | 1644 | /// \brief Use previously constructed node set |
| | 1645 | /// |
| | 1646 | /// Use previously constructed node set, and specify the node |
| | 1647 | /// label map and a functor which converts the label map values to |
| | 1648 | /// std::string. |
| | 1649 | template <typename Map, typename Converter> |
| | 1650 | GraphReader& useNodes(const Map& map, |
| | 1651 | const Converter& converter = Converter()) { |
| | 1652 | checkConcept<concepts::ReadMap<Node, typename Map::Value>, Map>(); |
| | 1653 | LEMON_ASSERT(!_use_nodes, "Multiple usage of useNodes() member"); |
| | 1654 | _use_nodes = true; |
| | 1655 | for (NodeIt n(_graph); n != INVALID; ++n) { |
| | 1656 | _node_index.insert(std::make_pair(converter(map[n]), n)); |
| | 1657 | } |
| | 1658 | return *this; |
| | 1659 | } |
| | 1660 | |
| | 1661 | /// \brief Use previously constructed edge set |
| | 1662 | /// |
| | 1663 | /// Use previously constructed edge set, and specify the edge |
| | 1664 | /// label map. |
| | 1665 | template <typename Map> |
| | 1666 | GraphReader& useEdges(const Map& map) { |
| | 1667 | checkConcept<concepts::ReadMap<Edge, typename Map::Value>, Map>(); |
| | 1668 | LEMON_ASSERT(!_use_edges, "Multiple usage of useEdges() member"); |
| | 1669 | _use_edges = true; |
| | 1670 | _writer_bits::DefaultConverter<typename Map::Value> converter; |
| | 1671 | for (EdgeIt a(_graph); a != INVALID; ++a) { |
| | 1672 | _edge_index.insert(std::make_pair(converter(map[a]), a)); |
| | 1673 | } |
| | 1674 | return *this; |
| | 1675 | } |
| | 1676 | |
| | 1677 | /// \brief Use previously constructed edge set |
| | 1678 | /// |
| | 1679 | /// Use previously constructed edge set, and specify the edge |
| | 1680 | /// label map and a functor which converts the label map values to |
| | 1681 | /// std::string. |
| | 1682 | template <typename Map, typename Converter> |
| | 1683 | GraphReader& useEdges(const Map& map, |
| | 1684 | const Converter& converter = Converter()) { |
| | 1685 | checkConcept<concepts::ReadMap<Edge, typename Map::Value>, Map>(); |
| | 1686 | LEMON_ASSERT(!_use_edges, "Multiple usage of useEdges() member"); |
| | 1687 | _use_edges = true; |
| | 1688 | for (EdgeIt a(_graph); a != INVALID; ++a) { |
| | 1689 | _edge_index.insert(std::make_pair(converter(map[a]), a)); |
| | 1690 | } |
| | 1691 | return *this; |
| | 1692 | } |
| | 1693 | |
| | 1694 | /// @} |
| | 1695 | |
| | 1696 | private: |
| | 1697 | |
| | 1698 | bool readLine() { |
| | 1699 | std::string str; |
| | 1700 | while(++line_num, std::getline(*_is, str)) { |
| | 1701 | line.clear(); line.str(str); |
| | 1702 | char c; |
| | 1703 | if (line >> std::ws >> c && c != '#') { |
| | 1704 | line.putback(c); |
| | 1705 | return true; |
| | 1706 | } |
| | 1707 | } |
| | 1708 | return false; |
| | 1709 | } |
| | 1710 | |
| | 1711 | bool readSuccess() { |
| | 1712 | return static_cast<bool>(*_is); |
| | 1713 | } |
| | 1714 | |
| | 1715 | void skipSection() { |
| | 1716 | char c; |
| | 1717 | while (readSuccess() && line >> c && c != '@') { |
| | 1718 | readLine(); |
| | 1719 | } |
| | 1720 | line.putback(c); |
| | 1721 | } |
| | 1722 | |
| | 1723 | void readNodes() { |
| | 1724 | |
| | 1725 | std::vector<int> map_index(_node_maps.size()); |
| | 1726 | int map_num, label_index; |
| | 1727 | |
| | 1728 | if (!readLine()) |
| | 1729 | throw DataFormatError("Cannot find map captions"); |
| | 1730 | |
| | 1731 | { |
| | 1732 | std::map<std::string, int> maps; |
| | 1733 | |
| | 1734 | std::string map; |
| | 1735 | int index = 0; |
| | 1736 | while (_reader_bits::readToken(line, map)) { |
| | 1737 | if (maps.find(map) != maps.end()) { |
| | 1738 | std::ostringstream msg; |
| | 1739 | msg << "Multiple occurence of node map: " << map; |
| | 1740 | throw DataFormatError(msg.str().c_str()); |
| | 1741 | } |
| | 1742 | maps.insert(std::make_pair(map, index)); |
| | 1743 | ++index; |
| | 1744 | } |
| | 1745 | |
| | 1746 | for (int i = 0; i < static_cast<int>(_node_maps.size()); ++i) { |
| | 1747 | std::map<std::string, int>::iterator jt = |
| | 1748 | maps.find(_node_maps[i].first); |
| | 1749 | if (jt == maps.end()) { |
| | 1750 | std::ostringstream msg; |
| | 1751 | msg << "Map not found in file: " << _node_maps[i].first; |
| | 1752 | throw DataFormatError(msg.str().c_str()); |
| | 1753 | } |
| | 1754 | map_index[i] = jt->second; |
| | 1755 | } |
| | 1756 | |
| | 1757 | { |
| | 1758 | std::map<std::string, int>::iterator jt = maps.find("label"); |
| | 1759 | if (jt == maps.end()) |
| | 1760 | throw DataFormatError("Label map not found in file"); |
| | 1761 | label_index = jt->second; |
| | 1762 | } |
| | 1763 | map_num = maps.size(); |
| | 1764 | } |
| | 1765 | |
| | 1766 | char c; |
| | 1767 | while (readLine() && line >> c && c != '@') { |
| | 1768 | line.putback(c); |
| | 1769 | |
| | 1770 | std::vector<std::string> tokens(map_num); |
| | 1771 | for (int i = 0; i < map_num; ++i) { |
| | 1772 | if (!_reader_bits::readToken(line, tokens[i])) { |
| | 1773 | std::ostringstream msg; |
| | 1774 | msg << "Column not found (" << i + 1 << ")"; |
| | 1775 | throw DataFormatError(msg.str().c_str()); |
| | 1776 | } |
| | 1777 | } |
| | 1778 | if (line >> std::ws >> c) |
| | 1779 | throw DataFormatError("Extra character on the end of line"); |
| | 1780 | |
| | 1781 | Node n; |
| | 1782 | if (!_use_nodes) { |
| | 1783 | n = _graph.addNode(); |
| | 1784 | _node_index.insert(std::make_pair(tokens[label_index], n)); |
| | 1785 | } else { |
| | 1786 | typename std::map<std::string, Node>::iterator it = |
| | 1787 | _node_index.find(tokens[label_index]); |
| | 1788 | if (it == _node_index.end()) { |
| | 1789 | std::ostringstream msg; |
| | 1790 | msg << "Node with label not found: " << tokens[label_index]; |
| | 1791 | throw DataFormatError(msg.str().c_str()); |
| | 1792 | } |
| | 1793 | n = it->second; |
| | 1794 | } |
| | 1795 | |
| | 1796 | for (int i = 0; i < static_cast<int>(_node_maps.size()); ++i) { |
| | 1797 | _node_maps[i].second->set(n, tokens[map_index[i]]); |
| | 1798 | } |
| | 1799 | |
| | 1800 | } |
| | 1801 | if (readSuccess()) { |
| | 1802 | line.putback(c); |
| | 1803 | } |
| | 1804 | } |
| | 1805 | |
| | 1806 | void readEdges() { |
| | 1807 | |
| | 1808 | std::vector<int> map_index(_edge_maps.size()); |
| | 1809 | int map_num, label_index; |
| | 1810 | |
| | 1811 | if (!readLine()) |
| | 1812 | throw DataFormatError("Cannot find map captions"); |
| | 1813 | |
| | 1814 | { |
| | 1815 | std::map<std::string, int> maps; |
| | 1816 | |
| | 1817 | std::string map; |
| | 1818 | int index = 0; |
| | 1819 | while (_reader_bits::readToken(line, map)) { |
| | 1820 | if (maps.find(map) != maps.end()) { |
| | 1821 | std::ostringstream msg; |
| | 1822 | msg << "Multiple occurence of edge map: " << map; |
| | 1823 | throw DataFormatError(msg.str().c_str()); |
| | 1824 | } |
| | 1825 | maps.insert(std::make_pair(map, index)); |
| | 1826 | ++index; |
| | 1827 | } |
| | 1828 | |
| | 1829 | for (int i = 0; i < static_cast<int>(_edge_maps.size()); ++i) { |
| | 1830 | std::map<std::string, int>::iterator jt = |
| | 1831 | maps.find(_edge_maps[i].first); |
| | 1832 | if (jt == maps.end()) { |
| | 1833 | std::ostringstream msg; |
| | 1834 | msg << "Map not found in file: " << _edge_maps[i].first; |
| | 1835 | throw DataFormatError(msg.str().c_str()); |
| | 1836 | } |
| | 1837 | map_index[i] = jt->second; |
| | 1838 | } |
| | 1839 | |
| | 1840 | { |
| | 1841 | std::map<std::string, int>::iterator jt = maps.find("label"); |
| | 1842 | if (jt == maps.end()) |
| | 1843 | throw DataFormatError("Label map not found in file"); |
| | 1844 | label_index = jt->second; |
| | 1845 | } |
| | 1846 | map_num = maps.size(); |
| | 1847 | } |
| | 1848 | |
| | 1849 | char c; |
| | 1850 | while (readLine() && line >> c && c != '@') { |
| | 1851 | line.putback(c); |
| | 1852 | |
| | 1853 | std::string source_token; |
| | 1854 | std::string target_token; |
| | 1855 | |
| | 1856 | if (!_reader_bits::readToken(line, source_token)) |
| | 1857 | throw DataFormatError("Source not found"); |
| | 1858 | |
| | 1859 | if (!_reader_bits::readToken(line, target_token)) |
| | 1860 | throw DataFormatError("Source not found"); |
| | 1861 | |
| | 1862 | std::vector<std::string> tokens(map_num); |
| | 1863 | for (int i = 0; i < map_num; ++i) { |
| | 1864 | if (!_reader_bits::readToken(line, tokens[i])) { |
| | 1865 | std::ostringstream msg; |
| | 1866 | msg << "Column not found (" << i + 1 << ")"; |
| | 1867 | throw DataFormatError(msg.str().c_str()); |
| | 1868 | } |
| | 1869 | } |
| | 1870 | if (line >> std::ws >> c) |
| | 1871 | throw DataFormatError("Extra character on the end of line"); |
| | 1872 | |
| | 1873 | Edge e; |
| | 1874 | if (!_use_edges) { |
| | 1875 | |
| | 1876 | typename NodeIndex::iterator it; |
| | 1877 | |
| | 1878 | it = _node_index.find(source_token); |
| | 1879 | if (it == _node_index.end()) { |
| | 1880 | std::ostringstream msg; |
| | 1881 | msg << "Item not found: " << source_token; |
| | 1882 | throw DataFormatError(msg.str().c_str()); |
| | 1883 | } |
| | 1884 | Node source = it->second; |
| | 1885 | |
| | 1886 | it = _node_index.find(target_token); |
| | 1887 | if (it == _node_index.end()) { |
| | 1888 | std::ostringstream msg; |
| | 1889 | msg << "Item not found: " << target_token; |
| | 1890 | throw DataFormatError(msg.str().c_str()); |
| | 1891 | } |
| | 1892 | Node target = it->second; |
| | 1893 | |
| | 1894 | e = _graph.addEdge(source, target); |
| | 1895 | _edge_index.insert(std::make_pair(tokens[label_index], e)); |
| | 1896 | } else { |
| | 1897 | typename std::map<std::string, Edge>::iterator it = |
| | 1898 | _edge_index.find(tokens[label_index]); |
| | 1899 | if (it == _edge_index.end()) { |
| | 1900 | std::ostringstream msg; |
| | 1901 | msg << "Edge with label not found: " << tokens[label_index]; |
| | 1902 | throw DataFormatError(msg.str().c_str()); |
| | 1903 | } |
| | 1904 | e = it->second; |
| | 1905 | } |
| | 1906 | |
| | 1907 | for (int i = 0; i < static_cast<int>(_edge_maps.size()); ++i) { |
| | 1908 | _edge_maps[i].second->set(e, tokens[map_index[i]]); |
| | 1909 | } |
| | 1910 | |
| | 1911 | } |
| | 1912 | if (readSuccess()) { |
| | 1913 | line.putback(c); |
| | 1914 | } |
| | 1915 | } |
| | 1916 | |
| | 1917 | void readAttributes() { |
| | 1918 | |
| | 1919 | std::set<std::string> read_attr; |
| | 1920 | |
| | 1921 | char c; |
| | 1922 | while (readLine() && line >> c && c != '@') { |
| | 1923 | line.putback(c); |
| | 1924 | |
| | 1925 | std::string attr, token; |
| | 1926 | if (!_reader_bits::readToken(line, attr)) |
| | 1927 | throw DataFormatError("Attribute name not found"); |
| | 1928 | if (!_reader_bits::readToken(line, token)) |
| | 1929 | throw DataFormatError("Attribute value not found"); |
| | 1930 | if (line >> c) |
| | 1931 | throw DataFormatError("Extra character on the end of line"); |
| | 1932 | |
| | 1933 | { |
| | 1934 | std::set<std::string>::iterator it = read_attr.find(attr); |
| | 1935 | if (it != read_attr.end()) { |
| | 1936 | std::ostringstream msg; |
| | 1937 | msg << "Multiple occurence of attribute " << attr; |
| | 1938 | throw DataFormatError(msg.str().c_str()); |
| | 1939 | } |
| | 1940 | read_attr.insert(attr); |
| | 1941 | } |
| | 1942 | |
| | 1943 | { |
| | 1944 | typename Attributes::iterator it = _attributes.lower_bound(attr); |
| | 1945 | while (it != _attributes.end() && it->first == attr) { |
| | 1946 | it->second->set(token); |
| | 1947 | ++it; |
| | 1948 | } |
| | 1949 | } |
| | 1950 | |
| | 1951 | } |
| | 1952 | if (readSuccess()) { |
| | 1953 | line.putback(c); |
| | 1954 | } |
| | 1955 | for (typename Attributes::iterator it = _attributes.begin(); |
| | 1956 | it != _attributes.end(); ++it) { |
| | 1957 | if (read_attr.find(it->first) == read_attr.end()) { |
| | 1958 | std::ostringstream msg; |
| | 1959 | msg << "Attribute not found in file: " << it->first; |
| | 1960 | throw DataFormatError(msg.str().c_str()); |
| | 1961 | } |
| | 1962 | } |
| | 1963 | } |
| | 1964 | |
| | 1965 | public: |
| | 1966 | |
| | 1967 | /// \name Execution of the reader |
| | 1968 | /// @{ |
| | 1969 | |
| | 1970 | /// \brief Start the batch processing |
| | 1971 | /// |
| | 1972 | /// This function starts the batch processing |
| | 1973 | void run() { |
| | 1974 | |
| | 1975 | LEMON_ASSERT(_is != 0, "This reader assigned to an other reader"); |
| | 1976 | |
| | 1977 | bool nodes_done = false; |
| | 1978 | bool edges_done = false; |
| | 1979 | bool attributes_done = false; |
| | 1980 | std::set<std::string> extra_sections; |
| | 1981 | |
| | 1982 | line_num = 0; |
| | 1983 | readLine(); |
| | 1984 | |
| | 1985 | while (readSuccess()) { |
| | 1986 | skipSection(); |
| | 1987 | try { |
| | 1988 | char c; |
| | 1989 | std::string section, caption; |
| | 1990 | line >> c; |
| | 1991 | _reader_bits::readToken(line, section); |
| | 1992 | _reader_bits::readToken(line, caption); |
| | 1993 | |
| | 1994 | if (line >> c) |
| | 1995 | throw DataFormatError("Extra character on the end of line"); |
| | 1996 | |
| | 1997 | if (section == "nodes" && !nodes_done) { |
| | 1998 | if (_nodes_caption.empty() || _nodes_caption == caption) { |
| | 1999 | readNodes(); |
| | 2000 | nodes_done = true; |
| | 2001 | } |
| | 2002 | } else if ((section == "edges" || section == "arcs") && |
| | 2003 | !edges_done) { |
| | 2004 | if (_edges_caption.empty() || _edges_caption == caption) { |
| | 2005 | readEdges(); |
| | 2006 | edges_done = true; |
| | 2007 | } |
| | 2008 | } else if (section == "attributes" && !attributes_done) { |
| | 2009 | if (_attributes_caption.empty() || _attributes_caption == caption) { |
| | 2010 | readAttributes(); |
| | 2011 | attributes_done = true; |
| | 2012 | } |
| | 2013 | } else { |
| | 2014 | if (extra_sections.find(section) != extra_sections.end()) { |
| | 2015 | std::ostringstream msg; |
| | 2016 | msg << "Multiple occurence of section " << section; |
| | 2017 | throw DataFormatError(msg.str().c_str()); |
| | 2018 | } |
| | 2019 | Sections::iterator it = _sections.find(section); |
| | 2020 | if (it != _sections.end()) { |
| | 2021 | extra_sections.insert(section); |
| | 2022 | it->second->process(*_is, line_num); |
| | 2023 | readLine(); |
| | 2024 | } else { |
| | 2025 | readLine(); |
| | 2026 | skipSection(); |
| | 2027 | } |
| | 2028 | } |
| | 2029 | } catch (DataFormatError& error) { |
| | 2030 | error.line(line_num); |
| | 2031 | throw; |
| | 2032 | } |
| | 2033 | } |
| | 2034 | |
| | 2035 | if (!nodes_done) { |
| | 2036 | throw DataFormatError("Section @nodes not found"); |
| | 2037 | } |
| | 2038 | |
| | 2039 | if (!edges_done) { |
| | 2040 | throw DataFormatError("Section @edges not found"); |
| | 2041 | } |
| | 2042 | |
| | 2043 | if (!attributes_done && !_attributes.empty()) { |
| | 2044 | throw DataFormatError("Section @attributes not found"); |
| | 2045 | } |
| | 2046 | |
| | 2047 | } |
| | 2048 | |
| | 2049 | /// @} |
| | 2050 | |
| | 2051 | }; |
| | 2052 | |
| | 2053 | /// \relates GraphReader |
| | 2054 | template <typename Graph> |
| | 2055 | GraphReader<Graph> graphReader(std::istream& is, Graph& graph) { |
| | 2056 | GraphReader<Graph> tmp(is, graph); |
| | 2057 | return tmp; |
| | 2058 | } |
| | 2059 | |
| | 2060 | /// \relates GraphReader |
| | 2061 | template <typename Graph> |
| | 2062 | GraphReader<Graph> graphReader(const std::string& fn, |
| | 2063 | Graph& graph) { |
| | 2064 | GraphReader<Graph> tmp(fn, graph); |
| | 2065 | return tmp; |
| | 2066 | } |
| | 2067 | |
| | 2068 | /// \relates GraphReader |
| | 2069 | template <typename Graph> |
| | 2070 | GraphReader<Graph> graphReader(const char* fn, Graph& graph) { |
| | 2071 | GraphReader<Graph> tmp(fn, graph); |
| | 2072 | return tmp; |
| | 2073 | } |