#!/usr/bin/python3

import configobj
import xml.etree.ElementTree as ET
import base64
import zlib
import io


'''
This is a trisurf instance manager written in python


Invoke with:
tsmgr [-t tape | -r snapshot.vtu] [-s subdirectory]

If tape is specified, the trisurf wilt start from tape with initial distribution, if snapshot is specified the trisurf will be restored from given snapshot file and simulation will continue.

'''



class Tape:
	'''Has all the info on the tape'''

	def __init__(self):
		return

	def readTape(self, tape='tape'):
		try:
			self.config=configobj.ConfigObj(tape)
		except:
			print("Error reading or parsing tape file!\n")
			exit(1)
	

	def setTape(self, string):
		self.config=configobj.ConfigObj(io.StringIO(string))
		return

	def getValue(self,key):
		return self.config[key]



class Runner:
	'''
	Class Runner consists of a single running or terminated instance of the trisurf
	'''
	def initFromTape(self, tape):
		self.tape=Tape()
		self.tape.readTape(tape)

	def initFromSnapshot(self, snapshotfile):
		try:
			tree = ET.parse(snapshotfile)
		except:
			print("Error reading snapshot file")
			exit(1)

		root = tree.getroot()
		tapetxt=root.find('tape')
		version=root.find('trisurfversion')
		#print("Reading snapshot made from: "+version.text)
		self.tape=Tape()
		#print(tapetxt.text)
		self.tape.setTape(tapetxt.text)

	def __init__(self, subdir='run0', tape='', snapshot=''):
		self.subdir=subdir
		if(tape!=''):
			self.initFromTape(tape)
		if(snapshot!=''):
			self.initFromSnapshot(snapshot)

		return

	def getStatus(self):
		pass

	def start(self):
		pass

	def stop(self):
		pass

	def __str__(self):
		return("Running instance")