Added MergeTree Tracking

Author: Himangshu Saikia

FileIO.cpp

@@ -0,0 +1,200 @@
+/*********************************************************************
+*  Author  : Himangshu Saikia
+*  Email   : [email protected]
+*  Project : Merge Tree Library
+*
+*********************************************************************
+*/
+
+#include "FileIO.h"
+
+namespace mtlib {
+
+	void FileIO::readRegionsFromCSV(const std::string & filename, std::map<dagkey, Region>& dagNodes)
+	{
+		dagNodes.clear();
+		std::ifstream f;
+		f.open(filename);
+
+		std::string line;
+
+		bool first = true;
+
+		int count = 0;
+
+		while (std::getline(f, line))
+		{
+			//std::cout << "Reading Line " << count << "\n";
+
+			std::istringstream iss(line);
+
+			if (first) {
+				first = false;
+
+				std::string token;
+
+				while (std::getline(iss, token, ',')) {
+					count++;
+				}
+
+				std::cout << "Histogram Size is " << count - 11 << "\n";
+
+				continue;
+			}
+
+			Region R;
+
+			R.hist.init(count - 11);
+
+			int sttype, par_id;
+			dagkey key;
+
+			char c; // comma
+
+			iss >> R.timestep >> c >> R.index >> c >> R.vindex >> c >> sttype >> c >> par_id >> c >> key >> c >> R.hist.hdf
+				>> c >> R.hist.vol >> c >> R.hist.cm.x >> c >> R.hist.cm.y >> c >> R.hist.cm.z;
+
+			for (auto& bin : R.hist.bins) {
+				iss >> c >> bin;
+			}
+
+			dagNodes[key] = R;
+
+		}
+
+		f.close();
+	}
+
+
+	void FileIO::readMergeTreesFromCSV(const std::string & filename, std::vector<std::map<int, int>>& unAugTrees)
+	{
+		unAugTrees.clear();
+
+		std::ifstream f;
+		f.open(filename);
+
+		std::string line;
+
+		bool first = true;
+
+		int count = 0;
+
+		while (std::getline(f, line))
+		{
+			if (first) {
+				first = false;
+				continue;
+			}
+
+			count++;
+
+			std::istringstream iss(line);
+			int t, id, vid, sttype, par_id, seq_id, st_vol;
+			float st_hdf;
+			Vertex3Df cm;
+			char c; // comma
+			iss >> t >> c >> id >> c >> vid >> c >> sttype >> c >> par_id >> c >> seq_id >> c >> st_hdf >> c >> st_vol >> c >> cm.x >> c >> cm.y >> c >> cm.z;
+
+			if (t >= unAugTrees.size()) {
+				unAugTrees.resize(t + 1);
+			}
+			unAugTrees[t].insert(std::pair<int, int>(id, par_id));
+
+		}
+
+		std::cout << "[Tracking::readMergeTreesFromCSV] Number of Merge Trees " << unAugTrees.size() << ". Num lines read " << count << "\n";
+
+		f.close();
+	}
+
+	void FileIO::writeCostMatrixToDisk(const std::string & filename, const std::vector<std::vector<double>>& costMatrix)
+	{
+		std::ofstream f;
+		f.open(filename);
+
+		f << costMatrix.size() << "\n";
+
+		for (size_t i = 0; i < costMatrix.size(); i++) {
+			for (size_t j = 0; j < costMatrix[i].size(); j++) {
+				f << costMatrix[i][j] << "\t";
+			}
+			f << "\n";
+		}
+		f.close();
+	}
+
+	void FileIO::readCostMatrixFromDisk(const std::string & filename, std::vector<std::vector<double>>& costMatrix)
+	{
+		std::ifstream f;
+		f.open(filename);
+
+		size_t n;
+		f >> n;
+
+		costMatrix.clear();
+		costMatrix.resize(n, std::vector<double>(n));
+
+		for (size_t i = 0; i < n; i++) {
+			for (size_t j = 0; j < n; j++) {
+				f >> costMatrix[i][j];
+			}
+		}
+
+		f.close();
+	}
+
+	void FileIO::drawMatches(const std::map<size_t, size_t>& matches, const std::string& prefix1, const std::string& prefix2, std::ofstream & file)
+	{
+		for (const auto& match : matches) {
+			file << prefix1 << match.first << "->" << prefix2 << match.second << "[color=\"#00b300\",style=\"setlinewidth(2)\"];\n";
+		}
+	}
+
+	void FileIO::addHeaderToStatisticsFile(std::ofstream & fs, const int hist_buckets)
+	{
+		fs << "Timestep,Merge_Tree_Id,Volume_Vertex,Subtree_Repr_Type,Merge_Tree_Parent_Id,DAG_Key,Subtree_Height_Difference,Subtree_Volume,Subtree_Center_of_Mass_x,Subtree_Center_of_Mass_y,Subtree_Center_of_Mass_z";
+		for (auto i = 0; i < hist_buckets; i++) {
+			fs << ",Hist_" << i;
+		}
+		fs << "\n";
+	}
+
+	void FileIO::addToStatistics(std::ofstream & fs, const int & timestep, const ConciseSMT & smt, const std::vector<Hist>& hists, const double globalRange)
+	{
+		for (auto i = 0; i < smt.cps.size() - 1; i++) {
+
+			int mtParentId = i;
+			if (smt.unAugIdxTree.find(i) != smt.unAugIdxTree.end()) {
+				mtParentId = smt.unAugIdxTree.find(i)->second;
+			}
+			else {
+				continue;
+			}
+
+			if (mtParentId == smt.unAugIdxTree.size()) {
+				mtParentId = -1; // global minimum
+			}
+
+			auto key = DagNode::makeKey(timestep, i);
+
+			fs << timestep << ","
+				<< i << ","
+				<< smt.cps[i].idx << ","
+				<< smt.cps[i].typ << ","
+				<< mtParentId << ","
+				<< key << ","
+				<< (hists[i].maxVal - hists[i].minVal) * globalRange << ","
+				<< hists[i].vol << ","
+				<< hists[i].cm.x << ","
+				<< hists[i].cm.y << ","
+				<< hists[i].cm.z;
+
+			for (const auto& bin : hists[i].bins) {
+				fs << "," << bin;
+			}
+
+			fs << "\n";
+		}
+	}
+
+}

