--- a/src/primitives.cpp Sun Jul 07 15:08:38 2013 +0300 +++ b/src/primitives.cpp Sun Jul 07 16:46:30 2013 +0300 @@ -1,17 +1,17 @@ /* * LDForge: LDraw parts authoring CAD * Copyright (C) 2013 Santeri Piippo - * + * * 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 3 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, see <http://www.gnu.org/licenses/>. */ @@ -33,129 +33,144 @@ static const str g_Other = QObject::tr( "Other" ); -static void populateCategories (); -static void loadPrimitiveCatgories (); +static void populateCategories(); +static void loadPrimitiveCatgories(); // ============================================================================= // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * // ============================================================================= -void loadPrimitives () { - print ("Loading primitives...\n"); +void loadPrimitives() +{ + print( "Loading primitives...\n" ); - loadPrimitiveCatgories (); + loadPrimitiveCatgories(); // Try to load prims.cfg - File conf (config::dirpath () + "prims.cfg", File::Read); - if (!conf) { + File conf( config::dirpath() + "prims.cfg", File::Read ); + + if( !conf ) + { // No prims.cfg, build it - PrimitiveLister::start (); - } else { - for (str line : conf) { - int space = line.indexOf (" "); - if (space == -1) + PrimitiveLister::start(); + } + else + { + for( str line : conf ) + { + int space = line.indexOf( " " ); + + if( space == -1 ) continue; Primitive info; - info.name = line.left (space); - info.title = line.mid (space + 1); + info.name = line.left( space ); + info.title = line.mid( space + 1 ); g_primitives << info; } - populateCategories (); + populateCategories(); } } // ============================================================================= // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * // ============================================================================= -void recursiveGetFilenames (QDir dir, vector<str>& fnames) { - QFileInfoList flist = dir.entryInfoList (); - for (const QFileInfo& info : flist) { - if (info.fileName () == "." || info.fileName () == "..") +void recursiveGetFilenames( QDir dir, vector<str>& fnames ) +{ + QFileInfoList flist = dir.entryInfoList(); + + for( const QFileInfo & info : flist ) + { + if( info.fileName() == "." || info.fileName() == ".." ) continue; // skip . and .. - if (info.isDir ()) - recursiveGetFilenames (QDir (info.absoluteFilePath ()), fnames); + if( info.isDir() ) + recursiveGetFilenames( QDir( info.absoluteFilePath() ), fnames ); else - fnames << info.absoluteFilePath (); + fnames << info.absoluteFilePath(); } } // ============================================================================= // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * // ============================================================================= -void PrimitiveLister::work () { +void PrimitiveLister::work() +{ g_activePrimLister = this; - m_prims.clear (); - - QDir dir (LDPaths::prims ()); - assert (dir.exists ()); + m_prims.clear(); - ulong baselen = dir.absolutePath ().length (); - + QDir dir( LDPaths::prims() ); + ulong baselen = dir.absolutePath().length(); + ulong i = 0; vector<str> fnames; - recursiveGetFilenames (dir, fnames); - emit starting (fnames.size ()); - ulong i = 0; - for (str fname : fnames) { - File f (fname, File::Read); + assert( dir.exists() ); + recursiveGetFilenames( dir, fnames ); + emit starting( fnames.size() ); + + for( str fname : fnames ) + { + File f( fname, File::Read ); Primitive info; - info.name = fname.mid (baselen + 1); // make full path relative - info.name.replace ('/', '\\'); // use DOS backslashes, they're expected + info.name = fname.mid( baselen + 1 ); // make full path relative + info.name.replace( '/', '\\' ); // use DOS backslashes, they're expected + info.cat = null; - if (!f.readLine (info.title)) + if( !f.readLine( info.title )) info.title = ""; - info.title = info.title.simplified (); - info.cat = null; + info.title = info.title.simplified(); - if (info.title[0] == '0') { - info.title.remove (0, 1); // remove 0 - info.title = info.title.simplified (); + if( info.title[0] == '0' ) + { + info.title.remove( 0, 1 ); // remove 0 + info.title = info.title.simplified(); } m_prims << info; - emit update (++i); + emit update( ++i ); } // Save to a config file - File conf (config::dirpath () + "prims.cfg", File::Write); - for (Primitive& info : m_prims) - fprint (conf, "%1 %2\n", info.name, info.title); + File conf( config::dirpath() + "prims.cfg", File::Write ); - conf.close (); + for( Primitive & info : m_prims ) + fprint( conf, "%1 %2\n", info.name, info.title ); + + conf.close(); g_primListerMutex = true; g_primitives = m_prims; - populateCategories (); + populateCategories(); g_primListerMutex = false; g_activePrimLister = null; - emit workDone (); + emit workDone(); } // ============================================================================= // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * // ============================================================================= -void PrimitiveLister::start () { - if (g_activePrimLister) +void PrimitiveLister::start() +{ + if( g_activePrimLister ) return; PrimitiveLister* lister = new PrimitiveLister; QThread* listerThread = new QThread; - lister->moveToThread (listerThread); - connect (lister, SIGNAL (starting (ulong)), g_win, SLOT (primitiveLoaderStart (ulong))); - connect (lister, SIGNAL (update (ulong)), g_win, SLOT (primitiveLoaderUpdate (ulong))); - connect (lister, SIGNAL (workDone ()), g_win, SLOT (primitiveLoaderEnd ())); - connect (listerThread, SIGNAL (started ()), lister, SLOT (work ())); - connect (listerThread, SIGNAL (finished ()), lister, SLOT (deleteLater ())); - listerThread->start (); + lister->moveToThread( listerThread ); + connect( lister, SIGNAL( starting( ulong )), g_win, SLOT( primitiveLoaderStart( ulong )) ); + connect( lister, SIGNAL( update( ulong )), g_win, SLOT( primitiveLoaderUpdate( ulong )) ); + connect( lister, SIGNAL( workDone() ), g_win, SLOT( primitiveLoaderEnd() )); + connect( listerThread, SIGNAL( started() ), lister, SLOT( work() )); + connect( listerThread, SIGNAL( finished() ), lister, SLOT( deleteLater() )); + listerThread->start(); } -static PrimitiveCategory* findCategory (str name) { - for (PrimitiveCategory& cat : g_PrimitiveCategories) - if (cat.name () == name) +static PrimitiveCategory* findCategory( str name ) +{ + for( PrimitiveCategory & cat : g_PrimitiveCategories ) + if( cat.name() == name ) return &cat; return null; @@ -164,52 +179,59 @@ // ============================================================================= // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * // ============================================================================= -static void populateCategories () { - for (PrimitiveCategory& cat : g_PrimitiveCategories) - cat.prims.clear (); +static void populateCategories() +{ + for( PrimitiveCategory & cat : g_PrimitiveCategories ) + cat.prims.clear(); PrimitiveCategory* unmatched = findCategory( g_Other ); - if (!unmatched) { + if( !unmatched ) + { // Shouldn't happen.. but catch it anyway. PrimitiveCategory cat; cat.setName( g_Other ); - unmatched = &(g_PrimitiveCategories << cat); + unmatched = &( g_PrimitiveCategories << cat ); } - for (Primitive& prim : g_primitives) { + for( Primitive & prim : g_primitives ) + { bool matched = false; // Go over the categories and their regexes, if and when there's a match, // the primitive's category is set to the category the regex beloings to. - for (PrimitiveCategory& cat : g_PrimitiveCategories) { - for (PrimitiveCategory::RegexEntry& entry : cat.regexes) { - switch (entry.type) { + for( PrimitiveCategory & cat : g_PrimitiveCategories ) + { + for( PrimitiveCategory::RegexEntry & entry : cat.regexes ) + { + switch( entry.type ) + { case PrimitiveCategory::Filename: // f-regex, check against filename - matched = entry.regex.exactMatch (prim.name); + matched = entry.regex.exactMatch( prim.name ); break; - + case PrimitiveCategory::Title: // t-regex, check against title - matched = entry.regex.exactMatch (prim.title); + matched = entry.regex.exactMatch( prim.title ); break; } - if (matched) { + if( matched ) + { prim.cat = &cat; break; } } // Drop out if a category was decided on. - if (prim.cat) + if( prim.cat ) break; } // If there was a match, add the primitive to the category. // Otherwise, add it to the list of unmatched primitives. - if (prim.cat) + if( prim.cat ) prim.cat->prims << prim; else unmatched->prims << prim; @@ -219,51 +241,58 @@ // ============================================================================= // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * // ============================================================================= -static void loadPrimitiveCatgories () { - g_PrimitiveCategories.clear (); - - File f (config::dirpath () + "primregexps.cfg", File::Read); +static void loadPrimitiveCatgories() +{ + g_PrimitiveCategories.clear(); - if (!f) - f.open (":/data/primitive-categories.cfg", File::Read); + File f( config::dirpath() + "primregexps.cfg", File::Read ); - if (!f) + if( !f ) + f.open( ":/data/primitive-categories.cfg", File::Read ); + + if( !f ) critical( QObject::tr( "Failed to open primitive categories!" )); - if (f) { + if( f ) + { PrimitiveCategory cat; - - for (str line : f) { + + for( str line : f ) + { int colon; - if (line.length () == 0 || line[0] == '#') + if( line.length() == 0 || line[0] == '#' ) continue; - if ((colon = line.indexOf (":")) == -1) { - if (cat.regexes.size () > 0) + if( ( colon = line.indexOf( ":" )) == -1 ) + { + if( cat.regexes.size() > 0 ) g_PrimitiveCategories << cat; - cat.regexes.clear (); - cat.prims.clear (); - cat.setName (line); - } else { - str cmd = line.left (colon); + cat.regexes.clear(); + cat.prims.clear(); + cat.setName( line ); + } + else + { + str cmd = line.left( colon ); PrimitiveCategory::Type type = PrimitiveCategory::Filename; - if (cmd == "f") { + + if( cmd == "f" ) type = PrimitiveCategory::Filename; - } elif (cmd == "t") { + elif( cmd == "t" ) type = PrimitiveCategory::Title; - } else + else continue; - QRegExp regex (line.mid (colon + 1)); + QRegExp regex( line.mid( colon + 1 )); PrimitiveCategory::RegexEntry entry = { regex, type }; cat.regexes << entry; } } - if (cat.regexes.size () > 0) + if( cat.regexes.size() > 0 ) g_PrimitiveCategories << cat; } @@ -285,106 +314,108 @@ for( int i = 0; i < segs; ++i ) { - double x0 = cos(( i * 2 * pi ) / divs ), - x1 = cos((( i + 1 ) * 2 * pi) / divs ), - z0 = sin(( i * 2 * pi ) / divs ), - z1 = sin((( i + 1 ) * 2 * pi ) / divs ); + double x0 = cos( ( i * 2 * pi ) / divs ), + x1 = cos( ( ( i + 1 ) * 2 * pi ) / divs ), + z0 = sin( ( i * 2 * pi ) / divs ), + z1 = sin( ( ( i + 1 ) * 2 * pi ) / divs ); LDObject* obj = null; switch( type ) { case Circle: - { - vertex v0( x0, 0.0f, z0 ), - v1( x1, 0.0f, z1 ); - - LDLine* line = new LDLine; - line->setVertex( 0, v0 ); - line->setVertex( 1, v1 ); - line->setColor( edgecolor ); - obj = line; - } - break; + { + vertex v0( x0, 0.0f, z0 ), + v1( x1, 0.0f, z1 ); + + LDLine* line = new LDLine; + line->setVertex( 0, v0 ); + line->setVertex( 1, v1 ); + line->setColor( edgecolor ); + obj = line; + } + break; case Cylinder: case Ring: case Cone: + { + double x2, x3, z2, z3; + double y0, y1, y2, y3; + + if( type == Cylinder ) { - double x2, x3, z2, z3; - double y0, y1, y2, y3; + x2 = x1; + x3 = x0; + z2 = z1; + z3 = z0; - if( type == Cylinder ) + y0 = y1 = 0.0f; + y2 = y3 = 1.0f; + } + else + { + x2 = x1 * ( num + 1 ); + x3 = x0 * ( num + 1 ); + z2 = z1 * ( num + 1 ); + z3 = z0 * ( num + 1 ); + + x0 *= num; + x1 *= num; + z0 *= num; + z1 *= num; + + if( type == Ring ) + y0 = y1 = y2 = y3 = 0.0f; + else { - x2 = x1; - x3 = x0; - z2 = z1; - z3 = z0; - - y0 = y1 = 0.0f; - y2 = y3 = 1.0f; - } else { - x2 = x1 * (num + 1); - x3 = x0 * (num + 1); - z2 = z1 * (num + 1); - z3 = z0 * (num + 1); - - x0 *= num; - x1 *= num; - z0 *= num; - z1 *= num; - - if( type == Ring ) - y0 = y1 = y2 = y3 = 0.0f; - else - { - y0 = y1 = 1.0f; - y2 = y3 = 0.0f; - } + y0 = y1 = 1.0f; + y2 = y3 = 0.0f; } - - vertex v0( x0, y0, z0 ), - v1( x1, y1, z1 ), - v2( x2, y2, z2 ), - v3( x3, y3, z3 ); - - LDQuad* quad = new LDQuad; - quad->setColor( maincolor ); - quad->setVertex( 0, v0 ); - quad->setVertex( 1, v1 ); - quad->setVertex( 2, v2 ); - quad->setVertex( 3, v3 ); - obj = quad; } - break; + + vertex v0( x0, y0, z0 ), + v1( x1, y1, z1 ), + v2( x2, y2, z2 ), + v3( x3, y3, z3 ); + + LDQuad* quad = new LDQuad; + quad->setColor( maincolor ); + quad->setVertex( 0, v0 ); + quad->setVertex( 1, v1 ); + quad->setVertex( 2, v2 ); + quad->setVertex( 3, v3 ); + obj = quad; + } + break; case Disc: case DiscNeg: + { + double x2, z2; + + if( type == Disc ) + x2 = z2 = 0.0f; + else { - double x2, z2; - - if( type == Disc ) - x2 = z2 = 0.0f; - else - { - x2 = ( x0 >= 0.0f ) ? 1.0f : -1.0f; - z2 = ( z0 >= 0.0f ) ? 1.0f : -1.0f; - } - - vertex v0( x0, 0.0f, z0 ), - v1( x1, 0.0f, z1 ), - v2( x2, 0.0f, z2 ); - - // Disc negatives need to go the other way around, otherwise - // they'll end up upside-down. - LDTriangle* seg = new LDTriangle; - seg->setColor( maincolor ); - seg->setVertex( type == Disc ? 0 : 2, v0 ); - seg->setVertex( 1, v1 ); - seg->setVertex( type == Disc ? 2 : 0, v2 ); - obj = seg; + x2 = ( x0 >= 0.0f ) ? 1.0f : -1.0f; + z2 = ( z0 >= 0.0f ) ? 1.0f : -1.0f; } - break; + + vertex v0( x0, 0.0f, z0 ), + v1( x1, 0.0f, z1 ), + v2( x2, 0.0f, z2 ); + + // Disc negatives need to go the other way around, otherwise + // they'll end up upside-down. + LDTriangle* seg = new LDTriangle; + seg->setColor( maincolor ); + seg->setVertex( type == Disc ? 0 : 2, v0 ); + seg->setVertex( 1, v1 ); + seg->setVertex( type == Disc ? 2 : 0, v2 ); + obj = seg; + } + break; default: break; @@ -405,10 +436,11 @@ type == Disc ? "Disc" : type == DiscNeg ? "Disc Negative" : type == Ring ? "Ring" : - "Cone"; + "Cone"; } -static const str g_radialNameRoots[] = { +static const str g_radialNameRoots[] = +{ "edge", "cyli", "disc", @@ -437,14 +469,14 @@ } // Compose some general information: prefix, fraction, root, ring number - str prefix = (divs == lores) ? "" : fmt( "%1/", divs ); + str prefix = ( divs == lores ) ? "" : fmt( "%1/", divs ); str frac = fmt( "%1-%2", numer, denom ); str root = g_radialNameRoots[type]; - str numstr = ( type == Ring || type == Cone ) ? fmt ( "%1", num ) : ""; + str numstr = ( type == Ring || type == Cone ) ? fmt( "%1", num ) : ""; // Truncate the root if necessary (7-16rin4.dat for instance). // However, always keep the root at least 2 characters. - int extra = ( frac.length() + numstr.length() + root.length() ) - 8; + int extra = ( frac.length() + numstr.length() + root.length()) - 8; root.chop( min<short>( max<short>( extra, 0 ), 2 )); // Stick them all together and return the result. @@ -471,11 +503,10 @@ ui.rb_cylinder->isChecked() ? Cylinder : ui.rb_disc->isChecked() ? Disc : ui.rb_ndisc->isChecked() ? DiscNeg : - ui.rb_ring->isChecked() ? Ring : - Cone; + ui.rb_ring->isChecked() ? Ring : Cone; // Make the description - str frac = ftoa(( (float) segs ) / divs ); + str frac = ftoa( ( ( float ) segs ) / divs ); str name = radialFileName( type, segs, divs, num ); str descr; @@ -483,7 +514,7 @@ if( frac.indexOf( "." ) == -1 ) frac += ".0"; - if (type == Ring || type == Cone) + if( type == Ring || type == Cone ) descr = fmt( "%1 %2 x %3", primitiveTypeName( type ), num, frac ); else descr = fmt( "%1 %2", primitiveTypeName( type ), frac ); @@ -503,4 +534,4 @@ g_win->save( f, false ); delete f; -} \ No newline at end of file +}