IonizationChamber

import datetime

from IonizationChamber import IonizationChamber 
from dmm import DMM
import config

class IonizationChamberStateMachine:
    def __init__(self, config):
        self.config = config
        self.deviceMeasurement = 0.0
        self.dmmMeasurement = 0.0
        
        self.nextState = self.initIonizationChamber


    def tick(self):
        self.nextState()


    def initIonizationChamber(self):
        self.chamber = IonizationChamber(config)
        self.chamber.connect()
        
        if config.useDMM:
            self.nextState = self.initDMM
        else:
            self.nextState = self.initOutputFile


    def initDMM(self):
        self.dmm = DMM(config)
        self.dmm.connect()
        
        self.nextState = self.initOutputFile


    def initOutputFile(self):
        self.logFile = open('data.csv', 'w')
        self.logFile.write("Time,Counter,DMM\n")
        
        self.nextState = self.getMeasurementFromIonizationChamber


    def getMeasurementFromIonizationChamber(self):
        self.deviceMeasurement = self.chamber.getMeasurement()
        
        if config.useDMM:
            self.nextState = self.getMeasurementFromDMM
        else:
            self.nextState = self.saveMeasurement


    def getMeasurementFromDMM(self):
        self.dmmMeasurement = self.dmm.getMeasurement()
        
        self.nextState = self.saveMeasurement


    def saveMeasurement(self):
        now = datetime.datetime.now()
        self.logFile.write("{0},{1},{2}\n".format(now, self.deviceMeasurement, self.dmmMeasurement))
        self.logFile.flush()
        
        self.nextState = self.showMeasurementToUser


    def showMeasurementToUser(self):
        print("{0}, {1}".format(self.deviceMeasurement, self.dmmMeasurement))
        
        self.nextState = self.getMeasurementFromIonizationChamber


if __name__=="__main__":
    machine = IonizationChamberStateMachine(config)

    while True:
        machine.tick()