예제 #1
0
 public ThreadMaster(ArtemisEngine engine, GameState gameState, TranspositionTable transpositionTable, IEngineConfig config)
 {
     this.gameState          = gameState;
     this.transpositionTable = transpositionTable;
     this.config             = config;
     threadsNum    = Environment.ProcessorCount;
     searchThreads = new SearchThread[threadsNum];
     for (int t = 0; t < threadsNum; t++)
     {
         GameState threadState = new GameState(gameState.PregeneratedAttacks, gameState.ZobristHashUtils);
         searchThreads[t] = new SearchThread(engine, this, threadState, transpositionTable, searchedNodes, config);
     }
 }
예제 #2
0
        public async Task <PVList> Search(CancellationToken ct)
        {
            Stopwatch watch = new Stopwatch();

            watch.Start();
            List <Task <PVList> > tasks = new List <Task <PVList> >();
            int currentThreadsNum       = config.Multithreading ? threadsNum : 1;

            for (int t = 0; t < currentThreadsNum; t++)
            {
                SearchThread thread = searchThreads[t];
                thread.LoadState(gameState);
                Task <PVList> engineTask = Task.Run(() => thread.Search(1, ct));
                tasks.Add(engineTask);
            }

            PVList[] foundPV = await Task.WhenAll(tasks);

            PVList       bestPV     = new PVList();
            SearchThread bestThread = null;

            for (int i = 0; i < foundPV.Length; i++)
            {
                if (foundPV[i] != null && foundPV[i].First != null && foundPV[i] > bestPV)
                {
                    bestPV     = foundPV[i];
                    bestThread = searchThreads[i];
                }
            }
            watch.Stop();

            bestThread.SearchStats.Time = watch.ElapsedMilliseconds;
            string statsJson = JsonConvert.SerializeObject(bestThread.SearchStats);

            Console.WriteLine(statsJson);
            Console.WriteLine(bestPV);

            searchedNodes.Clear();
            return(bestPV);
        }