Class: Aims::OutputParser

Inherits:
Object
  • Object
show all
Defined in:
lib/aims/output.rb

Overview

Parse an AIMS output file and generate an AimsOutput object Invoke with

output = Aims::OutputParser.parse(filename)

Class Method Summary collapse

Class Method Details

.parse(filename) ⇒ Object 



390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
 
# File 'lib/aims/output.rb', line 390

def OutputParser.parse(filename)
  
  n_atoms = 0
  vectors = []
  retval = AimsOutput.new
  retval.original_file = filename
  
  File.open(filename, 'r') do |f|
    f.each_line{|line|
      case line
        when /Found k-point grid:/
          retval.k_grid = line.split(":")[1].strip
          
        when /Computational steps:/
          retval.computational_steps = OutputParser.parse_computational_steps(f)
        
        when /Detailed time accounting/
          retval.timings = OutputParser.parse_detailed_time_accounting(f)
          
        when /Begin self-consistency loop/
          retval.geometry_step.sc_iterations << SCIteration.new
          
        when /Begin self-consistency iteration/
          retval.geometry_step.sc_iterations << SCIteration.new
                        
        when /End self-consistency iteration/, /End scf initialization - timings/
          retval.sc_iteration.timings = OutputParser.parse_sc_timings(f)

        when /Change of charge density/
          retval.sc_iteration.d_rho = line.split(' ')[6].to_f
      
        when /Change of sum of eigenvalues/
          retval.sc_iteration.d_eev = line.split(' ')[7].to_f
          
        when /Change of total energy/
          retval.sc_iteration.d_etot = line.split(' ')[6].to_f
        
        when /\|\ Total energy corrected/
          retval.geometry_step.total_corrected_energy = line.split(' ')[5].to_f
        
        when /\|\ Total energy uncorrected/
          retval.geometry_step.total_energy = line.split(' ')[5].to_f
          
        when /\|\ Number\ of\ atoms/
            n_atoms = line.split(' ')[5].to_i
            retval.n_atoms = n_atoms
            
        when  /\|\ Chemical potential/
          retval.geometry_step.chemical_potential = line.split(' ')[8].to_f
            
        when /Input\ geometry\:/
          line = f.readline
          if line=~/\|\ Unit\ cell\:/
              3.times { 
                line = f.readline
                fields = line.split(' ')
                vectors << Vector[fields[1].to_f, fields[2].to_f, fields[3].to_f]
              }
          end
          2.times {f.readline} 
          retval.geometry_steps << GeometryStep.new 
          retval.geometry_step.step_num = 0
          retval.geometry_step.geometry = OutputParser.parse_input_geometry(f, n_atoms)
          retval.geometry_step.geometry.lattice_vectors = vectors

        when /\ Updated\ atomic\ structure\:/
          last_step_num = retval.geometry_step.step_num
          retval.geometry_steps << GeometryStep.new 
          retval.geometry_step.step_num = last_step_num + 1
          retval.geometry_step.geometry = OutputParser.parse_updated_geometry(f, n_atoms)
        #  retval.geometry_step.geometry.lattice_vectors = vectors
          
        when /\ Final\ atomic\ structure\:/
          retval.geometry_step.geometry = OutputParser.parse_updated_geometry(f, n_atoms)
         # retval.geometry_step.geometry.lattice_vectors = vectors
        when /\  Total\ atomic\ forces/
          line = f.readline
          until line =~ /---/
              fields = line.split(' ')
              retval.geometry_step.forces << Vector[fields[2].to_f, fields[3].to_f, fields[4].to_f]
              line = f.readline
          end
        when /Present geometry is converged./
          retval.geometry_converged = true
      end
    }
  end

  return retval
  
end

.parse_atom(line) ⇒ Object 



304
305
306
307
308
309
310
 
# File 'lib/aims/output.rb', line 304

def OutputParser.parse_atom(line)
  fields = line.split(' ')
  a = Atom.new
  a.x, a.y, a.z = fields[1].to_f, fields[2].to_f, fields[3].to_f
  a.species = fields[4]
  a
end

.parse_atom_frac(line) ⇒ Object 



300
301
302
 
# File 'lib/aims/output.rb', line 300

def OutputParser.parse_atom_frac(line)
  nil
end

.parse_computational_steps(io) ⇒ Object 



378
379
380
381
382
383
384
385
386
387
388
 
# File 'lib/aims/output.rb', line 378

def OutputParser.parse_computational_steps(io)
  line = io.readline
  steps = []
  desc, value = line.split(":")
  until value.nil?
    steps << {:description => desc.sub("|", " ").strip, :value => value.to_f}
    line = io.readline
    desc, value = line.split(":")
  end
  steps 
end

.parse_detailed_time_accounting(io) ⇒ Object 



361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
 
# File 'lib/aims/output.rb', line 361

def OutputParser.parse_detailed_time_accounting(io)
  line = io.readline
  timings = Timings.new
  desc, times = line.split(":")
  until times.nil?
    fields = times.split(" ")
    description = desc.sub("|", "").strip
    cpu = fields[0].to_f
    wall = fields[2].to_f
    timings.add_cpu_time(description, cpu)
    timings.add_wall_time(description, wall)
    line = io.readline
    desc, times = line.split(":")
  end
  timings 
end

.parse_input_geometry(io, n_atoms) ⇒ Object 



288
289
290
291
292
293
294
295
296
297
298
 
# File 'lib/aims/output.rb', line 288

def OutputParser.parse_input_geometry(io, n_atoms)
  atoms = []
  n_atoms.times do 
    fields = io.readline.split(' ')
    a = Atom.new
    a.x, a.y, a.z = fields[4].to_f, fields[5].to_f, fields[6].to_f
    a.species = fields[3]
    atoms << a
  end
  Geometry.new(atoms, nil, :dont_make_bonds)
end

.parse_lattice_vector(line) ⇒ Object 



312
313
314
315
 
# File 'lib/aims/output.rb', line 312

def OutputParser.parse_lattice_vector(line)
  fields = line.split(' ')
  [fields[1].to_f, fields[2].to_f, fields[3].to_f]      
end

.parse_sc_timings(io) ⇒ Object 



345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
 
# File 'lib/aims/output.rb', line 345

def OutputParser.parse_sc_timings(io)
  line = io.readline
  timings = Timings.new
  until line =~ /---/
    desc, times = line.split(":")
    fields = times.split(" ")
    description = desc.sub("|", "").strip
    cpu = fields[0].to_f
    wall = fields[2].to_f
    timings.add_cpu_time(description, cpu)
    timings.add_wall_time(description, wall)
    line = io.readline
  end
  timings
end

.parse_updated_geometry(io, n_atoms) ⇒ Object 



317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
 
# File 'lib/aims/output.rb', line 317

def OutputParser.parse_updated_geometry(io, n_atoms)
  io.readline
  
  vectors = []
  atoms = []
  continue = TRUE
  
  begin
    line = io.readline
    case line
    when /lattice_vector/
      vectors << OutputParser.parse_lattice_vector(line)
    when /atom\b/
      atoms << OutputParser.parse_atom(line)
    when /atom_frac/
      OutputParser.parse_atom_frac(line)
    when /^\s*$/
      # do nothing
    else 
      continue = FALSE
    end
  end while continue 
  
  vectors = nil if vectors.empty?
  
  Geometry.new(atoms, vectors, :dont_make_bonds)
end