HierarchyModel.h

#ifndef __HIERARCHICALMODEL_H_
#define __HIERARCHICALMODEL_H_

#include <QStandardItemModel>
#include <QStandardItem>
#include <QModelIndex>
#include <QString>
#include "nlohmann/json.hpp"

//
// treemodel for hierarchical build
//
using json = nlohmann::json;

class CaseHierarchyModel : public QStandardItemModel{


public:

	CaseHierarchyModel();
	CaseHierarchyModel(QString);
	CaseHierarchyModel(QString,bool);
	~CaseHierarchyModel();

	void initialize(QString);
	void initializeFromFile(QString);
	void loadFile(QString filename); // load json file to store json object 
	void saveFile(QString filename); // save caseHiearchy to file


	void generateEntries(json ha); // build entries from hierarchy json obj
	void generateEntries(); // regenerate from cseHierarchy

	void addNode(QString); // add node to current node (m_currentItem)

	void removeNode(QStandardItem* ); // remove node
	void removeCurrentNode(); // remove current Node

	int changeCurrentNode(QModelIndex idx,QString newName); // change current node's name, return 0 is success, 1 is there is same name

	void removeNodeRecursivelyInJson(QString node); // remove a node and its child
	
	bool checkNodename(QString); //check node name exists in the build file (1 : exists, 0 : not existing) 
	bool checkCaseExists(QString); // check if the node has datasetfile entries
	bool isRoot(QString);// ceck if the tag is root node
	bool checkValidity(); // check the hierarchy is valid (currently checking if end nodes has at least one case)

	json getHierarchy(){return m_CaseHierarchy;}; // hierarchy 

	QStringList getFileList(QString node); // get the list of files of a node
	QStringList getRootComponents(); // get component list
	QStringList getAllCasePaths(); // get all actual case files list (for sanity check)
	QStandardItem* getRoot(){return m_rootNode;};
	void setCurrentTag(QModelIndex i);  
	QString getCurrentTag(){return m_currentTag;};
	QString getCurrentType();
	QStandardItem* getCurrentItem(){return m_currentItem;};

	void setFiles(QString nodename, QStringList ql);
	//void onItemChanged(const QModelIndex &);
	QString toString(); // dump json

	void update(); // set things updated
protected:
	void expandNode(QStandardItem*,json);	
	QStringList readCSV(QString filename);// load csv file
private:
	QString m_projectName; // Project target node
	json m_CaseHierarchy;  // hierarchy json object 
	QStandardItem* m_rootNode; //root node ("target")
	QString m_currentTag; // current tag of a node
	QStandardItem* m_currentItem; //current selected node's item
};
#endif