def main():
scpm.init_logger()
sampler = Sampler()
try:
sampler.open()
sampler.start()
try:
calibration = scpm.load_calibration_file()
except:
calibration = None
adc_data = sampler.adc_data_queue.get()
except KeyboardInterrupt:
sampler.terminate()
else:
sampler.terminate()
adc_data = scpm.join_frames_and_widen(adc_data)
split_adc_data = scpm.split_channels(adc_data)
adc_rms = scpm.calculate_adc_rms(split_adc_data)
if calibration is not None:
calcd_data = scpm.calculate_calibrated_power(
split_adc_data, adc_rms, calibration)
print("")
scpm.print_power(calcd_data)
voltage, current = scpm.convert_adc_to_numpy_float(split_adc_data)
voltage, current = scpm.shift_phase(voltage, current, calibration)
scpm.plot(voltage, current, calibration)
logging.shutdown()
def main():
scpm.init_logger()
sampler = Sampler()
try:
sampler.open()
sampler.start()
try:
calibration = scpm.load_calibration_file()
except:
calibration = None
adc_data = sampler.adc_data_queue.get()
except KeyboardInterrupt:
sampler.terminate()
else:
sampler.terminate()
adc_data = scpm.join_frames_and_widen(adc_data)
split_adc_data = scpm.split_channels(adc_data)
adc_rms = scpm.calculate_adc_rms(split_adc_data)
if calibration is not None:
calcd_data = scpm.calculate_calibrated_power(split_adc_data,
adc_rms, calibration)
print("")
scpm.print_power(calcd_data)
voltage, current = scpm.convert_adc_to_numpy_float(split_adc_data)
voltage, current = scpm.shift_phase(voltage, current, calibration)
scpm.plot(voltage, current, calibration)
logging.shutdown()
def __init__(self):
self.wavfile = None
self.sox_process = None # a subprocess.Popen object
self.wavfile_name = None
self.sampler = Sampler()
# Initialise zmq context and socket
self.context = zmq.Context()
self.socket = self.context.socket(zmq.PUB)
def main():
scpm.init_logger()
sampler = Sampler()
wu = WattsUp()
try:
sampler.open()
sampler.start()
wu.open()
calibration = scpm.load_calibration_file()
log.info("Comparing SCPM data with WattsUp data..."
" press CTRL+C to stop.")
while True:
wu_data = wu.get() # blocking
if wu_data is None:
log.warn("wu_data is none!")
continue
time.sleep(config.RECORD_SECONDS)
# Now get ADC data recorded at wu_data.time
adc_data = scpm.find_time(sampler.adc_data_queue, wu_data.time)
if adc_data:
print("")
adc_data = scpm.join_frames_and_widen(adc_data)
split_adc_data = scpm.split_channels(adc_data)
adc_rms = scpm.calculate_adc_rms(split_adc_data)
calcd_data = scpm.calculate_calibrated_power(
split_adc_data, adc_rms, calibration)
scpm.print_power(calcd_data, wu_data)
else:
log.warn("Could not find ADC data for time {}".format(
wu_data.time))
except KeyboardInterrupt:
pass
except Exception:
log.exception("")
wu.terminate()
sampler.terminate()
raise
wu.terminate()
sampler.terminate()
def main():
scpm.init_logger()
sampler = Sampler()
wu = WattsUp()
try:
sampler.open()
sampler.start()
wu.open()
scpm.calibrate(sampler.adc_data_queue, wu)
except KeyboardInterrupt:
pass
except Exception:
log.exception("")
wu.terminate()
sampler.terminate()
raise
wu.terminate()
sampler.terminate()
def main():
scpm.init_logger()
sampler = Sampler()
wu = WattsUp()
try:
sampler.open()
sampler.start()
wu.open()
calibration = scpm.load_calibration_file()
log.info("Comparing SCPM data with WattsUp data..."
" press CTRL+C to stop.")
while True:
wu_data = wu.get() # blocking
if wu_data is None:
log.warn("wu_data is none!")
continue
time.sleep(config.RECORD_SECONDS)
# Now get ADC data recorded at wu_data.time
adc_data = scpm.find_time(sampler.adc_data_queue, wu_data.time)
if adc_data:
print("")
adc_data = scpm.join_frames_and_widen(adc_data)
split_adc_data = scpm.split_channels(adc_data)
adc_rms = scpm.calculate_adc_rms(split_adc_data)
calcd_data = scpm.calculate_calibrated_power(split_adc_data,
adc_rms, calibration)
scpm.print_power(calcd_data, wu_data)
else:
log.warn("Could not find ADC data for time {}"
.format(wu_data.time))
except KeyboardInterrupt:
pass
except Exception:
log.exception("")
wu.terminate()
sampler.terminate()
raise
wu.terminate()
sampler.terminate()
def main():
scpm.init_logger()
sampler = Sampler()
wu = WattsUp()
try:
sampler.open()
sampler.start()
wu.open()
scpm.calibrate(sampler.adc_data_queue, wu)
except KeyboardInterrupt:
pass
except Exception:
log.exception("")
wu.terminate()
sampler.terminate()
raise
wu.terminate()
sampler.terminate()
class Recorder(object):
def __init__(self):
self.wavfile = None
self.sox_process = None # a subprocess.Popen object
self.wavfile_name = None
self.sampler = Sampler()
# Initialise zmq context and socket
self.context = zmq.Context()
self.socket = self.context.socket(zmq.PUB)
def start(self):
calibration = scpm.load_calibration_file()
self.sampler.open()
self.sampler.start()
log.info("Recording power data. Press CTRL+C to stop.")
# Bind to port
self.socket.bind("tcp://*:%s" % config.BROADCAST_PORT)
prev_conv_time = None # the last time sox was run
while True:
raw_adc_data = self.sampler.adc_data_queue.get()
if raw_adc_data is None:
log.warn("adc_data was None!")
continue
# Calculate real power, apparent power, v_rms
adc_data = scpm.join_frames_and_widen(raw_adc_data)
split_adc_data = scpm.split_channels(adc_data)
adc_rms = scpm.calculate_adc_rms(split_adc_data)
power = scpm.calculate_calibrated_power(split_adc_data, adc_rms,
calibration)
# Dump power data to disk
with open(config.DAT_FILENAME, 'a') as data_file:
data_file.write('{:.1f} {:.2f} {:.2f} {:.2f}\n'
.format(adc_data.time, power.real_power,
power.apparent_power, power.volts_rms))
# Broadcast latest values
self.socket.send_string('{:.1f} {:.2f} {:.2f} {:.2f}'
.format(adc_data.time, power.real_power,
power.apparent_power, power.volts_rms))
# Check if it's time to create a new FLAC file
t = datetime.datetime.utcfromtimestamp(adc_data.time)
if self.wavfile is None or t.hour == (prev_conv_time.hour+1)%24:
self._close_and_compress_wavfile()
self.wavfile_name = (config.FLAC_FILENAME_PREFIX +
'{:.6f}'.format(adc_data.time)
.replace('.', '_'))
self.wavfile = scpm.get_wavfile(self.wavfile_name)
else:
# Poll the sox process to see if it has finished.
# We don't really need to do this but it should
# prevent the sox process shell from appearing as a
# zombie process.
self._check_sox_return_code_no_wait()
prev_conv_time = t
# Write uncompressed wav data to disk
self.wavfile.writeframes(b''.join(raw_adc_data.data))
def _close_and_compress_wavfile(self):
if self.wavfile is None:
return
self.wavfile.close()
# Check if the previous conversion process has completed
self._blocking_check_sox_process_has_completed()
# Run new sox process to compress wav file
# sox is a high quality audio conversion program
# the "rate -v -L" option forces very high quality
# rate conversion with linear phase.
cmd = ("sox --no-dither {filename}.wav"
" --compression 8 {filename}.flac"
" rate -v -L {downsampled_rate}"
" && rm {filename}.wav"
.format(filename=self.wavfile_name,
downsampled_rate=config.DOWNSAMPLED_RATE))
log.info("Running: " + cmd)
self.sox_process = subprocess.Popen(cmd, shell=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
def _check_sox_return_code_no_wait(self):
if (self.sox_process is not None
and self.sox_process.returncode is None):
self.sox_process.poll()
if self.sox_process.returncode is not None:
if self.sox_process.returncode == 0:
log.info("Previous sox conversion successfully completed!")
else:
log.warn("Previous sox conversion FAILED!")
# Log sox output
stdout = self.sox_process.stdout.read()
stderr = self.sox_process.stderr.read()
if stdout:
log.debug("sox stdout: {}".format(stdout))
if stderr:
log.debug("sox stderr: {}".format(stderr))
def _blocking_check_sox_process_has_completed(self):
if self.sox_process is None:
return
self._check_sox_return_code_no_wait()
if self.sox_process.returncode is None:
log.warn("Previous sox conversion has not terminated yet!"
" Waiting...")
self.sox_process.wait()
self._check_sox_return_code_no_wait()
def terminate(self):
log.info("Recorder shutdown.\n")
self._close_and_compress_wavfile()
if self.sampler is not None:
self.sampler.terminate()
self._blocking_check_sox_process_has_completed()
self.socket.close()
self.context.term()
logging.shutdown()
def __init__(self):
self.wavfile = None
self.sox_process = None # a subprocess.Popen object
self.wavfile_name = None
self.sampler = Sampler()
class Recorder(object):
def __init__(self):
self.wavfile = None
self.sox_process = None # a subprocess.Popen object
self.wavfile_name = None
self.sampler = Sampler()
def start(self):
calibration = scpm.load_calibration_file()
self.sampler.open()
self.sampler.start()
log.info("Recording power data. Press CTRL+C to stop.")
prev_conv_time = None # the last time sox was run
while True:
raw_adc_data = self.sampler.adc_data_queue.get()
if raw_adc_data is None:
log.warn("adc_data was None!")
continue
# Calculate real power, apparent power, v_rms
adc_data = scpm.join_frames_and_widen(raw_adc_data)
split_adc_data = scpm.split_channels(adc_data)
adc_rms = scpm.calculate_adc_rms(split_adc_data)
power = scpm.calculate_calibrated_power(split_adc_data, adc_rms,
calibration)
# Dump power data to disk
with open(config.DAT_FILENAME, 'a') as data_file:
data_file.write('{:.1f} {:.2f} {:.2f} {:.2f}\n'.format(
adc_data.time, power.real_power, power.apparent_power,
power.volts_rms))
# Check if it's time to create a new FLAC file
t = datetime.datetime.utcfromtimestamp(adc_data.time)
if self.wavfile is None or t.hour == (prev_conv_time.hour +
1) % 24:
self._close_and_compress_wavfile()
self.wavfile_name = (
config.FLAC_FILENAME_PREFIX +
'{:.6f}'.format(adc_data.time).replace('.', '_'))
self.wavfile = scpm.get_wavfile(self.wavfile_name)
else:
# Poll the sox process to see if it has finished.
# We don't really need to do this but it should
# prevent the sox process shell from appearing as a
# zombie process.
self._check_sox_return_code_no_wait()
prev_conv_time = t
# Write uncompressed wav data to disk
self.wavfile.writeframes(b''.join(raw_adc_data.data))
def _close_and_compress_wavfile(self):
if self.wavfile is None:
return
self.wavfile.close()
# Check if the previous conversion process has completed
self._blocking_check_sox_process_has_completed()
# Run new sox process to compress wav file
# sox is a high quality audio conversion program
# the "rate -v -L" option forces very high quality
# rate conversion with linear phase.
cmd = ("sox --no-dither {filename}.wav"
" --compression 8 {filename}.flac"
" rate -v -L {downsampled_rate}"
" && rm {filename}.wav".format(
filename=self.wavfile_name,
downsampled_rate=config.DOWNSAMPLED_RATE))
log.info("Running: " + cmd)
self.sox_process = subprocess.Popen(cmd,
shell=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
def _check_sox_return_code_no_wait(self):
if (self.sox_process is not None
and self.sox_process.returncode is None):
self.sox_process.poll()
if self.sox_process.returncode is not None:
if self.sox_process.returncode == 0:
log.info("Previous sox conversion successfully completed!")
else:
log.warn("Previous sox conversion FAILED!")
# Log sox output
stdout = self.sox_process.stdout.read()
stderr = self.sox_process.stderr.read()
if stdout:
log.debug("sox stdout: {}".format(stdout))
if stderr:
log.debug("sox stderr: {}".format(stderr))
def _blocking_check_sox_process_has_completed(self):
if self.sox_process is None:
return
self._check_sox_return_code_no_wait()
if self.sox_process.returncode is None:
log.warn("Previous sox conversion has not terminated yet!"
" Waiting...")
self.sox_process.wait()
self._check_sox_return_code_no_wait()
def terminate(self):
log.info("Recorder shutdown.\n")
self._close_and_compress_wavfile()
if self.sampler is not None:
self.sampler.terminate()
self._blocking_check_sox_process_has_completed()
logging.shutdown()
def __init__(self):
self.wavfile = None
self.sox_process = None # a subprocess.Popen object
self.wavfile_name = None
self.sampler = Sampler()