FileIO.h

@@ -0,0 +1,38 @@
+/*********************************************************************
+*  Author  : Himangshu Saikia
+*  Email   : [email protected]
+*  Project : Merge Tree Library
+*
+*********************************************************************
+*/
+
+#pragma once
+#include <vector>
+#include <map>
+#include <iomanip>
+#include <limits>
+#include <sstream>
+#include <fstream>
+#include "TrackInfo.h"
+#include "MergeTree.h"
+#include "Dag.h"
+
+namespace mtlib {
+
+	class FileIO {
+	public:
+		static void readRegionsFromCSV(const std::string& filename, std::map<dagkey, Region>& dagNodes);
+		static void readMergeTreesFromCSV(const std::string& filename, std::vector<std::map<int, int> >& unAugTrees);
+		static void writeCostMatrixToDisk(const std::string& filename, const std::vector<std::vector<double> >& costMatrix);
+		static void readCostMatrixFromDisk(const std::string& filename, std::vector<std::vector<double> >& costMatrix);
+		static void drawMatches(const std::map<size_t, size_t>& matches, const std::string& prefix1, const std::string& prefix2, std::ofstream& file);
+		static void addHeaderToStatisticsFile(std::ofstream & fs, const int hist_buckets);
+		static void addToStatistics(
+			std::ofstream& fs,
+			const int& timestep,
+			const ConciseSMT& smt,
+			const std::vector < Hist >& hists,
+			const double globalRange
+		);
+	};
+}