import numpy as np
import time
import pdb
import weakref
from ctypes import *
from nidaqmx import Task

import atexit

# Simple class that can be used to control one Digital output line
class DO(Task):
    def __init__(self, dev):
        Task.__init__(self)
        self.do_channels.add_do_chan(dev)
        self.write(False)
        atexit.register(self.clean)

    def clean(self):
        self.low()
        self.close()

    def high(self):
        self.write(True)

    def low(self):
        self.write(False)


if __name__ == '__main__':
    dig_0 = DO('/cDAQ1Mod1/port0/line0')
    dig_0.high()
    time.sleep(5)

    dig_1 = DO('/cDAQ1Mod1/port0/line1')
    dig_1.high()
    time.sleep(5)