xbmc/utils/MusicInfoScraper.h

   1 #pragma once
   2
   3 /*
   4  *      Copyright (C) 2005-2008 Team XBMC
   5  *      http://www.xbmc.org
   6  *
   7  *  This Program is free software; you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License as published by
   9  *  the Free Software Foundation; either version 2, or (at your option)
  10  *  any later version.
  11  *
  12  *  This Program is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15  *  GNU General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU General Public License
  18  *  along with XBMC; see the file COPYING.  If not, write to
  19  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  20  *  http://www.gnu.org/copyleft/gpl.html
  21  *
  22  */
  23
  24 #include "MusicAlbumInfo.h"
  25 #include "MusicArtistInfo.h"
  26 #include "addons/Scraper.h"
  27 #include "Thread.h"
  28 #include "FileSystem/FileCurl.h"
  29
  30 namespace XFILE { class CFileCurl; }
  31
  32 namespace MUSIC_GRABBER
  33 {
  34 class CMusicInfoScraper : public CThread
  35 {
  36 public:
  37   CMusicInfoScraper(const ADDON::ScraperPtr &scraper);
  38   virtual ~CMusicInfoScraper(void);
  39   void FindAlbuminfo(const CStdString& strAlbum, const CStdString& strArtist = "");
  40   void LoadAlbuminfo(int iAlbum);
  41   void FindArtistinfo(const CStdString& strArtist);
  42   void LoadArtistinfo(int iArtist);
  43   bool Completed();
  44   bool Successfull();
  45   void Cancel();
  46   bool IsCanceled();
  47   int GetAlbumCount() const;
  48   int GetArtistCount() const;
  49   CMusicAlbumInfo& GetAlbum(int iAlbum);
  50   CMusicArtistInfo& GetArtist(int iArtist);
  51   std::vector<CMusicArtistInfo>& GetArtists()
  52   {
  53     return m_vecArtists;
  54   }
  55   std::vector<CMusicAlbumInfo>& GetAlbums()
  56   {
  57     return m_vecAlbums;
  58   }
  59   void SetScraperInfo(const ADDON::ScraperPtr& scraper)
  60   {
  61     m_scraper = scraper;
  62   }
  63   /*!
  64    \brief Checks whether we have a valid scraper.  If not, we try the fallbackScraper
  65    First tests the current scraper for validity by loading it.  If it is not valid we
  66    attempt to load the fallback scraper.  If this is also invalid we return false.
  67    \param fallbackScraper name of scraper to use as a fallback
  68    \return true if we have a valid scraper (or the default is valid).
  69    */
  70   bool CheckValidOrFallback(const CStdString &fallbackScraper);
  71 protected:
  72   void FindAlbuminfo();
  73   void LoadAlbuminfo();
  74   void FindArtistinfo();
  75   void LoadArtistinfo();
  76   virtual void OnStartup();
  77   virtual void Process();
  78   std::vector<CMusicAlbumInfo> m_vecAlbums;
  79   std::vector<CMusicArtistInfo> m_vecArtists;
  80   CStdString m_strAlbum;
  81   CStdString m_strArtist;
  82   int m_iAlbum;
  83   int m_iArtist;
  84   bool m_bSuccessfull;
  85   bool m_bCanceled;
  86   XFILE::CFileCurl m_http;
  87   ADDON::ScraperPtr m_scraper;
  88 };
  89
  90 }