MuseScore/omr/importpdf.cpp
2016-05-25 21:55:40 -04:00

322 lines
12 KiB
C++

//=============================================================================
// MuseScore
// Linux Music Score Editor
// $Id: importmidi.cpp 2721 2010-02-15 19:41:28Z wschweer $
//
// Copyright (C) 2002-2011 Werner Schweer and others
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License version 2.
//
// 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., 675 Mass Ave, Cambridge, MA 02139, USA.
//=============================================================================
#include "importpdf.h"
#include "libmscore/score.h"
#include "omr/omr.h"
#include "libmscore/part.h"
#include "libmscore/staff.h"
#include "libmscore/measure.h"
#include "libmscore/rest.h"
#include "omr/omrpage.h"
#include "libmscore/segment.h"
#include "libmscore/layoutbreak.h"
#include "libmscore/page.h"
#include "libmscore/clef.h"
#include "libmscore/bracket.h"
#include "libmscore/mscore.h"
#include "libmscore/chord.h"
#include "libmscore/note.h"
#include "libmscore/utils.h"
#include "libmscore/timesig.h"
#include "libmscore/keysig.h"
#include "libmscore/spacer.h"
namespace Ms {
//---------------------------------------------------------
// OmrState
//---------------------------------------------------------
class OmrState {
public:
Score* score = 0;
Fraction timesig { 4, 4};
int tick = 0;
void importPdfPage(OmrPage* omrPage, int top);
int importPdfSystem(OmrSystem* omrSystem);
void importPdfMeasure(OmrMeasure* m, const OmrSystem* omrSystem);
};
//---------------------------------------------------------
// importPdfMeasure
//---------------------------------------------------------
void OmrState::importPdfMeasure(OmrMeasure* m, const OmrSystem* omrSystem)
{
Measure* measure = new Measure(score);
measure->setTick(tick);
if (m->timesig()) {
timesig = m->timesig()->timesig;
score->sigmap()->add(tick, SigEvent(timesig));
}
measure->setTimesig(timesig);
measure->setLen(timesig);
TDuration d(TDuration::DurationType::V_MEASURE);
Rest* rest;
Segment* s = measure->getSegment(Segment::Type::ChordRest, tick);
for (int staffIdx = 0; staffIdx < omrSystem->staves().size(); ++staffIdx) {
rest = new Rest(score, d);
rest->setDuration(timesig);
rest->setTrack(staffIdx*4);
s->add(rest);
}
#if 0
for (int staffIdx = 0; staffIdx < omrSystem->staves().size(); ++staffIdx) {
if (tick == 0) {
const OmrStaff& omrStaff = omrSystem->staves()[staffIdx];
int keySigType = omrStaff.keySig().type;
KeySig* ks = new KeySig(score);
//ks->setSig(keySigType, keySigType);
ks->setTrack(staffIdx * VOICES);
Segment* s = measure->getSegment(Segment::Type::KeySig, 0);
s->add(ks);
//score->staff(staffIdx)->setKey(0, keySigType);
}
if (m->timesig()) {
TimeSig* ts = new TimeSig(score);
Segment* s = measure->getSegment(Segment::Type::TimeSig, tick);
ts->setSig(timesig);
ts->setTrack(staffIdx * VOICES);
s->add(ts);
}
Fraction nl;
QList<OmrChord>& chords = m->chords()[staffIdx];
if (timesig == Fraction(3,8)) {
if (chords.size() == 1) {
chords[0].duration = TDuration(timesig);
}
else if (chords.size() == 3) {
int i = 0;
for (;i < 3; ++i) {
if (chords[i].duration.fraction() != Fraction(1,4))
break;
}
if (i == 3) {
for (i = 0;i < 3; ++i) {
chords[i].duration = TDuration(Fraction(1, 8));
}
}
}
}
foreach(const OmrChord& omrChord, chords) {
nl += omrChord.duration.fraction();
}
bool notesOk = nl == timesig;
if (notesOk) {
int ltick = 0;
foreach(const OmrChord& omrChord, chords) {
Chord* chord = new Chord(score);
chord->setDurationType(omrChord.duration);
chord->setDuration(omrChord.duration.fraction());
chord->setTrack(staffIdx * VOICES);
Segment* s = measure->getSegment(Segment::Type::ChordRest, tick + ltick);
s->add(chord);
//int keyType = score->staff(staffIdx)->key(tick + ltick).accidentalType();
foreach (OmrNote* omrNote, omrChord.notes) {
Note* note = new Note(score);
//ClefType clef = score->staff(staffIdx)->initialClef()._concertClef;
//int pitch = line2pitch(omrNote->line, clef, keyType);
//note->setPitch(pitch);
note->setTpcFromPitch();
chord->add(note);
}
ltick += omrChord.duration.ticks();
}
}
else {
TDuration d(TDuration::DurationType::V_MEASURE);
Segment* s = measure->getSegment(Segment::Type::ChordRest, measure->tick());
Rest* rest = new Rest(score, d);
rest->setDuration(timesig);
rest->setTrack(staffIdx * VOICES);
s->add(rest);
}
}
#endif
score->measures()->add(measure);
tick += measure->timesig().ticks();
}
//---------------------------------------------------------
// importPdfSystem
//---------------------------------------------------------
int OmrState::importPdfSystem(OmrSystem* omrSystem)
{
for (int i = 0; i < omrSystem->measures().size(); ++i) {
OmrMeasure* m = &omrSystem->measures()[i];
importPdfMeasure(m, omrSystem);
}
if(score->lastMeasure()){
LayoutBreak* b = new LayoutBreak(score);
b->setLayoutBreakType(LayoutBreak::Type::LINE);
score->lastMeasure()->add(b);
}
return tick;
}
//---------------------------------------------------------
// importPdfPage
//---------------------------------------------------------
void OmrState::importPdfPage(OmrPage* omrPage, int top)
{
TDuration d(TDuration::DurationType::V_MEASURE);
int tick = 0;
int nsystems = omrPage->systems().size();
if(nsystems == 0) return;
//int n = nsystems == 0 ? 1 : nsystems;
for (int k = 0; k < nsystems; ++k) {
int numMeasures = 1;
if (k < nsystems) {
tick = importPdfSystem(omrPage->system(k));
}
else {
Measure* measure;
for (int i = 0; i < numMeasures; ++i) {
measure = new Measure(score);
measure->setTick(tick);
Rest* rest = new Rest(score, d);
rest->setDuration(Fraction(4,4));
rest->setTrack(0);
Segment* s = measure->getSegment(Segment::Type::ChordRest, tick);
s->add(rest);
rest = new Rest(score, d);
rest->setDuration(Fraction(4,4));
rest->setTrack(4);
s->add(rest);
score->measures()->add(measure);
tick += MScore::division * 4;
}
if (k < (nsystems-1)) {
LayoutBreak* b = new LayoutBreak(score);
b->setLayoutBreakType(LayoutBreak::Type::LINE);
measure->add(b);
}
}
}
Measure* measure = score->lastMeasure();
if (measure) {
LayoutBreak* b = new LayoutBreak(score);
b->setLayoutBreakType(LayoutBreak::Type::PAGE);
measure->add(b);
}
measure = score->firstMeasure();
if (measure) {
MStaff *ms = measure->mstaves()[0];
if (!ms->_vspacerUp){
Spacer* spacer = new Spacer(score);
spacer->setSpacerType(SpacerType::UP);
spacer->setTrack(0);
measure->add(spacer);
}
Spacer* sp = ms->_vspacerUp;
sp->layout();
sp->setPos(sp->rxpos(), top);
}
}
//---------------------------------------------------------
// importPdf
//---------------------------------------------------------
Score::FileError importPdf(MasterScore* score, const QString& path)
{
Omr* omr = new Omr(path, score);
if (!omr->readPdf()) {
delete omr;
return Score::FileError::FILE_BAD_FORMAT;
}
score->setOmr(omr);
qreal sp = omr->spatiumMM();
if (sp == 0.0)
sp = 1.5;
score->setSpatium(sp * DPMM);
score->style()->set(StyleIdx::lastSystemFillLimit, 0.0);
score->style()->set(StyleIdx::staffLowerBorder, 0.0);
score->style()->set(StyleIdx::measureSpacing, 1.0);
PageFormat pF;
pF.copy(*score->pageFormat());
pF.setEvenLeftMargin(5.0 * DPMM / DPI);
pF.setEvenTopMargin(0);
pF.setEvenBottomMargin(0);
pF.setOddLeftMargin(5.0 * DPMM / DPI);
pF.setOddTopMargin(0);
pF.setOddBottomMargin(0);
score->setPageFormat(pF);
score->style()->set(StyleIdx::minSystemDistance, Spatium(omr->systemDistance()));
score->style()->set(StyleIdx::maxSystemDistance, Spatium(omr->systemDistance()));
score->style()->set(StyleIdx::akkoladeDistance, Spatium(omr->staffDistance()));
//incomplete implementation for musescore skeletion creation
Part* part = new Part(score);
OmrPage* omrPage = omr->pages().front();
//may need to identify maximal number of staff per system and then high some of those
if (omrPage->systems().size() > 0) {
for (int i = 0; i < omrPage->systems().front().staves().size(); i++) {
Staff* staff = new Staff(score);
staff->setPart(part);
part->insertStaff(staff, -1);
score->staves().append(staff);
}
}
score->appendPart(part);
OmrState state;
state.score = score;
foreach (OmrPage* omrPage, omr->pages()) {
OmrStaff staff = omrPage->systems().last().staves().first();
int top = staff.top();
state.importPdfPage(omrPage, top);
}
//---create bracket
//score->staff(0)->setBracket(0, BracketType::BRACE);
//score->staff(0)->setBracketSpan(0, 2);
score->setShowOmr(true);
omr->page(0)->readHeader(score);
score->rebuildMidiMapping();
return Score::FileError::FILE_NO_ERROR;
}
}