TagCounter.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#include <string.h>
 
#if __DCL_WINDOWS
#include <windows.h>
#endif
 
#include <dcl/String.h>
#include <dcl/Charset.h>
#include <dcl/File.h>
#include <dcl/Dir.h>
#include <dcl/Writer.h>
 
#include "APEv2.h"
 
#include "main.h"
#include "TagCounter.h"
 
#define __TRACE_THIS    0
#if __TRACE_THIS
#define __DCL_TRACE1_N  __DCL_TRACE1
#define __DCL_TRACE2_N  __DCL_TRACE2
#define __DCL_TRACE3_N  __DCL_TRACE3
#else
#define __DCL_TRACE1_N(fmt, arg)
#define __DCL_TRACE2_N(fmt, arg1, arg2)
#define __DCL_TRACE3_N(fmt, arg1, arg2, arg3)
#endif
 
__DCL_BEGIN_NAMESPACE
 
#if __DCL_DEBUG
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("media/TagCounter.cpp");
#endif
 
void TagCounter::read(const String& _filename, TagCounter& _counter)
{
    File file(_filename);
    char buf[160];   // 128 + 32
 
    int idv2 = 0;
    int ape = 0;
    int idv1 = 0;
 
    // 파일의 시작에서 ID3v2와 APEv2를 검사한다.
    // "ID3" "APETAGEX"
    size_t n = file.read(buf, 32);
    if (n == 32) {
        if (memcmp(buf, "ID3", 3) == 0) {
            idv2 = buf[3];
        }
        else if (memcmp(buf, "APETAGEX", 8) == 0) {
            ape = APEv2::word(&buf[8]) / 1000;
        }
    }
 
    // 파일의 끝에서 APEv2와 ID3v1을 검사한다.
    file.seek(-160, File::END);
    n = file.read(buf, 160);
    if (n == 160) {
        if (ape == 0) {
            if (memcmp(buf, "APETAGEX", 8) == 0) {
                ape = APEv2::word(&buf[8]) / 1000;
            }
            else if (memcmp(&buf[128], "APETAGEX", 8) == 0) {
                ape = APEv2::word(&buf[128 + 8]) / 1000;
            }
        }
 
        if (memcmp(&buf[32], "TAG", 3) == 0) {
            idv1 = 1;
        }
    }
 
    switch (idv2) {
    case 2:
        _counter.ID3v22++;
        break;
    case 3 :
        _counter.ID3v23++;
        break;
    case 4 :
        _counter.ID3v24++;
    }
 
    switch (ape) {
    case 1:
        _counter.APEv1++;
        break;
    case 2:
        _counter.APEv2++;
    }
 
    if (idv1 == 1) {
        _counter.ID3v1++;
    }
}
 
void TagCounter::readDir(const String& _path, TagCounter& _counter,
    const MainArguments& _args)
{
    Dir dir(_path);
    Dir::Entry entry;
 
    while (dir.read(entry)) {
        String name = entry.name();
        if (entry.isDir()) {
            if (name.compare(L"..", name.length()) != 0) {
                String path = dir.path() + name;
                TagCounter counter;
                TagCounter::readDir(path, counter, _args);
                _counter += counter;
                if (_args.verbose()) {
                    _args.output() << L"[" << path << L"] " << counter.toString() << endl;
                }
            }
        }
        else {
            if (name.toLowerCase().endsWith(L".mp3")) {
                try {
                    TagCounter counter;
                    TagCounter::read(dir.path() + name, counter);
                    _counter += counter;
                    if (_args.verbose()) {
                        _args.output() << L"[  " << name << L"] " << counter.toString() << endl;
                    }
                }
                catch (Exception* e) {
                    _counter.error++;
                    _args.errout() << e->toStringAll() << endl;
                    e->destroy();
                }
            }
        }
    }
}
 
__DCL_END_NAMESPACE