mirror of
https://github.com/triqs/dft_tools
synced 2024-12-23 21:03:45 +01:00
11d394fd5b
* moved the plovasp C++ code to c++/triqs_dft_tools/converters/vasp * added global header triqs_dft_tools/triqs_dft_tools.hpp * python dir based on single cmakelist file * registered C++ tests for plovasp * corrected imports for py3 tests for plovasp * corrected block order in sigma_from_file and srvo3_Gloc * exchanged ref files for sigma_from_file, srvo3_Gloc, SrVO3.ref.h5 * moved vasp converter bash scripts from dir shells to bin dir
124 lines
4.4 KiB
Python
124 lines
4.4 KiB
Python
import tempfile
|
|
# -*- coding: utf-8 -*-
|
|
# seems to be executed at the level of the conf.py
|
|
# so we need to link the lib at that place...
|
|
"""
|
|
"""
|
|
import os
|
|
import codecs
|
|
from os import path
|
|
from subprocess import Popen,PIPE
|
|
from docutils import nodes
|
|
from docutils.parsers.rst import Directive
|
|
from docutils.parsers.rst import directives
|
|
from sphinx.errors import SphinxError
|
|
|
|
class TriqsExampleError(SphinxError):
|
|
category = 'triqs_example error'
|
|
|
|
class TriqsExampleRun:
|
|
#here = os.path.abspath(__file__)
|
|
#pycon = os.path.join(os.path.dirname(here),'pycon.py')
|
|
config = dict(
|
|
)
|
|
@classmethod
|
|
def builder_init(cls,app):
|
|
#cls.config.update(app.builder.config.autorun_languages)
|
|
#cls.config.update(app.builder.config.autocompile_opts)
|
|
pass
|
|
|
|
class TriqsExample(Directive):
|
|
has_content = True
|
|
required_arguments = 1
|
|
optional_arguments = 0
|
|
final_argument_whitespace = False
|
|
option_spec = {
|
|
'linenos': directives.flag,
|
|
}
|
|
|
|
def run(self):
|
|
document = self.state.document
|
|
filename = self.arguments[0]
|
|
if not document.settings.file_insertion_enabled:
|
|
return [document.reporter.warning('File insertion disabled',
|
|
line=self.lineno)]
|
|
env = document.settings.env
|
|
if filename.startswith('/') or filename.startswith(os.sep):
|
|
rel_fn = filename[1:]
|
|
else:
|
|
docdir = path.dirname(env.doc2path(env.docname, base=None))
|
|
rel_fn = path.normpath(path.join(docdir, filename))
|
|
try:
|
|
fn = path.join(env.srcdir, rel_fn)
|
|
except UnicodeDecodeError:
|
|
# the source directory is a bytestring with non-ASCII characters;
|
|
# let's try to encode the rel_fn in the file system encoding
|
|
rel_fn = rel_fn.encode(sys.getfilesystemencoding())
|
|
fn = path.join(env.srcdir, rel_fn)
|
|
|
|
encoding = self.options.get('encoding', env.config.source_encoding)
|
|
try:
|
|
f = codecs.open(fn, 'rU', encoding)
|
|
lines = f.readlines()
|
|
f.close()
|
|
except (IOError, OSError):
|
|
return [document.reporter.warning(
|
|
'Include file %r not found or reading it failed' % filename,
|
|
line=self.lineno)]
|
|
except UnicodeError:
|
|
return [document.reporter.warning(
|
|
'Encoding %r used for reading included file %r seems to '
|
|
'be wrong, try giving an :encoding: option' %
|
|
(encoding, filename))]
|
|
|
|
config = TriqsExampleRun.config
|
|
|
|
# Get configuration values for the language
|
|
input_encoding = 'utf8' #config.get(language+'_input_encoding','ascii')
|
|
output_encoding = 'utf8' #config.get(language+'_output_encoding','ascii')
|
|
show_source = True
|
|
|
|
# Build the code text
|
|
code = ''.join(lines).strip()
|
|
filename_clean = filename.rsplit('.',1)[0]
|
|
if filename_clean.startswith('./') : filename_clean = filename_clean[2:]
|
|
#print "Running the example ....",filename_clean
|
|
#print "Root ?", env.doc2path(env.docname, base=None)
|
|
|
|
import subprocess as S
|
|
error = True
|
|
try :
|
|
stdout =''
|
|
#resout = S.check_output("./example_bin/doc_%s"%(filename_clean) ,stderr=S.STDOUT,shell=True)
|
|
resout = S.check_output("./%s/doc_%s"%(docdir,filename_clean) ,stderr=S.STDOUT,shell=True)
|
|
if resout :
|
|
stdout = '---------- Result is -------\n' + resout.strip()
|
|
error = False
|
|
except S.CalledProcessError as E :
|
|
stdout ='---------- RunTime error -------\n'
|
|
stdout += E.output
|
|
|
|
# Process output
|
|
if stdout:
|
|
stdout = stdout.decode(output_encoding,'ignore')
|
|
out = ''.join(stdout).decode(output_encoding)
|
|
else:
|
|
out = '' #.join(stderr).decode(output_encoding)
|
|
|
|
# Get the original code with prefixes
|
|
code_out = '\n'.join((code,out))
|
|
|
|
if error : # report on console
|
|
print(" Error in processing ")
|
|
print(code_out)
|
|
|
|
literal = nodes.literal_block(code_out,code_out)
|
|
literal['language'] = 'c'
|
|
literal['linenos'] = 'linenos' in self.options
|
|
return [literal]
|
|
|
|
def setup(app):
|
|
app.add_directive('triqs_example', TriqsExample)
|
|
app.connect('builder-inited',TriqsExampleRun.builder_init)
|
|
|