mirror of
https://github.com/GoldenCheetah/GoldenCheetah.git
synced 2026-02-15 00:49:55 +00:00
Restructure source directory
Introducing a directory structure to make it a bit less daunting for new developers and perhaps even old hands. The main folders all start with an upper character, so src files are now located in; * Core - Core data structures * Gui - Main GUI elements * Metrics - Models and Metrics * FileIO - Device and File I/O * Charts - All the chart types * Cloud - Working with Web Resources * Train - Anything Train View specific * ANT - Our ANT+ Stack * Resources - Images, Translations, Web etc Apologies to anyone who needs to merge across this update.
This commit is contained in:
Mark Liversedge
384
src/FileIO/SrmRideFile.cpp
Normal file
384
src/FileIO/SrmRideFile.cpp
Normal file
@@ -0,0 +1,384 @@
|
||||
/*
|
||||
* Copyright (c) 2007 Sean C. Rhea (srhea@srhea.net)
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License as published by the Free
|
||||
* Software Foundation; either version 2 of the License, or (at your option)
|
||||
* any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful, but WITHOUT
|
||||
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||||
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
||||
* more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License along
|
||||
* with this program; if not, write to the Free Software Foundation, Inc., 51
|
||||
* Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
*/
|
||||
|
||||
//
|
||||
// Decodes .srm files. Adapted from (buggy) description by Stephan Mantler
|
||||
// (http://www.stephanmantler.com/?page_id=86).
|
||||
//
|
||||
|
||||
#include "SrmRideFile.h"
|
||||
#include <QDataStream>
|
||||
#include <QDate>
|
||||
#include <QFile>
|
||||
#include <QStringList>
|
||||
#include <assert.h>
|
||||
#include <cmath>
|
||||
#include <stdio.h>
|
||||
#ifdef WIN32
|
||||
#include <winsock.h>
|
||||
#else
|
||||
#include <arpa/inet.h>
|
||||
#endif
|
||||
|
||||
#define PI M_PI
|
||||
|
||||
static quint8 readByte(QDataStream &in)
|
||||
{
|
||||
quint8 value;
|
||||
in >> value;
|
||||
return value;
|
||||
}
|
||||
|
||||
static quint16 readShort(QDataStream &in)
|
||||
{
|
||||
quint16 value;
|
||||
in >> value;
|
||||
return value;
|
||||
}
|
||||
|
||||
static qint16 readSignedShort(QDataStream &in)
|
||||
{
|
||||
qint16 value;
|
||||
in >> value;
|
||||
return value;
|
||||
}
|
||||
|
||||
static quint32 readLong(QDataStream &in)
|
||||
{
|
||||
quint32 value;
|
||||
in >> value;
|
||||
return value;
|
||||
}
|
||||
|
||||
static qint32 readSignedLong(QDataStream &in)
|
||||
{
|
||||
qint32 value;
|
||||
in >> value;
|
||||
return value;
|
||||
}
|
||||
|
||||
struct marker
|
||||
{
|
||||
quint32 start, end;
|
||||
QString note;
|
||||
};
|
||||
|
||||
struct blockhdr
|
||||
{
|
||||
QDateTime dt;
|
||||
quint32 chunkcnt;
|
||||
};
|
||||
|
||||
static int srmFileReaderRegistered =
|
||||
RideFileFactory::instance().registerReader(
|
||||
"srm", "SRM training files", new SrmFileReader());
|
||||
|
||||
RideFile *SrmFileReader::openRideFile(QFile &file, QStringList &errorStrings, QList<RideFile*>*) const
|
||||
{
|
||||
if (!file.open(QFile::ReadOnly)) {
|
||||
errorStrings << QString("can't open file %1").arg(file.fileName());
|
||||
return NULL;
|
||||
}
|
||||
QDataStream in(&file);
|
||||
in.setByteOrder( QDataStream::LittleEndian );
|
||||
|
||||
RideFile *result = new RideFile;
|
||||
result->setDeviceType("SRM");
|
||||
result->setFileFormat("SRM training files (srm)");
|
||||
result->setTag("Sport", "Bike" );
|
||||
|
||||
char magic[4];
|
||||
in.readRawData(magic, sizeof(magic));
|
||||
if( strncmp(magic, "SRM", 3)){
|
||||
errorStrings << QString("Unrecognized file type, missing magic." );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
int version = magic[3] - '0';
|
||||
switch( version ){
|
||||
case 5:
|
||||
case 6:
|
||||
case 7:
|
||||
case 9:
|
||||
// ok
|
||||
break;
|
||||
|
||||
default:
|
||||
errorStrings << QString("Unsupported SRM file format version: %1")
|
||||
.arg(version);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
quint16 dayssince1880 = readShort(in);
|
||||
quint16 wheelcirc = readShort(in);
|
||||
quint8 recint1 = readByte(in);
|
||||
quint8 recint2 = readByte(in);
|
||||
quint16 blockcnt = readShort(in);
|
||||
quint16 markercnt = readShort(in);
|
||||
readByte(in); // padding
|
||||
quint8 commentlen = readByte(in);
|
||||
|
||||
if( commentlen > 70 )
|
||||
commentlen = 70;
|
||||
|
||||
char comment[71];
|
||||
in.readRawData(comment, sizeof(comment) - 1);
|
||||
comment[commentlen] = '\0';
|
||||
result->setTag("Notes", QString(comment) );
|
||||
|
||||
// assert propper markercnt to avoid segfaults
|
||||
if( in.status() != QDataStream::Ok ){
|
||||
errorStrings << QString("failed to read file header" );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
result->setRecIntSecs(((double) recint1) / recint2);
|
||||
unsigned recintms = (unsigned) round(result->recIntSecs() * 1000.0);
|
||||
|
||||
result->setTag("Wheel Circumference", QString("%1").arg(wheelcirc) );
|
||||
|
||||
QDate date(1880, 1, 1);
|
||||
date = date.addDays(dayssince1880);
|
||||
|
||||
QVector<marker> markers(markercnt + 1);
|
||||
for (int i = 0; i <= markercnt; ++i) {
|
||||
char mcomment[256];
|
||||
size_t mcommentlen = version < 6 ? 3 : 255;
|
||||
assert( mcommentlen < sizeof(mcomment) );
|
||||
in.readRawData(mcomment, mcommentlen );
|
||||
mcomment[mcommentlen] = '\0';
|
||||
|
||||
quint8 active = readByte(in);
|
||||
quint32 start = version < 9 ? readShort(in) : readLong(in );;
|
||||
quint32 end = version < 9 ? readShort(in) : readLong(in );
|
||||
quint16 avgwatts = readShort(in);
|
||||
quint16 avghr = readShort(in);
|
||||
quint16 avgcad = readShort(in);
|
||||
quint16 avgspeed = readShort(in);
|
||||
quint16 pwc150 = readShort(in);
|
||||
|
||||
// data fixup: Although the data chunk index in srm files starts
|
||||
// with 1, some srmwin wrote files referencing index 0.
|
||||
if( end < 1 ) end = 1;
|
||||
if( start < 1 ) start = 1;
|
||||
|
||||
// data fixup: some srmwin versions wrote markers with start > end
|
||||
if( end < start ){
|
||||
markers[i].start = end;
|
||||
markers[i].end = start;
|
||||
} else {
|
||||
markers[i].start = start;
|
||||
markers[i].end = end;
|
||||
}
|
||||
|
||||
markers[i].note = QString( mcomment);
|
||||
|
||||
if( i == 0 ){
|
||||
result->setTag("Athlete Name", QString(mcomment) );
|
||||
}
|
||||
|
||||
(void) active;
|
||||
(void) avgwatts;
|
||||
(void) avghr;
|
||||
(void) avgcad;
|
||||
(void) avgspeed;
|
||||
(void) pwc150;
|
||||
(void) wheelcirc;
|
||||
}
|
||||
|
||||
// fail early to tell devs whats wrong with file
|
||||
if( in.status() != QDataStream::Ok ){
|
||||
errorStrings << QString("failed to read marker" );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
blockhdr *blockhdrs = new blockhdr[blockcnt+1];
|
||||
quint32 blockchunkcnt = 0;
|
||||
for (int i = 0; i < blockcnt; ++i) {
|
||||
// In the .srm files generated by Rainer Clasen's srmcmd,
|
||||
// hsecsincemidn is a *signed* 32-bit integer. I haven't seen a
|
||||
// negative value in any .srm files generated by srmwin.exe, but
|
||||
// since the number of hundredths of a second in a day is << 2^31,
|
||||
// it seems safe to always treat this number as signed.
|
||||
qint32 hsecsincemidn = readLong(in);
|
||||
blockhdrs[i].chunkcnt = version < 9 ? readShort(in) : readLong(in);
|
||||
blockhdrs[i].dt = QDateTime(date);
|
||||
blockhdrs[i].dt = blockhdrs[i].dt.addMSecs(hsecsincemidn * 10);
|
||||
blockchunkcnt += blockhdrs[i].chunkcnt;
|
||||
}
|
||||
|
||||
// fail early to tell devs whats wrong with file
|
||||
if( in.status() != QDataStream::Ok ){
|
||||
errorStrings << QString("failed to read block headers" );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
quint16 zero = readShort(in);
|
||||
quint16 slope = readShort(in);
|
||||
quint32 datacnt = version < 9 ? readShort(in) : readLong(in);
|
||||
readByte(in); // padding
|
||||
|
||||
// fail early to tell devs whats wrong with file
|
||||
if( in.status() != QDataStream::Ok ){
|
||||
errorStrings << QString("failed to read calibration data" );
|
||||
return NULL;
|
||||
}
|
||||
|
||||
result->setTag("Slope", QString("%1")
|
||||
.arg( 140.0 / 42781 * slope, 0, 'f', 2) );
|
||||
result->setTag("Zero Offset", QString("%1").arg(zero) );
|
||||
|
||||
// SRM5 files have no blocks - synthesize one
|
||||
if( blockcnt < 1 ){
|
||||
blockcnt = 0;
|
||||
blockhdrs[0].chunkcnt = datacnt;
|
||||
blockhdrs[0].dt = QDateTime(date);
|
||||
}
|
||||
|
||||
// datacnt might overflow at 64k - so, use sum from blocks, instead
|
||||
if( datacnt > blockchunkcnt ){
|
||||
blockchunkcnt = datacnt;
|
||||
errorStrings << QString("inconsistent chunk count total=%1, blocks=%2")
|
||||
.arg(datacnt)
|
||||
.arg(blockchunkcnt);
|
||||
}
|
||||
|
||||
quint32 blkidx = 0;
|
||||
int blknum = 0, mrknum = 0, interval = 0;
|
||||
double km = 0.0, secs = 0.0;
|
||||
|
||||
if (markercnt > 0)
|
||||
mrknum = 1;
|
||||
|
||||
for (quint32 i = 0; i < blockchunkcnt; ++i) {
|
||||
int cad, hr, watts;
|
||||
double kph, alt, lon = 0, lat = 0;
|
||||
double temp=-255;
|
||||
if (version < 7) {
|
||||
quint8 ps[3];
|
||||
in.readRawData((char*) ps, sizeof(ps));
|
||||
cad = readByte(in);
|
||||
hr = readByte(in);
|
||||
kph = (((((unsigned) ps[1]) & 0xf0) << 3)
|
||||
| (ps[0] & 0x7f)) * 3.0 / 26.0;
|
||||
watts = (ps[1] & 0x0f) | (ps[2] << 0x4);
|
||||
alt = 0.0;
|
||||
}
|
||||
else {
|
||||
watts = readShort(in);
|
||||
cad = readByte(in);
|
||||
hr = readByte(in);
|
||||
|
||||
qint32 kph_tmp = readSignedLong(in);
|
||||
kph = kph_tmp < 0 ? 0 : kph_tmp * 3.6 / 1000.0;
|
||||
|
||||
alt = readSignedLong(in);
|
||||
temp = 0.1 * readSignedShort(in);
|
||||
|
||||
if( version == 9 ){
|
||||
lat = readSignedLong(in) * 180.0 / 0x7fffffff;
|
||||
lon = readSignedLong(in) * 180.0 / 0x7fffffff;
|
||||
}
|
||||
}
|
||||
|
||||
if( in.status() != QDataStream::Ok ){
|
||||
errorStrings << QString("premature end of file" );
|
||||
break;
|
||||
}
|
||||
|
||||
if (i == 0) {
|
||||
result->setStartTime(blockhdrs[blknum].dt);
|
||||
}
|
||||
if (mrknum < markers.size() && i == markers[mrknum].end) {
|
||||
++interval;
|
||||
++mrknum;
|
||||
}
|
||||
|
||||
// markers count from 1
|
||||
if ((i > 0) && (mrknum < markers.size()) && (i == markers[mrknum].start - 1))
|
||||
++interval;
|
||||
|
||||
km += result->recIntSecs() * kph / 3600.0;
|
||||
|
||||
double nm = watts / 2.0 / PI / cad * 60.0;
|
||||
result->appendPoint(secs, cad, hr, km, kph, nm, watts, alt, lon, lat, 0.0, 0.0, temp, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, interval);
|
||||
|
||||
++blkidx;
|
||||
if ((blkidx == blockhdrs[blknum].chunkcnt) && (blknum + 1 < blockcnt)) {
|
||||
QDateTime end = blockhdrs[blknum].dt.addMSecs(
|
||||
recintms * blockhdrs[blknum].chunkcnt);
|
||||
++blknum;
|
||||
blkidx = 0;
|
||||
QDateTime start = blockhdrs[blknum].dt;
|
||||
qint64 endms =
|
||||
((qint64) end.toTime_t()) * 1000 + end.time().msec();
|
||||
qint64 startms =
|
||||
((qint64) start.toTime_t()) * 1000 + start.time().msec();
|
||||
double diff_secs = (startms - endms + recintms) / 1000.0;
|
||||
if (diff_secs < result->recIntSecs()) {
|
||||
errorStrings << QString("ERROR: time goes backwards by %1 s"
|
||||
" on trans " "to block %2"
|
||||
).arg(diff_secs).arg(blknum);
|
||||
secs += result->recIntSecs(); // for lack of a better option
|
||||
}
|
||||
else {
|
||||
secs += diff_secs;
|
||||
}
|
||||
}
|
||||
else {
|
||||
secs += result->recIntSecs();
|
||||
}
|
||||
}
|
||||
|
||||
// assert some points were found. prevents segfault when looking at
|
||||
// the overall markers[0].start/.end
|
||||
// note: we're not checking in.status() to cope with truncated files
|
||||
|
||||
if( result->dataPoints().size() < 1 ){
|
||||
errorStrings << QString("file contains no data points");
|
||||
return NULL;
|
||||
}
|
||||
|
||||
double last = 0.0;
|
||||
for (int i = 1; i < markers.size(); ++i) {
|
||||
const marker &marker = markers[i];
|
||||
quint32 start = qMin(marker.start, (quint32)result->dataPoints().size()) - 1;
|
||||
double start_secs = result->dataPoints()[start]->secs;
|
||||
quint32 end = qMin(marker.end - 1, (quint32)result->dataPoints().size() - 1);
|
||||
double end_secs = result->dataPoints()[end]->secs + result->recIntSecs();
|
||||
if( last < start_secs )
|
||||
result->addInterval(RideFileInterval::DEVICE, last, start_secs, "");
|
||||
QString note = QString("%1").arg(i);
|
||||
if( marker.note.length() )
|
||||
note += QString(" ") + marker.note;
|
||||
if( start_secs <= end_secs )
|
||||
result->addInterval(RideFileInterval::DEVICE, start_secs, end_secs, note );
|
||||
last = end_secs;
|
||||
}
|
||||
if (!markers.empty() && markers.last().end < (quint32)result->dataPoints().size()) {
|
||||
double start_secs = result->dataPoints().last()->secs + result->recIntSecs();
|
||||
result->addInterval(RideFileInterval::DEVICE, last, start_secs, "");
|
||||
}
|
||||
|
||||
file.close();
|
||||
return result;
|
||||
}
|
||||
|
||||
// vi:expandtab tabstop=4 shiftwidth=4
|
||||
Reference in New Issue
Block a user